python webdriver 简单框架_python webdriver 简单应用-CSDN博客

本文链接：https://blog.csdn.net/fqlike/article/details/51026992

以百度搜索为例
一、首先写百度的元素定位类，后期可以用excel表代替，命名为locators.py

# -*- coding: utf-8 -*-
from selenium.webdriver.common.by import By
class BaiduLocator(object):
    Search_Input = (By.ID,"kw")
    Search_Submit_Button  = (By.ID,"su")
#other locator ....
class BaiduUserLocator(object):
    """Baidu User Locator"""
    pass

二、写关于百度页面的操作类,命名为page.py

# -*- coding: utf-8 -*-
from locators import *
import sys
reload(sys)
sys.setdefaultencoding('utf8')

class BasePage(object):
    """基类"""
    def __init__(self,deriver):
        self.driver = driver
class BaiduSearchPage(BasePage):
    """百度搜索，所有页面都继承于BasePage"""
    def input_search(self):
        #输入关键词
        search_input = self.driver.find_element(*BaiduLocator.Search_Input)
        search_input.clear()
        search_input.send_keys("selenium")
    def click_su(self):
        #点击搜索
        su_button = self.driver.find_element(*BaiduLocator.Search_Submit_Button).click()
    def is_title_match(self):
        #断言
        return "selenium" in self.driver.title

三、引入unittest测试框架，生成测试用例，此文件命名为baidusearch.py

# -*- coding: utf-8 -*-
from selenium import webdriver
import page,unittest,time
reload(sys)
sys.setdefaultencoding('utf8')

class Baidu(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Chrome() #选择浏览器，根据需要自己选择
        #self.driver = webdriver.Firefox()
        self.driver.implicitly_wait(30)
        self.base_url = "http://www.baidu.com"
        self.driver.get(self.base_url)
        self.verificationErrors = []
        self.accept_next_alert = True
    def test_baidu_search(self):
        '''baidu search test'''
        search_page = page.BaiduSearchPage(self.driver) #初始化百度页面
        search_page.input_search() #输入关键词
        time.sleep(1) #等待1秒钟
        search_page.click_su() #点击搜索
        self.assertTrue(search_page.is_title_match(),"selenium is not in title") #断言

    def tearDown(self):
        self.driver.quit() #清环境
if __name__ == "__main__":
    unittest.main()