rhino/binding/web/tests/en-factory/selenium_test.py

Go to the documentation of this file.

#!/usr/bin/python3
 
import os
import sys
import threading
import time
from argparse import ArgumentParser
from http.server import HTTPServer, SimpleHTTPRequestHandler
 
from selenium import webdriver
from selenium.common.exceptions import WebDriverException
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from webdriver_manager.chrome import ChromeDriverManager
 
 
class SimpleHttpServer(threading.Thread):
    def __init__(self, host='localhost', port=4001, path='.'):
        self._base_url = f'http://{host}:{port}'
        self._root_path = path
        self._cwd = os.getcwd()
        self._server = HTTPServer((host, port), SimpleHTTPRequestHandler)
        super().__init__(daemon=True)
 
    @property
    def base_url(self):
        return self._base_url
 
    def run(self):
        os.chdir(self._root_path)
        print(f'starting server on port {self._server.server_port}')
        self._server.serve_forever()
 
    def stop(self):
        os.chdir(self._cwd)
        self._server.shutdown()
        self._server.socket.close()
        print(f'stopping server on port {self._server.server_port}')
 
 
def run_unit_test_selenium(url, access_key, absolute_ic_audio_file, absolute_ooc_audio_file):
    desired_capabilities = DesiredCapabilities.CHROME
    desired_capabilities['goog:loggingPrefs'] = {'browser': 'ALL'}
    opts = Options()
    opts.headless = True
    driver = webdriver.Chrome(ChromeDriverManager().install(), desired_capabilities=desired_capabilities, options=opts)
 
    driver.get(url)
    assert "unit test" in driver.title
 
    wait = WebDriverWait(driver, 3600)
 
    driver.find_element_by_id("inContextAudioFile").send_keys(absolute_ic_audio_file)
    wait.until(EC.visibility_of_element_located((By.ID, "inContextAudioLoaded")))
 
    driver.find_element_by_id("oocAudioFile").send_keys(absolute_ooc_audio_file)
    wait.until(EC.visibility_of_element_located((By.ID, "oocAudioLoaded")))
 
    driver.find_element_by_id("accessKey").send_keys(access_key)
    driver.find_element_by_id("submit").click()
    wait.until(EC.visibility_of_element_located((By.ID, "testComplete")))
 
    test_result = 1
    test_message = "Tests failed"
    for entry in driver.get_log('browser'):
        print(entry['message'])
        if 'Test passed!' in entry['message']:
            test_message = "Tests passed"
            test_result = 0
 
    driver.close()
    print(test_message)
    return test_result
 
 
def main():
    parser = ArgumentParser()
 
    parser.add_argument(
        '--access_key',
        required=True)
    parser.add_argument(
        '--audio_file_in_context',
        required=True)
    parser.add_argument(
        '--audio_file_out_context',
        required=True)
 
    args = parser.parse_args()
 
    absolute_ic_audio_file = os.path.abspath(args.audio_file_in_context)
    absolute_ooc_audio_file = os.path.abspath(args.audio_file_out_context)
 
    simple_server = SimpleHttpServer(port=4005, path=os.path.join(os.path.dirname(__file__), '..', '..'))
    test_url = f'{simple_server.base_url}/rhino-web-en-factory/test/index.html'
    simple_server.start()
    time.sleep(4)
 
    result = 0
    try:
        result = run_unit_test_selenium(test_url,
                                        args.access_key,
                                        absolute_ic_audio_file,
                                        absolute_ooc_audio_file)
    except Exception as e:
        print(e)
        result = 1
    finally:
        simple_server.stop()
        sys.exit(result)
 
 
if __name__ == '__main__':
    main()