Examples with TestSampler org.apache.jmeter.junit.stubs.TestSampler used on opensource projects

Example 46 with TestSampler

use of org.apache.jmeter.junit.stubs.TestSampler in project jmeter by apache.

the class TestOnceOnlyController method testInsideInnerLoop.

@Test
public void testInsideInnerLoop() throws Exception {
    // Test plan with OnlyOnceController inside inner loop
    // Set up the test plan
    LoopController controller = new LoopController();
    final int outerLoopCount = 4;
    controller.setLoops(outerLoopCount);
    // OnlyOnce samples
    OnceOnlyController sub_1 = new OnceOnlyController();
    sub_1.addTestElement(new TestSampler("one"));
    sub_1.addTestElement(new TestSampler("two"));
    controller.addTestElement(sub_1);
    controller.addIterationListener(sub_1);
    // Outer sample
    controller.addTestElement(new TestSampler("three"));
    // Inner loop
    LoopController sub_2 = new LoopController();
    final int innerLoopCount = 3;
    sub_2.setLoops(innerLoopCount);
    // Sample in inner loop
    sub_2.addTestElement(new TestSampler("four"));
    // OnlyOnce inside inner loop
    OnceOnlyController sub_3 = new OnceOnlyController();
    sub_3.addTestElement(new TestSampler("five"));
    sub_3.addTestElement(new TestSampler("six"));
    sub_2.addTestElement(sub_3);
    sub_2.addIterationListener(sub_3);
    // Sample in inner loop
    sub_2.addTestElement(new TestSampler("seven"));
    controller.addTestElement(sub_2);
    // Compute the expected sample names
    String[] onlyOnceOrder = new String[] { "one", "two" };
    String[] order = new String[] { "three", "four", "five", "six", "seven", "four", "seven", "four", "seven" };
    // Outer only once + ("three" + "only once five and six" + ("four" + "seven") * innerLoopCount) * outerLoopCount;
    int expectedNoSamples = 2 + (1 + 2 + (1 + 1) * innerLoopCount) * outerLoopCount;
    String[] expectedSamples = new String[expectedNoSamples];
    // The only once samples
    System.arraycopy(onlyOnceOrder, 0, expectedSamples, 0, onlyOnceOrder.length);
    // The outer sample and the inner loop samples
    final int onceOnlySamples = onlyOnceOrder.length;
    for (int i = 0; i < order.length * outerLoopCount; i++) {
        expectedSamples[onceOnlySamples + i] = order[i % order.length];
    }
    // Execute the test pan
    controller.setRunningVersion(true);
    sub_1.setRunningVersion(true);
    sub_2.setRunningVersion(true);
    sub_3.setRunningVersion(true);
    controller.initialize();
    int counter = 0;
    TestElement sampler = null;
    while ((sampler = controller.next()) != null) {
        assertEquals(expectedSamples[counter], sampler.getPropertyAsString(TestElement.NAME));
        counter++;
    }
    assertEquals(expectedNoSamples, counter);
}