|
1 | 1 | package test.listeners; |
2 | 2 |
|
3 | | -import org.testng.Assert; |
4 | | -import org.testng.IAlterSuiteListener; |
5 | | -import org.testng.TestNG; |
| 3 | +import org.testng.*; |
6 | 4 | import org.testng.annotations.Test; |
| 5 | +import org.testng.internal.collections.Pair; |
7 | 6 | import org.testng.xml.XmlSuite; |
8 | 7 | import org.testng.xml.XmlTest; |
9 | 8 | import test.SimpleBaseTest; |
10 | 9 |
|
| 10 | +import java.util.ArrayList; |
11 | 11 | import java.util.Arrays; |
12 | 12 | import java.util.List; |
| 13 | +import java.util.Optional; |
13 | 14 |
|
14 | 15 | public class AlterSuiteListenerTest extends SimpleBaseTest { |
15 | 16 |
|
16 | 17 | public static final String ALTER_SUITE_LISTENER = "AlterSuiteListener"; |
17 | 18 |
|
18 | 19 | @Test |
19 | 20 | public void executionListenerWithXml() { |
20 | | - XmlSuite suite = runTest(AlterSuiteListener1SampleTest.class, AlterSuiteNameListener.class.getName()); |
| 21 | + XmlSuite suite = runTest(AlterSuiteListener1SampleTest.class, AlterSuiteNameListener.class.getName()).second(); |
21 | 22 | Assert.assertEquals(suite.getName(), AlterSuiteNameListener.class.getSimpleName()); |
22 | 23 | } |
23 | 24 |
|
24 | 25 | @Test |
25 | 26 | public void executionListenerWithoutListener() { |
26 | | - XmlSuite suite = runTest(AlterSuiteListener1SampleTest.class, null/*Donot add the listener*/); |
| 27 | + XmlSuite suite = runTest(AlterSuiteListener1SampleTest.class).second(); |
27 | 28 | Assert.assertEquals(suite.getName(), ALTER_SUITE_LISTENER); |
28 | 29 | } |
29 | 30 |
|
30 | 31 | @Test |
31 | 32 | public void executionListenerWithXml2() { |
32 | | - XmlSuite suite = runTest(AlterSuiteListener1SampleTest.class, AlterXmlTestsInSuiteListener.class.getName()); |
| 33 | + XmlSuite suite = runTest(AlterSuiteListener1SampleTest.class, AlterXmlTestsInSuiteListener.class.getName()).second(); |
33 | 34 | Assert.assertEquals(suite.getTests().size(), 2); |
34 | 35 | } |
35 | 36 |
|
| 37 | + @Test(description = "GITHUB-2469") |
| 38 | + public void executionListenerWithXml3() { |
| 39 | + Pair<TestNG, XmlSuite> retObjects = runTest(AlterSuiteListener1SampleTest.class, |
| 40 | + AlterXmlTestWithParameterInSuiteListener.class.getName(), AlteredXmlSuiteReadListener.class.getName()); |
| 41 | + TestNG tng = retObjects.first(); |
| 42 | + XmlSuite suite = retObjects.second(); |
| 43 | + Assert.assertEquals(suite.getTests().size(), 2); |
| 44 | + List<ISuiteListener> listeners = Optional.ofNullable(tng.getSuiteListeners()).orElse(new ArrayList<>()); |
| 45 | + Assert.assertFalse(listeners.isEmpty()); |
| 46 | + for (ISuiteListener iSuiteListener : |
| 47 | + listeners) { |
| 48 | + if (iSuiteListener instanceof AlteredXmlSuiteReadListener) { |
| 49 | + AlteredXmlSuiteReadListener alteredXmlSuiteReadListener = (AlteredXmlSuiteReadListener) iSuiteListener; |
| 50 | + XmlSuite xmlSuite = alteredXmlSuiteReadListener.currentSuiteOnStart.getXmlSuite(); |
| 51 | + List<XmlTest> tests = xmlSuite.getTests(); |
| 52 | + int i = 1; |
| 53 | + for (XmlTest xmlTest: |
| 54 | + tests) { |
| 55 | + Assert.assertEquals(xmlTest.getParameter("param"), String.valueOf(i)); |
| 56 | + i++; |
| 57 | + } |
| 58 | + } |
| 59 | + } |
| 60 | + } |
36 | 61 |
|
37 | | - private XmlSuite runTest(Class<?> listenerClass, String listenerName) { |
| 62 | + private Pair<TestNG, XmlSuite> runTest(Class<?> listenerClass, String... listenerNames) { |
38 | 63 | XmlSuite s = createXmlSuite(ALTER_SUITE_LISTENER); |
39 | 64 | createXmlTest(s, "Test", listenerClass.getName()); |
40 | | - boolean addListener = (listenerName != null); |
41 | 65 |
|
42 | | - if (addListener) { |
| 66 | + |
| 67 | + for (String listenerName: |
| 68 | + listenerNames) { |
43 | 69 | s.addListener(listenerName); |
44 | 70 | } |
45 | 71 | TestNG tng = create(); |
46 | 72 | tng.setXmlSuites(Arrays.asList(s)); |
47 | 73 | tng.run(); |
48 | | - return s; |
| 74 | + Pair<TestNG, XmlSuite> returnObj = new Pair<>(tng, s); |
| 75 | + return returnObj; |
49 | 76 | } |
50 | 77 |
|
51 | 78 | public static class AlterSuiteListener1SampleTest { |
@@ -78,4 +105,37 @@ public void alter(List<XmlSuite> suites) { |
78 | 105 | } |
79 | 106 | } |
80 | 107 |
|
| 108 | + public static class AlterXmlTestWithParameterInSuiteListener implements IAlterSuiteListener { |
| 109 | + |
| 110 | + @Override |
| 111 | + public void alter(List<XmlSuite> suites) { |
| 112 | + XmlSuite suite = suites.get(0); |
| 113 | + List<XmlTest> tests = suite.getTests(); |
| 114 | + XmlTest test = tests.get(0); |
| 115 | + |
| 116 | + List<XmlTest> newXmlTests = new ArrayList<>(); |
| 117 | + XmlTest newXmlTest = (XmlTest) test.clone(); |
| 118 | + newXmlTest.setName("name_1"); |
| 119 | + newXmlTest.addParameter("param", "1"); |
| 120 | + newXmlTests.add(newXmlTest); |
| 121 | + |
| 122 | + newXmlTest = (XmlTest) test.clone(); |
| 123 | + newXmlTest.setName("name_2"); |
| 124 | + newXmlTest.addParameter("param", "2"); |
| 125 | + newXmlTests.add(newXmlTest); |
| 126 | + |
| 127 | + suite.setTests(newXmlTests); |
| 128 | + } |
| 129 | + } |
| 130 | + |
| 131 | + public static class AlteredXmlSuiteReadListener implements ISuiteListener { |
| 132 | + |
| 133 | + public ISuite currentSuiteOnStart; |
| 134 | + |
| 135 | + @Override |
| 136 | + public void onStart(ISuite suite) { |
| 137 | + currentSuiteOnStart = suite; |
| 138 | + } |
| 139 | + } |
| 140 | + |
81 | 141 | } |
0 commit comments