IE support added

Author: chavarera

s_tool/driver.py

@@ -3,6 +3,7 @@
 from selenium import webdriver
 from webdriver_manager.chrome import ChromeDriverManager
 from webdriver_manager.firefox import GeckoDriverManager
+from webdriver_manager.microsoft import IEDriverManager
 from webdriver_manager.utils import ChromeType
 
 from s_tool.exceptions import SToolException
@@ -27,7 +28,7 @@ class SeleniumDriver:
     """SeleniumDriver class to manage driver object and all utility functions at one place"""
 
     def __init__(self, browser=None, headless=False):
-        self.browser_list = ["chrome", "chromium", "firefox"]
+        self.browser_list = ["chrome", "chromium", "firefox", "ie"]
         self.driver = None
         self.browser = browser
         self.headless = headless
@@ -76,6 +77,11 @@ def _load_driver(self):
                 executable_path=GeckoDriverManager().install(), options=options
             )
 
+        if browser == "ie":
+            self.driver = webdriver.Ie(
+                executable_path=IEDriverManager().install(), options=options
+            )
+
         # Maximize window to give normal browser feel
         self.driver.maximize_window()
 

tests/test_driver.py

@@ -31,6 +31,8 @@ def test_base():
         URL = "https://www.google.com/"
         obj.get(URL)
 
+        # Verify driver
+        assert "ie" in obj.browser_list
         # session
         assert type(obj.session()) is str