test: Refactor AnswerNode tests, extract configuration to variables and update node initialization.

pull/20921/head
GuanMu 10 months ago
parent 868542dea8
commit f433b8a175

@ -144,12 +144,7 @@ def test_execute_answer_with_outputs():
variable_pool.add(["start", "weather"], "sunny") variable_pool.add(["start", "weather"], "sunny")
variable_pool.add(["start", "score"], 85) variable_pool.add(["start", "score"], 85)
node = AnswerNode( node_config = {
id=str(uuid.uuid4()),
graph_init_params=init_params,
graph=graph,
graph_runtime_state=GraphRuntimeState(variable_pool=variable_pool, start_at=time.perf_counter()),
config={
"id": "answer", "id": "answer",
"data": { "data": {
"title": "Answer with outputs", "title": "Answer with outputs",
@ -160,9 +155,18 @@ def test_execute_answer_with_outputs():
{"variable": "status", "type": "string", "value_selector": ["start", "weather"]}, {"variable": "status", "type": "string", "value_selector": ["start", "weather"]},
], ],
}, },
}, }
node = AnswerNode(
id=str(uuid.uuid4()),
graph_init_params=init_params,
graph=graph,
graph_runtime_state=GraphRuntimeState(variable_pool=variable_pool, start_at=time.perf_counter()),
config=node_config,
) )
node.init_node_data(node_config["data"])
# Mock db.session.close() # Mock db.session.close()
db.session.close = MagicMock() db.session.close = MagicMock()
@ -228,12 +232,7 @@ def test_execute_answer_with_empty_outputs():
conversation_variables=[], conversation_variables=[],
) )
node = AnswerNode( node_config = {
id=str(uuid.uuid4()),
graph_init_params=init_params,
graph=graph,
graph_runtime_state=GraphRuntimeState(variable_pool=variable_pool, start_at=time.perf_counter()),
config={
"id": "answer", "id": "answer",
"data": { "data": {
"title": "No outputs", "title": "No outputs",
@ -241,9 +240,18 @@ def test_execute_answer_with_empty_outputs():
"answer": "Simple answer", "answer": "Simple answer",
"outputs": [], "outputs": [],
}, },
}, }
node = AnswerNode(
id=str(uuid.uuid4()),
graph_init_params=init_params,
graph=graph,
graph_runtime_state=GraphRuntimeState(variable_pool=variable_pool, start_at=time.perf_counter()),
config=node_config,
) )
node.init_node_data(node_config["data"])
# Mock db.session.close() # Mock db.session.close()
db.session.close = MagicMock() db.session.close = MagicMock()

Loading…
Cancel
Save