Samples for jUnit parallel.

Author: gaurav-bstack

junit/JUnitParallelSample.java

@@ -0,0 +1,83 @@
+import static org.junit.Assert.*;
+
+import java.io.File;
+import java.io.IOException;
+import java.net.URL;
+import java.util.LinkedList;
+
+import org.apache.commons.io.FileUtils;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.junit.runners.Parameterized;
+import org.openqa.selenium.By;
+import org.openqa.selenium.OutputType;
+import org.openqa.selenium.Platform;
+import org.openqa.selenium.TakesScreenshot;
+import org.openqa.selenium.WebDriver;
+import org.openqa.selenium.WebElement;
+import org.openqa.selenium.remote.Augmenter;
+import org.openqa.selenium.remote.DesiredCapabilities;
+import org.openqa.selenium.remote.RemoteWebDriver;
+
+@RunWith(Parallelized.class)
+public class JUnitParallelSample {
+  private String platform;
+  private String browserName;
+  private String browserVersion;
+
+  @Parameterized.Parameters
+  public static LinkedList getEnvironments() throws Exception {
+    LinkedList<String[]> env = new LinkedList();
+
+    env.add(new String[]{Platform.XP.toString(), "chrome", "27"});
+    env.add(new String[]{Platform.WINDOWS.toString(),"firefox","20"});
+    env.add(new String[]{Platform.WINDOWS.toString(),"ie","7"});
+    env.add(new String[]{Platform.WINDOWS.toString(),"opera","12.14"});
+//add more browsers here
+
+    return env;
+  }
+
+  public JUnitParallelSample(String platform, String browserName, String browserVersion) {
+    this.platform = platform;
+    this.browserName = browserName;
+    this.browserVersion = browserVersion;
+  }
+
+  private WebDriver driver;
+
+  @Before
+  public void setUp() throws Exception {
+    DesiredCapabilities capability = new DesiredCapabilities();
+    capability.setCapability("platform", platform);
+    capability.setCapability("browser", browserName);
+    capability.setCapability("browserVersion", browserVersion);
+    capability.setCapability("build", "JUnit-Parallel");
+    driver = new RemoteWebDriver(new URL("http://USERNAME:ACCESS_KEY@hub.browserstack.com/wd/hub"), capability);  
+  }
+
+  @Test
+  public void testSimple() throws Exception {
+    driver.get("http://www.google.com");
+    String title = driver.getTitle();
+    System.out.println("Page title is: " + title);
+    assertEquals("Google", title);
+    WebElement element = driver.findElement(By.name("q"));
+    element.sendKeys("Browser Stack");
+    element.submit();
+    driver = new Augmenter().augment(driver);
+    File srcFile = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);
+    try {
+      FileUtils.copyFile(srcFile, new File("Screenshot.png"));
+    } catch (IOException e) {
+      e.printStackTrace();
+    }
+  }
+
+  @After
+  public void tearDown() throws Exception {
+    driver.quit();
+  }
+}

junit/Parallelized.java

@@ -0,0 +1,41 @@
+package com.browserstack;
+
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.Executors;
+import java.util.concurrent.TimeUnit;
+
+import org.junit.runners.Parameterized;
+import org.junit.runners.model.RunnerScheduler;
+
+public class Parallelized extends Parameterized {
+
+  private static class ThreadPoolScheduler implements RunnerScheduler {
+    private ExecutorService executor; 
+
+    public ThreadPoolScheduler() {
+      String threads = System.getProperty("junit.parallel.threads", "16");
+      int numThreads = Integer.parseInt(threads);
+      executor = Executors.newFixedThreadPool(numThreads);
+    }
+
+    @Override
+    public void finished() {
+      executor.shutdown();
+      try {
+        executor.awaitTermination(10, TimeUnit.MINUTES);
+      } catch (InterruptedException exc) {
+        throw new RuntimeException(exc);
+      }
+    }
+
+    @Override
+    public void schedule(Runnable childStatement) {
+      executor.submit(childStatement);
+    }
+  }
+
+  public Parallelized(Class klass) throws Throwable {
+    super(klass);
+    setScheduler(new ThreadPoolScheduler());
+  }
+}