Новая организация селениум тестов

Author: PeeachPie

tests/selenium/selenium_session.py

@@ -3,9 +3,29 @@
 
 from selenium.webdriver import Chrome
 
+HOST = 'http://web:5000'
+ROOT_DIR = '/usr/src/project'
+
+from selenium.webdriver.chrome.options import Options
+
+def chrome_options(audio_file=None):
+    chrome_options = Options()
+
+    chrome_options.add_argument('--no-sandbox')
+    chrome_options.add_argument('--headless')
+    chrome_options.add_argument(f'--unsafely-treat-insecure-origin-as-secure={HOST}')
+    chrome_options.add_argument("--disable-user-media-security")
+    chrome_options.add_argument("--use-fake-device-for-media-stream")
+    chrome_options.add_argument("--use-fake-ui-for-media-stream")
+    
+    if audio_file is not None:
+        chrome_options.add_argument(f'--use-file-for-fake-audio-capture={audio_file}')
+
+    return chrome_options
+
 class SeleniumSession:
-    def __init__(self, host, config, chrome_options, requires_init=True):
-        self.__prepare_session(host, config, chrome_options, requires_init)
+    def __init__(self, config, chrome_options, requires_init=True):
+        self.__prepare_session(HOST, config, chrome_options, requires_init)
 
     def __init_driver(self, chrome_options):
         self.driver = Chrome(options=chrome_options)
@@ -30,7 +50,7 @@ def __registrate(self, config):
 
     def __prepare_session(self, host, config, chrome_options, requires_init):
         self.host = host
-        
+
         self.__init_driver(chrome_options)
 
         if requires_init:

tests/selenium/test_simple_training.py

@@ -1,41 +1,22 @@
-import os
 import pytest
 from time import sleep
 
-from selenium.webdriver.chrome.options import Options
-
+from selenium_session import ROOT_DIR, chrome_options
 from training_session import Training
 
 from app.config import Config
 
-HOST = 'http://web:5000'
-ROOT_DIR = os.getcwd()
-
 CONFIG_PATH = f'{ROOT_DIR}/app_conf/testing.ini'
 AUDIO_FILE = f"{ROOT_DIR}/simple_phrases_russian.wav"
-PRESENTATION_FILE = f"{ROOT_DIR}/test_data/test_presentation_file_0.pdf"
 
+PRESENTATION_FILE = f"{ROOT_DIR}/test_data/test_presentation_file_0.pdf"
 ESTIMATED_PROCESSING_TIME_IN_SECONDS = 100
 
-def chrome_options():
-    chrome_options = Options()
-
-    chrome_options.add_argument('--no-sandbox')
-    chrome_options.add_argument('--headless')
-    chrome_options.add_argument(f'--unsafely-treat-insecure-origin-as-secure={HOST}')
-
-    chrome_options.add_argument("--disable-user-media-security")
-    chrome_options.add_argument("--use-fake-device-for-media-stream")
-    chrome_options.add_argument("--use-fake-ui-for-media-stream")
-    chrome_options.add_argument(f'--use-file-for-fake-audio-capture={AUDIO_FILE}')
-
-    return chrome_options
-    
 @pytest.fixture(scope='module')
 def training_session():
     Config.init_config(CONFIG_PATH)
 
-    training_session = Training(HOST, Config.c, chrome_options())
+    training_session = Training(Config.c, chrome_options(AUDIO_FILE))
 
     yield training_session
 

tests/selenium/training_session.py

@@ -1,15 +1,15 @@
 from time import sleep
 
-from selenium_session import SeleniumSession
+from selenium_session import SeleniumSession, HOST
 
 from selenium.webdriver.common.by import By
 from selenium.webdriver.support import expected_conditions as EC
 from selenium.webdriver.support.ui import WebDriverWait
 
 
 class Training(SeleniumSession):
-    def __init__(self, host, config, chrome_options, requires_init=True):
-        super().__init__(host, config, chrome_options, requires_init)
+    def __init__(self, config, chrome_options, requires_init=True):
+        super().__init__(config, chrome_options, requires_init)
 
     def upload_presentation(self, presentation_path):
         self.driver.get(f'{self.host}/upload_presentation/')