带有 @BeforeClass 的 TestNG 数据提供者

Question

I am trying to run a class with multiple tests under two different conditions. Basically I have a bunch of tests related to a search. I am adding new functionality of a new search strategy, and in the meantime want to run the already written tests under both configurations. As we have multiple classes each with multiple tests I want to streamline this process as much as possible. Ideally it'd be great to do the setup in a @BeforeClass with a data provider so that all tests in the class are basically run twice under the different configurations, but doesn't look like this is possible.

Right now I have:

public class SearchTest1 {
    @Test(dataProvider = "SearchType")
    public void test1(SearchType searchType) {
        setSearchType(searchType);
        //Do the test1 logic
    }

    @Test(dataProvider = "SearchType")
    public void test2(SearchType searchType) {
        setSearchType(searchType);
        //Do the test2 logic
    }

    @DataProvider(name = "SearchType")
    public Object[][] createData() {
        return new Object[][]{
            new Object[] {SearchType.scheme1, SearchType.scheme2}
        }
    }
}

Is there a better way to do this?

Answer 1

If you want to avoid having to annotate each and every method with the data provider, you can use a Factory instead.

public class SearchTest1 {
    private final SearchType searchType;

    public SearchTest1( SearchType searchType ) {
       this.searchType = searchType;
    }

    @Test
    public void test2() {
        //Do the test2 logic
    }
    ...
}

And your factory class will be:

public class SearchTestFactory {
   @Factory
   public Object [] createInstances() {
      return new Object[] { new SeartchTest1( SearchType.ONE ), new SearchTest1( SearchType.TWO ) };
   }
}

See more on this here .

Then you can either have one factory that enumerates every test class or a separate factory for each, the first one is obviously less flexible, the second one means slightly more code.

Answer 2

You can use parameters in @BeforeClass. Just use (with some cleanup)

context.getCurrentXmlTest().getParameters()

      @SuppressWarnings("deprecation")
      @BeforeClass
      public void setUp(ITestContext context) {
          System.out.println(context.getCurrentXmlTest().getAllParameters());
      }