Commit 5839dce9 authored by Leonardo Menezes's avatar Leonardo Menezes
Browse files

added test to OptimizeIndexController

parent 1d171aed
Loading
Loading
Loading
Loading
+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]
  }

}