added more checks in the code

DiamondLightSource · Oct 17, 2024 · d747a89 · d747a89
1 parent 7576392
commit d747a89
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/blueapi/service/main.py b/src/blueapi/service/main.py
@@ -2,7 +2,6 @@
 from contextlib import asynccontextmanager
 from typing import Any
 
-from dotenv import load_dotenv
 from fastapi import (
     BackgroundTasks,
     Body,

diff --git a/tests/unit_tests/client/test_rest.py b/tests/unit_tests/client/test_rest.py
@@ -107,6 +107,8 @@ def test_refresh_if_signature_expired(rest: BlueapiRestClient):
         mock_verify_token.side_effect = jwt.ExpiredSignatureError
         mock_refresh_token.return_value = {"access_token": "new_token"}
         result = rest.get_plans()
+        mock_verify_token.assert_called_once()
+        mock_refresh_token.assert_called_once()
         assert result == PlanResponse(plans=[PlanModel.from_plan(plan)])