def setUpClass(cls):
logger.debug("setUp...")
# self.driver = Chrome.normal()
cls.driver = Chrome.headless()
# cls.driver = webdriver.Firefox()
cls.driver.implicitly_wait(10)
cls.driver.maximize_window()
page = LoginPage(cls.driver)
page.login()
def fetch_links():
driver = Chrome.headless()
page = LoginPage(driver)
page.login()
links = []
link_elements = driver.find_elements_by_xpath(
'//a[not(contains(@href,"javascript"))]')
for element in link_elements:
link = element.get_attribute("href")
if link:
link = link.split('/')[-3:]
link = [os.path.join(link[0], link[1]), link[2]]
links.append(link)
return links
