Simple API endpoint (#6)

* added simple router

* included basic router to the main

spacy_keyword_extraction_api/api_router.py

@@ -0,0 +1,16 @@
+import logging
+
+from fastapi import APIRouter
+
+
+LOGGER = logging.getLogger(__name__)
+
+
+def create_api_router() -> APIRouter:
+    router = APIRouter()
+
+    @router.get("/v1/extract-keywords")
+    def extract_keywords(text: str):
+        return text.split()
+
+    return router

spacy_keyword_extraction_api/main.py

@@ -3,14 +3,17 @@
 from fastapi import FastAPI
 from fastapi.staticfiles import StaticFiles
 
+from spacy_keyword_extraction_api.api_router import create_api_router
+
 
 LOGGER = logging.getLogger(__name__)
 
 
 def create_app():
     app = FastAPI()
 
-    app = FastAPI(docs_url=None, redoc_url=None)
+    app.include_router(create_api_router())
+
     app.mount('/', StaticFiles(directory='static', html=True), name='static')
 
     return app

tests/unit_test/api_router_test.py

@@ -0,0 +1,18 @@
+from fastapi.testclient import TestClient
+from fastapi import FastAPI
+
+from spacy_keyword_extraction_api.api_router import create_api_router
+
+
+def create_test_client():
+    app = FastAPI()
+    app.include_router(create_api_router())
+    client = TestClient(app)
+    return client
+
+
+class TestExtractKeyword:
+    def test_should_return_keywords(self):
+        client = create_test_client()
+        response = client.get('/v1/extract-keywords', params={'text': 'some text'})
+        assert response.json() == ['some', 'text']

tests/unit_test/main_test.py

@@ -7,3 +7,9 @@ def test_read_main():
     client = TestClient(create_app())
     response = client.get("/")
     assert response.status_code == 200
+
+
+def test_extract_keywords_endpoint_is_available():
+    client = TestClient(create_app())
+    response = client.get("/v1/extract-keywords", params={'text': 'some text'})
+    assert response.status_code == 200