mirror of
https://github.com/httprunner/httprunner.git
synced 2026-05-12 02:21:29 +08:00
75 lines
2.2 KiB
Python
75 lines
2.2 KiB
Python
import multiprocessing
|
|
import time
|
|
import unittest
|
|
|
|
import requests
|
|
from tests.api_server import FLASK_APP_PORT, HTTPBIN_HOST, HTTPBIN_PORT
|
|
from tests.api_server import app as flask_app
|
|
from tests.api_server import gen_md5, gen_random_string, get_sign, httpbin_app
|
|
|
|
|
|
def run_flask():
|
|
flask_app.run(port=FLASK_APP_PORT)
|
|
|
|
|
|
def run_httpbin():
|
|
if httpbin_app:
|
|
httpbin_app.run(host=HTTPBIN_HOST, port=HTTPBIN_PORT)
|
|
|
|
|
|
class ApiServerUnittest(unittest.TestCase):
|
|
""" Test case class that sets up an HTTP server which can be used within the tests
|
|
"""
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
cls.host = "http://127.0.0.1:5000"
|
|
cls.flask_process = multiprocessing.Process(
|
|
target=run_flask
|
|
)
|
|
cls.httpbin_process = multiprocessing.Process(
|
|
target=run_httpbin
|
|
)
|
|
cls.flask_process.start()
|
|
cls.httpbin_process.start()
|
|
time.sleep(0.1)
|
|
cls.api_client = requests.Session()
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
cls.flask_process.terminate()
|
|
cls.httpbin_process.terminate()
|
|
|
|
def get_token(self, user_agent, device_sn, os_platform, app_version):
|
|
url = "%s/api/get-token" % self.host
|
|
headers = {
|
|
'Content-Type': 'application/json',
|
|
'User-Agent': user_agent,
|
|
'device_sn': device_sn,
|
|
'os_platform': os_platform,
|
|
'app_version': app_version
|
|
}
|
|
data = {
|
|
'sign': get_sign(user_agent, device_sn, os_platform, app_version)
|
|
}
|
|
|
|
resp = self.api_client.post(url, json=data, headers=headers)
|
|
resp_json = resp.json()
|
|
self.assertTrue(resp_json["success"])
|
|
self.assertIn("token", resp_json)
|
|
self.assertEqual(len(resp_json["token"]), 16)
|
|
return resp_json["token"]
|
|
|
|
def get_authenticated_headers(self):
|
|
user_agent = 'iOS/10.3'
|
|
device_sn = gen_random_string(15)
|
|
os_platform = 'ios'
|
|
app_version = '2.8.6'
|
|
|
|
token = self.get_token(user_agent, device_sn, os_platform, app_version)
|
|
headers = {
|
|
'device_sn': device_sn,
|
|
'token': token
|
|
}
|
|
return headers
|