Added /getReports endpoint and tests

endpoints/__init__.py

@@ -4,3 +4,4 @@ from .user import get_me, create_bot_token, get_bot_tokens, delete_bot_token
 from .parser import load_report
 from .indicators import get_indicators_from_group, get_indicator_groups
 from .watcher import create_watcher
+from .parser import get_reports

endpoints/parser.py

@@ -9,6 +9,7 @@ from starlette.requests import Request
 from integrations.s3 import save_report_to_s3
 from models import Report, IndicatorGroup
 from models.indicator import IndicatorGroupPD
+from models.report import ReportPD
 from parsers.pdf_parser import process_pdf
 from parsers.text_parser import CollectedData
 from responses.errors import ReportNotPresented
@@ -39,3 +40,13 @@ async def load_report(request: Request):
     background.add_task(save_report_to_s3, temp.name, report_id)
     background.add_task(os.remove, temp.name)
     return OkResponse({"report_id": report_id, "indicator_group": group_pd}, background=background)
+
+
+async def get_reports(request: Request):
+    """ Returns user's reports """
+    reports = await Report.filter(owner=request.state.user).all()
+    as_pd = []
+    for report in reports:
+        report.owner = request.state.user
+        as_pd.append(ReportPD.from_orm(report))
+    return OkResponse(as_pd)

endpoints/routes.py

@@ -18,7 +18,8 @@ api_routes = [
     Route("/getBotTokens", get_bot_tokens, methods=["GET"]),
     Route("/deleteBotToken", delete_bot_token, methods=["GET"]),
     Route("/createWatcher", create_watcher, methods=["POST"]),
-    Route("/getIndicatorGroups", get_indicator_groups, methods=["GET"])
+    Route("/getIndicatorGroups", get_indicator_groups, methods=["GET"]),
+    Route("/getReports", get_reports, methods=["GET"])
 ]
 
 admin_routes = []

tests/test_indicator.py

@@ -95,3 +95,15 @@ async def test_indicator_groups_endpoint(sti_auth, user):
 
     assert resp.status_code == 200
     assert len(resp.json()["data"]) == len(groups)
+
+
+def test_reports_endpoint(sti_auth, user, mts_report):
+    resp = sti_auth.post("/api/loadReport", files={"file": open(mts_report, "rb")})
+
+    assert resp.status_code == 200
+    report_id = resp.json()["data"]["report_id"]
+
+    resp_reports = sti_auth.get("/api/getReports")
+    assert 1 == len(resp_reports.json()["data"])
+    print(resp_reports.json())
+    assert resp_reports.json()["data"][0]["id"] == report_id