2 files changed, 12 insertions, 3 deletions
diff --git a/src/main/java/com/mavlushechka/a1qa/driverUtils/WebDriverFactory.java b/src/main/java/com/mavlushechka/a1qa/driverUtils/WebDriverFactory.java
index fc311b0..e259526 100644
--- a/src/main/java/com/mavlushechka/a1qa/driverUtils/WebDriverFactory.java
+++ b/src/main/java/com/mavlushechka/a1qa/driverUtils/WebDriverFactory.java
@@ -1,5 +1,6 @@
 package com.mavlushechka.a1qa.driverUtils;
 
+import com.mavlushechka.a1qa.models.Browser;
 import com.mavlushechka.a1qa.utils.JSONParser;
 import com.mavlushechka.a1qa.utils.LoggerUtils;
 import com.mavlushechka.a1qa.utils.StringUtils;
@@ -16,12 +17,12 @@ public class WebDriverFactory {
     }
 
     protected static WebDriver createWebDriver() {
-        switch (StringUtils.convertToConstantCase(JSONParser.parseData("config", "browser.name"))) {
-            case "CHROME" -> {
+        switch (Browser.valueOf(StringUtils.convertToConstantCase(JSONParser.parseData("config", "browser.name")))) {
+            case CHROME -> {
                 LoggerUtils.info("Creating ChromeDriver.");
                 return createChromeDriver();
             }
-            case "FIREFOX" -> {
+            case FIREFOX -> {
                 LoggerUtils.info("Creating FirefoxDriver.");
                 return createFirefoxDriver();
             }
diff --git a/src/main/java/com/mavlushechka/a1qa/models/Browser.java b/src/main/java/com/mavlushechka/a1qa/models/Browser.java
new file mode 100644
index 0000000..013236b
--- /dev/null
+++ b/src/main/java/com/mavlushechka/a1qa/models/Browser.java
@@ -0,0 +1,8 @@
+package com.mavlushechka.a1qa.models;
+
+
+public enum Browser {
+
+    CHROME, FIREFOX
+
+}