clickalllinksonpage.py

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.ui import Select
from selenium.common.exceptions import NoSuchElementException
import unittest, time, re, sys

baseurl = sys.argv[1]

class Clickalllinksonpage(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Firefox()
        self.driver.implicitly_wait(30)
        self.base_url = ""
        self.verificationErrors = []
        self.accept_next_alert = True
	self.links = []
	    
    def test_clickalllinksonpage(self):
        driver = self.driver
        driver.get(baseurl)
    	links = driver.find_elements_by_xpath('//a')
    	size = len(links)
    	print(size)
        for i in range(0, size-1):
            links = driver.find_elements_by_xpath('//a')
            print(links[i].get_attribute("href"))
            links[i].click()
            time.sleep(5)
            print(str(driver.title))
            #go back
            driver.get(baseurl)
            self.driver.implicitly_wait(30)
            
    def is_element_present(self, how, what):
        try: self.driver.find_element(by=how, value=what)
        except NoSuchElementException, e: return False
        return True
    
    def is_alert_present(self):
        try: self.driver.switch_to_alert()
        except NoAlertPresentException, e: return False
        return True
    
    def close_alert_and_get_its_text(self):
        try:
            alert = self.driver.switch_to_alert()
            alert_text = alert.text
            if self.accept_next_alert:
                alert.accept()
            else:
                alert.dismiss()
            return alert_text
        finally: self.accept_next_alert = True
    
    def tearDown(self):
        self.driver.quit()
        self.assertEqual([], self.verificationErrors)

if __name__ == "__main__":
    unittest.main(argv=[sys.argv[0]])