| #! /usr/bin/env python3 |
| # |
| # BitBake Toaster functional tests implementation |
| # |
| # Copyright (C) 2017 Intel Corporation |
| # |
| # SPDX-License-Identifier: GPL-2.0-only |
| # |
| |
| import os |
| import logging |
| import subprocess |
| import signal |
| import re |
| |
| from tests.browser.selenium_helpers_base import SeleniumTestCaseBase |
| from selenium.webdriver.common.by import By |
| from selenium.common.exceptions import NoSuchElementException |
| |
| logger = logging.getLogger("toaster") |
| toaster_processes = [] |
| |
| class SeleniumFunctionalTestCase(SeleniumTestCaseBase): |
| wait_toaster_time = 10 |
| |
| @classmethod |
| def setUpClass(cls): |
| # So that the buildinfo helper uses the test database' |
| if os.environ.get('DJANGO_SETTINGS_MODULE', '') != \ |
| 'toastermain.settings_test': |
| raise RuntimeError("Please initialise django with the tests settings: " |
| "DJANGO_SETTINGS_MODULE='toastermain.settings_test'") |
| |
| # Wait for any known toaster processes to exit |
| global toaster_processes |
| for toaster_process in toaster_processes: |
| try: |
| os.waitpid(toaster_process, os.WNOHANG) |
| except ChildProcessError: |
| pass |
| |
| # start toaster |
| cmd = "bash -c 'source toaster start'" |
| start_process = subprocess.Popen( |
| cmd, |
| cwd=os.environ.get("BUILDDIR"), |
| shell=True) |
| toaster_processes = [start_process.pid] |
| if start_process.wait() != 0: |
| port_use = os.popen("lsof -i -P -n | grep '8000 (LISTEN)'").read().strip() |
| message = '' |
| if port_use: |
| process_id = port_use.split()[1] |
| process = os.popen(f"ps -o cmd= -p {process_id}").read().strip() |
| message = f"Port 8000 occupied by {process}" |
| raise RuntimeError(f"Can't initialize toaster. {message}") |
| |
| builddir = os.environ.get("BUILDDIR") |
| with open(os.path.join(builddir, '.toastermain.pid'), 'r') as f: |
| toaster_processes.append(int(f.read())) |
| with open(os.path.join(builddir, '.runbuilds.pid'), 'r') as f: |
| toaster_processes.append(int(f.read())) |
| |
| super(SeleniumFunctionalTestCase, cls).setUpClass() |
| cls.live_server_url = 'http://localhost:8000/' |
| |
| @classmethod |
| def tearDownClass(cls): |
| super(SeleniumFunctionalTestCase, cls).tearDownClass() |
| |
| global toaster_processes |
| |
| cmd = "bash -c 'source toaster stop'" |
| stop_process = subprocess.Popen( |
| cmd, |
| cwd=os.environ.get("BUILDDIR"), |
| shell=True) |
| # Toaster stop has been known to hang in these tests so force kill if it stalls |
| try: |
| if stop_process.wait(cls.wait_toaster_time) != 0: |
| raise Exception('Toaster stop process failed') |
| except Exception as e: |
| if e is subprocess.TimeoutExpired: |
| print('Toaster stop process took too long. Force killing toaster...') |
| else: |
| print('Toaster stop process failed. Force killing toaster...') |
| stop_process.kill() |
| for toaster_process in toaster_processes: |
| os.kill(toaster_process, signal.SIGTERM) |
| |
| |
| def get_URL(self): |
| rc=self.get_page_source() |
| project_url=re.search(r"(projectPageUrl\s:\s\")(.*)(\",)",rc) |
| return project_url.group(2) |
| |
| |
| def find_element_by_link_text_in_table(self, table_id, link_text): |
| """ |
| Assume there're multiple suitable "find_element_by_link_text". |
| In this circumstance we need to specify "table". |
| """ |
| try: |
| table_element = self.get_table_element(table_id) |
| element = table_element.find_element(By.LINK_TEXT, link_text) |
| except NoSuchElementException: |
| print('no element found') |
| raise |
| return element |
| |
| def get_table_element(self, table_id, *coordinate): |
| if len(coordinate) == 0: |
| #return whole-table element |
| element_xpath = "//*[@id='" + table_id + "']" |
| try: |
| element = self.driver.find_element(By.XPATH, element_xpath) |
| except NoSuchElementException: |
| raise |
| return element |
| row = coordinate[0] |
| |
| if len(coordinate) == 1: |
| #return whole-row element |
| element_xpath = "//*[@id='" + table_id + "']/tbody/tr[" + str(row) + "]" |
| try: |
| element = self.driver.find_element(By.XPATH, element_xpath) |
| except NoSuchElementException: |
| return False |
| return element |
| #now we are looking for an element with specified X and Y |
| column = coordinate[1] |
| |
| element_xpath = "//*[@id='" + table_id + "']/tbody/tr[" + str(row) + "]/td[" + str(column) + "]" |
| try: |
| element = self.driver.find_element(By.XPATH, element_xpath) |
| except NoSuchElementException: |
| return False |
| return element |