feat: add timeout for start of SUT

Add possibility to define timeout for SUT in ini file.
WithSecureOpenSource · Jan 4, 2022 · c296d90 · c296d90
1 parent 2c7106f
commit c296d90
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/change_analyzer/main.py b/change_analyzer/main.py
@@ -1,5 +1,5 @@
+import configparser
 import uuid
-from configparser import ConfigParser
 from datetime import datetime
 import argparse
 
@@ -14,7 +14,7 @@
 from change_analyzer.wrappers.sequence_recorder import SequenceRecorder
 
 
-CONFIG = ConfigParser()
+CONFIG = configparser.ConfigParser()
 
 
 def reset() -> WebDriver:
@@ -26,6 +26,12 @@ def reset() -> WebDriver:
             "deviceName": "WindowsPC",
         }
         driver = webdriver.Remote(CONFIG["driver"]["command_executor"], capabilities)
+        try:
+            timeout = CONFIG.get("driver", "timeout")
+            print(f"{timeout} second timeout defined.")
+            driver.implicitly_wait(timeout)
+        except configparser.NoOptionError:
+            print("No timeout defined.")
 
     if CONFIG["driver"]["platform"] == "web":
         web_options = webdriver.ChromeOptions()