diff --git a/shierfeng-fastapi-backend/module_business/service/usage_service.py b/shierfeng-fastapi-backend/module_business/service/usage_service.py index efb6014..c4d5340 100644 --- a/shierfeng-fastapi-backend/module_business/service/usage_service.py +++ b/shierfeng-fastapi-backend/module_business/service/usage_service.py @@ -130,9 +130,7 @@ class UsageService: @staticmethod def _success(usage: BizUsageRecord) -> CrudResponseModel: remaining_packages = usage.remaining_packages_after - flow_state = ( - PalletFlowState.AT_SITE_EMPTY if remaining_packages == 0 else PalletFlowState.AT_SITE_PARTIAL - ) + flow_state = PalletFlowState.AT_SITE_EMPTY if remaining_packages == 0 else PalletFlowState.AT_SITE_PARTIAL return CrudResponseModel( is_success=True, message='使用成功', diff --git a/shierfeng-fastapi-backend/tests/module_business/test_usage_service.py b/shierfeng-fastapi-backend/tests/module_business/test_usage_service.py index 31d567f..a08c42d 100644 --- a/shierfeng-fastapi-backend/tests/module_business/test_usage_service.py +++ b/shierfeng-fastapi-backend/tests/module_business/test_usage_service.py @@ -581,9 +581,7 @@ class TestUsageControllerContract(unittest.TestCase): response_schema = schema['components']['schemas'][post_response_schema['$ref'].rsplit('/', 1)[-1]] self.assertIn('data', response_schema['properties']) self.assertIn('data', response_schema['required']) - data_schema = schema['components']['schemas'][ - response_schema['properties']['data']['$ref'].rsplit('/', 1)[-1] - ] + data_schema = schema['components']['schemas'][response_schema['properties']['data']['$ref'].rsplit('/', 1)[-1]] self.assertEqual( {'usageId', 'remainingPackages', 'flowState'}, set(data_schema['properties']), @@ -1472,9 +1470,7 @@ class TestUsageIdempotency(unittest.IsolatedAsyncioTestCase): ), ) expected_state = ( - PalletFlowState.AT_SITE_EMPTY - if remaining_after == 0 - else PalletFlowState.AT_SITE_PARTIAL + PalletFlowState.AT_SITE_EMPTY if remaining_after == 0 else PalletFlowState.AT_SITE_PARTIAL ) self.assertIsInstance(result.result, UsageSuccessResponseModel) self.assertEqual(