mirror of
https://github.com/httprunner/httprunner.git
synced 2026-05-12 02:21:29 +08:00
116 lines
3.9 KiB
Python
116 lines
3.9 KiB
Python
import requests
|
|
import random
|
|
from .base import ApiServerUnittest
|
|
|
|
class TestApiServer(ApiServerUnittest):
|
|
def setUp(self):
|
|
super(TestApiServer, self).setUp()
|
|
self.host = "http://127.0.0.1:5000"
|
|
self.api_client = requests.Session()
|
|
self.clear_users()
|
|
|
|
def tearDown(self):
|
|
super(TestApiServer, self).tearDown()
|
|
|
|
def clear_users(self):
|
|
url = "%s/api/users" % self.host
|
|
return self.api_client.delete(url)
|
|
|
|
def get_users(self):
|
|
url = "%s/api/users" % self.host
|
|
return self.api_client.get(url)
|
|
|
|
def create_user(self, uid, name, password):
|
|
url = "%s/api/users/%d" % (self.host, uid)
|
|
data = {
|
|
'name': name,
|
|
'password': password
|
|
}
|
|
return self.api_client.post(url, json=data)
|
|
|
|
def get_user(self, uid):
|
|
url = "%s/api/users/%d" % (self.host, uid)
|
|
return self.api_client.get(url)
|
|
|
|
def update_user(self, uid, name, password):
|
|
url = "%s/api/users/%d" % (self.host, uid)
|
|
data = {
|
|
'name': name,
|
|
'password': password
|
|
}
|
|
return self.api_client.put(url, json=data)
|
|
|
|
def delete_user(self, uid):
|
|
url = "%s/api/users/%d" % (self.host, uid)
|
|
return self.api_client.delete(url)
|
|
|
|
def test_clear_users(self):
|
|
resp = self.clear_users()
|
|
self.assertEqual(200, resp.status_code)
|
|
self.assertEqual(True, resp.json()['success'])
|
|
|
|
def test_create_user_not_existed(self):
|
|
resp = self.create_user(1000, 'user1', '123456')
|
|
self.assertEqual(201, resp.status_code)
|
|
self.assertEqual(True, resp.json()['success'])
|
|
|
|
def test_create_user_existed(self):
|
|
resp = self.create_user(1000, 'user1', '123456')
|
|
resp = self.create_user(1000, 'user1', '123456')
|
|
self.assertEqual(500, resp.status_code)
|
|
|
|
def test_get_users_empty(self):
|
|
resp = self.get_users()
|
|
self.assertEqual(200, resp.status_code)
|
|
self.assertEqual(resp.json()['count'], 0)
|
|
|
|
def test_get_users_not_empty(self):
|
|
resp = self.create_user(1000, 'user1', '123456')
|
|
resp = self.get_users()
|
|
self.assertEqual(200, resp.status_code)
|
|
self.assertEqual(resp.json()['count'], 1)
|
|
|
|
resp = self.create_user(1001, 'user2', '123456')
|
|
resp = self.get_users()
|
|
self.assertEqual(200, resp.status_code)
|
|
self.assertEqual(resp.json()['count'], 2)
|
|
|
|
def test_get_user_not_existed(self):
|
|
resp = self.get_user(1000)
|
|
self.assertEqual(404, resp.status_code)
|
|
self.assertEqual(resp.json()['success'], False)
|
|
|
|
def test_get_user_existed(self):
|
|
self.create_user(1000, 'user1', '123456')
|
|
resp = self.get_user(1000)
|
|
self.assertEqual(200, resp.status_code)
|
|
self.assertEqual(resp.json()['success'], True)
|
|
|
|
def test_update_user_not_existed(self):
|
|
resp = self.update_user(1000, 'user1', '123456')
|
|
self.assertEqual(404, resp.status_code)
|
|
self.assertEqual(resp.json()['success'], False)
|
|
|
|
def test_update_user_existed(self):
|
|
self.create_user(1000, 'user1', '123456')
|
|
resp = self.update_user(1000, 'user2', '123456')
|
|
self.assertEqual(200, resp.status_code)
|
|
self.assertEqual(resp.json()['data']['name'], 'user2')
|
|
|
|
def test_delete_user_not_existed(self):
|
|
resp = self.delete_user(1000)
|
|
self.assertEqual(404, resp.status_code)
|
|
self.assertEqual(resp.json()['success'], False)
|
|
|
|
def test_delete_user_existed(self):
|
|
self.create_user(1000, 'leo', '123456')
|
|
resp = self.delete_user(1000)
|
|
self.assertEqual(200, resp.status_code)
|
|
self.assertEqual(resp.json()['success'], True)
|
|
|
|
def test_get_response_with_status_code(self):
|
|
status_code = random.randint(200, 511)
|
|
url = "%s/status_code/%d" % (self.host, status_code)
|
|
resp = self.api_client.get(url)
|
|
self.assertEqual(status_code, resp.status_code)
|