package test.methodinterceptors;
import org.testng.Assert;
import org.testng.ITestResult;
import org.testng.TestListenerAdapter;
import org.testng.TestNG;
import org.testng.annotations.Test;
import org.testng.xml.Parser;
import org.testng.xml.XmlSuite;
import org.xml.sax.SAXException;
import test.SimpleBaseTest;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.util.List;
import javax.xml.parsers.ParserConfigurationException;
public class MethodInterceptorTest extends SimpleBaseTest {
private String XML =
"<!DOCTYPE suite SYSTEM \"http://beust.com/testng/testng-1.0.dtd\" >" +
"" +
"<suite name=\"Single\" verbose=\"0\">" +
"" +
"<listeners>" +
" <listener class-name=\"test.methodinterceptors.NullMethodInterceptor\" />" +
"</listeners>" +
"" +
" <test name=\"Single\" >" +
" <classes>" +
" <class name=\"test.methodinterceptors.FooTest\" />" +
" </classes>" +
" </test>" +
"" +
"</suite>";
@Test
public void noMethodsShouldRun() {
TestNG tng = create();
tng.setTestClasses(new Class[] { FooTest.class });
testNullInterceptor(tng);
}
private void testNullInterceptor(TestNG tng) {
tng.setMethodInterceptor(new NullMethodInterceptor());
TestListenerAdapter tla = new TestListenerAdapter();
tng.addListener(tla);
tng.run();
Assert.assertEquals(tla.getPassedTests().size(), 0);
Assert.assertEquals(tla.getFailedTests().size(), 0);
Assert.assertEquals(tla.getSkippedTests().size(), 0);
}
private void testFast(boolean useInterceptor) {
TestNG tng = create();
tng.setTestClasses(new Class[] { FooTest.class });
if (useInterceptor) {
tng.setMethodInterceptor(new FastTestsFirstInterceptor());
}
TestListenerAdapter tla = new TestListenerAdapter();
// tng.setParallel("methods");
tng.addListener(tla);
tng.run();
Assert.assertEquals(tla.getPassedTests().size(), 3);
ITestResult first = tla.getPassedTests().get(0);
String method = "zzzfast";
if (useInterceptor) {
Assert.assertEquals(first.getMethod().getMethodName(), method);
} else {
Assert.assertNotSame(first.getMethod().getMethodName(), method);
}
}
@Test
public void fastShouldRunFirst() {
testFast(true /* use interceptor */);
}
@Test
public void fastShouldNotRunFirst() {
testFast(false /* don't use interceptor */);
}
@Test
public void nullMethodInterceptorWorksInTestngXml()
throws IOException, ParserConfigurationException, SAXException {
File f = File.createTempFile("testng-tests-", "");
f.deleteOnExit();
BufferedWriter bw = new BufferedWriter(new FileWriter(f));
bw.write(XML);
bw.close();
try {
List<XmlSuite> xmlSuites = new Parser(f.getAbsolutePath()).parseToList();
TestNG tng = create();
tng.setXmlSuites(xmlSuites);
testNullInterceptor(tng);
}
catch(Exception ex) {
ex.printStackTrace();
}
}
@Test(timeOut = 1000)
public void shouldNotLockUpWithInterceptorThatRemovesMethods() {
TestNG tng = create(LockUpInterceptorSampleTest.class);
tng.setParallel("methods");
tng.run();
}
}