src/test/java/test/ReturnValueTest.java

package test;

import org.testng.Assert;
import org.testng.TestListenerAdapter;
import org.testng.TestNG;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;
import org.testng.xml.XmlSuite;
import org.testng.xml.XmlTest;

import java.util.Arrays;

public class ReturnValueTest extends SimpleBaseTest {

  private XmlSuite m_suite;
  private XmlTest m_test;

  @BeforeMethod
  public void before() {
    m_suite = createXmlSuite("suite");
    m_test = createXmlTest(m_suite, "test", ReturnValueSampleTest.class.getName());
  }

  @Test
  public void suiteReturnValueTestShouldBeRun() {
    m_suite.setAllowReturnValues(true);
    runTest(true);
  }
  
  @Test
  public void suiteReturnValueTestShouldNotBeRun() {
    runTest(false);
  }

  @Test
  public void testReturnValueTestShouldBeRun() {
    m_test.setAllowReturnValues(true);
    runTest(true);
  }
  
  private void runTest(boolean allowed) {
    TestNG tng = create();
    tng.setXmlSuites(Arrays.asList(m_suite));
    TestListenerAdapter tla = new TestListenerAdapter();
    tng.addListener(tla);
    tng.run();

    if (allowed) {
      Assert.assertEquals(tla.getFailedTests().size(), 0);
      Assert.assertEquals(tla.getSkippedTests().size(), 0);
      assertTestResultsEqual(tla.getPassedTests(), Arrays.asList("shouldRun"));
    } else {
      Assert.assertEquals(tla.getFailedTests().size(), 0);
      Assert.assertEquals(tla.getPassedTests().size(), 0);
      Assert.assertEquals(tla.getSkippedTests().size(), 0);
    }
  }
  
}