From 7bb312d34efcd16bf5d369356f7ce5243071bbdc Mon Sep 17 00:00:00 2001
From: foefl <f.foerster@d-opt.com>
Date: Thu, 27 Mar 2025 10:04:34 +0100
Subject: [PATCH] change setup APIfor session to ensure correct state

---
 src/delta_barth/management.py | 7 ++++---
 tests/conftest.py             | 4 +++-
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/src/delta_barth/management.py b/src/delta_barth/management.py
index 359fe12..15badc4 100644
--- a/src/delta_barth/management.py
+++ b/src/delta_barth/management.py
@@ -11,10 +11,11 @@ from delta_barth.session import Session
 SESSION: Final[Session] = Session(HTTP_BASE_CONTENT_HEADERS)
 
 
-def set_data_path(
-    path: str,
+def setup(
+    data_path: str,
 ) -> None:  # pragma: no cover
-    SESSION.set_data_path(path)
+    SESSION.set_data_path(data_path)
+    SESSION.setup()
 
 
 def set_credentials(
diff --git a/tests/conftest.py b/tests/conftest.py
index f399f4e..a1b4735 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -74,8 +74,9 @@ def exmpl_api_sales_prognosis_output() -> pd.DataFrame:
 
 # ** sessions
 @pytest.fixture(scope="function")
-def session(credentials, api_base_url) -> delta_barth.session.Session:
+def session(credentials, api_base_url, tmp_path) -> delta_barth.session.Session:
     session = delta_barth.session.Session(HTTP_BASE_CONTENT_HEADERS)
+    session.set_data_path(str(tmp_path))
     session.set_base_url(api_base_url)
     session.set_credentials(
         username=credentials["user"],
@@ -83,6 +84,7 @@ def session(credentials, api_base_url) -> delta_barth.session.Session:
         database=credentials["db"],
         mandant=credentials["mandant"],
     )
+    session.setup()
 
     return session