Merge pull request #5 from rooeque/gatlingDI

change gatling adapters to use DI instead of directly calling measurement controllers
Workday · Sep 13, 2018 · 321d59c · 321d59c
2 parents 1a671b9 + 8a6f5ab
commit 321d59c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/src/main/scala/com/workday/warp/adapters/gatling/WarpFunSpec.scala b/src/main/scala/com/workday/warp/adapters/gatling/WarpFunSpec.scala
@@ -1,8 +1,9 @@
 package com.workday.warp.adapters.gatling
 
 import com.workday.warp.adapters.gatling.traits.{HasDefaultTestName, HasWarpHooks}
-import com.workday.warp.collectors.{AbstractMeasurementCollectionController, DefaultMeasurementCollectionController}
+import com.workday.warp.collectors.AbstractMeasurementCollectionController
 import com.workday.warp.common.CoreConstants.{UNDEFINED_TEST_ID => DEFAULT_TEST_ID}
+import com.workday.warp.inject.WarpGuicer
 import io.gatling.http.funspec.GatlingHttpFunSpec
 
 /**
@@ -14,7 +15,7 @@ abstract class WarpFunSpec(val testId: String) extends GatlingHttpFunSpec with H
 
   def this() = this(DEFAULT_TEST_ID)
 
-  val controller: AbstractMeasurementCollectionController = new DefaultMeasurementCollectionController(this.canonicalName)
+  val controller: AbstractMeasurementCollectionController = WarpGuicer.getController(this.canonicalName, tags = List.empty)
 
   before {
     beforeStart()

diff --git a/src/main/scala/com/workday/warp/adapters/gatling/WarpSimulation.scala b/src/main/scala/com/workday/warp/adapters/gatling/WarpSimulation.scala
@@ -1,8 +1,9 @@
 package com.workday.warp.adapters.gatling
 
 import com.workday.warp.adapters.gatling.traits.{HasDefaultTestName, HasWarpHooks}
-import com.workday.warp.collectors.{AbstractMeasurementCollectionController, DefaultMeasurementCollectionController}
+import com.workday.warp.collectors.AbstractMeasurementCollectionController
 import com.workday.warp.common.CoreConstants.{UNDEFINED_TEST_ID => DEFAULT_TEST_ID}
+import com.workday.warp.inject.WarpGuicer
 import io.gatling.core.Predef.Simulation
 
 /**
@@ -14,7 +15,7 @@ abstract class WarpSimulation(val testId: String) extends Simulation with HasDef
 
   def this() = this(DEFAULT_TEST_ID)
 
-  val controller: AbstractMeasurementCollectionController = new DefaultMeasurementCollectionController(this.canonicalName)
+  val controller: AbstractMeasurementCollectionController = WarpGuicer.getController(this.testId, tags = List.empty)
 
   before {
     beforeStart()