Used TestNG DataProvider to do MxN parallelisation

Author: anirudha13

pom.xml

@@ -0,0 +1,69 @@
+package browserstack;
+
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
+
+import org.testng.annotations.AfterClass;
+import org.testng.annotations.BeforeClass;
+import org.testng.annotations.DataProvider;
+import org.testng.annotations.Test;
+
+import browserstack.driverconfig.Platform;
+import browserstack.driverconfig.WebDriverFactory;
+import browserstack.providers.ManagedWebDriver;
+import browserstack.providers.WebDriverManager;
+import io.cucumber.testng.FeatureWrapper;
+import io.cucumber.testng.PickleWrapper;
+import io.cucumber.testng.TestNGCucumberRunner;
+
+/**
+ * Created with IntelliJ IDEA.
+ *
+ * @author Anirudha Khanna
+ */
+public class RunWebDriverCucumberTests {
+
+    private TestNGCucumberRunner testNGCucumberRunner;
+    private WebDriverFactory webDriverFactory;
+    private WebDriverManager webDriverManager;
+
+    @BeforeClass(alwaysRun = true)
+    public void setUpClass() {
+        testNGCucumberRunner = new TestNGCucumberRunner(this.getClass());
+        this.webDriverFactory = WebDriverFactory.getInstance();
+        this.webDriverManager = WebDriverManager.getInstance();
+    }
+
+    @Test(groups = "cucumber", description = "Runs Cucumber Feature", dataProvider = "scenarios")
+    public void feature(PickleWrapper pickleWrapper, FeatureWrapper featureWrapper, ManagedWebDriver managedWebDriver) {
+        managedWebDriver.setTestName(pickleWrapper.getPickle().getName());
+        webDriverManager.setThreadLocalWebDriver(managedWebDriver);
+        testNGCucumberRunner.runScenario(pickleWrapper.getPickle());
+    }
+
+    @DataProvider(name = "scenarios", parallel = true)
+    public Object[][] scenarios() {
+        List<Object[]> browserScenarioParams = new ArrayList<>();
+
+        Object[][] scenarios = testNGCucumberRunner.provideScenarios();
+        for (Object[] testParams : scenarios) {
+            for (Platform platform : webDriverFactory.getPlatforms()) {
+                Object[] newTestParams = Arrays.copyOf(testParams, testParams.length + 1);
+                ManagedWebDriver managedWebDriver = new ManagedWebDriver(platform, webDriverFactory);
+                newTestParams[testParams.length] = managedWebDriver;
+                browserScenarioParams.add(newTestParams);
+            }
+        }
+        return browserScenarioParams.toArray(new Object[0][0]);
+    }
+
+    @AfterClass(alwaysRun = true)
+    public void tearDownClass() {
+        if (testNGCucumberRunner == null) {
+            return;
+        }
+        testNGCucumberRunner.finish();
+    }
+
+}

src/test/java/browserstack/RunWebDriverCucumberTests.java

@@ -1,7 +1,14 @@
 package browserstack.driverconfig;
 
-import com.fasterxml.jackson.databind.ObjectMapper;
-import com.fasterxml.jackson.dataformat.yaml.YAMLFactory;
+import java.io.IOException;
+import java.net.MalformedURLException;
+import java.net.URL;
+import java.nio.file.Paths;
+import java.util.Arrays;
+import java.util.LinkedHashMap;
+import java.util.List;
+import java.util.Map;
+
 import org.apache.commons.lang3.StringUtils;
 import org.openqa.selenium.WebDriver;
 import org.openqa.selenium.chrome.ChromeDriver;
@@ -21,14 +28,8 @@
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
-import java.io.IOException;
-import java.net.MalformedURLException;
-import java.net.URL;
-import java.nio.file.Paths;
-import java.util.Arrays;
-import java.util.LinkedHashMap;
-import java.util.List;
-import java.util.Map;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.dataformat.yaml.YAMLFactory;
 
 public class WebDriverFactory {
 
@@ -95,25 +96,29 @@ private WebDriverConfiguration parseWebDriverConfig() {
         return webDriverConfiguration;
     }
 
-    public WebDriver createWebDriverForPlatform(Platform platform, String testName, String[] specificCapabilitiesKeys) throws MalformedURLException {
+    public WebDriver createWebDriverForPlatform(Platform platform, String testName, String[] specificCapabilitiesKeys) {
         WebDriver webDriver = null;
-        Map<String, Object> specificCapabilitiesMap = new LinkedHashMap<>();
-        if (webDriverConfiguration.getSpecificCapabilities() != null && specificCapabilitiesKeys.length > 0) {
-            Arrays.stream(specificCapabilitiesKeys)
-                    .forEach(specificCapabilityKey -> webDriverConfiguration
-                            .getSpecificCapabilities()
-                            .getSpecificCapabilities(specificCapabilityKey)
-                            .forEach(caps -> caps.getCapabilityMap().forEach(specificCapabilitiesMap::put)));
-        }
-        switch (this.webDriverConfiguration.getDriverType()) {
-            case onPremDriver:
-                webDriver = createOnPremWebDriver(platform, specificCapabilitiesMap);
-                break;
-            case onPremGridDriver:
-                webDriver = createOnPremGridWebDriver(platform, specificCapabilitiesMap);
-                break;
-            case cloudDriver:
-                webDriver = createRemoteWebDriver(platform, testName, specificCapabilitiesMap);
+        try {
+            Map<String, Object> specificCapabilitiesMap = new LinkedHashMap<>();
+            if (webDriverConfiguration.getSpecificCapabilities() != null && specificCapabilitiesKeys.length > 0) {
+                Arrays.stream(specificCapabilitiesKeys)
+                  .forEach(specificCapabilityKey -> webDriverConfiguration
+                                                      .getSpecificCapabilities()
+                                                      .getSpecificCapabilities(specificCapabilityKey)
+                                                      .forEach(caps -> caps.getCapabilityMap().forEach(specificCapabilitiesMap::put)));
+            }
+            switch (this.webDriverConfiguration.getDriverType()) {
+                case onPremDriver:
+                    webDriver = createOnPremWebDriver(platform, specificCapabilitiesMap);
+                    break;
+                case onPremGridDriver:
+                    webDriver = createOnPremGridWebDriver(platform, specificCapabilitiesMap);
+                    break;
+                case cloudDriver:
+                    webDriver = createRemoteWebDriver(platform, testName, specificCapabilitiesMap);
+            }
+        } catch (Throwable throwable) {
+            throw new Error(throwable);
         }
         return webDriver;
     }
@@ -126,6 +131,10 @@ public DriverType getDriverType() {
         return webDriverConfiguration.getDriverType();
     }
 
+    public boolean isCloudDriver() {
+        return webDriverConfiguration.getDriverType() == DriverType.cloudDriver;
+    }
+
     public List<Platform> getPlatforms() {
         return this.webDriverConfiguration.getActivePlatforms();
     }
@@ -262,4 +271,4 @@ private String createBuildName(String buildPrefix) {
         return buildName + "-" + buildSuffix;
     }
 
-}
+}