apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-06-17 15:25:17 +02:00
Code Issues Projects Releases Packages Wiki Activity

fix tests

Browse source
This commit is contained in:
Adil Hafeez 2025-12-25 14:14:10 -08:00
parent f7d9453b9c
commit 1b2deb154e
No known key found for this signature in database
GPG key ID: 9B18EF7691369645
1 changed files with 16 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

29
cli/test/test_config_generator.py
View file

@ -59,13 +59,13 @@ tracing:
random_sampling: 100
"""
arch_config_schema = ""
with open("../../config/arch_config_schema.yaml", "r") as file:
with open("../config/arch_config_schema.yaml", "r") as file:
arch_config_schema = file.read()
m_open = mock.mock_open()
# Provide enough file handles for all open() calls in validate_and_render_schema
m_open.side_effect = [
mock.mock_open(read_data="").return_value,
# Removed empty read - was causing validation failures
mock.mock_open(read_data=arch_config).return_value, # ARCH_CONFIG_FILE
mock.mock_open(
read_data=arch_config_schema
@ -78,7 +78,7 @@ tracing:
mock.mock_open().return_value, # ARCH_CONFIG_FILE_RENDERED (write)
]
with mock.patch("builtins.open", m_open):
with mock.patch("config_generator.Environment"):
with mock.patch("planoai.config_generator.Environment"):
validate_and_render_schema()
@ -108,7 +108,7 @@ agents:
listeners:
- name: tmobile
type: agent
router: arch_agent_v2
router: plano_orchestrator_v1
agents:
- name: simple_tmobile_rag_agent
description: t-mobile virtual assistant for device contracts.
@ -132,13 +132,13 @@ listeners:
model: openai/gpt-4o
"""
arch_config_schema = ""
with open("../../config/arch_config_schema.yaml", "r") as file:
with open("../config/arch_config_schema.yaml", "r") as file:
arch_config_schema = file.read()
m_open = mock.mock_open()
# Provide enough file handles for all open() calls in validate_and_render_schema
m_open.side_effect = [
mock.mock_open(read_data="").return_value,
# Removed empty read - was causing validation failures
mock.mock_open(read_data=arch_config).return_value, # ARCH_CONFIG_FILE
mock.mock_open(
read_data=arch_config_schema
@ -319,26 +319,29 @@ def test_validate_and_render_schema_tests(monkeypatch, arch_config_test_case):
expected_error = arch_config_test_case.get("expected_error")
arch_config_schema = ""
with open("../../config/arch_config_schema.yaml", "r") as file:
with open("../config/arch_config_schema.yaml", "r") as file:
arch_config_schema = file.read()
m_open = mock.mock_open()
# Provide enough file handles for all open() calls in validate_and_render_schema
m_open.side_effect = [
mock.mock_open(read_data="").return_value,
mock.mock_open(read_data=arch_config).return_value, # ARCH_CONFIG_FILE
mock.mock_open(
read_data=arch_config
).return_value, # validate_prompt_config: ARCH_CONFIG_FILE
mock.mock_open(
read_data=arch_config_schema
).return_value, # ARCH_CONFIG_SCHEMA_FILE
mock.mock_open(read_data=arch_config).return_value, # ARCH_CONFIG_FILE
).return_value, # validate_prompt_config: ARCH_CONFIG_SCHEMA_FILE
mock.mock_open(
read_data=arch_config
).return_value, # validate_and_render_schema: ARCH_CONFIG_FILE
mock.mock_open(
read_data=arch_config_schema
).return_value, # ARCH_CONFIG_SCHEMA_FILE
).return_value, # validate_and_render_schema: ARCH_CONFIG_SCHEMA_FILE
mock.mock_open().return_value, # ENVOY_CONFIG_FILE_RENDERED (write)
mock.mock_open().return_value, # ARCH_CONFIG_FILE_RENDERED (write)
]
with mock.patch("builtins.open", m_open):
with mock.patch("config_generator.Environment"):
with mock.patch("planoai.config_generator.Environment"):
if expected_error:
# Test expects an error
with pytest.raises(Exception) as excinfo: