From c516316000944adb31a81f07a66181df1bdbf653 Mon Sep 17 00:00:00 2001
From: Guillem Hernandez Sola <guillem@itnove.com>
Date: Fri, 24 Nov 2017 17:07:38 +0100
Subject: [PATCH] browser selection

---
 .../testng/startUsingWebDriver/BaseTest.java  | 23 +++++++++++--------
 1 file changed, 14 insertions(+), 9 deletions(-)

diff --git a/src/test/java/com/itnove/trainings/testng/startUsingWebDriver/BaseTest.java b/src/test/java/com/itnove/trainings/testng/startUsingWebDriver/BaseTest.java
index 3308c6b..b5b7361 100644
--- a/src/test/java/com/itnove/trainings/testng/startUsingWebDriver/BaseTest.java
+++ b/src/test/java/com/itnove/trainings/testng/startUsingWebDriver/BaseTest.java
@@ -3,6 +3,7 @@ package com.itnove.trainings.testng.startUsingWebDriver;
 
 import org.openqa.selenium.JavascriptExecutor;
 import org.openqa.selenium.chrome.ChromeDriver;
+import org.openqa.selenium.firefox.FirefoxDriver;
 import org.openqa.selenium.interactions.Actions;
 import org.openqa.selenium.remote.DesiredCapabilities;
 import org.openqa.selenium.remote.RemoteWebDriver;
@@ -26,15 +27,19 @@ public class BaseTest {
 
     @BeforeMethod
     public void setUp() throws IOException {
-        DesiredCapabilities capabilities = DesiredCapabilities.chrome();
-        System.setProperty("webdriver.chrome.driver", "src" + File.separator + "main" + File.separator + "resources" + File.separator + "chromedriver-linux");
-        driver = new ChromeDriver(capabilities);
-        //DesiredCapabilities capabilities = DesiredCapabilities.firefox();
-        //System.setProperty("webdriver.gecko.driver",
-        //        "src" + File.separator + "main"
-        //                + File.separator + "resources"
-        //                + File.separator + "geckodriver-linux");
-       // driver = new FirefoxDriver(capabilities);
+        String browser = System.getProperty("browser");
+        if (browser != null && browser.equalsIgnoreCase("chrome")) {
+            DesiredCapabilities capabilities = DesiredCapabilities.chrome();
+            System.setProperty("webdriver.chrome.driver", "src" + File.separator + "main" + File.separator + "resources" + File.separator + "chromedriver-linux");
+            driver = new ChromeDriver(capabilities);
+        } else {
+            DesiredCapabilities capabilities = DesiredCapabilities.firefox();
+            System.setProperty("webdriver.gecko.driver",
+                    "src" + File.separator + "main"
+                            + File.separator + "resources"
+                            + File.separator + "geckodriver-linux");
+            driver = new FirefoxDriver(capabilities);
+        }
         wait = new LocalRemoteWebDriverWait(driver, timeOut);
         hover = new Actions(driver);
         driver.manage().deleteAllCookies();