2nd version

PollingSchedulerTest.scala

package com.zeebox.context.feedsingest.server

import com.typesafe.config.Config
import com.zeebox.context.feedsingest.config.TrapitConfig
import com.zeebox.context.feedsingest.trapit.HasTrapitConfig
import com.zeebox.core.config.ConfigLoader
import org.junit.runner.RunWith
import org.specs2.runner.JUnitRunner
import org.specs2.mutable.SpecificationWithJUnit
import org.specs2.mock.Mockito
import org.specs2.mock.mockito.MockitoFunctions
import com.zeebox.context.feedsingest.utils.ConcurrencyUtils
import org.joda.time.DateTime
import com.zeebox.core.monitoring.MonitoringService

object PollingSchedulerTest extends Mockito {

  trait DummyTrapitConfig extends HasTrapitConfig {
    implicit val config: TrapitConfig = {
      val loader: Config = ConfigLoader("feeds.trapit.daily_poll_allocation" -> new Integer(5000))
      TrapitConfig.apply(loader)
    }
  }

  abstract class TestPollingScheduler extends LivePollingScheduler with DummyTrapitConfig {
    val monitoringService = mock[MonitoringService]
    def queueCount: Option[Int] = Some(1000)
  }
}

@RunWith(classOf[JUnitRunner])
class PollingSchedulerTest extends SpecificationWithJUnit
with Mockito
with MockitoFunctions
with ConcurrencyUtils {

  import PollingSchedulerTest._

  "The Polling Scheduler" should {

    "Correctly calculate poll times from offsets (1)" in {
      val f = new TestPollingScheduler{
        override def now: DateTime = DateTime.parse("2013-01-01T00:00:00.000Z")
      }
      f.nextPoll().get.toString() must_== "2013-01-01T04:48:00.000Z"
    }

    "Correctly calculate poll times from offsets (2)" in {
      val f = new TestPollingScheduler{
        override def now: DateTime = DateTime.parse("2013-01-01T04:48:00.000Z")
      }
      f.nextPoll().get.toString() must_== "2013-01-01T09:36:00.000Z"
    }

    "Correctly calculate poll times from offsets, rolling to next day" in {
      val f = new TestPollingScheduler{
        override def now: DateTime = DateTime.parse("2013-01-01T19:12:00.000Z")
      }
      f.nextPoll().get.toString() must_== "2013-01-02T00:00:00.000Z"
    }

    "Correctly calculate poll times from offsets, rolling to next day (hour in)" in {
      val f = new TestPollingScheduler{
        override def now: DateTime = DateTime.parse("2013-01-01T20:12:00.000Z")
      }
      f.nextPoll().get.toString() must_== "2013-01-02T01:00:00.000"
    }
  }
}