Merge pull request #296 from KopysovDB/master

Add random choice support for WeightedSwitchController

Author: sebastianlorenzo88

jmeter-java-dsl/src/main/java/us/abstracta/jmeter/javadsl/core/controllers/DslWeightedSwitchController.java

@@ -44,6 +44,7 @@
 public class DslWeightedSwitchController extends BaseController<DslWeightedSwitchController> {
 
   public static final long DEFAULT_WEIGHT = 100;
+  private boolean isRandomChoice = false;
 
   public DslWeightedSwitchController() {
     super("Weighted Switch Controller", WeightedSwitchControllerGui.class, new ArrayList<>());
@@ -80,6 +81,21 @@ public DslWeightedSwitchController child(long weight, DslSampler child) {
     return addWeightedChild(weight, child);
   }
 
+  /**
+   * Sets the random choice mode for child elements.
+   * <p>
+   * When random choice mode is enabled (randomChoice = true), the controller will select child elements
+   * randomly, according to their weights. When disabled (randomChoice = false), selection will occur
+   * sequentially based on weights.
+   *
+   * @param randomChoice true - to enable random selection, false - for sequential selection
+   * @return current controller instance for method chaining
+   */
+  public DslWeightedSwitchController randomChoice(boolean randomChoice){
+    this.isRandomChoice = randomChoice;
+    return this;
+  }
+
   private DslWeightedSwitchController addWeightedChild(long weight, ThreadGroupChild child) {
     children.add(new WeightedChild(weight, child));
     return this;
@@ -144,6 +160,7 @@ public HashTree buildTreeUnder(HashTree parent, BuildTreeContext context) {
       }
     }
     controller.setData(model);
+    controller.setIsRandomChoice(isRandomChoice);
     return ret;
   }
 

jmeter-java-dsl/src/test/java/us/abstracta/jmeter/javadsl/core/controllers/DslWeightedSwitchControllerTest.java

@@ -44,6 +44,29 @@ public void shouldOnlyExecuteGivenPercentOfTheTimesWhenInPlan() throws Exception
             new WeightedLabel(label3, weight2)));
   }
 
+  @Test
+  public void shouldHandleRandomChoiceWhenInPlan() throws Exception {
+    int threads = 1;
+    int iterations = 20;
+    long weight1 = 60;
+    long weight2 = 30;
+    String label3 = "sample3";
+
+    TestPlanStats stats = testPlan(
+            threadGroup(threads, iterations,
+                    weightedSwitchController()
+                            .child(weight1, httpSampler(SAMPLE_1_LABEL, wiremockUri))
+                            .children(httpSampler(SAMPLE_2_LABEL, wiremockUri))
+                            .child(weight2, httpSampler(label3, wiremockUri))
+                            .randomChoice(true)
+            )
+    ).run();
+
+    Map<String, Long> sampleCounts = buildSampleCountsMap(stats);
+    assertThat(sampleCounts.keySet()).containsExactlyInAnyOrder(SAMPLE_1_LABEL, SAMPLE_2_LABEL, label3);
+  }
+
+
   private Map<String, Long> buildSampleCountsMap(TestPlanStats stats) {
     return stats.labels().stream()
         .collect(Collectors.toMap(s -> s, s -> stats.byLabel(s).samplesCount()));