kg.apc.jmeter.JMeterPluginsUtils Java Examples

private void trySettingLoadFromProperty() {
    String loadProp = JMeterUtils.getProperty(DATA_PROPERTY);
    log.debug("Loading property: {}={}", DATA_PROPERTY, loadProp);
    if (!StringUtils.isEmpty(loadProp)) {
        log.info("GUI load profile will be ignored as property {} is defined", DATA_PROPERTY);
        PowerTableModel dataModel = new PowerTableModel(VariableThroughputTimer.columnIdentifiers, VariableThroughputTimer.columnClasses);

        String[] chunks = loadProp.split("\\)");

        for (String chunk : chunks) {
            try {
                parseChunk(chunk, dataModel);
            } catch (RuntimeException e) {
                log.warn("Wrong load chunk {} will be ignored", chunk, e);
            }
        }

        log.info("Setting load profile from property {}: {}", DATA_PROPERTY, loadProp);
        overrideProp = JMeterPluginsUtils.tableModelRowsToCollectionProperty(dataModel, VariableThroughputTimer.DATA_PROPERTY);
    }
}
 

@Test
public void testScheduleThread() {
    System.out.println("scheduleThread");
    HashTree hashtree = new HashTree();
    hashtree.add(new LoopController());
    JMeterThread thread = new JMeterThread(hashtree, null, null);

    CollectionProperty prop = JMeterPluginsUtils.tableModelRowsToCollectionProperty(dataModel, UltimateThreadGroup.DATA_PROPERTY);
    instance.setData(prop);
    instance.testStarted();

    instance.scheduleThread(thread);

    assertTrue(thread.getStartTime() > 0);
    assertTrue(thread.getEndTime() > thread.getStartTime());
}
 

private void init() {
    setLayout(new BorderLayout(0, 5));
    setBorder(makeBorder());

    Container topPanel = makeTitlePanel();

    add(JMeterPluginsUtils.addHelpLinkToPanel(topPanel, WIKIPAGE), BorderLayout.NORTH);
    add(topPanel, BorderLayout.NORTH);

    JPanel mainPanel = new HorizontalPanel();

    mainPanel.setBorder(BorderFactory.createTitledBorder("File Rotating Rules"));
    mainPanel.add(new JLabel("Limit samples count in file: ", JLabel.RIGHT));
    maxSamplesCount = new JTextField(20);
    mainPanel.add(maxSamplesCount);

    topPanel.add(mainPanel);
}
 

@Override
public void configure(TestElement tg) {
    super.configure(tg);
    ParallelHTTPSampler utg = (ParallelHTTPSampler) tg;
    JMeterProperty threadValues = utg.getData();
    if (threadValues instanceof NullProperty) {
        log.warn("Received null property instead of collection");
        return;
    }

    CollectionProperty columns = (CollectionProperty) threadValues;

    tableModel.removeTableModelListener(this);
    JMeterPluginsUtils.collectionPropertyToTableModelRows(columns, tableModel);
    tableModel.addTableModelListener(this);
    buttons.checkDeleteButtonStatus();
    updateUI();
}
 

/**
 * Modifies a given TestElement to mirror the data in the gui components.
 *
 * @see org.apache.jmeter.gui.JMeterGUIComponent#modifyTestElement(TestElement)
 */
@Override
public void modifyTestElement(TestElement c) {
    super.modifyTestElement(c);
    if (c instanceof ResultCollector) {
        ResultCollector rc = (ResultCollector) c;
        rc.setFilename(getFile());

        rc.setProperty(new StringProperty(
                CorrectedResultCollector.FILENAME, getFile()));
        CollectionProperty rows = JMeterPluginsUtils
                .tableModelRowsToCollectionProperty(tableModel,
                        DATA_PROPERTY);
        rc.setProperty(rows);
        collector = rc;
    }
}
 

private void init() {
    setLayout(new BorderLayout(0, 5));
    setBorder(makeBorder());
    add(JMeterPluginsUtils.addHelpLinkToPanel(makeTitlePanel(), WIKIPAGE), BorderLayout.NORTH);

    Container topPanel = makeTitlePanel();

    add(topPanel, BorderLayout.NORTH);

    JPanel mainPanel = new HorizontalPanel();

    JLabel lbl = new JLabel(MSG, JLabel.CENTER);
    topPanel.add(lbl);

    generateParentSamples = new JCheckBox();
    mainPanel.add(generateParentSamples);
    mainPanel.add(new JLabel("Generate parent sample", JLabel.RIGHT));

    topPanel.add(mainPanel);
}
 

@Override
public void configure(TestElement el) {
    super.configure(el);
    setFile(el.getPropertyAsString(CorrectedResultCollector.FILENAME));
    JMeterProperty fileValues = el.getProperty(DATA_PROPERTY);

    if (!(fileValues instanceof NullProperty)) {
        CollectionProperty columns = (CollectionProperty) fileValues;

        tableModel.removeTableModelListener(this);
        JMeterPluginsUtils.collectionPropertyToTableModelRows(columns,
                tableModel, columnClasses);
        tableModel.addTableModelListener(this);
        updateUI();
    } else {
        log.warn("Received null property instead of collection");
    }
    checkDeleteButtonStatus();
    checkMergeButtonStatus();

    startTimeRef = 0;
}
 

@Test
public void testScheduleThreadAll() {
    System.out.println("scheduleThreadAll");
    HashTree hashtree = new HashTree();
    hashtree.add(new LoopController());

    CollectionProperty prop = JMeterPluginsUtils.tableModelRowsToCollectionProperty(dataModel, UltimateThreadGroup.DATA_PROPERTY);
    instance.setData(prop);
    instance.testStarted();

    for (int n = 0; n < instance.getNumThreads(); n++) {
        JMeterThread thread = new JMeterThread(hashtree, null, null);
        thread.setThreadNum(n);
        instance.scheduleThread(thread);
    }
}
 

private static void parseChunk(String chunk, PowerTableModel model) {
    log.debug("Parsing chunk: " + chunk);
    String[] parts = chunk.split("[(,]");
    String loadVar = parts[0].trim();

    if (loadVar.equalsIgnoreCase("spawn")) {
        Integer[] row = new Integer[5];
        row[START_THREADS_CNT_FIELD_NO] = Integer.parseInt(parts[1].trim());
        row[INIT_DELAY_FIELD_NO] = JMeterPluginsUtils.getSecondsForShortString(parts[2]);
        row[STARTUP_TIME_FIELD_NO] = JMeterPluginsUtils.getSecondsForShortString(parts[3]);
        row[HOLD_LOAD_FOR_FIELD_NO] = JMeterPluginsUtils.getSecondsForShortString(parts[4]);
        row[SHUTDOWN_TIME_FIELD_NO] = JMeterPluginsUtils.getSecondsForShortString(parts[5]);
        model.addRow(row);
    } else {
        throw new RuntimeException("Unknown load type: " + parts[0]);
    }
}
 

private void createGui() {
    setLayout(new BorderLayout(0, 5));

    setBorder(makeBorder());
    add(JMeterPluginsUtils.addHelpLinkToPanel(makeTitlePanel(), getWikiPage()), BorderLayout.NORTH);

    final JTabbedPane tabbedPane = new JTabbedPane();
    if (isProxyEnabled()) {
        tabbedPane.add("Proxy", createProxyPanel());
    }
    tabbedPane.add(browserName(), createBrowserPanel());
    if (isExperimentalEnabled()) {
        tabbedPane.add("Experimental", createExperimentalPanel());
    }

    add(tabbedPane, BorderLayout.CENTER);
}
 

@Override
public TestElement createTestElement() {
    ResultCollector modelNew = getModel();
    if (modelNew == null) {
        modelNew = new CompositeResultCollector();
        ((CompositeResultCollector) modelNew).setCompositeModel(compositeModel);
        setModel(modelNew);
    }
    modifyTestElement(modelNew);
    modelNew.setComment(JMeterPluginsUtils.getWikiLinkText("CompositeGraph"));
    return modelNew;
}
 

@Override
public void updateUI() {
    super.updateUI();

    if (tableModel != null) {
        VariableThroughputTimer utgForPreview = new VariableThroughputTimer();
        utgForPreview.setData(JMeterPluginsUtils.tableModelRowsToCollectionPropertyEval(tableModel, VariableThroughputTimer.DATA_PROPERTY));
        updateChart(utgForPreview);
    }
}
 

@Test
public void testGetData_broken_rename() {
    CollectionProperty prop = JMeterPluginsUtils.tableModelRowsToCollectionProperty(dataModel, UltimateThreadGroup.EXTERNAL_DATA_PROPERTY);
    instance.setProperty(prop);
    JMeterProperty result = instance.getData();
    assertNotNull(instance.getProperty(UltimateThreadGroup.DATA_PROPERTY));
    assertTrue(instance.getProperty(UltimateThreadGroup.EXTERNAL_DATA_PROPERTY) instanceof NullProperty);
    assertFalse(result instanceof NullProperty);
    assertEquals(prop.getStringValue(), result.getStringValue());
}
 

@Override
public void configure(TestElement te) {
    super.configure(te);
    CorrectedResultCollector rc = (CorrectedResultCollector) te;
    JMeterProperty regexpValues = rc.getProperty(REGEXPS_PROPERTY);
    if (!(regexpValues instanceof NullProperty)) {
        JMeterPluginsUtils.collectionPropertyToTableModelRows((CollectionProperty) regexpValues, tableModel, columnClasses);
        regExps = (CollectionProperty) regexpValues;
    } else {
        log.warn("Received null property instead of collection");
    }
}
 

@Override
public void updateUI() {
    super.updateUI();

    if (tableModel != null) {
        CorrectedResultCollector crc = new CorrectedResultCollector();
        crc.setProperty(JMeterPluginsUtils
                .tableModelRowsToCollectionPropertyEval(tableModel,
                        DATA_PROPERTY));
    }
}
 

@Override
public void modifyTestElement(TestElement te) {
    super.modifyTestElement(te);
    if (grid.isEditing()) {
        grid.getCellEditor().stopCellEditing();
    }

    if (te instanceof PerfMonCollector) {
        PerfMonCollector pmte = (PerfMonCollector) te;
        CollectionProperty rows = JMeterPluginsUtils.tableModelRowsToCollectionProperty(tableModel, PerfMonCollector.DATA_PROPERTY);
        pmte.setData(rows);
    }
    super.configureTestElement(te);
}
 

@Override
public TestElement createTestElement() {
    UDPSampler sampler = new UDPSampler();
    modifyTestElement(sampler);
    sampler.setComment(JMeterPluginsUtils.getWikiLinkText(WIKIPAGE));
    return sampler;
}
 

@Override
public TestElement createTestElement() {
    XMLFormatPostProcessor te = new XMLFormatPostProcessor();
    modifyTestElement(te);
    te.setComment(JMeterPluginsUtils.getWikiLinkText(WIKIPAGE));
    return te;
}
 

public DummySubPostProcessorGui() {
    setLayout(new BorderLayout(0, 5));
    setBorder(makeBorder());

    add(JMeterPluginsUtils.addHelpLinkToPanel(makeTitlePanel(), WIKIPAGE), BorderLayout.NORTH);

    this.dummyPanel = new DummyPanel();
    add(dummyPanel.init(), BorderLayout.CENTER);
    dummyPanel.initFields();
}
 

@Override
public void modifyTestElement(TestElement te) {
    super.modifyTestElement(te);
    if (grid.isEditing()) {
        grid.getCellEditor().stopCellEditing();
    }

    if (te instanceof CorrectedResultCollector) {
        CorrectedResultCollector rc = (CorrectedResultCollector) te;
        CollectionProperty rows = JMeterPluginsUtils.tableModelRowsToCollectionProperty(tableModel, REGEXPS_PROPERTY);
        rc.setProperty(rows);
    }
}
 

private void init() {
    setLayout(new BorderLayout());
    setBorder(makeBorder());
    add(JMeterPluginsUtils.addHelpLinkToPanel(makeTitlePanel(), WIKIPAGE), BorderLayout.NORTH);

    add(createVariablePanel(), BorderLayout.CENTER);
}
 

@Test
public void testRun() throws InterruptedException {
    JMXMonCollector instance = new TestJMXMonCollector(this);
    instance.setData(JMeterPluginsUtils.tableModelRowsToCollectionProperty(dataModel, JMXMonCollector.DATA_PROPERTY));
    instance.testStarted();

    setQueryResult(ATTRIBUTE1, 1);
    setQueryResult(ATTRIBUTE2, 1);
    instance.processConnectors();
    assertLastSample(PROBE1, 1);
    assertEquals(null, latestSamples.get(PROBE2)); // Delta can not produce values at first loop

    setQueryResult(ATTRIBUTE1, -2);
    setQueryResult(ATTRIBUTE2, 2);
    instance.processConnectors();
    assertLastSample(PROBE1, -2);
    assertLastSample(PROBE2, 1);

    setQueryResult(ATTRIBUTE1, 13);
    setQueryResult(ATTRIBUTE2, 1);
    instance.processConnectors();
    assertLastSample(PROBE1, 13);
    assertLastSample(PROBE2, -1);

    instance.testEnded();
    assertSampleGeneratorThreadIsStoped();

}
 

/**
 * making this once to be efficient and avoid manipulating strings in switch
 * operators
 */
private void compileColumns() {
    log.debug("Compiling columns string: " + getColumns());
    String[] chunks = JMeterPluginsUtils.replaceRNT(getColumns()).split("\\|");
    log.debug("Chunks " + chunks.length);
    compiledFields = new int[chunks.length];
    compiledVars = new int[chunks.length];
    compiledConsts = new ByteBuffer[chunks.length];
    for (int n = 0; n < chunks.length; n++) {
        int fieldID = availableFieldNames.indexOf(chunks[n]);
        if (fieldID >= 0) {
            //log.debug(chunks[n] + " field id: " + fieldID);
            compiledFields[n] = fieldID;
        } else {
            compiledFields[n] = -1;
            compiledVars[n] = -1;
            if (chunks[n].contains(VAR_PREFIX)) {
                log.debug(chunks[n] + " is sample variable");
                String varN = chunks[n].substring(VAR_PREFIX.length());
                try {
                    compiledVars[n] = Integer.parseInt(varN);
                } catch (NumberFormatException e) {
                    log.error("Seems it is not variable spec: " + chunks[n]);
                    compiledConsts[n] = ByteBuffer.wrap(chunks[n].getBytes());
                }
            } else {
                log.debug(chunks[n] + " is const");
                if (chunks[n].length() == 0) {
                    //log.debug("Empty const, treated as |");
                    chunks[n] = "|";
                }

                compiledConsts[n] = ByteBuffer.wrap(chunks[n].getBytes());
            }
        }
    }
}
 

@Override
public void configure(TestElement te) {
    super.configure(te);
    MonitoringResultsCollector mrc = (MonitoringResultsCollector) te;
    JMeterProperty samplerValues = mrc.getSamplerSettings();
    if (!(samplerValues instanceof NullProperty)) {
        JMeterPluginsUtils.collectionPropertyToTableModelRows((CollectionProperty) samplerValues, tableModel, getColumnClasses());
    } else {
        log.warn("Received null property instead of collection");
    }
}
 

@Override
public void modifyTestElement(TestElement te) {
    super.modifyTestElement(te);
    if (grid.isEditing()) {
        grid.getCellEditor().stopCellEditing();
    }

    if (te instanceof MonitoringResultsCollector) {
        MonitoringResultsCollector mrc = (MonitoringResultsCollector) te;
        CollectionProperty rows = JMeterPluginsUtils.tableModelRowsToCollectionProperty(tableModel, MonitoringResultsCollector.DATA_PROPERTY);
        mrc.setData(rows);
    }
    super.configureTestElement(te);
}
 

private void initGui() {
    setBorder(makeBorder());
    setLayout(new BorderLayout());
    add(JMeterPluginsUtils.addHelpLinkToPanel(makeTitlePanel(), getWikiPage()), BorderLayout.NORTH);
    container = getGraphPanelContainer();
    container.add(createGraphPanel(), BorderLayout.CENTER);
    add(container, BorderLayout.CENTER);
    reloadLabelToColorMapping();
    addMouseClickColorChangeListener();
}
 

@Override
public void modifyTestElement(TestElement tg) {
    if (grid.isEditing()) {
        grid.getCellEditor().stopCellEditing();
    }

    if (tg instanceof UltimateThreadGroup) {
        UltimateThreadGroup utg = (UltimateThreadGroup) tg;
        CollectionProperty rows = JMeterPluginsUtils.tableModelRowsToCollectionProperty(tableModel, UltimateThreadGroup.DATA_PROPERTY);
        utg.setData(rows);
        utg.setSamplerController((LoopController) loopPanel.createTestElement());
    }
    super.configureTestElement(tg);
}
 

@Override
public TestElement createTestElement() {
    TestElement te = new AutoStop();
    modifyTestElement(te);
    te.setComment(JMeterPluginsUtils.getWikiLinkText(WIKIPAGE));
    return te;
}
 

protected void openFile() throws IOException {
    String filename = getFilename();
    FileOutputStream fos = new FileOutputStream(filename, !isOverwrite());
    fileChannel = fos.getChannel();

    String header = JMeterPluginsUtils.replaceRNT(getFileHeader());
    if (!header.isEmpty()) {
        syncWrite(ByteBuffer.wrap(header.getBytes()));
    }
}
 

@Test
public void testGetNumThreads() {
    System.out.println("getNumThreads");
    CollectionProperty prop = JMeterPluginsUtils.tableModelRowsToCollectionProperty(dataModel, UltimateThreadGroup.DATA_PROPERTY);
    instance.setData(prop);
    instance.testStarted();

    int expResult = 15;
    int result = instance.getNumThreads();
    assertEquals(expResult, result);
}