import org.apache.jorphan.collections.HashTree; //导入依赖的package包/类
@Test
public void loadControl1() throws Exception {
   log.info("start loadControl1()");

   // Initialize Properties, logging, locale, etc.
   JMeterUtils.loadJMeterProperties(JMETER_HOME + "/bin/jmeter.properties");
   JMeterUtils.setJMeterHome(JMETER_HOME);
   JMeterUtils.initLocale();

   // Initialize JMeter SaveService
   SaveService.loadProperties();

   // Load existing .jmx Test Plan
   ClassLoader classLoader = Thread.currentThread().getContextClassLoader();
   URL url = classLoader.getResource("tutorial/jmeter/Tutorial6.jmx");
   File file = new File(url.toURI());
   HashTree testPlanTree = SaveService.loadTree(file);

   // Run JMeter Test
   StandardJMeterEngine jmeter = new StandardJMeterEngine();
   jmeter.configure(testPlanTree);
   jmeter.run();
}
 

import org.apache.jorphan.collections.HashTree; //导入依赖的package包/类
public void start() {
    log.debug("Start debugging");
    frontend.started();

    HashTree hashTree = getSelectedTree();
    StandardJMeterEngine.register(new StateListener()); // oh, dear, they use static field then clean it...
    engine = new DebuggerEngine(JMeterContextService.getContext());
    engine.setStepper(this);
    JMeter.convertSubTree(hashTree);
    engine.configure(hashTree);
    try {
        engine.runTest();
    } catch (JMeterEngineException e) {
        log.error("Failed to pauseContinue debug", e);
        stop();
    }
}
 

import org.apache.jorphan.collections.HashTree; //导入依赖的package包/类
@Test
public void runRealEngine() throws Exception {
    TestTreeProvider prov = new TestProvider();

    HashTree hashTree = prov.getTestTree();
    JMeter.convertSubTree(hashTree);

    StandardJMeterEngine engine = new StandardJMeterEngine();
    engine.configure(hashTree);
    engine.runTest();
    while (engine.isActive()) {
        Thread.sleep(1000);
    }
}
 

import org.apache.jorphan.collections.HashTree; //导入依赖的package包/类
/**
 * Start JMeter engine
 * @param noTimer ignore timers 
 */
private void startEngine(boolean ignoreTimer) {
    GuiPackage gui = GuiPackage.getInstance();
    HashTree testTree = gui.getTreeModel().getTestPlan();
    JMeter.convertSubTree(testTree);
    testTree.add(testTree.getArray()[0], gui.getMainFrame());
    log.debug("test plan before cloning is running version: "
            + ((TestPlan) testTree.getArray()[0]).isRunningVersion());
    TreeCloner cloner = cloneTree(testTree, ignoreTimer);
    engine = new StandardJMeterEngine();
    engine.configure(cloner.getClonedTree());
    try {
        engine.runTest();
    } catch (JMeterEngineException e) {
        JOptionPane.showMessageDialog(gui.getMainFrame(), e.getMessage(), 
                JMeterUtils.getResString("error_occurred"), JOptionPane.ERROR_MESSAGE); //$NON-NLS-1$
    }
    log.debug("test plan after cloning and running test is running version: "
            + ((TestPlan) testTree.getArray()[0]).isRunningVersion());
}
 

import org.apache.jorphan.collections.HashTree; //导入依赖的package包/类
@Test
public void runDebugEngine() throws Exception {
    TestProvider prov = new TestProvider();

    Debugger sel = new Debugger(prov, new FrontendMock());
    AbstractThreadGroup tg = prov.getTG(0);
    sel.selectThreadGroup(tg);
    HashTree testTree = sel.getSelectedTree();

    DebuggingThreadGroup tg2 = (DebuggingThreadGroup) getFirstTG(testTree);
    LoopController samplerController = (LoopController) tg2.getSamplerController();
    samplerController.setLoops(1);
    samplerController.setContinueForever(false);

    JMeter.convertSubTree(testTree);

    DebuggerEngine engine = new DebuggerEngine(JMeterContextService.getContext());
    StepTriggerCounter hook = new StepTriggerCounter();
    engine.setStepper(hook);
    engine.configure(testTree);
    engine.runTest();
    while (engine.isActive()) {
        Thread.sleep(1000);
    }
    assertEquals(88, hook.cnt);
}
 

import org.apache.jorphan.collections.HashTree; //导入依赖的package包/类
@Test
public void testThreadStopping() throws Exception {
    DirectoryListingConfig config = new DirectoryListingConfig();

    File rootDir = TestDirectoryListingConfigActionTest.createFileTree();

    setDirectoryConfig(config, rootDir.getAbsolutePath(), VARIABLE_NAME, true, false, false, false, true, false);

    final HashTree hashTree = new HashTree();
    hashTree.add(new LoopController());

    JMeterThread thread = new JMeterThread(hashTree, null, null);
    JMeterContextService.getContext().setThread(thread);

    testFlow(config);

    try {
        config.iterationStart(null);
    } catch (JMeterStopThreadException ex) {
        assertEquals("All files in the directory have been passed.", ex.getMessage());
    }
}
 

import org.apache.jorphan.collections.HashTree; //导入依赖的package包/类
private void selectThreadGroup(AbstractThreadGroup tg) {
    debugger.selectThreadGroup(tg);
    treeModel.clearTestPlan();
    HashTree origTree = debugger.getSelectedTree();
    TreeCloner cloner = new TreeCloner();
    origTree.traverse(cloner);
    HashTree selectedTree = cloner.getClonedTree();

    // Hack to resolve ModuleControllers from JMeter.java
    SearchClass<ReplaceableController> replaceableControllers = new SearchClass<>(ReplaceableController.class);
    selectedTree.traverse(replaceableControllers);
    Collection<ReplaceableController> replaceableControllersRes = replaceableControllers.getSearchResults();
    for (ReplaceableController replaceableController : replaceableControllersRes) {
        replaceableController.resolveReplacementSubTree((JMeterTreeNode) treeModel.getRoot());
    }

    JMeter.convertSubTree(selectedTree);
    try {
        treeModel.addSubTree(selectedTree, (JMeterTreeNode) treeModel.getRoot());
    } catch (IllegalUserActionException e) {
        throw new RuntimeException(e);
    }

    // select TG for visual convenience
    SearchByClass<DebuggingThreadGroup> tgs = new SearchByClass<>(DebuggingThreadGroup.class);
    selectedTree.traverse(tgs);
    for (DebuggingThreadGroup forSel : tgs.getSearchResults()) {
        Wrapper<AbstractThreadGroup> wtg = new ThreadGroupWrapper();
        wtg.setWrappedElement(forSel);
        selectTargetInTree(wtg);
    }
}
 

import org.apache.jorphan.collections.HashTree; //导入依赖的package包/类
@Test
public void runVariablesDebugEngine() throws Exception {
    TestProvider prov = new TestProvider("/com/blazemeter/jmeter/debugger/vars.jmx", "vars.jmx");

    Debugger sel = new Debugger(prov, new FrontendMock());
    AbstractThreadGroup tg = prov.getTG(0);
    sel.selectThreadGroup(tg);
    HashTree testTree = sel.getSelectedTree();

    TestSampleListener listener = new TestSampleListener();
    testTree.add(testTree.getArray()[0], listener);

    DebuggingThreadGroup tg2 = (DebuggingThreadGroup) getFirstTG(testTree);
    LoopController samplerController = (LoopController) tg2.getSamplerController();
    samplerController.setLoops(1);
    samplerController.setContinueForever(false);

    JMeter.convertSubTree(testTree);

    DebuggerEngine engine = new DebuggerEngine(JMeterContextService.getContext());
    StepTriggerCounter hook = new StepTriggerCounter();
    engine.setStepper(hook);
    engine.configure(testTree);
    engine.runTest();
    while (engine.isActive()) {
        Thread.sleep(1000);
    }
    assertEquals(8, hook.cnt);

    assertEquals(3, listener.events.size());
    for (SampleEvent event : listener.events) {
        SampleResult res = event.getResult();
        String label = res.getSampleLabel();
        assertTrue("Label: " + label + " must end with '123'", label.endsWith("123"));
        assertFalse("Variable ${VAR} must be changed to '123' value. label: " + label, label.contains("${VAR}"));
        assertTrue("label: '" + label + "' response: '" + res.getResponseMessage() +"'", res.isSuccessful());
    }
}
 

import org.apache.jorphan.collections.HashTree; //导入依赖的package包/类
@Test
public void runVariablesInAssertions() throws Exception {
    TestProvider prov = new TestProvider("/com/blazemeter/jmeter/debugger/debug.jmx", "debug.jmx");

    Debugger sel = new Debugger(prov, new FrontendMock());
    AbstractThreadGroup tg = prov.getTG(0);
    sel.selectThreadGroup(tg);
    HashTree testTree = sel.getSelectedTree();

    TestSampleListener listener = new TestSampleListener();
    testTree.add(testTree.getArray()[0], listener);

    DebuggingThreadGroup tg2 = (DebuggingThreadGroup) getFirstTG(testTree);
    LoopController samplerController = (LoopController) tg2.getSamplerController();
    samplerController.setLoops(1);
    samplerController.setContinueForever(false);

    JMeter.convertSubTree(testTree);

    DebuggerEngine engine = new DebuggerEngine(JMeterContextService.getContext());
    StepTriggerCounter hook = new StepTriggerCounter();
    engine.setStepper(hook);
    engine.configure(testTree);
    engine.runTest();
    while (engine.isActive()) {
        Thread.sleep(1000);
    }
    assertEquals(4, hook.cnt);

    assertEquals(1, listener.events.size());
    SampleEvent event = listener.events.get(0);
    SampleResult result = event.getResult();
    AssertionResult[] assertionResults = result.getAssertionResults();
    assertEquals(1, assertionResults.length);

    AssertionResult assertionRes = assertionResults[0];
    assertNull(assertionRes.getFailureMessage());
}
 

import org.apache.jorphan.collections.HashTree; //导入依赖的package包/类
@Test
public void runVariablesInControllers() throws Exception {
    TestProvider prov = new TestProvider("/com/blazemeter/jmeter/debugger/loops.jmx", "loops.jmx");

    Debugger sel = new Debugger(prov, new FrontendMock());
    AbstractThreadGroup tg = prov.getTG(0);
    sel.selectThreadGroup(tg);
    HashTree testTree = sel.getSelectedTree();

    TestSampleListener listener = new TestSampleListener();
    testTree.add(testTree.getArray()[0], listener);

    DebuggingThreadGroup tg2 = (DebuggingThreadGroup) getFirstTG(testTree);
    LoopController samplerController = (LoopController) tg2.getSamplerController();
    samplerController.setLoops(1);
    samplerController.setContinueForever(false);

    JMeter.convertSubTree(testTree);

    DebuggerEngine engine = new DebuggerEngine(JMeterContextService.getContext());
    StepTriggerCounter hook = new StepTriggerCounter();
    engine.setStepper(hook);
    engine.configure(testTree);
    engine.runTest();
    while (engine.isActive()) {
        Thread.sleep(1000);
    }
    assertEquals(12, hook.cnt);

    assertEquals(3, listener.events.size());
}
 

import org.apache.jorphan.collections.HashTree; //导入依赖的package包/类
/**
 * Extract from tree (included test plan) all Test Elements located in a Test Fragment
 * @param tree HashTree included Test Plan
 * @return HashTree Subset within Test Fragment or Empty HashTree
 */
private HashTree getProperBranch(HashTree tree) {
    for (Object o : new LinkedList<>(tree.list())) {
        TestElement item = (TestElement) o;

        //if we found a TestPlan, then we are on our way to the TestFragment
        if (item instanceof TestPlan)
        {
            return getProperBranch(tree.getTree(item));
        }

        if (item instanceof TestFragmentController)
        {
            return tree.getTree(item);
        }
    }
    log.warn("No Test Fragment was found in included Test Plan, returning empty HashTree");
    return new HashTree();
}
 

import org.apache.jorphan.collections.HashTree; //导入依赖的package包/类
@RequestMapping(value = "/testplan", method = RequestMethod.GET, produces = "application/json")
 public @ResponseBody TestPlanModel getTestPlan(@QueryParam(value = "name") String name){
 
 	if(name == null || name.isEmpty()){
 		return new TestPlanModel();
 	}else{
 		File testplan = new File("./testplans/" + name);
try {
	HashTree tree = SaveService.loadTree(testplan);
	return new TestPlanModel(tree);
	
} catch (IOException e) {
	log.error("Couldn't read file for tesplan {}. Message is {}", testplan, e.getMessage());
}
 		
 		return new TestPlanModel();
 	}
 	
 }
 

import org.apache.jorphan.collections.HashTree; //导入依赖的package包/类
public TestElementModel newElementByBeanName(String name) {
	if (name == null)
		return null;
	
	try {
		TestElementModel model = new TestElementModel();
		
		TestElement ele = (TestElement) context.getBean(name);
		model.setTestElement(ele);
		//model.setType(TestElementModel.getElementType(ele));
		model.setHashTree(new HashTree());
		
		return model;
					
	} catch (Exception e) {
		log.error("Couldn't initialize element {} because of exception {} reason {}", 
				name,e.getClass().getName(), e.getMessage());
		
		return null;
	}
	
	
}
 

import org.apache.jorphan.collections.HashTree; //导入依赖的package包/类
/**
 * {@inheritDoc}
 */
@Override
public Object unmarshal(HierarchicalStreamReader reader, UnmarshallingContext context) {
    ScriptWrapper wrap = new ScriptWrapper();
    wrap.version = reader.getAttribute(ATT_VERSION);
    ConversionHelp.setInVersion(wrap.version);// Make sure decoding
                                                // follows input file
    reader.moveDown();
    // Catch errors and rethrow as ConversionException so we get location details
    try {
        wrap.testPlan = (HashTree) context.convertAnother(wrap, getNextType(reader));
    } catch (NoClassDefFoundError | Exception e) {
        throw createConversionException(e);
    }
    return wrap;
}
 

import org.apache.jorphan.collections.HashTree; //导入依赖的package包/类
/** {@inheritDoc} */
@Override
public Object unmarshal(HierarchicalStreamReader reader, UnmarshallingContext context) {
    boolean isKey = true;
    Object current = null;
    HashTree tree = (HashTree) createCollection(context.getRequiredType());
    while (reader.hasMoreChildren()) {
        reader.moveDown();
        Object item = readItem(reader, context, tree);
        if (isKey) {
            tree.add(item);
            current = item;
            isKey = false;
        } else {
            tree.set(current, (HashTree) item);
            isKey = true;
        }
        reader.moveUp();
    }
    return tree;
}
 

import org.apache.jorphan.collections.HashTree; //导入依赖的package包/类
/**
 * Adds a feature to the ThreadGroup attribute of the RemoteJMeterEngineImpl
 * object.
 *
 * @param testTree
 *            the feature to be added to the ThreadGroup attribute
 */
@Override
public void rconfigure(HashTree testTree, String host, File jmxBase, String scriptName) throws RemoteException {
    log.info("Creating JMeter engine on host "+host+" base '"+jmxBase+"'");
    try {
        log.info("Remote client host: " + getClientHost());
    } catch (ServerNotActiveException e) {
        // ignored
    }
    synchronized(LOCK) { // close window where another remote client might jump in
        if (backingEngine != null && backingEngine.isActive()) {
            log.warn("Engine is busy - cannot create JMeter engine");
            throw new IllegalStateException("Engine is busy - please try later");
        }
        ownerThread = Thread.currentThread();
        backingEngine = new StandardJMeterEngine(host);
        backingEngine.configure(testTree); // sets active = true
    }
    FileServer.getFileServer().setScriptName(scriptName);
    FileServer.getFileServer().setBase(jmxBase);
}
 

import org.apache.jorphan.collections.HashTree; //导入依赖的package包/类
private JMeterEngine getClientEngine(String address, HashTree testTree) {
    JMeterEngine engine;
    try {
        engine = createEngine(address);
        engine.configure(testTree);
        if (!remoteProps.isEmpty()) {
            engine.setProperties(remoteProps);
        }
        return engine;
    } catch (Exception ex) {
        log.error("Failed to create engine at " + address, ex);
        JMeterUtils.reportErrorToUser(ex.getMessage(),
                JMeterUtils.getResString("remote_error_init") + ": " + address); // $NON-NLS-1$ $NON-NLS-2$
        return null;
    }
}
 

import org.apache.jorphan.collections.HashTree; //导入依赖的package包/类
@Override
public void addNode(Object node, HashTree subTree) {
	log.debug("Adding node {} to tree.", node.getClass().getSimpleName());			

	try {
		
		jsonGen.writeStartObject();
		jsonGen.writeObjectField("testElement", node);
		jsonGen.writeStringField("type", TestElementModel.getElementType(node).toString());
		jsonGen.writeArrayFieldStart("hashTree");
						
	} catch (IOException e) {
		
		log.error("Cannot add node {}. Exception type: {}, message {}.",
				node.getClass().getSimpleName(), e.getClass().getName(), e.getMessage());
	}			

}
 

import org.apache.jorphan.collections.HashTree; //导入依赖的package包/类
public void saveScenario(OutputStream outStream, int recordingIndex) throws IOException
{
    LOG.info("Start {} scenario saving", recordingIndex);
    LOG.info("Cloning template tree");
    HashTree hashTree = (HashTree)mainHashTreeTemplate.clone();
    LOG.info("Searching for main nodes and trees");
    findMainNodesAndTrees(hashTree);

    RecordingController recordingController = recordingControllers.get(recordingIndex);
    HashTree recordingControllerSubTree = transactionControllerSubTree.add(recordingController);

    LOG.info("Extracting test elements");
    List<TestElement> samples = extractAppropriateTestElements(recordingController);

    LOG.info("Placing test elements");
    placeAndProcessTestElements(recordingControllerSubTree, samples);
    LOG.info("Saving into out stream");
    SaveService.saveTree(hashTree, outStream);
}
 

import org.apache.jorphan.collections.HashTree; //导入依赖的package包/类
private void placeAndProcessTestElements(HashTree hashTree, List<TestElement> samples)
{
    for (TestElement element : samples)
    {
        List<TestElement> descendants = findAndRemoveHeaderManagers(element);
        HashTree parent = hashTree.add(element);
        descendants.forEach(parent::add);

        if (element instanceof HTTPSamplerBase)
        {
            HTTPSamplerBase http = (HTTPSamplerBase)element;

            LOG.info("Start sampler processing");
            scriptProcessor.processScenario(http, parent, vars, this);
            LOG.info("Stop sampler processing");
        }
    }
}