Added enum for project options

Author: Nilesh Rakshe

src/main/java/com/mindstix/cb/utils/Constants.java

@@ -15,4 +15,16 @@ public final class Constants {
  */
 private Constants() {
 }
+
+public static enum Platform {
+LOCAL, BROWSERSTACK, SAUCELABS
+}
+
+public static enum Browser {
+CHROME, FIREFOX, IE
+}
+
+public static enum Mode {
+NORMAL, HEADLESS, GRID, ICOGNITO
+}
 }

src/main/java/com/mindstix/cb/utils/DriverUtility.java

@@ -52,14 +52,15 @@ public static WebDriver loadDriver(String browserName) {
 InputStream driverInput = null;
 WebDriver webDriver = null;
 try {
-switch (System.getProperty("env.platform")) {
-case "local":
+String platform = System.getProperty("env.platform");
+switch (Constants.Platform.valueOf(platform.toUpperCase())) {
+case LOCAL:
 webDriver = initDriver(browserName);
 break;
-case "browserStack":
+case BROWSERSTACK:
 webDriver = initBrowserStack();
 break;
-case "sauceLabs":
+case SAUCELABS:
 webDriver = initSauceLabs();
 break;
 default:
@@ -94,14 +95,14 @@ public void closeWebDriver(WebDriver webDriver) {
  */
 private static WebDriver initDriver(String browserType) {
 WebDriver webDriver;
-switch (browserType) {
-case "chrome":
+switch (Constants.Browser.valueOf(browserType.toUpperCase())) {
+case CHROME:
 webDriver = initChromeDriver();
 break;
-case "firefox":
+case FIREFOX:
 webDriver = initFirefoxDriver();
 break;
-case "IE":
+case IE:
 webDriver = initIEDriver();
 break;
 default:
@@ -122,19 +123,19 @@ private static WebDriver initChromeDriver() {
 ChromeOptions options = null;
 WebDriver webDriver = null;
 if (mode != null) {
-switch (mode) {
-case "normal":
+switch (Constants.Mode.valueOf(mode.toUpperCase())) {
+case NORMAL:
 DesiredCapabilities handlSSLErr = DesiredCapabilities.chrome();
 handlSSLErr.setCapability(CapabilityType.ACCEPT_SSL_CERTS, true);
 webDriver = new ChromeDriver(handlSSLErr);
 break;
-case "headless":
+case HEADLESS:
 options = new ChromeOptions();
 options.addArguments("--headless");
 options.addArguments("--start-maximized");
 webDriver = new ChromeDriver(options);
 break;
-case "grid":
+case GRID:
 String hubIPAddress = System.getProperty("env.hubIP");
 DesiredCapabilities dc = DesiredCapabilities.chrome();
 try {
@@ -143,7 +144,7 @@ private static WebDriver initChromeDriver() {
 throw new RuntimeException(e.getMessage());
 }
 break;
-case "incognito":
+case ICOGNITO:
 options = new ChromeOptions();
 options.addArguments("--incognito");
 webDriver = new ChromeDriver(options);