implement auth

ki/test/test_login_endpoint.py

@@ -1,42 +1,25 @@
-from alembic import command
 import json
+
 import unittest
 
-from app import app, db, migrate
-from ki.actions import seed
+from ki.test.ApiTest import ApiTest
 
 
-class TestLoginEndpoint(unittest.TestCase):
-    def setUp(self):
-        app.debug = True
-        app.config["SQLALCHEMY_DATABASE_URI"] = "sqlite:///:memory:"
-        self.client = app.test_client()
-
-        with app.app_context():
-            config = migrate.get_config()
-            command.upgrade(config, "head")
-
-        seed(True)
-
-    def tearDown(self):
-        db.drop_all()
-        db.engine.dispose()
-
+class TestLoginEndpoint(ApiTest):
     def test_login(self):
-        response1_data = self.login()
-        response2_data = self.login()
+        response1_data = self.login("peter", "geheim")
+        response2_data = self.login("peter", "geheim")
         self.assertNotEqual(response1_data["token"], response2_data["token"])
 
-    def login(self):
-        response = self.client.post("/users/login",
-                                    data=json.dumps({
-                                        "username": "peter",
-                                        "password": "geheim"
-                                    }),
-                                    content_type="application/json")
-        self.assertEqual(response.status_code, 200)
-        self.assertIn("token", response.json)
-        return response.json
+    def test_login_wrong_credentails(self):
+        login_data = {"username": "peter", "password": "123456"}
+        login_response = self.client.post("/users/login", data=json.dumps(login_data), content_type="application/json")
+        self.assertEqual(login_response.status_code, 403)
+
+    def test_login_unknown_user(self):
+        login_data = {"username": "karl", "password": "123456"}
+        login_response = self.client.post("/users/login", data=json.dumps(login_data), content_type="application/json")
+        self.assertEqual(login_response.status_code, 403)
 
 
 if __name__ == "main":