regexextractorgui.java

测试工具

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * 
 */

package org.apache.jmeter.extractor.gui;

import java.awt.BorderLayout;
import java.awt.Component;
import java.awt.GridBagConstraints;
import java.awt.GridBagLayout;
import java.util.List;

import javax.swing.BorderFactory;
import javax.swing.Box;
import javax.swing.ButtonGroup;
import javax.swing.JPanel;
import javax.swing.JRadioButton;

import org.apache.jmeter.extractor.RegexExtractor;
import org.apache.jmeter.processor.gui.AbstractPostProcessorGui;
import org.apache.jmeter.testelement.TestElement;
import org.apache.jmeter.util.JMeterUtils;
import org.apache.jorphan.gui.JLabeledTextField;

/**
 * Regular Expression Extractor Post-Processor GUI
 */
public class RegexExtractorGui extends AbstractPostProcessorGui {
	private JLabeledTextField regexField;

	private JLabeledTextField templateField;

	private JLabeledTextField defaultField;

	private JLabeledTextField matchNumberField;

	private JLabeledTextField refNameField;

	private JRadioButton useBody;

	private JRadioButton useHeaders;
    
    private JRadioButton useURL;

    private JRadioButton useCode;

    private JRadioButton useMessage;

	private ButtonGroup group;

	public RegexExtractorGui() {
		super();
		init();
	}

	public String getLabelResource() {
		return "regex_extractor_title"; //$NON-NLS-1$
	}

	public void configure(TestElement el) {
		super.configure(el);
		if (el instanceof RegexExtractor){
			RegexExtractor re = (RegexExtractor) el;
			useHeaders.setSelected(re.useHeaders());
			useBody.setSelected(re.useBody());
	        useURL.setSelected(re.useUrl());
	        useCode.setSelected(re.useCode());
	        useMessage.setSelected(re.useMessage());
			regexField.setText(re.getRegex());
			templateField.setText(re.getTemplate());
			defaultField.setText(re.getDefaultValue());
			matchNumberField.setText(re.getMatchNumberAsString());
			refNameField.setText(re.getRefName());
		}
	}

	/**
	 * @see org.apache.jmeter.gui.JMeterGUIComponent#createTestElement()
	 */
	public TestElement createTestElement() {
		RegexExtractor extractor = new RegexExtractor();
		modifyTestElement(extractor);
		return extractor;
	}

	/**
	 * Modifies a given TestElement to mirror the data in the gui components.
	 * 
	 * @see org.apache.jmeter.gui.JMeterGUIComponent#modifyTestElement(TestElement)
	 */
	public void modifyTestElement(TestElement extractor) {
		super.configureTestElement(extractor);
		if (extractor instanceof RegexExtractor) {
			RegexExtractor regex = (RegexExtractor) extractor;
			regex.setUseField(group.getSelection().getActionCommand());			
			regex.setRefName(refNameField.getText());
			regex.setRegex(regexField.getText());
			regex.setTemplate(templateField.getText());
			regex.setDefaultValue(defaultField.getText());
			regex.setMatchNumber(matchNumberField.getText());
		}
	}
    
    /**
     * Implements JMeterGUIComponent.clearGui
     */
    public void clearGui() {
        super.clearGui();
        
        useBody.setSelected(true);
        
        regexField.setText(""); //$NON-NLS-1$
        templateField.setText(""); //$NON-NLS-1$
        defaultField.setText(""); //$NON-NLS-1$
        refNameField.setText(""); //$NON-NLS-1$
        matchNumberField.setText(""); //$NON-NLS-1$
    }    

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

		Box box = Box.createVerticalBox();
		box.add(makeTitlePanel());
		box.add(makeSourcePanel());
		add(box, BorderLayout.NORTH);
		add(makeParameterPanel(), BorderLayout.CENTER);
	}

	private JPanel makeSourcePanel() {
		JPanel panel = new JPanel();
		panel.setBorder(BorderFactory.createTitledBorder(JMeterUtils.getResString("regex_source"))); //$NON-NLS-1$

		useBody = new JRadioButton(JMeterUtils.getResString("regex_src_body")); //$NON-NLS-1$
		useHeaders = new JRadioButton(JMeterUtils.getResString("regex_src_hdrs")); //$NON-NLS-1$
        useURL = new JRadioButton(JMeterUtils.getResString("regex_src_url")); //$NON-NLS-1$
        useCode = new JRadioButton(JMeterUtils.getResString("assertion_code_resp")); //$NON-NLS-1$
        useMessage = new JRadioButton(JMeterUtils.getResString("assertion_message_resp")); //$NON-NLS-1$

		group = new ButtonGroup();
		group.add(useBody);
		group.add(useHeaders);
        group.add(useURL);
        group.add(useCode);
        group.add(useMessage);

		panel.add(useBody);
		panel.add(useHeaders);
        panel.add(useURL);
        panel.add(useCode);
        panel.add(useMessage);

		useBody.setSelected(true);
		
		// So we know which button is selected
		useBody.setActionCommand(RegexExtractor.USE_BODY);
		useHeaders.setActionCommand(RegexExtractor.USE_HDRS);
		useURL.setActionCommand(RegexExtractor.USE_URL);
		useCode.setActionCommand(RegexExtractor.USE_CODE);
		useMessage.setActionCommand(RegexExtractor.USE_MESSAGE);
		
		return panel;
	}

	private JPanel makeParameterPanel() {
		regexField = new JLabeledTextField(JMeterUtils.getResString("regex_field")); //$NON-NLS-1$
		templateField = new JLabeledTextField(JMeterUtils.getResString("template_field")); //$NON-NLS-1$
		defaultField = new JLabeledTextField(JMeterUtils.getResString("default_value_field")); //$NON-NLS-1$
		refNameField = new JLabeledTextField(JMeterUtils.getResString("ref_name_field")); //$NON-NLS-1$
		matchNumberField = new JLabeledTextField(JMeterUtils.getResString("match_num_field")); //$NON-NLS-1$

		JPanel panel = new JPanel(new GridBagLayout());
		GridBagConstraints gbc = new GridBagConstraints();
		initConstraints(gbc);
		addField(panel, refNameField, gbc);
		resetContraints(gbc);
		addField(panel, regexField, gbc);
		resetContraints(gbc);
		addField(panel, templateField, gbc);
		resetContraints(gbc);
		addField(panel, matchNumberField, gbc);
		resetContraints(gbc);
		gbc.weighty = 1;
		addField(panel, defaultField, gbc);
		return panel;
	}

	private void addField(JPanel panel, JLabeledTextField field, GridBagConstraints gbc) {
		List item = field.getComponentList();
		panel.add((Component) item.get(0), gbc.clone());
		gbc.gridx++;
		gbc.weightx = 1;
		gbc.fill=GridBagConstraints.HORIZONTAL;
		panel.add((Component) item.get(1), gbc.clone());
	}

	// Next line
	private void resetContraints(GridBagConstraints gbc) {
		gbc.gridx = 0;
		gbc.gridy++;
		gbc.weightx = 0;
        gbc.fill=GridBagConstraints.NONE;
	}

	private void initConstraints(GridBagConstraints gbc) {
		gbc.anchor = GridBagConstraints.NORTHWEST;
		gbc.fill = GridBagConstraints.NONE;
		gbc.gridheight = 1;
		gbc.gridwidth = 1;
		gbc.gridx = 0;
		gbc.gridy = 0;
		gbc.weightx = 0;
		gbc.weighty = 0;
	}
}