引用单元测试模块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了