From 7d52214e9124ec20e72a25362438249dfc5e50d0 Mon Sep 17 00:00:00 2001 From: Alex Zorkin Date: Tue, 18 Feb 2025 18:21:37 -0800 Subject: [PATCH] fix: import cleanup --- .../lcfs/tests/fuel_export/test_fuel_exports_services.py | 8 +------- .../lcfs/tests/fuel_supply/test_fuel_supplies_services.py | 2 -- 2 files changed, 1 insertion(+), 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 index 63929939f..7662a2837 100644 --- 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 index 18de80b7d..19a4dfef2 100644 --- 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