/*
* Copyright 2000-2015 JetBrains s.r.o.
*
* 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 com.theoryinpractice.testng.configuration;
import com.intellij.execution.CantRunException;
import com.intellij.execution.testframework.JavaTestLocator;
import com.intellij.execution.testframework.TestSearchScope;
import com.intellij.execution.testframework.sm.runner.SMTestProxy;
import com.intellij.psi.PsiClass;
import com.intellij.psi.PsiMethod;
import com.intellij.psi.search.GlobalSearchScope;
import com.intellij.testFramework.fixtures.LightCodeInsightFixtureTestCase;
import com.theoryinpractice.testng.model.TestData;
import com.theoryinpractice.testng.model.TestNGTestObject;
import com.theoryinpractice.testng.model.TestType;
import com.theoryinpractice.testng.ui.actions.RerunFailedTestsAction;
import org.testng.annotations.AfterMethod;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
@SuppressWarnings("UndeclaredTests")
public class DetectClassesToRunTest extends LightCodeInsightFixtureTestCase {
@BeforeMethod
@Override
protected void setUp() throws Exception {
super.setUp();
myFixture.addClass("package org.testng.annotations; @interface Test {String[] dependsOnMethods() default {};}");
myFixture.addClass("package org.testng.annotations; @interface BeforeClass {}");
myFixture.addClass("package org.testng.annotations; @interface BeforeGroups { String[] value() default {};}");
}
@AfterMethod
@Override
protected void tearDown() throws Exception {
super.tearDown();
}
@Test
public void testNonRelatedBeforeClassIncluded() throws Exception {
final PsiClass configClass = myFixture.addClass("package p; public class AConfig {@org.testng.annotations.BeforeClass public void setup(){}}");
final PsiClass testClass = myFixture.addClass("package p; public class ATest {@org.testng.annotations.Test public void testOne(){}}");
//should not be included in resulted xml
myFixture.addClass("package p; public class BTest {}");
doTestPackageConfiguration(configClass, testClass);
}
@Test
public void testNonRelatedIncludedWhenConfigIsLocatedInSuperclassInAnotherPackage() throws Exception {
myFixture.addClass("package a; public class AConfig {@org.testng.annotations.BeforeClass public void setup(){}}");
final PsiClass emptyClassWithSuperConfig = myFixture.addClass("package p; import a.AConfig; public class BConfig extends AConfig {}");
final PsiClass testClass = myFixture.addClass("package p; public class ATest {@org.testng.annotations.Test public void testOne(){}}");
doTestPackageConfiguration(emptyClassWithSuperConfig, testClass);
}
@Test
public void testBeforeClassIsIncludedIfRunOnlyOneMethod() throws Exception {
final PsiClass aClass =
myFixture.addClass("package a; public class AConfig {" +
"@org.testng.annotations.BeforeClass public void setup(){}\n" +
"@org.testng.annotations.Test public void testOne(){}\n" +
"}");
doTestMethodConfiguration(aClass, aClass.getMethods()[1]);
}
@Test
public void testOneMethodWhenAnnotationIsOnBaseClassOnly() throws Exception {
myFixture.addClass("package a; @org.testng.annotations.Test public class BaseClass {}");
final PsiClass aClass =
myFixture.addClass("package a; public class ATest extends BaseClass {" +
" public void testOne(){}\n" +
"}");
doTestMethodConfiguration(aClass, aClass.getMethods());
}
@Test
public void testOneMethodWithDependencies() throws Exception {
final PsiClass aClass =
myFixture.addClass("package a; public class ATest {" +
" @org.testng.annotations.Test\n" +
" public void testTwo(){}\n " +
" @org.testng.annotations.Test(dependsOnMethods=\"testTwo\")\n" +
" public void testOne(){}\n" +
"}");
doTestMethodConfiguration(aClass, aClass.getMethods());
}
@Test
public void testDependsOnGroupDontIncludeForeignClass() throws Exception {
final PsiClass aClass =
myFixture.addClass("package a; public class ATest {" +
" @org.testng.annotations.Test(groups = { \"g1\" })\n" +
" public void testTwo(){}\n " +
" @org.testng.annotations.Test(dependsOnGroups = {\"g1\" })\n" +
" public void testOne(){}\n" +
"}");
myFixture.addClass("package a; public class ForeignTest {" +
" @org.testng.annotations.Test(groups = { \"g1\" })\n" +
" public void testForth(){}\n " +
"}");
doTestClassConfiguration(aClass);
}
@Test
public void testBeforeGroups() throws Exception {
final PsiClass aClass =
myFixture.addClass("package a; public class ATest {" +
" @org.testng.annotations.Test(groups = { \"g1\" })\n" +
" public void testOne(){}\n" +
"}");
final PsiClass configClass = myFixture.addClass("package a; public class ConfigTest {" +
" @org.testng.annotations.BeforeGroups(groups = { \"g1\" })\n" +
" public void testTwo(){}\n " +
"}");
doTestMethodConfiguration(aClass, configClass, configClass.getMethods()[0], aClass.getMethods());
}
public void testRerunFailedTestWithDependency() throws Exception {
final PsiClass aClass =
myFixture.addClass("package a; public class ATest {" +
" @org.testng.annotations.Test()\n" +
" public void testTwo(){}\n " +
" @org.testng.annotations.Test(dependsOnMethods = \"testTwo\")\n" +
" public void testOne(String s){}\n" + //parameterized
"}");
final LinkedHashMap<PsiClass, Map<PsiMethod, List<String>>> classes = new LinkedHashMap<>();
classes.put(aClass, new HashMap<>());
final GlobalSearchScope projectScope = GlobalSearchScope.projectScope(getProject());
final SMTestProxy testProxy = new SMTestProxy("testOne", false, "java:test://a.ATest.testOne[a]");
testProxy.setLocator(new JavaTestLocator());
RerunFailedTestsAction.includeFailedTestWithDependencies(classes, projectScope, getProject(), testProxy);
assertEquals(1, classes.size());
final Map<PsiMethod, List<String>> params = classes.get(aClass);
assertContainsElements(params.keySet(), aClass.getMethods());
final List<String> paramsToRerun = params.get(aClass.findMethodsByName("testOne", false)[0]);
assertEquals(1, paramsToRerun.size());
assertContainsElements(paramsToRerun, "a");
}
public void testRerunFailedParameterized() throws Exception {
@SuppressWarnings("TestNGDataProvider") final PsiClass aClass =
myFixture.addClass("package a; " +
"import org.testng.annotations.DataProvider;\n" +
"import org.testng.annotations.Test;\n" +
"\n" +
"import static org.testng.Assert.assertEquals;\n" +
"\n" +
"public class ATest {\n" +
"\n" +
" @DataProvider\n" +
" public Object[][] testData() {\n" +
" return new Object[][]{\n" +
" {1},\n" +
" {2},\n" +
" };\n" +
" }\n" +
"\n" +
" @Test(dataProvider = \"testData\")\n" +
" public void test(int in) {\n" +
" assertEquals(in, 0);\n" +
" }\n" +
"}\n");
final LinkedHashMap<PsiClass, Map<PsiMethod, List<String>>> classes = new LinkedHashMap<>();
classes.put(aClass, new HashMap<>());
final GlobalSearchScope projectScope = GlobalSearchScope.projectScope(getProject());
final SMTestProxy testProxy = new SMTestProxy("test", false, "java:test://a.ATest.test[0]");
testProxy.setLocator(new JavaTestLocator());
RerunFailedTestsAction.includeFailedTestWithDependencies(classes, projectScope, getProject(), testProxy);
final SMTestProxy testProxy2 = new SMTestProxy("test", false, "java:test://a.ATest.test[1]");
testProxy2.setLocator(new JavaTestLocator());
RerunFailedTestsAction.includeFailedTestWithDependencies(classes, projectScope, getProject(), testProxy2);
assertEquals(1, classes.size());
final Map<PsiMethod, List<String>> params = classes.get(aClass);
final PsiMethod[] tests = aClass.findMethodsByName("test", false);
assertContainsElements(params.keySet(), tests);
final List<String> paramsToRerun = params.get(tests[0]);
assertEquals(2, paramsToRerun.size());
assertContainsElements(paramsToRerun, "0", "1");
}
private void doTestMethodConfiguration(PsiClass aClass, PsiMethod... expectedMethods) throws CantRunException {
doTestMethodConfiguration(aClass, null, null, expectedMethods);
}
private void doTestMethodConfiguration(PsiClass aClass, PsiClass secondaryClass, PsiMethod configMethod, PsiMethod... expectedMethods) throws CantRunException {
final TestNGConfiguration configuration =
new TestNGConfiguration("testOne", getProject(), TestNGConfigurationType.getInstance().getConfigurationFactories()[0]);
final TestData data = configuration.getPersistantData();
data.TEST_OBJECT = TestType.METHOD.getType();
data.METHOD_NAME = "testOne";
data.setScope(TestSearchScope.SINGLE_MODULE);
configuration.setModule(data.setMainClass(aClass));
final TestNGTestObject testObject = TestNGTestObject.fromConfig(configuration);
assertNotNull(testObject);
final LinkedHashMap<PsiClass, Map<PsiMethod, List<String>>> classes = new LinkedHashMap<>();
testObject.fillTestObjects(classes);
assertContainsElements(classes.keySet(), aClass);
final Map<PsiMethod, List<String>> methods = classes.get(aClass);
assertContainsElements(methods.keySet(), expectedMethods);
if (secondaryClass != null) {
final Map<PsiMethod, List<String>> configMethods = classes.get(secondaryClass);
assertTrue(configMethods != null);
assertTrue(configMethods.containsKey(configMethod));
}
}
private void doTestClassConfiguration(PsiClass aClass) throws CantRunException {
final TestNGConfiguration configuration =
new TestNGConfiguration("TestA", getProject(), TestNGConfigurationType.getInstance().getConfigurationFactories()[0]);
final TestData data = configuration.getPersistantData();
data.TEST_OBJECT = TestType.CLASS.getType();
data.setScope(TestSearchScope.SINGLE_MODULE);
configuration.setModule(data.setMainClass(aClass));
final TestNGTestObject testObject = TestNGTestObject.fromConfig(configuration);
assertNotNull(testObject);
final LinkedHashMap<PsiClass, Map<PsiMethod, List<String>>> classes = new LinkedHashMap<>();
testObject.fillTestObjects(classes);
assertContainsElements(classes.keySet(), aClass);
assertEquals(1, classes.size());
}
private void doTestPackageConfiguration(PsiClass... containingClasses) throws CantRunException {
final TestNGConfiguration configuration =
new TestNGConfiguration("p", getProject(), TestNGConfigurationType.getInstance().getConfigurationFactories()[0]);
final TestData data = configuration.getPersistantData();
data.TEST_OBJECT = TestType.PACKAGE.getType();
data.PACKAGE_NAME = "p";
data.setScope(TestSearchScope.WHOLE_PROJECT);
final TestNGTestObject testObject = TestNGTestObject.fromConfig(configuration);
assertNotNull(testObject);
final LinkedHashMap<PsiClass, Map<PsiMethod, List<String>>> classes = new LinkedHashMap<>();
testObject.fillTestObjects(classes);
assertContainsElements(classes.keySet(), containingClasses);
for (PsiClass psiClass : containingClasses) {
assertEmpty(classes.get(psiClass).keySet());
}
}
}