使用ScalaTest和Future进行集成测试

时间:2015-08-23 15:07:11

标签: scala unit-testing asynchronous integration-testing future

我试图围绕异步方法构建我的应用程序,所以我使用Future和Scala。有关详细信息,请elastic4s lib查询Elasticsearch服务器。

我从这个模式开始,所以我首先创建了这个方法:

    def insert(person: Person) = {
      client.execute {
        index into "index" / "type" source person
      } onComplete {
        case Success(s) => {
          logger.info("{} has been inserted successfully", interaction.toString)
          something_to_do()
         }
        case Failure(t) => logger.warn("An error has occured: {}", t.getMessage)
     }
   }

如果我理解得很好,这种方法是不可测试的。 当我读到这个post时,我必须创建一个返回Future的方法,我必须使用ScalaFutures(用于ScalaTest)来测试我的异步方法。

所以我执行这个模式,并且我已经创建了两个这样的方法:

def insert(person: Person): Future[IndexResponse] = {
    client.execute {
        index into "index" / "type" source person
    }
}

def insertToEs(person: Person): Unit = {
    insert(person) onComplete {
        case Success(s) => {
            logger.info("{} has been inserted successfully", person.toString)
            something_to_do()
        }
        case Failure(t) => logger.warn("An error has occured: {}", t.getMessage)
    }
}

所以现在测试我的第一个方法很容易,但是如何测试insertToEs方法呢?

更一般地说,我想创建集成测试:Kafka(嵌入式)到Elasticsearch(嵌入式)。

我有这个代码调用以前的insertToEs方法:

def receive: Unit = {
    ...
    val iterator = stream.iterator()
    while(iterator.hasNext) {
        insertToEs(iterator.next.message)
    }
}

我已经在接收方法上创建了一个测试,但看起来集成测试在方法执行之前结束了。

@RunWith(classOf[JUnitRunner])
class ToEsSpec extends FlatSpec with Matchers with BeforeAndAfterAll with ScalaFutures {

  override def beforeAll = {
      run_kafka_server
      run_elasticsearch_server
  }

  override def afterAll = {
      shutdown
  }

  it should "receive message from kafka and insert to es" {

      send_to_kafka(json)

      receive

      assert if inserted in es
  }
}
你能告诉我一下吗?感谢

1 个答案:

答案 0 :(得分:3)

请参阅http://doc.scalatest.org/2.2.4/index.html#org.scalatest.concurrent.Futures

在我的一些测试用例中,我使用了以下两种模式: