Loading test/controllers/OptimizeIndexControllerSpec.scala 0 → 100644 +53 −0 Original line number Diff line number Diff line package controllers import elastic.{ElasticClient, ElasticResponse} import exceptions.MissingRequiredParamException import models.CerebroRequest import org.specs2.Specification import org.specs2.mock.Mockito import play.api.libs.json.Json import play.api.test.FakeApplication import scala.concurrent.duration.Duration import scala.concurrent.{Await, Future} object OptimizeIndexControllerSpec extends Specification with Mockito { def is = s2""" OptimizeIndexController should ${step(play.api.Play.start(FakeApplication()))} invoke optimizeIndex $optimizeIndex should throw exception if indices param is missing $missingIndices ${step(play.api.Play.stop(FakeApplication()))} """ val controller = new OptimizeIndexController def optimizeIndex = { val expectedResponse = Json.parse( """ |{ | "_shards": { | "total": 10, | "successful": 5, | "failed": 0 | } |} """.stripMargin ) val body = Json.obj("host" -> "somehost", "indices" -> "a,b,c") val client = mock[ElasticClient] client.optimizeIndex("a,b,c", "somehost") returns Future.successful(ElasticResponse(200, expectedResponse)) val response = Await.result(controller.processElasticRequest(CerebroRequest(body), client), Duration("1s")) there was one(client).optimizeIndex("a,b,c", "somehost") response.body mustEqual expectedResponse response.status mustEqual 200 } def missingIndices = { val body = Json.obj("host" -> "somehost") val client = mock[ElasticClient] controller.processElasticRequest(CerebroRequest(body), client) must throwA[MissingRequiredParamException] } } Loading
test/controllers/OptimizeIndexControllerSpec.scala 0 → 100644 +53 −0 Original line number Diff line number Diff line package controllers import elastic.{ElasticClient, ElasticResponse} import exceptions.MissingRequiredParamException import models.CerebroRequest import org.specs2.Specification import org.specs2.mock.Mockito import play.api.libs.json.Json import play.api.test.FakeApplication import scala.concurrent.duration.Duration import scala.concurrent.{Await, Future} object OptimizeIndexControllerSpec extends Specification with Mockito { def is = s2""" OptimizeIndexController should ${step(play.api.Play.start(FakeApplication()))} invoke optimizeIndex $optimizeIndex should throw exception if indices param is missing $missingIndices ${step(play.api.Play.stop(FakeApplication()))} """ val controller = new OptimizeIndexController def optimizeIndex = { val expectedResponse = Json.parse( """ |{ | "_shards": { | "total": 10, | "successful": 5, | "failed": 0 | } |} """.stripMargin ) val body = Json.obj("host" -> "somehost", "indices" -> "a,b,c") val client = mock[ElasticClient] client.optimizeIndex("a,b,c", "somehost") returns Future.successful(ElasticResponse(200, expectedResponse)) val response = Await.result(controller.processElasticRequest(CerebroRequest(body), client), Duration("1s")) there was one(client).optimizeIndex("a,b,c", "somehost") response.body mustEqual expectedResponse response.status mustEqual 200 } def missingIndices = { val body = Json.obj("host" -> "somehost") val client = mock[ElasticClient] controller.processElasticRequest(CerebroRequest(body), client) must throwA[MissingRequiredParamException] } }