Polling Scheduler test case

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)
    }
  }

  class TestPollingScheduler extends LivePollingScheduler with DummyTrapitConfig {
    val monitoringService = mock[MonitoringService]

    override def getMidnight: DateTime = DateTime.parse("2013-1-1").toDateMidnight.toDateTime

    def queueCount: Option[Int] = Some(1000)

    override def now: DateTime = {
      DateTime.parse("2013-1-1T00:00:00")
    }
  }

}

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

  import PollingSchedulerTest._

  "The Polling Scheduler" should {

    "Correctly calculate poll times from tick" in {
      val f = new TestPollingScheduler()

      f.getMidnight.toString() must_== "2013-01-01T00:00:00.000Z"
      //TODO RFC If the next poll is midnight, should we re-schedule it for immediate execution?
      f.nextPoll(0).get.toString() must_== "2013-01-01T00:00:00.000Z"
      f.nextPoll(1).get.toString() must_== "2013-01-01T04:48:00.000Z"
      f.nextPoll(2).get.toString() must_== "2013-01-01T09:36:00.000Z"
      f.nextPoll(3).get.toString() must_== "2013-01-01T14:24:00.000Z"
      f.nextPoll(4).get.toString() must_== "2013-01-01T19:12:00.000Z"
      f.nextPoll(5).get.toString() must_== "2013-01-02T00:00:00.000Z"
      f.nextPoll(6) must_== None
      f.nextPoll(7) must_== None
    }

    "Correctly calculate poll times from offsets" in {
      val f = new TestPollingScheduler()
      f.now.toString() must_== "2013-01-01T00:00:00.000Z"
      f.getMidnight.toString() must_== "2013-01-01T00:00:00.000Z"
      f.nextPoll(DateTime.parse("2013-01-01T00:00:00.000Z")).get.toString() must_== "2013-01-01T00:00:00.000Z"
      f.nextPoll(DateTime.parse("2013-01-01T04:47:00.000Z")).get.toString() must_== "2013-01-01T04:48:00.000Z"
      f.nextPoll(DateTime.parse("2013-01-01T09:35:00.000Z")).get.toString() must_== "2013-01-01T09:36:00.000Z"
      f.nextPoll(DateTime.parse("2013-01-01T14:23:00.000Z")).get.toString() must_== "2013-01-01T14:24:00.000Z"
      f.nextPoll(DateTime.parse("2013-01-01T19:11:00.000Z")).get.toString() must_== "2013-01-01T19:12:00.000Z"
      f.nextPoll(DateTime.parse("2013-01-02T00:00:00.000Z")) must_== None
    }
  }
}