功能：实现客户现场使用记录

shierfeng-fastapi-backend/tests/module_business/test_master_service.py

@@ -503,13 +503,13 @@ class TestMasterDao(unittest.IsolatedAsyncioTestCase):
             self.assertIn(f'{table}.{id_field} = 9', sql)
 
     async def test_locking_master_gets_use_current_reads_and_refresh_identity_map(self) -> None:
-        strain_lock_parameter = inspect.signature(MasterDao.get_strain).parameters.get('for_update')
-        self.assertIsNotNone(strain_lock_parameter)
-        self.assertIs(inspect.Parameter.KEYWORD_ONLY, strain_lock_parameter.kind)
-        self.assertFalse(strain_lock_parameter.default)
-
-        for resource in ('warehouse', 'strain'):
+        for resource in ('warehouse', 'strain', 'line'):
             with self.subTest(resource=resource):
+                lock_parameter = inspect.signature(getattr(MasterDao, f'get_{resource}')).parameters.get('for_update')
+                self.assertIsNotNone(lock_parameter)
+                self.assertIs(inspect.Parameter.KEYWORD_ONLY, lock_parameter.kind)
+                self.assertFalse(lock_parameter.default)
+
                 result = MagicMock()
                 result.scalar_one_or_none.return_value = None
                 self.db.execute = AsyncMock(return_value=result)