#32 Verify Gatling simulations whether edgegrid-signer-gatling really includes correct headers in HTTP request.

Author: mgawinec

edgegrid-signer-gatling/pom.xml

@@ -32,6 +32,10 @@
             <groupId>org.scala-lang</groupId>
             <artifactId>scala-library</artifactId>
         </dependency>
+        <dependency>
+            <groupId>com.github.tomakehurst</groupId>
+            <artifactId>wiremock</artifactId>
+        </dependency>
     </dependencies>
 
     <build>

edgegrid-signer-gatling/src/test/scala/com/akamai/edgegrid/signer/gatling/EdgeGridSignerSimulation1.scala

@@ -17,24 +17,46 @@
 package com.akamai.edgegrid.signer.gatling
 
 import com.akamai.edgegrid.signer.ahc.AsyncHttpClientEdgeGridSignatureCalculator
+import com.github.tomakehurst.wiremock.WireMockServer
+import com.github.tomakehurst.wiremock.client.WireMock._
+import com.github.tomakehurst.wiremock.core.WireMockConfiguration
 import io.gatling.core.Predef._
 import io.gatling.http.Predef._
 
 
 class EdgeGridSignerSimulation1 extends Simulation {
 
   val httpConf = http
-    .baseURL("http://localhost")
+    .baseURL("http://" + testdata.SERVICE_MOCK_HOST)
+
+  val wireMockServer = new WireMockServer(WireMockConfiguration.wireMockConfig().port(testdata.SERVICE_MOCK_PORT))
 
   val testScenario = scenario("Test scenario")
     .exec(
       http("fakeRequest")
         .get("/test")
         .signatureCalculator(new AsyncHttpClientEdgeGridSignatureCalculator(testdata.testCredential))
+        .check(status.is(201))
+    )
+
+  before {
+    wireMockServer.start()
+    wireMockServer.stubFor(get(urlPathEqualTo("/test"))
+      .withHeader("Authorization", matching(".*"))
+      .withHeader("Host", equalTo(testdata.SERVICE_MOCK))
+      .willReturn(aResponse.withStatus(201)
+        .withHeader("Content-Type", "text/xml")
+        .withBody("<response>Some content</response>")))
+  }
+
+  setUp(testScenario.inject(atOnceUsers(1)))
+    .protocols(httpConf)
+    .assertions(
+      global.successfulRequests.percent.is(100)
     )
 
-  setUp(
-    testScenario.inject(atOnceUsers(1))
-  ).protocols(httpConf)
+  after {
+    wireMockServer.stop()
+  }
 
 }

edgegrid-signer-gatling/src/test/scala/com/akamai/edgegrid/signer/gatling/EdgeGridSignerSimulation2.scala

@@ -17,22 +17,44 @@
 package com.akamai.edgegrid.signer.gatling
 
 import com.akamai.edgegrid.signer.ahc.AsyncHttpClientEdgeGridSignatureCalculator
+import com.github.tomakehurst.wiremock.WireMockServer
+import com.github.tomakehurst.wiremock.client.WireMock._
+import com.github.tomakehurst.wiremock.core.WireMockConfiguration
 import io.gatling.core.Predef._
 import io.gatling.http.Predef._
 
-class EdgeGridSignerSimulation2 extends Simulation{
+class EdgeGridSignerSimulation2 extends Simulation {
 
   val httpConf = http
 
+  val wireMockServer = new WireMockServer(WireMockConfiguration.wireMockConfig().port(testdata.SERVICE_MOCK_PORT))
+
   val testScenario = scenario("Test scenario")
     .exec(
       http("fakeRequest")
-        .get("https://" + testdata.testCredential.getHost + "/test")
+        .get("http://" + testdata.testCredential.getHost + "/test")
         .signatureCalculator(new AsyncHttpClientEdgeGridSignatureCalculator(testdata.testCredential))
     )
 
-  setUp(
-    testScenario.inject(atOnceUsers(1))
-  ).protocols(httpConf)
+  before {
+    wireMockServer.start()
+    wireMockServer.stubFor(get(urlPathEqualTo("/test"))
+      .withHeader("Authorization", matching(".*"))
+      .withHeader("Host", equalTo(testdata.SERVICE_MOCK))
+      .willReturn(aResponse.withStatus(201)
+        .withHeader("Content-Type", "text/xml")
+        .withBody("<response>Some content</response>")))
+  }
+
+  setUp(testScenario.inject(atOnceUsers(1)))
+    .protocols(httpConf)
+    .assertions(
+      global.successfulRequests.percent.is(100)
+    )
+
+  after {
+    wireMockServer.stop()
+  }
+
 
 }

edgegrid-signer-gatling/src/test/scala/com/akamai/edgegrid/signer/gatling/TestData.scala

@@ -26,4 +26,9 @@ package object testdata {
     .clientSecret("SOMESECRET")
     .host("localhost:9089").build
 
+  val SERVICE_MOCK_HOST = "localhost"
+  val SERVICE_MOCK_PORT = 9089
+  val SERVICE_MOCK = testdata.SERVICE_MOCK_HOST + ":" + testdata.SERVICE_MOCK_PORT
+
+
 }