unittest框架实战(二)

#coding=utf-8

from selenium import webdriver
import MySQLdb

class My_driver(object):
    '''各种驱动类'''
    def __init__(self):
        pass

    def ff_driver(self):
        '''火狐浏览器驱动'''
        self.driver = webdriver.Firefox()
        return self.driver

    def Chrome_driver(self):
        '''谷歌浏览器
        '''
        self.driver = webdriver.Chrome()
        return self.driver

    def mysql_driver(self):
        '''Mysql 数据库驱动'''
        conn = MySQLdb.connect(host = '**.***.***.***', user='root', password='root', charset='utf8')
        self.cursor = conn.cursor()

        return self.cursor

驱动文件,没什么好说的