引用单元测试模块unittest框架(三)

先做一下driver.py文件,用于存放驱动,需要什么浏览器就做什么浏览器,数据库的链接也可以放在这里

from selenium import webdriver

class Driver_base(object):

    def driver(self):
        self.driver = webdriver.Firefox()
        return self.driver

if __name__ == '__main__':
    driver_1 = Driver_base()
    driver_1.driver()

我只需要用到火狐就ok了

data.py文件用来存放测试数据

在想用什么存放数据的时候去网上看有很多说法,excel,.txt, CSV......等等

个人感觉还是用python自己的文件是最棒的,直接把数据赋值给一个个变量;其他文件需要的时候直接去import,调用里面的变量更加的方便,当然,使用方便了在数据的管理上做不到excel那种,所以在数据存放的时候一定要做好数据的规划.

Mytest.py

这个文件用于继承unittest模块,构建自己的模块,添加一些自己想要的东西

#coding=utf-8

import unittest, os, sys
sys.path.append('..')
from Base.driver import Driver_base

class Myunittest(unittest.TestCase):

    def setUp(self):
        self.driver = Driver_base().driver()
        self.driver.implicitly_wait(10)
        self.driver.maximize_window()
        print 'TestCase Go Go Go......'

    def tearDown(self):
        self.driver.quit()
        print 'TestCase EndEndEnd......'


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

base.py

这个文件用于基层的封装,比如定位方法,元素操作,等等

#coding=utf-8
from selenium.webdriver.support.wait import WebDriverWait
from selenium import webdriver
from selenium.webdriver.common.by import By
import unittest, random
from selenium.webdriver.support.select import Select
class Base(object):

    def __init__(self, driver):
        self.driver = driver
#打开网址
    def open_browser(self, url):
        self.driver.get(url)
#定位一个元素
    def element(self, *loc):
        return self.driver.find_element(*loc)
#定位一组元素
    def elements(self, *loc):
        return self.driver.find_element(*loc)
#点击
    def wait_element_click(self, *loc):
        driver = self.driver
        return WebDriverWait(driver, 10).until(lambda driver: driver.find_element(*loc)).click()
#输入
    def wait_element_sendkeys(self, text, *loc):
        driver = self.driver
        return WebDriverWait(driver, 10).until(lambda driver: driver.find_element(*loc)).send_keys(text)
#进入iframe
    def switch_frame(self, ID):
        driver = self.driver
        return driver.switch_to.frame(ID)
#下拉框选择
    def select_list(self, text, *loc):
        driver = self.driver
        return WebDriverWait(driver, 10).until\
        (lambda driver:Select(driver.find_element(*loc))).select_by_visible_text(text)
还有很多没有完成,随后慢慢的添加进去就ok了