aboutsummaryrefslogtreecommitdiffhomepage
path: root/tests/base.py
blob: b9a026f2e1e122992a32350c92d606e5bd19ec3a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
# use pip install selenium first
# ensure you have both chome driver & chrome installed

import unittest
from selenium import webdriver
from selenium.common.exceptions import NoSuchElementException

class Base(unittest.TestCase):
    """Include test cases on a given url"""

    def start(self, path='main'):
        self.driver.get('http://localhost:8080/' + path)
    def xpath(self, path):
        return self.driver.find_element_by_xpath('/html/body/'+path)
    def body_text(self):
        return self.driver.find_element_by_xpath('/html/body').text

    def setUp(self):
        """Start web driver"""
        chrome_options = webdriver.ChromeOptions()
        chrome_options.add_argument('--no-sandbox')
        chrome_options.add_argument('--headless')
        chrome_options.add_argument('--disable-gpu')
        self.driver = webdriver.Chrome(options=chrome_options)
        self.driver.implicitly_wait(10)

    def tearDown(self):
        """Stop web driver"""
        self.driver.quit()