diff --git a/test_script.py b/test_script.py index afc7753..41b6448 100644 --- a/test_script.py +++ b/test_script.py @@ -26,7 +26,7 @@ def test_case_1(self): """Find and click top-right button""" try: self.driver.get('https://www.oursky.com/') - el = self.driver.find_element_by_class_name('btn-header') + el = self.driver.find_element_by_class_name('header__cta') el.click() except NoSuchElementException as ex: self.fail(ex.msg) @@ -35,7 +35,7 @@ def test_case_2(self): """Find and click Learn more button""" try: self.driver.get('https://www.oursky.com/') - el = self.driver.find_element_by_xpath(".//*[@id='tag-line-wrap']/span/a") + el = self.driver.find_element_by_xpath(".//*[@id='landing__btn']/span") el.click() except NoSuchElementException as ex: self.fail(ex.msg)