专栏首页AlgorithmDog的专栏Akka 使用系列之二: 测试

Akka 使用系列之二: 测试

通过上一篇文章,我们已经大致了解怎么使用 Akka,期待细致用法。这篇文章将介绍如何用 Akka-testkit 对 Akka 程序进行测试。

并行程序是最难调试的程序类型之一,因此做好测试是相当重要的事情。为了减轻 Akka 的程序的测试难度, Akka 官方专门开发了一个测试工具包 Akka-testkit。

1 Actor 的测试需求

对于一个 Actor, 我们要测什么呢?不同的文章有不同的说法,比如 http://rerun.me/2014/09/29/akka-notes-logging-and-testing/ 就把 Actor 测试需求分为:1)发送消息给 Actors, 2)测试内部状态,3)测试日志和 4)带参数 Actor 的测试。我个人认为,对于一个 Actor, 我们要测的有三个方面:1)Actor 接收消息之后,是否返回正确的消息,2)Actor 接收消息之后,是否正确地改变内部状态和执行内部行为,3)Actor 接收消息之后,是否正确地发消息给后续 Actor。当然这是我的一家之言,有什么不完善的地方,欢迎大家讨论。下面是一个简单的示例图。

下面是 studentActor 的一段代码,反应了 studentActor 接受到早上时间消息之后的动作,包括:1)给环境或者闹钟回应“关闭闹钟”,2)内部变量 DayInSchool 加 1,3)向老师发送问题消息。这段代码将包含所有要测试的元素,后面我们将示例怎么用 Akka-testkit 测试这段代码。

  def receive = {
      case time:Long => {

        val originalSender = sender;
        sender ! "关闭闹钟" 

        DayInSchool += 1;
        log.info("DayInSchool is %d".format(DayInSchool))

        remoteTeacherRef ! "历史上规模最大的众筹行动是什么?";
      }
    }

2 不适用的 Scalatest

Scalatest 是 Scala 开发者们最常见的测试工具,其用法非常简便。下面是一个 Scalatest 的简单示例。

@RunWith(classOf[JUnitRunner])
class TeacherServiceTest1 extends FunSuite with BeforeAndAfter{
  test("countAnswer"){
     assert(1==1)
  }
}

但是我们无法使用 scalatest 测试 Actor。原因在于:1)Scalatest 无法捕捉被测 Actor 回应的消息,因此无法测试被测 Actor 是否正确回应消息; 2)Scalatest 无法获取被测 Actor 的内部状态,因此无法测试被测 Actor 内部状态的改变是否正确; 3) Scalatest 无法捕捉被测 Actor 对外发送的消息,因此无法测试被测 Actor 对外发送的消息是否正确。因此有必要针对 Akka 开发一套测试工具, Akka-testkit 测试包应运而生。

3 Akka-testkit 的使用

Maven 项目要使用 Akka-testkit,需要在 pom.xml 文件中加入 akka-testkit 包,如下所示。

<dependency>
    <groupId>com.typesafe.akka</groupId>
    <artifactId>akka-testkit_2.10</artifactId>
    <version>2.2.5</version>
</dependency>

然后编写单元测试的代码,其基本范例如下。

@RunWith(classOf[JUnitRunner]) //
class StudentActorTest
  extends TestKit(ActorSystem("SummerSchool",
    ConfigFactory.parseString("""一些配置""")))
  with WordSpecLike
  with BeforeAndAfterAll{
   
     "A actor " must {
       "acts in this way" in {
          测试代码写在这里。
        }
     }

  }

Akka-testkit 的主要工具包括, 1) testProbe 用于测试 Actor 回应和发送消息,testActor 用于简便情况下测试 Actor 回应消息,和 2) testActorRef 用于测试 Actor 内部状态的改变。

3.1 回应消息的测试

对于被测 Actor 是否正确地回应消息,可以用 testProbe 测试。首先将 testProbe 给被测 Actor 发送消息,再看 testProbe 是否接受到期望的回应消息。下面是一个示例。

 //test its responses
  "The countAnswer " must {
    "response a correct answer order" in {
      val studentActor 
          = system.actorOf(Props(new StudentActor(teacherActor )))
      val testProb   = new TestProbe(system);
      testProb.send(studentActor, 7.toLong);
      //testProbe 给 studentActor 发送 “早上 7 点啦” 的消息
      testProb.send(studentActor, 7.toLong)

      testProb.expectMsg("关闭闹钟")
      //测试 testProbe 是否收到预期回应消息
      testProb.expectMsg("关闭闹钟")
    }
  }

除了使用 testProbe 之外,Akka-testkit 还提供一种简便方法: 使用 testActor。 如果测试类实现特质 ImplicitSender, studentActorRef ! 7.toLong 发送给 studentActor 的消息 7.toLong 就是从 testActor 来的。然后在调用 expectMsg(“关闭闹钟”) 就可以测试 testActor 是否收到 studentActor 回应消息 “关闭闹钟” 了。具体代码如下所示。

class StudentActorTest
  extends TestKit(ActorSystem("SummerSchool",
    ConfigFactory.parseString("""一些配置""")))
  with ImplicitSender //加这句,把testActor 设置为消息发出 Actor
  with WordSpecLike
  with BeforeAndAfterAll{

  "StudentActor" must {
    "response correctly" in {
      val studentActorRef 
      = system.actorOf(Props(new StudentActor(teacherActor )))
      studentActorRef ! 7.toLong; 
      //testActor 发出 7.toLong 消息给 studentActor
      expectMsg("关闭闹钟") 
      //testActor 应该收到 studentActor 回应消息 "关闭闹钟"
    }
  }

}

我们可以看出,使用 testActor 的代码比使用 testProbe 的简便。但是,一个东西的用法越是简便,功能便越缺失。testActor 最大的缺失是只能接受被测 Actor 发来的一个回应消息。比如下面的代码就会报错。

"StudentActor" must {
    "response correctly" in {
      val studentActorRef 
      = system.actorOf(Props(new StudentActor(teacherActor )))
      studentActorRef ! 7.toLong; 
      studentActorRef ! 8.toLong; 
      expectMsg("关闭闹钟") 
      expectMsg("关闭闹钟") 
    }

3.2 内部状态的测试

对于被测 Actor 内部状态的改变,可以用 TestActorRef 进行测试。TestActorRef.underlyingActor 可以探测被测 Actor 的内部,用于测试被测 Actor 内部状态是否符合预期。 下面是一个示例。

"StudentActor" must {
    "increase the DayInSchool" in {
      val testActorRef 
       = TestActorRef(new StudentActor(teacherActor)) 
      // 建立 Actor 的TestActorRef
      testActorRef ! 7.toLong; 
      assert(testActorRef.underlyingActor.DayInSchool == 1); 
      // TestActorRef.underlyingActor 探测 DayInSchool 变量是否符合预期。
    }
}

3.3 发出消息的测试

对于被测 Actor 是否正确地发出消息,也可以用 testProbe 测试。首先将 testProbe 设置为被测 Actor 发出消息的目标,然后让被测 Actor 发出消息,再看 testProbe 是否接受到期望的消息。下面是一个示例。

"StudentActor " must{
    val questionReceiver 
      = TestProbe()
    val studentActorRef 
      = system.actorOf(Props(new StudentActor(questionReceiver.ref))) 
    // 设置为 studentActor 发送消息的目标

    "send a question after waking up" in {
        studentActorRef ! 7.toLong 
        studentActorRef ! 7.toLong
        /*给 studentActor 发送消息“7点啦” ,
         *studentActor 会给老师(这里的 questionReceiver.ref) 发送问题
         */

        questionReceiver.expectMsg("历史上规模最大的众筹行动是什么?")
        questionReceiver.expectMsg("历史上规模最大的众筹行动是什么?")
        //模拟老师的 testProbe 是否收到预期问题
    }
}

4 总结

Akka-testkit 是 Akka 官方推出的 Akka 测试工具包,用于减轻 Akka 程序的测试难度。Akka-testkit 的主要工具包括, 1) testProbe 用于测试被测 Actor 回应和发送消息,testActor 用于简便情况下测试被测 Actor 回应消息,和 2) testActorRef 用于测试被测 Actor 内部状态的改变。完整的项目代码已经上传到 Github 上了。被测 Actor 是 org.algorithmdog.akkalearning.StudentActor, 测试类是 org.algorithmdog.akkalearning.StudentActorTest。

这篇文章难产了很长一段时间,对不住支持我的读者们。对不起。Akka 和 Actor 模型对我来说是一个全新的东西,花了比较多的时间学习和熟悉。学习之后,觉得第一篇写得太不清楚了,准备重构第一篇。对于这篇文章质量,我个人比较满意的,甚至敢认为这篇文章应该是国内关于 Akka-testkit 最清楚的文章之一(ps:大牛们轻喷)。

Akka 系列系列文章

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • Akka 使用系列之四: Future

    这篇文章介绍 Akka 的同步机制,以及 Spark 和 Akka 的恩怨情仇。 ? 1 Akka 中的 Future Akka 中的...

    AlgorithmDog
  • Akka 使用系列之一: 快速入门

    最近在看 Spark 相关的资料,准备整理一个 Spark 系列。Akka 是 Spark 实现内部通讯的组件,Spark 启动过程的第一步便是建立 A...

    AlgorithmDog
  • Akka 使用系列之三: 层次结构和容错机制

    这篇文章介绍 Akka 层次结构,以及基于层次结构的容错机制。 ? 1. Akka 的层次结构 我们需要实现一个翻译模块,其功能是输入中...

    AlgorithmDog
  • Excel图表学习61: 编写一个给多个数据系列添加趋势线的加载宏

    在《Excel图表学习60:给多个数据系列添加趋势线》中,我们手工给多个散点图系列添加了一条趋势线,如下图1所示。

    fanjy
  • 苦涩的技术我该怎么学?Akka 实战

    关注“一猿小讲”公众号的小伙伴都清楚,上次《技术再深入一点又何妨?一脸懵B的聊Actor》,我们在“懵 B”的状态下,聊了聊 Actor 模型的理论知识。稍微再...

    一猿小讲
  • Generative Adversarial Networks

    上一篇讲述了VAEs(变分自编码器),那么这次继续学习一下另一个生成模型——GANs。这里建议如果没有看VAEs的请点击传送门:,因为有所关联,所以如果直接看这...

    可爱见见
  • Docker Data Center系列(二)- UCP安装指南

    安装过程中UCP节点需要能下载docker官网的资源,如果不能访问,可通过其它机器下载软件包,然后执行离线安装。

    羽客
  • 如何给老婆解释什么是Restful

    果不其然,看着看着,她又对我发难了,“Restful是什么呀,老公?是restaurant的形容词吗,突然就觉得好饿了啊......”

    Java3y
  • JAVA 设计模式 组合模式

    用途 组合模式 (Component) 将对象组合成树形结构以表示“部分-整体”的层次结构。 组合模式使得用户对单个对象和组合对象的使用具有唯一性。 组合模式是...

    静默虚空
  • java泛型详解

    输出是啥?正确答案是true。上面的代码中涉及到了泛型,而输出的结果缘由是类型擦除。

    提莫队长

扫码关注云+社区

领取腾讯云代金券