1 package test.thread; 2 3 import org.testng.Assert; 4 import org.testng.TestNG; 5 import org.testng.annotations.Test; 6 7 import test.SimpleBaseTest; 8 9 public class DataProviderThreadPoolSizeTest extends SimpleBaseTest { 10 11 @Test 12 public void shouldUseDefaultDataProviderThreadCount() { 13 TestNG tng = create(DataProviderThreadPoolSizeSampleTest.class); 14 tng.setGroups("parallel"); 15 tng.run(); 16 Assert.assertEquals(DataProviderThreadPoolSizeSampleTest.getThreadCount(), 10); 17 } 18 19 @Test 20 public void shouldNotUseThreadsIfNotUsingParallel() { 21 TestNG tng = create(DataProviderThreadPoolSizeSampleTest.class); 22 tng.setGroups("sequential"); 23 tng.run(); 24 Assert.assertEquals(DataProviderThreadPoolSizeSampleTest.getThreadCount(), 1); 25 } 26 27 @Test 28 public void shouldUseSpecifiedDataProviderThreadCount() { 29 TestNG tng = create(DataProviderThreadPoolSizeSampleTest.class); 30 tng.setGroups("parallel"); 31 tng.setDataProviderThreadCount(3); 32 tng.run(); 33 Assert.assertEquals(DataProviderThreadPoolSizeSampleTest.getThreadCount(), 3); 34 } 35 } 36
