Add Models to JWTRoutes class & init_app method #119

Author: joegasewicz

tests/test_extensions.py

@@ -15,6 +15,9 @@ class TestExtension:
     config = {
                 "IGNORED_ROUTES": IGNORED_ROUTES,
                 "WHITE_LIST_ROUTES": WHITE_LIST_ROUTES,
+                "SECRET_KEY": "a sectrect key",
+                "JWT_ROUTER_API_NAME": "api/v1",
+                "ENTITY_KEY": "user_id",
             }
 
     def test_init_extensions(self, MockEntityModel):
@@ -24,6 +27,8 @@ def test_init_extensions(self, MockEntityModel):
         assert config.whitelist_routes == self.WHITE_LIST_ROUTES
         assert config.ignored_routes == self.IGNORED_ROUTES
         assert config.entity_models == [MockEntityModel]
+        assert config.entity_key == "user_id"
+        assert config.api_name == "api/v1"
 
         config = {**self.config, "ENTITY_MODELS": [MockEntityModel]}
         con = extensions.init_extensions(config)