selenium之使用unittest測試框架

# 測試角色權限管理頁面功能
from selenium import webdriver
from login_page import LoginPage
import random, time, unittest


class TestRole(unittest.TestCase):
    def setUp(self):
        self.login_obj = LoginPage()
        self.driver = webdriver.Chrome()
        driver = self.driver
        login_obj = self.login_obj
        time.sleep(2)
        driver = login_obj.login(driver)  # 登錄
        driver.find_element_by_id('4').click()  # 進入監控管理
        time.sleep(1)
        # 進入角色管理
        driver.find_element_by_xpath('//a[@url="manager/role/role_list.do"]').click()
        time.sleep(2)

        # 定義變量
        random_num_to_str = str(random.randint(1000, 9999))  # 隨機數，保證不重復
        self.role_name = 'role_' + random_num_to_str  # 角色名，限制10個字符
        self.spec_context = 'spec_' + random_num_to_str  # 備注信息
        self.modify_spec_context = self.spec_context + '_modify'  # 修改后的備注信息

    def tearDown(self):
        '''關閉瀏覽器'''
        self.login_obj.quit()

    def test_role(self):
        driver = self.driver

        def create_role():
            '''新增角色'''
            driver.find_element_by_id('addRow').click()
            driver.find_element_by_id('role_name').send_keys(self.role_name)
            driver.find_element_by_id('spec').send_keys(self.spec_context)
            time.sleep(3)
            driver.find_element_by_id('userSave').click()
            time.sleep(3)

        def query_role_by_name():
            '''查詢角色'''
            driver.find_element_by_id('searchKey').send_keys(self.role_name)
            driver.find_element_by_id('search').click()
            time.sleep(3)

        def edit_role():
            '''編輯角色'''
            driver.find_element_by_xpath("//button[@title='編輯']").click()  # 查找title='編輯'的button
            driver.find_element_by_id('spec').clear()
            driver.find_element_by_id('spec').send_keys(self.modify_spec_context)
            time.sleep(2)
            driver.find_element_by_id('userSave').click()
            time.sleep(3)

        def bind_menu():
            '''菜單綁定'''
            driver.find_element_by_xpath("//a[@title='菜單綁定']").click()
            driver.find_element_by_xpath("//div[@class='checkBoxUp']/div[1]/i").click()  # 全選菜單
            time.sleep(3)
            driver.find_element_by_id('userSave').click()
            time.sleep(3)

        def del_role():
            '''刪除角色'''
            driver.find_element_by_xpath("//a[@title='刪除']").click()
            time.sleep(1)
            driver.find_element_by_class_name("aui_state_highlight").click()
            time.sleep(3)

        create_role()
        query_role_by_name()
        edit_role()
        bind_menu()
        del_role()


if __name__ == '__main__':
    '''自動執行以'test'開頭的測試方法'''
    unittest.main()

 到這里完成了一條完整的測試用例，但如果有多條測試用例需要一起執行，后續說明。