// $Header: /home/cvs/jakarta-jmeter/src/core/org/apache/jmeter/gui/action/Load.java,v 1.23 2004/02/13 02:21:36 sebb Exp $
/*
* Copyright 2001-2004 The Apache Software Foundation.
*
* Licensed 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.gui.action;
import java.awt.event.ActionEvent;
import java.io.File;
import java.io.FileInputStream;
import java.io.InputStream;
import java.util.HashSet;
import java.util.Set;
import javax.swing.JFileChooser;
import javax.swing.tree.TreePath;
import junit.framework.TestCase;
import org.apache.jmeter.exceptions.IllegalUserActionException;
import org.apache.jmeter.gui.GuiPackage;
import org.apache.jmeter.gui.tree.JMeterTreeNode;
import org.apache.jmeter.gui.util.FileDialoger;
import org.apache.jmeter.save.SaveService;
import org.apache.jmeter.testelement.TestPlan;
import org.apache.jmeter.util.JMeterUtils;
import org.apache.jorphan.collections.HashTree;
import org.apache.jorphan.logging.LoggingManager;
import org.apache.log.Logger;
/**
* @author Michael Stover
* @version $Revision: 1.23 $
*/
public class Load implements Command
{
transient private static Logger log = LoggingManager.getLoggerForClass();
private static Set commands = new HashSet();
static {
commands.add("open");
commands.add("merge");
}
public Load()
{
}
public Set getActionNames()
{
return commands;
}
public void doAction(ActionEvent e)
{
boolean merging = e.getActionCommand().equals("merge");
if (!merging)
{
ActionRouter.getInstance().doActionNow(
new ActionEvent(e.getSource(),e.getID(),"close"));
}
JFileChooser chooser =
FileDialoger.promptToOpenFile(new String[] { ".jmx" });
if (chooser == null)
{
return;
}
boolean isTestPlan = false;
InputStream reader = null;
File f = null;
try
{
f = chooser.getSelectedFile();
if (f != null)
{
if (merging){
log.info("Merging file: " + f);
} else {
log.info("Loading file: " + f);
}
reader = new FileInputStream(f);
HashTree tree = SaveService.loadSubTree(reader);
isTestPlan = insertLoadedTree(e.getID(), tree);
}
}
catch (Exception ex)
{
String msg = ex.getMessage();
if (msg == null)
{
msg = "Unexpected error - see log for details";
log.warn("Unexpected error", ex);
}
JMeterUtils.reportErrorToUser(msg);
}
finally
{
GuiPackage.getInstance().updateCurrentGui();
GuiPackage.getInstance().getMainFrame().repaint();
}
if (isTestPlan && f != null)
{
(
(Save) ActionRouter.getInstance().getAction(
"save",
"org.apache.jmeter.gui.action.Save")).setTestPlanFile(
f.getAbsolutePath());
}
}
/**
* Returns a boolean indicating whether the loaded tree was a full test plan
*/
public boolean insertLoadedTree(int id, HashTree tree)
throws Exception, IllegalUserActionException
{
//convertTree(tree);
if (tree == null)
{
throw new Exception("Error in TestPlan - see log file");
}
boolean isTestPlan = tree.getArray()[0] instanceof TestPlan;
HashTree newTree = GuiPackage.getInstance().addSubTree(tree);
GuiPackage.getInstance().updateCurrentGui();
GuiPackage.getInstance().getMainFrame().getTree().setSelectionPath(
new TreePath(((JMeterTreeNode) newTree.getArray()[0]).getPath()));
tree = GuiPackage.getInstance().getCurrentSubTree();
ActionRouter.getInstance().actionPerformed(
new ActionEvent(
tree.get(tree.getArray()[tree.size() - 1]),
id,
CheckDirty.SUB_TREE_LOADED));
return isTestPlan;
}
/**
*
* @version $Revision: 1.23 $ Last updated: $Date: 2004/02/13 02:21:36 $
*/
public static class Test extends TestCase
{
File testFile1,
testFile2,
testFile3,
testFile4,
testFile5,
testFile6,
testFile7,
testFile8,
testFile9,
testFile10,
testFile11,
testFile12,
testFile13;
static Load loader = new Load();
public Test(String name)
{
super(name);
}
public void setUp()
{
testFile1 = //TODO: not used - why?
new File(
System.getProperty("user.dir") + "/testfiles",
"Test Plan.jmx");
testFile2 = //TODO: not used - why?
new File(
System.getProperty("user.dir") + "/testfiles",
"Modification Manager.jmx");
testFile3 =
new File(
System.getProperty("user.dir") + "/testfiles",
"proxy.jmx");
testFile4 =
new File(
System.getProperty("user.dir") + "/testfiles",
"AssertionTestPlan.jmx");
testFile5 =
new File(
System.getProperty("user.dir") + "/testfiles",
"AuthManagerTestPlan.jmx");
testFile6 =
new File(
System.getProperty("user.dir") + "/testfiles",
"HeaderManagerTestPlan.jmx");
testFile7 =
new File(
System.getProperty("user.dir") + "/testfiles",
"InterleaveTestPlan.jmx");
testFile8 =
new File(
System.getProperty("user.dir") + "/testfiles",
"InterleaveTestPlan2.jmx");
testFile9 =
new File(
System.getProperty("user.dir") + "/testfiles",
"LoopTestPlan.jmx");
testFile10 =
new File(
System.getProperty("user.dir") + "/testfiles",
"OnceOnlyTestPlan.jmx");
testFile11 =
new File(
System.getProperty("user.dir") + "/testfiles",
"ProxyServerTestPlan.jmx");
testFile12 =
new File(
System.getProperty("user.dir") + "/testfiles",
"SimpleTestPlan.jmx");
testFile13 = //TODO: not used - why?
new File(
System.getProperty("user.dir") + "/testfiles",
"URLRewritingExample.jmx");
}
public void testFile3() throws Exception
{
assertTree(getTree(testFile3));
}
private void assertTree(HashTree tree) throws Exception
{
assertTrue(
tree.getArray()[0]
instanceof org.apache.jmeter.testelement.TestPlan);
}
public void testFile4() throws Exception
{
assertTree(getTree(testFile4));
}
public void testFile5() throws Exception
{
assertTree(getTree(testFile5));
}
public void testFile6() throws Exception
{
assertTree(getTree(testFile6));
}
public void testFile7() throws Exception
{
assertTree(getTree(testFile7));
}
public void testFile8() throws Exception
{
assertTree(getTree(testFile8));
}
public void testFile9() throws Exception
{
assertTree(getTree(testFile9));
}
public void testFile10() throws Exception
{
assertTree(getTree(testFile10));
}
public void testFile11() throws Exception
{
assertTree(getTree(testFile11));
}
public void testFile12() throws Exception
{
assertTree(getTree(testFile12));
}
private HashTree getTree(File f) throws Exception
{
FileInputStream reader = new FileInputStream(f);
HashTree tree = SaveService.loadSubTree(reader);
return tree;
}
}
}