HOME/Articles/

selenium GetTextOrAttribute (snippet)

Article Outline

Python web automation, selenium example 'GetTextOrAttribute'

Modules used in program:

  • import time
  • import unittest

GetTextOrAttribute

Python selenium example: GetTextOrAttribute

from selenium                           import webdriver

from selenium.webdriver.support.ui      import WebDriverWait

from selenium.webdriver.common.by       import By

from selenium.webdriver.support.select  import Select

import unittest

import time


class GetTextOrAttribute(unittest.TestCase):

    def setUp(self):
        global driver
        driver = webdriver.Firefox()
        driver.get("http://travelingtony.weebly.com")
        driver.maximize_window()

    def test_GetTextOrAttribute(self):
        seePhotosButtonLocator = "(//span[@class='wsite-button-inner'])[1]"
        seePhotosButtonElement = WebDriverWait(driver, 10).\
                                 until(lambda driver: driver.find_element_by_xpath(seePhotosButtonLocator))
        text = seePhotosButtonElement.text
        print(text  #Prints "CHECK OUT MY COOLEST PHOTOS")
        classAttribute = seePhotosButtonElement.get_attribute("class")
        print(classAttribute #Prints "wsite-button-inner")

    def tearDown(self):
        driver.quit()

if __name__ == "__main__":
   unittest.main()