Added init() to the RoomManager

CoSMoSoftware · Jun 13, 2019 · ea7d527 · ea7d527
1 parent b5735e1
commit ea7d527
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/pom.xml b/pom.xml
@@ -6,7 +6,7 @@
 
   <groupId>io.cosmosoftware.kite</groupId>
   <artifactId>kite-extras</artifactId>
-  <version>0.1.22</version>
+  <version>0.1.23</version>
 
   <properties>
     <kite.log4j.version>1.2.17</kite.log4j.version>

diff --git a/src/main/java/io/cosmosoftware/kite/manager/RoomManager.java b/src/main/java/io/cosmosoftware/kite/manager/RoomManager.java
@@ -189,5 +189,8 @@ private String getRandomRoomId(int roomIdLen) {
     return Integer.toString((int) Math.floor(Math.random() * Math.pow(10, roomIdLen)));
   }
 
+  public static void init() {
+    roomManagers.clear();
+  }
 
 }
diff --git a/src/main/java/io/cosmosoftware/kite/steps/StepPhase.java b/src/main/java/io/cosmosoftware/kite/steps/StepPhase.java
@@ -35,7 +35,7 @@ public boolean shouldProcess(TestStep step) {
     StepPhase stepPhase = step.getStepPhase();
     switch(stepPhase) {
       case DEFAULT:
-        return this == RAMPUP;
+        return this == RAMPUP || stepPhase == this;
       case RAMPUP:
       case LOADREACHED:
         return stepPhase == this;