012 python接口 数据驱动ddt

‘‘‘
时间：2018/11/05
目录: 
　　一: 安装ddt
　　二: 禅道登录 
        1 工程结构
        2 unittest文件
        3 实现功能文件
        4 执行结果
　　三: 使用excel
        1 安装xlrd
        2 准备测试数据
　　四: 禅道登录
        1 工程结构 
        2 unittest文件
        3 实现功能文件
        4 read_excel.py
        5 执行结果        
‘‘‘

# coding:utf-8
import unittest
import ddt
import requests

from DataDriver.chandao_login_api import *

data = [{"user": "admin", "psw": "123456", "expect": True},
        {"user": "admin2", "psw": "123456", "expect": False},
        {"user": "admin", "psw": "123456", "expect": True},
        {"user": "admin4", "psw": "123123", "expect": False}
        ]


@ddt.ddt
class Test(unittest.TestCase):
    def setUp(self):
        self.session = requests.session()

    def tearDown(self):
        self.session.cookies.clear()    # 清除缓存 - cookies
        self.session.close()            # 去除警告 - 关闭进程

    @ddt.data(*data)    # *data : 分组传送 - 字典类型
    def test_001(self, testData):
        strUser = testData["user"]
        strPass = testData["psw"]
        strResult = Login(self.session, strUser, strPass)
        bResult = DecideResult(strResult)
        self.assertTrue(bResult == testData["expect"])


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

# coding:utf-8
import requests


url = "http://127.0.0.1/zentao/user-login.htm"  # url经常变 : 做全局变量 - 做配置文件

def Login(session, strUser, strPass):
    ‘‘‘
    登录禅道
    :param session:     session
    :param strUser:     用户名
    :param strPass:     密码
    :return:            请求结果 - 服务端
    ‘‘‘

    body = {
        "account": strUser,
        "password": strPass,
        "keepLogin[]": "on",
        "referer": "/zentao/my.html"
    }
    r = session.post(url, data = body)
    strResult = r.content.decode("utf-8")
    return strResult


def DecideResult(strResult):
    ‘‘‘
    判断结果 - 登录禅道
    :param strResult:  请求结果 - 服务端
    :return:           True - 登录成功 
                       Fasle - 登录失败 
    ‘‘‘
    if "登录失败" in strResult:
        print("登录失败")
        return False
    elif "parent.location" in strResult:
        print("登录成功")
        return True
    else:
        print("登录失败 - 其他情况")
        return False


if __name__ == ‘__main__‘:
    s = requests.session()
    result = Login(s, "admin", "123456")
    DecideResult(result)

# coding:utf-8
import unittest
import ddt
import requests
from DataDriver.chandao_login_api import *
from DataDriver.read_excel import ExcelUtil


data = ExcelUtil("testData.xlsx", "Sheet1")
data = data.dict_data()
print(data)

@ddt.ddt
class Test(unittest.TestCase):
    def setUp(self):
        self.session = requests.session()

    def tearDown(self):
        self.session.cookies.clear()    # 清除缓存 - cookies
        self.session.close()            # 去除警告 - 关闭进程

    @ddt.data(*data)    # *data : 分组传送 - 字典类型
    def test_001(self, testData):

        strUser = testData["user"]
        strPass = testData["psw"]
        strResult = Login(self.session, strUser, strPass)
        bResult = DecideResult(strResult)
        self.assertTrue(str(bResult) == testData["expect"])


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

# coding:utf-8
import requests


url = "http://127.0.0.1/zentao/user-login.htm"  # url经常变 : 做全局变量 - 做配置文件

def Login(session, strUser, strPass):
    ‘‘‘
    登录禅道
    :param session:     session
    :param strUser:     用户名
    :param strPass:     密码
    :return:            请求结果 - 服务端
    ‘‘‘

    body = {
        "account": strUser,
        "password": strPass,
        "keepLogin[]": "on",
        "referer": "/zentao/my.html"
    }
    r = session.post(url, data = body)
    strResult = r.content.decode("utf-8")
    return strResult


def DecideResult(strResult):
    ‘‘‘
    判断结果 - 登录禅道
    :param strResult:  请求结果 - 服务端
    :return:
    ‘‘‘
    if "登录失败" in strResult:
        print("登录失败")
        return False
    elif "parent.location" in strResult:
        print("登录成功")
        return True
    else:
        print("登录失败 - 其他情况")
        return False


if __name__ == ‘__main__‘:
    s = requests.session()
    result = Login(s, "admin", "123456")
    DecideResult(result)

# coding:utf-8
import xlrd
class ExcelUtil():
    def __init__(self, excelPath, sheetName="Sheet1"):
        self.data = xlrd.open_workbook(excelPath)
        self.table = self.data.sheet_by_name(sheetName)

        # 获取第一行作为key值
        self.keys = self.table.row_values(0)
        # 获取总行数
        self.rowNum = self.table.nrows
        # 获取总列数
        self.colNum = self.table.ncols

    def dict_data(self):
        if self.rowNum <= 1:
            print("总行数小于1")
        else:
            r = []
            j=1
            for i in range(self.rowNum-1):
                s = {}
                # 从第二行取对应values值
                values = self.table.row_values(j)
                for x in range(self.colNum):
                    s[self.keys[x]] = values[x]
                r.append(s)
                j+=1
            return r

if __name__ == "__main__":
    filepath = "testData.xlsx"
    sheetName = "Sheet1"
    data = ExcelUtil(filepath, sheetName)
    print(data.dict_data())