test: resolve mock JWT invalid. mock API call

cemsbv · Jun 19, 2024 · fb0c31c · fb0c31c
1 parent 36dd6a4
commit fb0c31c
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/tests/model/test_fitcore.py b/tests/model/test_fitcore.py
@@ -33,12 +33,6 @@ def test_fitcore_model_generator() -> None:
         + datetime.timedelta(days=80),
     )
 
-    # generate fitcore model
-    model = FitCoreModelGenerator(
-        series=series,
-        client=client,
-    )
-
     # mock API calls
     if not real_http:
         with requests_mock.Mocker() as m:
@@ -55,8 +49,20 @@ def test_fitcore_model_generator() -> None:
             )
             m.post(BASE_URL + f"simpleKoppejan/predict", json={"settlement": [0] * 500})
 
+            # generate fitcore model
+            model = FitCoreModelGenerator(
+                series=series,
+                client=client,
+            )
+
             model.plot_fill_settlement_time(log_time=False)
     else:
+        # generate fitcore model
+        model = FitCoreModelGenerator(
+            series=series,
+            client=client,
+        )
+
         model.plot_fill_settlement_time(log_time=False)
 
     # Show the plots