retrieval of sales prognosis data

tests/api/test_common.py

@@ -1,3 +1,5 @@
+from datetime import datetime as Datetime
+
 import pytest
 
 from delta_barth.api import common
@@ -50,7 +52,7 @@ def test_combine_route(base, route, expect):
 
 def test_assert_login():
     with pytest.raises(ApiConnectionError):
-        common._assert_login()
+        common._assert_login_status()
 
 
 @pytest.mark.api_con_required
@@ -88,8 +90,9 @@ def test_login_logout(credentials, api_base_url):
     resp = common.logout(
         base_url=api_base_url,
     )
+    assert resp.error is None
     assert HTTP_CURRENT_CONNECTION.session_token is None
-    assert "DelecoToken" not in HTTP_CURRENT_CONNECTION
+    assert "DelecoToken" not in HTTP_CURRENT_CONNECTION.headers
     resp = common.login(
         base_url=api_base_url,
         user_name=credentials["user"],
@@ -100,3 +103,28 @@ def test_login_logout(credentials, api_base_url):
     assert resp.error is not None
     assert resp.error.status_code == 409
     assert resp.error.message == "Nutzer oder Passwort falsch."
+
+
+@pytest.mark.api_con_required
+def test_get_sales_prognosis_data(credentials, api_base_url):
+    resp = common.login(
+        base_url=api_base_url,
+        user_name=credentials["user"],
+        password=credentials["pwd"],
+        database=credentials["db"],
+        mandant=credentials["mandant"],
+    )
+    assert resp.error is None
+    date = Datetime(2022, 6, 1)
+    resp = common.get_sales_prognosis_data(api_base_url, date)
+    assert resp.error is None
+    assert len(resp.daten) > 0
+    date = Datetime(2030, 1, 1)
+    resp = common.get_sales_prognosis_data(api_base_url, date)
+    assert resp.error is None
+    assert len(resp.daten) == 0
+    # close connection
+    resp = common.logout(
+        base_url=api_base_url,
+    )
+    assert resp.error is None