fix: import cleanup

bcgov · Feb 19, 2025 · 7d52214 · 7d52214
1 parent 8008a85
commit 7d52214
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 9 deletions.
diff --git a/backend/lcfs/tests/fuel_export/test_fuel_exports_services.py b/backend/lcfs/tests/fuel_export/test_fuel_exports_services.py
@@ -16,6 +16,7 @@
 from lcfs.db.models.compliance.FuelExport import FuelExport
 from lcfs.db.base import ActionTypeEnum, UserTypeEnum
 from lcfs.db.models.user.Role import RoleEnum
+from types import SimpleNamespace
 
 # Mock common data for reuse
 mock_fuel_type = FuelTypeSchema(
@@ -38,9 +39,6 @@
 
 @pytest.mark.anyio
 async def test_get_fuel_export_options_success(fuel_export_service, mock_repo):
-    # (If needed, set a dummy request here as well)
-    from types import SimpleNamespace
-
     dummy_user = SimpleNamespace(id=1, role_names=[RoleEnum.GOVERNMENT])
     dummy_request = MagicMock()
     dummy_request.user = dummy_user
@@ -55,8 +53,6 @@ async def test_get_fuel_export_options_success(fuel_export_service, mock_repo):
 @pytest.mark.anyio
 async def test_get_fuel_export_list_success(fuel_export_service, mock_repo):
     # Set up a dummy request with a valid user
-    from types import SimpleNamespace
-
     dummy_user = SimpleNamespace(id=1, role_names=[RoleEnum.GOVERNMENT])
     dummy_request = MagicMock()
     dummy_request.user = dummy_user
@@ -94,8 +90,6 @@ async def test_get_fuel_export_list_success(fuel_export_service, mock_repo):
 @pytest.mark.anyio
 async def test_get_fuel_exports_paginated_success(fuel_export_service, mock_repo):
     # Set up a dummy request with a valid user
-    from types import SimpleNamespace
-
     dummy_user = SimpleNamespace(id=1, role_names=[RoleEnum.GOVERNMENT])
     dummy_request = MagicMock()
     dummy_request.user = dummy_user

diff --git a/backend/lcfs/tests/fuel_supply/test_fuel_supplies_services.py b/backend/lcfs/tests/fuel_supply/test_fuel_supplies_services.py
@@ -83,8 +83,6 @@ async def test_get_fuel_supply_list(fuel_supply_service):
     service, mock_repo, _ = fuel_supply_service
 
     # Create a dummy request with a user that supports attribute access.
-    from types import SimpleNamespace
-
     dummy_user = SimpleNamespace(id=1, role_names=[RoleEnum.GOVERNMENT])
     dummy_request = MagicMock()
     dummy_request.user = dummy_user