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

Author: PeeachPie

tests/selenium/selenium_session.py

@@ -0,0 +1,42 @@
+import requests
+from time import sleep
+
+from selenium.webdriver import Chrome
+
+class SeleniumSession:
+    def __init__(self, host, 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)
+        self.session = requests.Session()
+
+        sleep(5)
+        
+    def __registrate(self, config):
+        self.session.request('POST',f'{self.host}/lti', data={
+            'lis_person_name_full': config.testing.lis_person_name_full,
+            'ext_user_username': config.testing.session_id,
+            'custom_task_id': config.testing.custom_task_id,
+            'custom_task_description': config.testing.custom_task_description,
+            'custom_attempt_count': config.testing.custom_attempt_count,
+            'custom_required_points': config.testing.custom_required_points,
+            'custom_criteria_pack_id': config.testing.custom_criteria_pack_id,
+            'roles': config.testing.roles,
+            'lis_outcome_service_url': config.testing.lis_outcome_service_url,
+            'lis_result_sourcedid': config.testing.lis_result_source_did,
+            'oauth_consumer_key': config.testing.oauth_consumer_key,
+        })
+
+    def __prepare_session(self, host, config, chrome_options, requires_init):
+        self.host = host
+        
+        self.__init_driver(chrome_options)
+
+        if requires_init:
+            self.driver.get(f'{self.host}/init/')
+
+        self.__registrate(config)
+
+    def end_session(self):
+        self.driver.quit()

tests/selenium/test_simple_training.py

@@ -0,0 +1,64 @@
+import os
+import pytest
+from time import sleep
+
+from selenium.webdriver.chrome.options import 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"
+
+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())
+
+    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()
+
+    sleep(5)
+
+def test_training_session_end(training_session):
+    training_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} секунд"

tests/selenium/test_training.py

@@ -0,0 +1,53 @@
+from time import sleep
+
+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, host, config, chrome_options, requires_init=True):
+        super().__init__(host, config, chrome_options, requires_init)
+
+    def upload_presentation(self, presentation_path):
+        self.driver.get(f'{self.host}/upload_presentation/')
+
+        file_input = WebDriverWait(self.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()
+    
+    def prepare_record(self):
+        WebDriverWait(self.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.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()
+
+    def end_training(self):
+        WebDriverWait(self.driver, 5).until(EC.element_to_be_clickable((By.ID, "done"))).click()
+
+        WebDriverWait(self.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()
+
+            feedback_elements = self.driver.find_elements(By.ID, 'feedback')
+
+            if feedback_elements and feedback_elements[0].text.startswith('Оценка за тренировку'):
+                feedback_flag = True
+                break
+            
+            sleep(step)
+
+        return feedback_flag