from alembic import command import json import unittest from app import app, migrate from ki.commands import seed 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() def test_login(self): response1_data = self.login() response2_data = self.login() 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 if __name__ == "main": unittest.main()