Commit c98572c0 authored by Leonardo Menezes's avatar Leonardo Menezes
Browse files

added test to GetIndexMappingController

parent 4b7e460d

app/controllers/GetIndexMapping.scala→app/controllers/GetIndexMappingController.scala

+1 −1
Original line number Diff line number Diff line 
package controllers



class GetIndexMapping extends ElasticActionController {

class GetIndexMappingController extends ElasticActionController {



  def processElasticRequest = (request, client) => client.getIndexMapping(request.get("index"), request.host)

conf/routes

+1 −1
Original line number Diff line number Diff line
@@ -16,7 +16,7 @@ POST /apis/clear_indices_cache @controllers.ClearIndexCacheControl 
POST       /apis/refresh_indices             @controllers.RefreshIndexController.execute

POST       /apis/delete_indices              @controllers.DeleteIndexController.execute

POST       /apis/get_index_settings          @controllers.GetIndexSettings.execute

POST       /apis/get_index_mapping           @controllers.GetIndexMapping.execute

POST       /apis/get_index_mapping           @controllers.GetIndexMappingController.execute

POST       /apis/update_cluster_settings     @controllers.PutClusterSettings.execute

POST       /apis/get_node_stats              @controllers.NodeStatsController.execute

POST       /apis/disable_shard_allocation    @controllers.DisableShardAllocationController.execute

test/controllers/GetIndexMappingControllerSpec.scala

0 → 100644
+51 −0
Original line number Diff line number Diff line 
package controllers



import elastic.{ElasticResponse, ElasticClient}

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 GetIndexMappingControllerSpec extends Specification with Mockito {



  def is =

    s2"""

    GetIndexMappingController should                   ${step(play.api.Play.start(FakeApplication()))}

      invoke getIndexMapping                           $getIndexMapping

      should throw exception if index param is missing $missingIndex

                                                       ${step(play.api.Play.stop(FakeApplication()))}

      """



  val controller = new GetIndexMappingController



  def getIndexMapping = {

    val expectedResponse = Json.parse(

      """

        |{

        |  "someIndex": {

        |    "mappings": {}

        |  }

        |}

      """.stripMargin

    )

    val body = Json.obj("host" -> "somehost", "index" -> "someIndex")

    val client = mock[ElasticClient]

    client.getIndexMapping("someIndex", "somehost") returns Future.successful(ElasticResponse(200, expectedResponse))

    val response = Await.result(controller.processElasticRequest(CerebroRequest(body), client), Duration("1s"))

    there was one(client).getIndexMapping("someIndex", "somehost")

    response.body mustEqual expectedResponse

    response.status mustEqual 200

  }



  def missingIndex = {

    val body = Json.obj("host" -> "somehost")

    val client = mock[ElasticClient]

    controller.processElasticRequest(CerebroRequest(body), client) must throwA[MissingRequiredParamException]

  }



}