一、簡介

本文從一個簡單的登錄接口測試入手，一步步調整優(yōu)化接口調用姿勢；
然后簡單討論了一下接口測試框架的要點；
最后介紹了一下我們目前正在使用的接口測試框架 pithy。
期望讀者可以通過本文對接口自動化測試有一個大致的了解。

二、引言

為什么要做接口自動化測試？

在當前互聯(lián)網(wǎng)產(chǎn)品迭代頻繁的背景下，回歸測試的時間越來越少，很難在每個迭代都對所有功能做完整回歸。

但接口自動化測試因其實現(xiàn)簡單、維護成本低，容易提高覆蓋率等特點，越來越受重視。

為什么要自己寫框架呢？

使用 requets unittest 很容易實現(xiàn)接口自動化測試，而且 requests 的api已經(jīng)非常人性化，非常簡單。

但通過封裝以后（特別是針對公司內特定接口），再加上對一些常用工具的封裝，可以進一步提高業(yè)務腳本編寫效率。

三、環(huán)境準備

確保本機已安裝 python2.7 以上版本，然后安裝如下庫：

pip install flaskpip install requests

后面我們會使用 flask 寫一個用來測試的接口，使用requests去測試。

四、測試接口準備

下面使用 flask 實現(xiàn)兩個 http 接口，一個登錄，另外一個查詢詳情，但需要登錄后才可以，新建一個 demo.py 文件（注意，不要使用windows記事本），把下面代碼 copy 進去，然后保存、關閉。

接口代碼

#!/usr/bin/python# coding=utf-8from flask import Flask, request, session, jsonify USERNAME = 'admin'PASSWORD = '123456'app = Flask(__name__)app.secret_key = 'pithy'@app.route('/login', methods=['GET', 'POST'])def login(): error = None if request.method == 'POST': if request.form['username'] != USERNAME: error = 'Invalid username' elif request.form['password'] != PASSWORD: error = 'Invalid password' else: session['logged_in'] = True return jsonify({'code': 200, 'msg': 'success'}) return jsonify({'code': 401, 'msg': error}), 401@app.route('/info', methods=['get'])def info(): if not session.get('logged_in'): return jsonify({'code': 401, 'msg': 'please login !!'}) return jsonify({'code': 200, 'msg': 'success', 'data': 'info'})if __name__ == '__main__': app.run(debug=True)

最后執(zhí)行如下命令：

python demo.py

響應如下：

* Running on http://127.0.0.1:5000/ (Press CTRL C to quit) * Restarting with sta

大家可以看到服務已經(jīng)起來了。

接口信息

登錄接口

請求url

/login

請求方法

post

請求參數(shù)

| 參數(shù)名稱 | 參數(shù)類型 | 參數(shù)說明 |

| :————: | :——-: | :——: |

| username | String | 登錄名稱 |

| password | String | 登錄密碼 |

響應信息

| 參數(shù)名稱 | 參數(shù)類型 | 參數(shù)說明 |

| :————: | :——-: | :——: |

| code | Integer | 結果code |

| msg | String | 結果信息 |

詳情接口

請求url

/info

請求方法

get

請求 cookies

| 參數(shù)名稱 | 參數(shù)類型 | 參數(shù)說明 |

| :————: | :——-: | :——: |

| session | String | session |

響應信息

| 參數(shù)名稱 | 參數(shù)類型 | 參數(shù)說明 |

| :————: | :——-: | :——: |

| code | Integer | 結果code |

| msg | String | 結果信息 |

| data | String | 數(shù)據(jù)信息 |

五、編寫接口測試

測試思路

• 使用 requests庫模擬發(fā)送 HTTP 請求。
• 使用 python 標準庫里 unittest 寫測試 case。

腳本實現(xiàn)

#!/usr/bin/python# coding=utf-8import requestsimport unittestclass TestLogin(unittest.TestCase): @classmethod def setUpClass(cls): cls.login_url = 'http://127.0.0.1:5000/login' cls.info_url = 'http://127.0.0.1:5000/info' cls.username = 'admin' cls.password = '123456' def test_login(self): """ 測試登錄 """ data = { 'username': self.username, 'password': self.password } response = requests.post(self.login_url, data=data).json() assert response['code'] == 200 assert response['msg'] == 'success' def test_info(self): """ 測試info接口 """ data = { 'username': self.username, 'password': self.password } response_cookies = requests.post(self.login_url, data=data).cookies session = response_cookies.get('session') assert session info_cookies = { 'session': session } response = requests.get(self.info_url, cookies=info_cookies).json() assert response['code'] == 200 assert response['msg'] == 'success' assert response['data'] == 'info'

六、優(yōu)化

封裝接口調用

寫完這個測試登錄腳本，你或許會發(fā)現(xiàn)，在整個項目的測試過程，登錄可能不止用到一次，如果每次都這么寫，會不會太冗余了？

對，確實太冗余了，下面做一下簡單的封裝，把登錄接口的調用封裝到一個方法里，把調用參數(shù)暴漏出來，示例腳本如下：

#!/usr/bin/python# coding=utf-8import requestsimport unittesttry: from urlparse import urljoinexcept ImportError: from urllib.parse import urljoinclass DemoApi(object): def __init__(self, base_url): self.base_url = base_url def login(self, username, password): """ 登錄接口 :param username: 用戶名 :param password: 密碼 """ url = urljoin(self.base_url, 'login') data = { 'username': username, 'password': password } return requests.post(url, data=data).json() def get_cookies(self, username, password): """ 獲取登錄cookies """ url = urljoin(self.base_url, 'login') data = { 'username': username, 'password': password } return requests.post(url, data=data).cookies def info(self, cookies): """ 詳情接口 """ url = urljoin(self.base_url, 'info') return requests.get(url, cookies=cookies).json()class TestLogin(unittest.TestCase): @classmethod def setUpClass(cls): cls.base_url = 'http://127.0.0.1:5000' cls.username = 'admin' cls.password = '123456' cls.app = DemoApi(cls.base_url) def test_login(self): """ 測試登錄 """ response = self.app.login(self.username, self.password) assert response['code'] == 200 assert response['msg'] == 'success' def test_info(self): """ 測試獲取詳情信息 """ cookies = self.app.get_cookies(self.username, self.password) response = self.app.info(cookies) assert response['code'] == 200 assert response['msg'] == 'success' assert response['data'] == 'info'

OK，在這一個版本中，我們不但在把登錄接口的調用封裝成了一個實例方法，實現(xiàn)了復用，而且還把 host（self.base_url）提取了出來。

但問題又來了，登錄之后，登錄接口的 http 響應會把 session 以 cookie 的形式 set 到客戶端，之后的接口都會使用此 session 去請求。

還有，就是在接口調用過程中，希望可以把日志打印出來，以便調試或者出錯時查看。

好吧，我們再來改一版。

保持 cookies &增加 log 信息

使用 requests 庫里的同一個 Session 對象 (它也會在同一個 Session 實例發(fā)出的所有請求之間保持 cookie )，即可解決上面的問題，示例代碼如下：

#!/usr/bin/python# coding=utf-8import unittestfrom pprint import pprintfrom requests.sessions import Sessiontry: from urlparse import urljoinexcept ImportError: from urllib.parse import urljoinclass DemoApi(object): def __init__(self, base_url): self.base_url = base_url # 創(chuàng)建session實例 self.session = Session() def login(self, username, password): """ 登錄接口 :param username: 用戶名 :param password: 密碼 """ url = urljoin(self.base_url, 'login') data = { 'username': username, 'password': password } response = self.session.post(url, data=data).json() print('n*****************************************') print(u'n1、請求url: n%s' % url) print(u'n2、請求頭信息:') pprint(self.session.headers) print(u'n3、請求參數(shù):') pprint(data) print(u'n4、響應:') pprint(response) return response def info(self): """ 詳情接口 """ url = urljoin(self.base_url, 'info') response = self.session.get(url).json() print('n*****************************************') print(u'n1、請求url: n%s' % url) print(u'n2、請求頭信息:') pprint(self.session.headers) print(u'n3、請求cookies:') pprint(dict(self.session.cookies)) print(u'n4、響應:') pprint(response) return responseclass TestLogin(unittest.TestCase): @classmethod def setUpClass(cls): cls.base_url = 'http://127.0.0.1:5000' cls.username = 'admin' cls.password = '123456' cls.app = DemoApi(cls.base_url) def test_login(self): """ 測試登錄 """ response = self.app.login(self.username, self.password) assert response['code'] == 200 assert response['msg'] == 'success' def test_info(self): """ 測試獲取詳情信息 """ self.app.login(self.username, self.password) response = self.app.info() assert response['code'] == 200 assert response['msg'] == 'success' assert response['data'] == 'info'

大功告成，我們把多個相關接口調用封裝到一個類中，使用同一個 requests Session 實例來保持 cookies，并且在調用過程中打印出了日志，我們所有目標都實現(xiàn)了。

但再看下腳本，又會感覺不太舒服，在每個方法里，都要寫一遍print 1、2、3… 要拼url、還要很多細節(jié)等等。

但其實我們 真正需要做的只是拼出關鍵的參數(shù)（url 參數(shù)、body 參數(shù)或者傳入 headers 信息），可不可以只需定義必須的信息，然后把其它共性的東西都封裝起來呢，統(tǒng)一放到一個地方去管理？

感謝每一個認真閱讀我文章的人！??！

如果下面這些資料用得到的話可以直接拿走：

1、自學開發(fā)或者測試必備的完整項目源碼與環(huán)境

2、測試工作中所有模板（測試計劃、測試用例、測試報告等）

3、軟件測試經(jīng)典面試題

4、Python/Java自動化測試實戰(zhàn).pdf

5、Jmeter/postman接口測試全套視頻獲取

我個人整理了我這幾年軟件測試生涯整理的一些技術資料，包含：電子書，簡歷模塊，各種工作模板，面試寶典，自學項目等。需要的評論區(qū)找我謝謝