Новая структура селениум тестов

Author: PeeachPie

tests/selenium/conftest.py

@@ -0,0 +1,18 @@
+import pytest
+
+from selenium_session import SeleniumSession, ROOT_DIR, chrome_options
+from app.config import Config
+
+CONFIG_PATH = f'{ROOT_DIR}/app_conf/testing.ini'
+AUDIO_FILE = f"{ROOT_DIR}/simple_phrases_russian.wav"
+
+@pytest.fixture(scope="class")
+def selenium_session(request):
+    Config.init_config(CONFIG_PATH)
+    session = SeleniumSession(Config.c, chrome_options(AUDIO_FILE))
+
+    request.cls.selenium_session = session
+
+    yield session
+
+    session.end_session()

tests/selenium/test_simple_training.py

@@ -1,45 +1,29 @@
 import pytest
 from time import sleep
 
-from selenium_session import ROOT_DIR, chrome_options
+from selenium_session import ROOT_DIR
 from training_session import Training
 
-from app.config import Config
-
-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"
 ESTIMATED_PROCESSING_TIME_IN_SECONDS = 100
 
-@pytest.fixture(scope='module')
-def training_session():
-    Config.init_config(CONFIG_PATH)
-
-    training_session = Training(Config.c, chrome_options(AUDIO_FILE))
-
-    yield training_session
-
-    training_session.end_session()
-
-def test_presentation_upload(training_session):
-    training_session.upload_presentation(PRESENTATION_FILE)
-
-def test_record_preparation(training_session):
-    training_session.prepare_record()
-
-    sleep(5)
-
-def test_button_next(training_session):
-    training_session.next_slide()
+@pytest.mark.usefixtures("selenium_session")
+class TestSimpleTraining:
+    def test_presentation_upload(self):
+        Training(self.selenium_session).upload_presentation(PRESENTATION_FILE)
 
-    sleep(5)
+    def test_record_preparation(self):
+        Training(self.selenium_session).prepare_record()
+        sleep(5)
 
-def test_training_session_end(training_session):
-    training_session.end_training()
+    def test_button_next(self):
+        Training(self.selenium_session).next_slide()
+        sleep(5)
 
-    sleep(5)
+    def test_training_session_end(self):
+        Training(self.selenium_session).end_training()
+        sleep(5)
 
-def test_training_feedback(training_session):
-    got_feegback = training_session.wait_for_feedback(ESTIMATED_PROCESSING_TIME_IN_SECONDS)
-    assert got_feegback, f"Проверка тренировки заняла более {ESTIMATED_PROCESSING_TIME_IN_SECONDS} секунд"
+    def test_training_feedback(self):
+        got_feedback = Training(self.selenium_session).wait_for_feedback(ESTIMATED_PROCESSING_TIME_IN_SECONDS)
+        assert got_feedback, f"Проверка тренировки заняла более {ESTIMATED_PROCESSING_TIME_IN_SECONDS} секунд"

tests/selenium/training_session.py

@@ -1,48 +1,48 @@
 from time import sleep
 
-from selenium_session import SeleniumSession, HOST
+from selenium_session import SeleniumSession
 
 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, config, chrome_options, requires_init=True):
-        super().__init__(config, chrome_options, requires_init)
+class Training:
+    def __init__(self, selenium: SeleniumSession):
+        self.selenium = selenium
 
     def upload_presentation(self, presentation_path):
-        self.driver.get(f'{self.host}/upload_presentation/')
+        self.selenium.driver.get(f'{self.selenium.host}/upload_presentation/')
 
-        file_input = WebDriverWait(self.driver, 20).until(EC.visibility_of_element_located((By.CSS_SELECTOR, "input[type=file]")))
+        file_input = WebDriverWait(self.selenium.driver, 20).until(EC.visibility_of_element_located((By.CSS_SELECTOR, "input[type=file]")))
         file_input.send_keys(presentation_path)
 
-        WebDriverWait(self.driver, 5).until(EC.element_to_be_clickable((By.ID, "button-submit"))).click()
+        WebDriverWait(self.selenium.driver, 5).until(EC.element_to_be_clickable((By.ID, "button-submit"))).click()
 
     def prepare_record(self):
-        WebDriverWait(self.driver, 10).until(EC.element_to_be_clickable((By.ID, "record"))).click()
+        WebDriverWait(self.selenium.driver, 10).until(EC.element_to_be_clickable((By.ID, "record"))).click()
 
-        WebDriverWait(self.driver, 10).until(EC.presence_of_element_located((By.ID, "model-timer")))
+        WebDriverWait(self.selenium.driver, 10).until(EC.presence_of_element_located((By.ID, "model-timer")))
 
-        WebDriverWait(self.driver, 10).until(EC.invisibility_of_element((By.ID, "model-timer")))
+        WebDriverWait(self.selenium.driver, 10).until(EC.invisibility_of_element((By.ID, "model-timer")))
 
     def next_slide(self):
-        WebDriverWait(self.driver, 10).until(EC.element_to_be_clickable((By.ID, "next"))).click()
+        WebDriverWait(self.selenium.driver, 10).until(EC.element_to_be_clickable((By.ID, "next"))).click()
 
     def end_training(self):
-        WebDriverWait(self.driver, 5).until(EC.element_to_be_clickable((By.ID, "done"))).click()
+        WebDriverWait(self.selenium.driver, 5).until(EC.element_to_be_clickable((By.ID, "done"))).click()
 
-        WebDriverWait(self.driver, 5).until(lambda d : d.switch_to.alert).accept()
+        WebDriverWait(self.selenium.driver, 5).until(lambda d : d.switch_to.alert).accept()
 
     def wait_for_feedback(self, seconds):
         feedback_flag = False
         step_count = 10
         step = seconds / step_count
 
         for _ in range(step_count):
-            self.driver.refresh()
+            self.selenium.driver.refresh()
 
-            feedback_elements = self.driver.find_elements(By.ID, 'feedback')
+            feedback_elements = self.selenium.driver.find_elements(By.ID, 'feedback')
 
             if feedback_elements and feedback_elements[0].text.startswith('Оценка за тренировку'):
                 feedback_flag = True