第二个python自动化练习

#Author:xiaoxiao

from selenium import webdriver
import unittest

class DownLoad(unittest.TestCase):
    #初始化工作
    def setUp(self):
        #打开浏览器
        self.driver = webdriver.Firefox()
        self.driver.implicitly_wait(10)
        self.driver.get("http://photo.163.com")
    def testDownLoaf(self):
        #找到了解更多按钮并点击
        knowbutton = self.driver.find_element_by_link_text("了解更多")
        knowbutton.click()
        #获取窗口句柄
        windowhanle = self.driver.current_window_handle
        windowhanles = self.driver.window_handles
        for handle in windowhanles:
            if handle != windowhanle:
                self.driver.switch_to.window(handle)
        #断言验证
        self.assertEqual(self.driver.current_url,"http://photo.163.com/cloudphotos/","点击错误")
        #获取iphone按钮并点击
        iphonebutton = self.driver.find_element_by_id("J-iphone")
        iphonebutton.click()
        con = self.driver.find_element_by_id("J-xbox-title").text
        #断言验证
        self.assertEqual(con,"iPhone版下载","验证iphone版下载错误")
    def tearDown(self):
        #退出浏览器
        self.driver.quit()

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