From fd3285b561294c780b86afaf462fb0c6bde330ef Mon Sep 17 00:00:00 2001
From: Taka <taka@vm-linux-dev>
Date: Sat, 18 Aug 2018 13:15:52 +0000
Subject: [PATCH] Added support for the IE WebDriver.

---
 conftest.py                          | 11 ++++++++++-
 vasl_templates/webapp/tests/utils.py |  2 ++
 2 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/conftest.py b/conftest.py
index 72039ee..d5326d5 100644
--- a/conftest.py
+++ b/conftest.py
@@ -142,8 +142,17 @@ def webdriver( request ):
         options = wb.ChromeOptions()
         options.set_headless( headless = pytest.config.option.headless ) #pylint: disable=no-member
         driver = wb.Chrome( chrome_options=options )
+    elif driver == "ie":
+        # NOTE: IE11 requires a registry key to be set:
+        #   https://github.com/SeleniumHQ/selenium/wiki/InternetExplorerDriver#required-configuration
+        options = wb.IeOptions()
+        if pytest.config.option.headless: #pylint: disable=no-member
+            raise RuntimeError( "IE WebDriver cannot be run headless." )
+        options.IntroduceInstabilityByIgnoringProtectedModeSettings = True
+        options.EnsureCleanSession = True
+        driver = wb.Ie( ie_options=options )
     else:
-        assert False, "Unknown webdriver: {}".format( driver )
+        raise RuntimeError( "Unknown webdriver: {}".format( driver ) )
 
     # set the browser size
     words = pytest.config.option.window_size.split( "x" ) #pylint: disable=no-member
diff --git a/vasl_templates/webapp/tests/utils.py b/vasl_templates/webapp/tests/utils.py
index 71bc7f8..88438ae 100644
--- a/vasl_templates/webapp/tests/utils.py
+++ b/vasl_templates/webapp/tests/utils.py
@@ -361,6 +361,8 @@ def get_clipboard() :
 
 def wait_for( timeout, func ):
     """Wait for a condition to become true."""
+    if os.name == "nt":
+        timeout *= 2 # Selenium runs pretty slow on Windows :-/
     start_time = time.time()
     while True:
         if func():