Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import os
- import unittest
- import time
- from selenium import webdriver
- class TestOne(unittest.TestCase):
- def setUp(self):
- self.base_route = os.environ['HOST_URL']
- print('HOST_URL title is {}'.format(self.base_route))
- options = webdriver.ChromeOptions()
- options.add_argument('--headless')
- options.add_argument("--no-sandbox")
- self.driver = webdriver.Chrome(options=options)
- self.driver.set_window_size(1120, 550)
- def test_url(self):
- self.driver.get(self.base_route)
- time.sleep(30)
- os.environ['PDSS_TITLE'] = self.driver.find_element_by_class_name('title').text
- title = self.driver.find_element_by_class_name('title').text
- assert os.environ['PDSS_TITLE'] in title
- def tearDown(self):
- self.driver.quit()
- if __name__ == '__main__':
- unittest.main()
Add Comment
Please, Sign In to add comment