在akka中测试计时器事件

时间:2014-11-13 23:14:05

标签: scala unit-testing testing akka

我有一个函数使用scheduleOnce来安排事件在将来某个时间发生,我想编写一个测试来检查:

  • 事件确实已安排
  • 计划在正确的时间
  • 当该事件最终触发时系统按预期行事

但我不希望测试实际上等几分钟什么都不做。

我应该如何最好地测试使用akka的Scheduler的代码?

2 个答案:

答案 0 :(得分:4)

以下是@lmm描述的模拟调度程序的示例。在这个例子中,我们真正测试了动作的完整调度和处理,作为两个独立的场景。给出一些条件的第一个测试(在我的例子中接收某种类型的消息)我们将安排一个回调,第二个是处理在该定时器关闭时被激发回来的消息。代码如下:

object TimerExampleActor{
  case object ClearState
  case class ScheduleStateClearing(duration:FiniteDuration)
}

class TimerExampleActor extends Actor{
  import TimerExampleActor._

  var state:List[Int] = Nil

  def receive = {
    case ScheduleStateClearing(d) =>
      scheduler.scheduleOnce(d, self, ClearState)(context.dispatcher)

    case ClearState => 
      state = Nil
  }

  def scheduler = context.system.scheduler
}

然后,使用specs2和mockito,我的测试用例如下:

class TimerExampleActorTest extends Specification with Mockito with NoTimeConversions{
  import TimerExampleActor._
  implicit val system = ActorSystem("test")

  trait testscope extends Scope{
    val mockScheduler = mock[Scheduler]
    val actor = TestActorRef(new TimerExampleActor{
      override def scheduler = mockScheduler 
    })
  }

  "A request to schedule state clearing" should{
    "schedule a callback for the message ClearState to self with the supplied duration" in new testscope{
      val dur = 1.minute
      actor ! ScheduleStateClearing(dur) 
      there was one(mockScheduler).scheduleOnce(dur, actor, ClearState)(actor.underlyingActor.context.dispatcher)      
    }
  }

  "A ClearState message received by the actor" should{
    "clear the interval 'state' List" in new testscope{
      actor.underlyingActor.state = List(1,2,3)
      actor ! ClearState
      actor.underlyingActor.state mustEqual Nil
    }
  }
}

你可以看到,当我在测试中创建actor实例时,我覆盖了我创建的方法,以获取调度程序的实例,允许我返回一个模拟。这不是测试这样的东西的唯一方法,但它肯定是你可以考虑的一个选择。

答案 1 :(得分:0)

使调度程序采用时间参数。在您的测试中使用比常规代码更短的时间。

或者......在测试时,你可以混合一个特殊的特性,根据需要修改你的课程(缩短等待时间。)