mirror of https://github.com/OpenVidu/openvidu.git
150 lines
6.2 KiB
Python
150 lines
6.2 KiB
Python
#!/usr/bin/python3
|
|
import unittest
|
|
from selenium import webdriver
|
|
from selenium.webdriver import firefox
|
|
from selenium.webdriver.firefox import service
|
|
from webdriver_manager.chrome import ChromeDriverManager
|
|
from webdriver_manager.firefox import GeckoDriverManager
|
|
from selenium.webdriver.chrome import options
|
|
from selenium.webdriver.chrome.service import Service as ChromeService
|
|
from selenium.webdriver.firefox.service import Service as FirefoxService
|
|
from selenium.webdriver.common.keys import Keys
|
|
from selenium.webdriver.common.by import By
|
|
from prettytable import from_html_one
|
|
import time
|
|
import os
|
|
|
|
class InfraSmokeTests(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.openvidu_url = os.getenv('OV_URL')
|
|
self.openvidu_password = os.getenv('OV_SECRET')
|
|
self.driver = None
|
|
|
|
def test_inspector(self):
|
|
self.inspector_check(browser="chrome")
|
|
self.inspector_check(browser="firefox")
|
|
self.inspector_check(browser="firefox", turn=True)
|
|
|
|
def inspector_check(self, browser="chrome", turn=False):
|
|
print('\n\n======================================================================')
|
|
print('|')
|
|
print('|')
|
|
print('| Testing OpenVidu with ' + browser + ' and force relay: ' + str(turn))
|
|
print('|')
|
|
print('|')
|
|
print('======================================================================')
|
|
if self.openvidu_url == None or self.openvidu_password == None:
|
|
raise(Exception("You must specify OV_URL and OV_SECRET environment variables"))
|
|
if browser == "chrome":
|
|
self.runChrome()
|
|
else:
|
|
self.runFirefox(turn)
|
|
|
|
url_test = self.openvidu_url + '/inspector'
|
|
self.driver.get(url_test)
|
|
|
|
elem = self.driver.find_element(By.ID, 'secret-input')
|
|
elem.send_keys(self.openvidu_password)
|
|
|
|
elem = self.driver.find_element(By.ID, 'login-btn')
|
|
elem.send_keys(Keys.RETURN)
|
|
|
|
# print('data:image/png;base64,' + self.driver.get_screenshot_as_base64())
|
|
elem = self.driver.find_element(By.ID,'menu-test-btn')
|
|
elem.send_keys(Keys.RETURN)
|
|
|
|
elem = self.driver.find_element(By.ID,'test-btn')
|
|
elem.send_keys(Keys.RETURN)
|
|
|
|
video_error = False
|
|
try:
|
|
self.driver.find_element(By.XPATH, "//*[contains(text(), 'Stream playing')]")
|
|
except:
|
|
video_error = True
|
|
finally:
|
|
# print('data:image/png;base64,' + self.driver.get_screenshot_as_base64())
|
|
if browser == "firefox":
|
|
self.print_candidates()
|
|
|
|
if video_error == True:
|
|
raise Exception('Error. No video detected')
|
|
|
|
print('Video detected.\n')
|
|
elem = self.driver.find_element(By.ID,'test-btn')
|
|
elem.send_keys(Keys.RETURN)
|
|
|
|
self.closeBrowser()
|
|
print('Sucess with ' + browser + ' and Force Turn: ' + str(turn) + '\n')
|
|
print('----------------------------------------------------------------------\n')
|
|
|
|
def runChrome(self):
|
|
self.options = webdriver.ChromeOptions()
|
|
self.options.add_argument("--use-fake-ui-for-media-stream")
|
|
self.options.add_argument("--disable-infobars")
|
|
self.options.add_argument("--ignore-certificate-errors")
|
|
self.options.add_argument("--start-maximized")
|
|
self.options.add_argument("--use-fake-device-for-media-stream")
|
|
self.options.add_argument("--no-sandbox")
|
|
|
|
self.driver = webdriver.Chrome(
|
|
service=ChromeService(ChromeDriverManager().install()),
|
|
options = self.options)
|
|
self.driver.implicitly_wait(15)
|
|
|
|
def runFirefox(self, turn=False):
|
|
print("Running firefox with Turn: ", turn)
|
|
self.options = webdriver.FirefoxOptions()
|
|
self.options.set_preference('media.navigator.permission.disabled', False)
|
|
self.options.set_preference('media.navigator.streams.fake', True)
|
|
self.options.set_preference('media.peerconnection.enabled', True)
|
|
self.options.set_preference('media.peerconnection.ice.obfuscate_host_addresses', False)
|
|
self.options.set_preference('media.peerconnection.identity.enabled', True)
|
|
self.options.set_preference('media.peerconnection.mtransport_process', True)
|
|
self.options.set_preference('media.peerconnection.ice.no_host', False)
|
|
self.options.set_preference('network.process.enabled', True)
|
|
self.options.set_preference('media.peerconnection.ice.relay_only', turn)
|
|
self.options.set_preference('media.peerconnection.turn.disable', not turn)
|
|
|
|
self.driver = webdriver.Firefox(
|
|
service=FirefoxService(GeckoDriverManager().install()),
|
|
options = self.options)
|
|
self.driver.implicitly_wait(5)
|
|
self.driver.maximize_window()
|
|
|
|
def print_candidates(self):
|
|
try:
|
|
# New tab
|
|
self.driver.execute_script("window.open('');")
|
|
# Switch to the new window
|
|
self.driver.switch_to.window(self.driver.window_handles[1])
|
|
# Open about:webrtc
|
|
self.driver.get('about:webrtc')
|
|
peer_conn_elems = self.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)
|
|
ice_stats_div_elems = self.driver.find_elements(By.XPATH, "//div[contains(@id, 'ice-stats')]")
|
|
for ice_stats_div in ice_stats_div_elems:
|
|
table_elems = ice_stats_div.find_elements(By.TAG_NAME, 'table')
|
|
ice_candidates_table = table_elems[0]
|
|
html_ice_table = '<table>' + ice_candidates_table.get_attribute('innerHTML') + '</table>'
|
|
print(from_html_one(html_ice_table))
|
|
# Go to main window
|
|
self.driver.switch_to.window(self.driver.window_handles[0])
|
|
except:
|
|
print('Error getting candidates')
|
|
|
|
def closeBrowser(self):
|
|
# close the browser window
|
|
self.driver.close()
|
|
self.driver.quit()
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|