python ddt數(shù)據(jù)驅動最簡實例代碼

更新時間：2019年02月22日 11:53:15 作者：三只松鼠

在本篇內容里我們給大家分享了關于python ddt數(shù)據(jù)驅動最簡實例代碼以及相關知識點，需要的朋友們跟著學習下。

在接口自動化測試中，往往一個接口的用例需要考慮正確的、錯誤的、異常的、邊界值等諸多情況，然后你需要寫很多個同樣代碼，參數(shù)不同的用例。如果測試接口很多，不但需要寫大量的代碼，測試數(shù)據(jù)和代碼柔合在一起，可維護性也會變的很差。數(shù)據(jù)驅動可以完美的將代碼和測試數(shù)據(jù)分開，將代碼進行分裝，提高復用性，測試數(shù)據(jù)維護在本地文件或數(shù)據(jù)庫。

使用python做接口自動化，首要任務是搭建一個自動化測試框架，其中unittest+ddt是一個不錯的選擇，下文主要介紹ddt在unittest下的使用。

ddt包含兩個方法裝飾器 ddt.data 和 ddt.file_data

一、ddt.data（直接輸入測試數(shù)據(jù)）

ddt.unpack 的作用是把參數(shù)中元祖或者列表的元素對應到多個參數(shù)上，沒有加 ddt.unpack 表示把元祖或者列表本身當成一個參數(shù)傳入。

下面這段代碼，ddt.data() 有三組測試數(shù)據(jù)，每組測試數(shù)據(jù)都會執(zhí)行一次 test_login() 。

import json
import unittest
from common.readConfig import readConfig
import requests
from ddt import ddt,data,file_data,unpack

@ddt  # 在測試類前必須首先聲明使用 ddt.ddt
class Mytest(unittest.TestCase):
  def setUp(self):
    # 獲取測試接口的url
    self.url = readConfig().getHttp('usercenter') + readConfig().get_UC('login')print('setup')

  @data(({"isRememberMe": True ,"password": "111111","username": "root"},200),
     ({"isRememberMe": True, "password": "1111111", "username": "root"},406),
     ({"isRememberMe": True, "password": "111111", "username": "rot"},406))
  @unpack
  # 后臺人員登錄
  def test_login(self,data,status):
    body = json.dumps(data)
    header = {"Content-Type":"application/json","Accept": "application/json","token": "1231"}
    re = requests.post(self.url,data=body,headers = header,verify = False)
    code = re.status_code
    print(re.text,re.status_code)
    # 斷言
    self.assertEqual(int(status),int(code))

  def tearDown(self):
    print('tearDown')

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

二、ddt.file_data (參數(shù)是文件名。文件可以是json 或者 yaml類型)

如果文件中是列表，每個列表的值會作為測試用例參數(shù)，同時作為測試用例方法名后綴顯示。如果文件中是字典，字典的key會作為測試用例方法的后綴顯示，字典的值會作為測試用例參數(shù)。下文兩種類型的文件分別舉一個例子。

新建文件testdata.json：

{
  "first": ["{'isRememberMe': True,'password': '111111','username': 'root'}", "200"],
  "second": ["{'isRememberMe': True, 'password': '1111111', 'username': 'root'}", "406"],
  "third": ["{'isRememberMe': True, 'password': '111111', 'username': 'rot'}", "406"]
}

新建文件testdata.yaml:

first: ["{'isRememberMe': True,'password': '111111','username': 'root'}","200"]

second: ["{'isRememberMe': True, 'password': '1111111', 'username': 'root'}", "406"]

third: ["{'isRememberMe': True, 'password': '111111', 'username': 'rot'}", "406"]

新建測試腳本：

import json
import unittest
from common.readConfig import readConfig
import requests
from ddt import ddt,data,file_data,unpack

@ddt  # 在測試類前必須首先聲明使用 ddt.ddt
class Mytest(unittest.TestCase):
  def setUp(self):
    # 獲取測試接口的url
    self.url = readConfig().getHttp('usercenter') + readConfig().get_UC('login')print('setup')

  @file_data('D:\\automation--interface\\testcase\\UC\\testdata.json')
  # @file_data('D:\\automation--interface\\testcase\\UC\\testdata.yaml')

  # 后臺人員登錄
  def test_login(self,data):
    body = json.dumps(eval(data[0]))
    status = data[1]
    header = {"Content-Type":"application/json","Accept": "application/json","token": "1231"}
    re = requests.post(self.url,data=body,headers = header,verify = False)
    code = re.status_code
    print(re.text,re.status_code)
    # 斷言
    self.assertEqual(int(status),int(code))

  def tearDown(self):
    print('tearDown')

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

以上代碼大家可以本地測試下，感謝同學們的學習和對腳本之家的支持。

您可能感興趣的文章: