ApiServer: add testcases for MD5 authentication

test/test_apiserver_v2.py

@@ -0,0 +1,151 @@
+import json
+import random
+import requests
+
+from .base import ApiServerUnittest
+
+
+class TestApiServerV2(ApiServerUnittest):
+
+    authentication = True
+
+    def setUp(self):
+        super(TestApiServerV2, self).setUp()
+        self.host = "http://127.0.0.1:5000"
+        self.api_client = requests.Session()
+        self.clear_users()
+
+    def tearDown(self):
+        super(TestApiServerV2, self).tearDown()
+
+    def test_index(self):
+        headers = self.prepare_headers()
+        resp = self.api_client.get(self.host, headers=headers)
+        self.assertEqual(200, resp.status_code)
+
+    def clear_users(self):
+        url = "%s/api/users" % self.host
+        return self.api_client.delete(url, headers=self.prepare_headers())
+
+    def get_users(self):
+        url = "%s/api/users" % self.host
+        return self.api_client.get(url, headers=self.prepare_headers())
+
+    def create_user(self, uid, name, password):
+        url = "%s/api/users/%d" % (self.host, uid)
+        data = {
+            'name': name,
+            'password': password
+        }
+        headers = self.prepare_headers(json.dumps(data))
+        return self.api_client.post(url, headers=headers, json=data)
+
+    def get_user(self, uid):
+        url = "%s/api/users/%d" % (self.host, uid)
+        return self.api_client.get(url, headers=self.prepare_headers())
+
+    def update_user(self, uid, name, password):
+        url = "%s/api/users/%d" % (self.host, uid)
+        data = {
+            'name': name,
+            'password': password
+        }
+        headers = self.prepare_headers(json.dumps(data))
+        return self.api_client.put(url, headers=headers, json=data)
+
+    def delete_user(self, uid):
+        url = "%s/api/users/%d" % (self.host, uid)
+        return self.api_client.delete(url, headers=self.prepare_headers())
+
+    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_customized_response_status_code(self):
+        status_code = random.randint(200, 511)
+        url = "%s/customize-response" % self.host
+        expected_response = {
+            'status_code': status_code,
+        }
+        resp = self.api_client.post(
+            url,
+            headers=self.prepare_headers(json.dumps(expected_response)),
+            json=expected_response
+        )
+        self.assertEqual(status_code, resp.status_code)
+
+    def test_get_customized_response_headers(self):
+        expected_response = {
+            'headers': {
+                'abc': 123,
+                'def': 456
+            }
+        }
+        url = "%s/customize-response" % self.host
+        resp = self.api_client.post(
+            url,
+            headers=self.prepare_headers(json.dumps(expected_response)),
+            json=expected_response
+        )
+        self.assertIn('abc', resp.headers)
+        self.assertIn('123', resp.headers['abc'])