| /* |
| * 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.netbeans.modules.testng.ant; |
| |
| import java.io.File; |
| import java.io.IOException; |
| import java.util.Arrays; |
| import java.util.Properties; |
| import java.util.Set; |
| import java.util.logging.Level; |
| import java.util.logging.Logger; |
| import javax.swing.event.ChangeListener; |
| import org.apache.tools.ant.module.api.AntTargetExecutor; |
| import org.apache.tools.ant.module.api.support.AntScriptUtils; |
| import org.apache.tools.ant.module.spi.AntSession; |
| import org.netbeans.api.project.Project; |
| import org.netbeans.api.project.ProjectManager; |
| import org.netbeans.modules.gsf.testrunner.api.RerunHandler; |
| import org.netbeans.modules.gsf.testrunner.api.RerunType; |
| import org.netbeans.modules.gsf.testrunner.api.TestSession; |
| import org.netbeans.modules.gsf.testrunner.api.Testcase; |
| import org.netbeans.modules.testng.api.TestNGSupport; |
| import org.netbeans.modules.testng.api.TestNGUtils; |
| import org.netbeans.modules.testng.spi.TestConfig; |
| import org.netbeans.modules.testng.spi.TestNGSupportImplementation; |
| import org.netbeans.spi.project.ActionProvider; |
| import org.netbeans.spi.project.SingleMethod; |
| import org.openide.filesystems.FileObject; |
| import org.openide.filesystems.FileUtil; |
| import org.openide.util.Lookup; |
| import org.openide.util.lookup.Lookups; |
| |
| /** |
| * |
| * @author answer |
| */ |
| public class TestNGExecutionManager implements RerunHandler { |
| |
| private File scriptFile = null; |
| private String[] targets = null; |
| private Properties properties; |
| private TestSession testSession; |
| private Lookup lookup = Lookup.EMPTY; |
| |
| private static final Logger LOGGER = Logger.getLogger(TestNGExecutionManager.class.getName()); |
| |
| public TestNGExecutionManager(AntSession session, TestSession testSession, Properties props) { |
| this.testSession = testSession; |
| this.properties = props; |
| properties.setProperty("ignore.failing.tests", "true"); //NOI18N |
| try { |
| scriptFile = session.getOriginatingScript(); |
| targets = session.getOriginatingTargets(); |
| |
| String testIncludes = properties.getProperty("test.includes");//NOI18N |
| if (testIncludes != null) { |
| FileObject testFO = testSession.getFileLocator().find(testIncludes); |
| lookup = Lookups.fixed(testFO); |
| } |
| |
| //"Run/Debug Focused Test Method" actions (test-method/debug-test-single-nb) in a nb module project |
| String testClass = properties.getProperty("test.class");//NOI18N |
| String testMethods = properties.getProperty("test.methods");//NOI18N |
| if(testClass != null) { |
| FileObject testFO = testSession.getFileLocator().find(testClass.replace('.', '/') + ".java"); //NOI18N |
| if (testMethods != null) { |
| SingleMethod methodSpec = new SingleMethod(testFO, testMethods); |
| lookup = Lookups.singleton(methodSpec); |
| } else { |
| lookup = Lookups.fixed(testFO); |
| } |
| } |
| |
| if (targets.length == 0) { |
| String className = properties.getProperty("classname"); //NOI18N |
| String methodName = properties.getProperty("methodname"); //NOI18N |
| if (className != null) { |
| FileObject testFO = testSession.getFileLocator().find(className.replace('.', '/') + ".java"); //NOI18N |
| if (methodName != null) { |
| SingleMethod methodSpec = new SingleMethod(testFO, methodName); |
| lookup = Lookups.singleton(methodSpec); |
| } else { |
| if(testFO != null) { |
| lookup = Lookups.fixed(testFO); |
| } |
| } |
| } |
| if (scriptFile.getName().equals("junit.xml")) { //NOI18N |
| if (methodName != null) { |
| targets = new String[]{SingleMethod.COMMAND_RUN_SINGLE_METHOD}; |
| } else { |
| targets = new String[]{ActionProvider.COMMAND_TEST_SINGLE}; |
| } |
| } else if (scriptFile.getName().equals("junit-debug.xml")) { //NOI18N |
| if (methodName != null) { |
| targets = new String[]{SingleMethod.COMMAND_DEBUG_SINGLE_METHOD}; |
| } else { |
| targets = new String[]{ActionProvider.COMMAND_DEBUG_TEST_SINGLE}; |
| } |
| } |
| } |
| } catch (Exception e) { |
| } |
| } |
| |
| public void rerun() { |
| if ((properties.getProperty("test.includes") != null && properties.getProperty("test.includes").endsWith(".xml")) || //NOI18N |
| (properties.getProperty("test.class") != null && properties.getProperty("test.class").endsWith(".xml"))) { //NOI18N |
| if (properties.getProperty("continue.after.failing.tests") == null) { //NOI18N |
| properties.setProperty("continue.after.failing.tests", "true"); //NOI18N |
| } |
| try { |
| runAnt(FileUtil.toFileObject(scriptFile), targets, properties); |
| } catch (IOException ex) { |
| LOGGER.log(Level.SEVERE, ex.getMessage(), ex); |
| } |
| } else { |
| Project project = testSession.getProject(); |
| if(ProjectManager.getDefault().isValid(project)) { |
| ActionProvider actionProvider = project.getLookup().lookup(ActionProvider.class); |
| String[] actionNames = getActionNames(targets); |
| if (actionProvider != null) { |
| if (Arrays.asList(actionProvider.getSupportedActions()).contains(actionNames[0]) |
| && actionProvider.isActionEnabled(actionNames[0], lookup)) { |
| actionProvider.invokeAction(actionNames[0], lookup); |
| } |
| } |
| } |
| } |
| } |
| |
| public void rerun(Set<Testcase> tests) { |
| Project p = testSession.getProject(); |
| TestNGSupportImplementation.TestExecutor exec = TestNGSupport.findTestNGSupport(p).createExecutor(p); |
| TestConfig conf = TestNGUtils.getTestConfig(p.getProjectDirectory(), true, null, null, null); |
| try { |
| exec.execute(TestNGSupport.Action.RUN_FAILED, conf); |
| } catch (IOException ex) { |
| LOGGER.log(Level.SEVERE, null, ex); |
| } |
| } |
| |
| public boolean enabled(RerunType type) { |
| switch (type){ |
| case ALL: |
| return true; |
| case CUSTOM: |
| Project p = testSession.getProject(); |
| if (TestNGSupport.isActionSupported(TestNGSupport.Action.RUN_FAILED, p)) { |
| return TestNGSupport.findTestNGSupport(p).createExecutor(p).hasFailedTests(); |
| } |
| return false; |
| default: |
| return false; |
| } |
| } |
| |
| // public boolean enabled() { |
| // if ((scriptFile == null) || (targets == null) || (targets.length == 0)) { |
| // return false; |
| // } |
| // |
| // Project project = testSession.getProject(); |
| // ActionProvider actionProvider = project.getLookup().lookup(ActionProvider.class); |
| // if (actionProvider != null) { |
| // boolean runSupported = false; |
| // for (String action : actionProvider.getSupportedActions()) { |
| // if (action.equals(targets[0])) { |
| // runSupported = true; |
| // break; |
| // } |
| // } |
| // if (runSupported && actionProvider.isActionEnabled(targets[0], lookup)) { |
| // return true; |
| // } |
| // } |
| // |
| // return false; |
| // } |
| |
| @Override |
| public void addChangeListener(ChangeListener listener) { |
| } |
| |
| @Override |
| public void removeChangeListener(ChangeListener listener) { |
| } |
| |
| private static void runAnt(FileObject antScript, String[] antTargets, Properties antProps) throws IOException { |
| AntTargetExecutor.Env execenv = new AntTargetExecutor.Env(); |
| Properties props = execenv.getProperties(); |
| props.putAll(antProps); |
| execenv.setProperties(props); |
| AntTargetExecutor.createTargetExecutor(execenv).execute(AntScriptUtils.antProjectCookieFor(antScript), antTargets); |
| } |
| |
| private String[] getActionNames(String[] targetNames) { |
| String[] actions = new String[targetNames.length]; |
| for (int i = 0; i < targetNames.length; i++) { |
| if (targetNames[i].equals("test-single")) { //NOI18N |
| actions[i] = ActionProvider.COMMAND_TEST_SINGLE; |
| } else if (targetNames[i].equals("debug-test")) { //NOI18N |
| actions[i] = ActionProvider.COMMAND_DEBUG_TEST_SINGLE; |
| } else if (targetNames[i].equals("test-unit")) { //NOI18N |
| actions[i] = ActionProvider.COMMAND_TEST; |
| } else if (targetNames[i].equals("test-method")) { //NOI18N |
| actions[i] = SingleMethod.COMMAND_RUN_SINGLE_METHOD; |
| } else if (targetNames[i].equals("debug-test-single-nb")) { //NOI18N |
| String testMethods = properties.getProperty("test.methods");//NOI18N |
| if (testMethods != null) { |
| actions[i] = SingleMethod.COMMAND_DEBUG_SINGLE_METHOD; |
| } else { |
| actions[i] = ActionProvider.COMMAND_DEBUG_TEST_SINGLE; |
| } |
| } else { |
| actions[i] = targetNames[i]; |
| } |
| } |
| return actions; |
| } |
| } |