openvidu/openvidu-server/docker/openvidu-deployment-tester/src/utils.py

import time
import os

from webdriver_manager.chrome import ChromeDriverManager
from webdriver_manager.firefox import GeckoDriverManager
from selenium.webdriver.chrome.service import Service as ChromeService
from selenium.webdriver.firefox.service import Service as FirefoxService
from selenium.webdriver.common.by import By
from selenium import webdriver
from bs4 import BeautifulSoup
from prettytable import from_html_one

def install_drivers(chrome_version, gecko_version):
    print("Downloading drivers...")
    # Download Chrome driver
    chrome_driver_manager = ChromeDriverManager(version=chrome_version)
    chrome_driver_manager.install()

    # Download Firefox driver
    firefox_driver_manager = GeckoDriverManager(version=gecko_version)
    firefox_driver_manager.install()

    with open("chrome_version", "w") as f:
        f.write(chrome_driver_manager.driver.get_version())
    with open("gecko_version", "w") as f:
        f.write(firefox_driver_manager.driver.get_version())

def authenticated_url(openvidu_url, openvidu_secret):
    separator = "://"
    basic_auth_url_str = "OPENVIDUAPP:" + openvidu_secret + "@"
    split_url = openvidu_url.split(separator)
    return split_url[0] + separator + basic_auth_url_str + split_url[1]

def runBrowser(browser, turn=False):
    driver=None
    if browser == "chrome":
        driver = runChrome()
    else:
        driver = runFirefox(turn)
    driver.maximize_window()
    driver.implicitly_wait(10)
    return driver

def runChrome():
    driver_location = os.environ.get("CHROME_DRIVER_LOCATION")

    # Load version from file
    chrome_version = None
    with open("chrome_version", "r") as f:
        chrome_version = f.read()

    print("Running chrome...")
    print("Chrome version: ", chrome_version)
    options = webdriver.ChromeOptions()
    options.add_argument("--use-fake-ui-for-media-stream")
    options.add_argument("--disable-infobars")
    options.add_argument("--ignore-certificate-errors")
    options.add_argument("--start-maximized")
    options.add_argument("--use-fake-device-for-media-stream")
    options.add_argument("--no-sandbox")

    return webdriver.Chrome(
        service=ChromeService(ChromeDriverManager(path=driver_location, version=chrome_version).install()),
        options = options)

def runFirefox(turn=False):
    driver_location = os.environ.get("GECKO_DRIVER_LOCATION")

    # Load version from file
    gecko_version = None
    with open("gecko_version", "r") as f:
        gecko_version = f.read()

    print("Running firefox with Turn: ", turn)
    print("Gecko version: ", gecko_version)
    options = webdriver.FirefoxOptions()
    options.set_preference('media.navigator.permission.disabled', True)
    options.set_preference('media.navigator.streams.fake', True)
    options.set_preference('media.peerconnection.enabled', True)
    options.set_preference('media.peerconnection.ice.obfuscate_host_addresses', False)
    options.set_preference('media.peerconnection.identity.enabled', True)
    options.set_preference('media.peerconnection.mtransport_process', True)
    options.set_preference('media.peerconnection.ice.no_host', False)
    options.set_preference('network.process.enabled', True)
    options.set_preference('media.peerconnection.ice.relay_only', turn)
    options.set_preference('media.peerconnection.turn.disable', not turn)

    return webdriver.Firefox(
        service=FirefoxService(GeckoDriverManager(path=driver_location, version=gecko_version).install()),
        options = options)

def print_candidates(driver):
    try:
        # New tab
        driver.execute_script("window.open('');")
        # Switch to the new window
        driver.switch_to.window(driver.window_handles[1])
        # Open about:webrtc
        driver.get('about:webrtc')
        peer_conn_elems = driver.find_elements(By.CLASS_NAME, "peer-connection")
        for peer_conn in peer_conn_elems:
            show_details_elems = peer_conn.find_elements(By.XPATH, "//*[contains(text(), 'show details')]")
            for show_details in show_details_elems:
                show_details.click()

        print("Waiting for candidates to be checked...")
        # Get ice stats
        time.sleep(15)
        # about:webrtc page refreshes each second, so we need to
        # safe the entire HTML in a variable to have a Snapshot of the situation
        about_webrtc_html = '<html>' + driver.find_element(By.TAG_NAME, 'html').get_attribute('innerHTML') + '</html>'
        # Search the tables using a parser and print all candidates
        soup = BeautifulSoup(about_webrtc_html, 'html.parser')
        for caption in soup.findAll('caption', {'data-l10n-id' : 'about-webrtc-trickle-caption-msg'}):
            print(from_html_one(str(caption.parent)))
        # Close about:webrtc
        driver.close()
        driver.switch_to.window(driver.window_handles[0])
        driver.close()
    except:
        print('[Warn] Some candidates may not appear in test result')
        driver.switch_to.window(driver.window_handles[0])
        driver.close()