pip install flaskpip install requests

#!/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)

python demo.py

#!/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):
        """
        測(cè)試登錄
        """
        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):
        """
        測(cè)試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'

#!/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):
        """
        測(cè)試登錄
        """
        response = self.app.login(self.username, self.password)        assert response['code'] == 200
        assert response['msg'] == 'success'
    def test_info(self):
        """
        測(cè)試獲取詳情信息
        """
        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'

#!/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實(shí)例
        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、請(qǐng)求url: \n%s' % url)
        print(u'\n2、請(qǐng)求頭信息:')
        pprint(self.session.headers)
        print(u'\n3、請(qǐng)求參數(shù):')
        pprint(data)
        print(u'\n4、響應(yīng):')
        pprint(response)        return response    def info(self):
        """
        詳情接口
        """
        url = urljoin(self.base_url, 'info')
        response = self.session.get(url).json()
        print('\n*****************************************')
        print(u'\n1、請(qǐng)求url: \n%s' % url)
        print(u'\n2、請(qǐng)求頭信息:')
        pprint(self.session.headers)
        print(u'\n3、請(qǐng)求cookies:')
        pprint(dict(self.session.cookies))
        print(u'\n4、響應(yīng):')
        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):
        """
        測(cè)試登錄
        """
        response = self.app.login(self.username, self.password)        assert response['code'] == 200
        assert response['msg'] == 'success'
    def test_info(self):
        """
        測(cè)試獲取詳情信息
        """
        self.app.login(self.username, self.password)
        response = self.app.info()        assert response['code'] == 200
        assert response['msg'] == 'success'
        assert response['data'] == 'info'