From 9dcf410fd2c7d21fcb3e8e8668448312fcdc046a Mon Sep 17 00:00:00 2001 From: QuantumGhost Date: Wed, 4 Jun 2025 11:51:06 +0800 Subject: [PATCH] chore(api): fix mypy violation for `scoped_session`. --- .../services/test_workflow_draft_variable_service.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/api/tests/integration_tests/services/test_workflow_draft_variable_service.py b/api/tests/integration_tests/services/test_workflow_draft_variable_service.py index 8288dc7bd8..466c390155 100644 --- a/api/tests/integration_tests/services/test_workflow_draft_variable_service.py +++ b/api/tests/integration_tests/services/test_workflow_draft_variable_service.py @@ -20,7 +20,7 @@ class TestWorkflowDraftVariableService(unittest.TestCase): def setUp(self): self._test_app_id = str(uuid.uuid4()) - self._session: Session = db.session + self._session: Session = db.session() sys_var = WorkflowDraftVariable.new_sys_variable( app_id=self._test_app_id, name="sys_var", @@ -95,6 +95,7 @@ class TestWorkflowDraftVariableService(unittest.TestCase): def test_get_node_variable(self): srv = self._get_test_srv() node_var = srv.get_node_variable(self._test_app_id, self._node1_id, "str_var") + assert node_var is not None assert node_var.id == self._node1_str_var_id assert node_var.name == "str_var" assert node_var.get_value() == build_segment("str_value") @@ -102,6 +103,7 @@ class TestWorkflowDraftVariableService(unittest.TestCase): def test_get_system_variable(self): srv = self._get_test_srv() sys_var = srv.get_system_variable(self._test_app_id, "sys_var") + assert sys_var is not None assert sys_var.id == self._sys_var_id assert sys_var.name == "sys_var" assert sys_var.get_value() == build_segment("sys_value") @@ -109,6 +111,7 @@ class TestWorkflowDraftVariableService(unittest.TestCase): def test_get_conversation_variable(self): srv = self._get_test_srv() conv_var = srv.get_conversation_variable(self._test_app_id, "conv_var") + assert conv_var is not None assert conv_var.id == self._conv_var_id assert conv_var.name == "conv_var" assert conv_var.get_value() == build_segment("conv_value")