Skip to content

Commit a2a162a

Browse files
fix: update orchestrator module import and references in test cases
1 parent 7b3a4ae commit a2a162a

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

src/tests/services/test_orchestrator.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
from unittest.mock import AsyncMock, MagicMock, patch
44

55
import pytest
6-
import orchestrator as orch_module
6+
import orchestrator
77
from orchestrator import (_HARMFUL_PATTERNS_COMPILED,
88
_SYSTEM_PROMPT_PATTERNS_COMPILED,
99
PLANNING_INSTRUCTIONS, RAI_HARMFUL_CONTENT_RESPONSE,
@@ -840,7 +840,7 @@ def test_get_orchestrator_singleton():
840840
mock_builder.return_value = mock_builder_instance
841841

842842
# Reset the singleton
843-
orch_module._orchestrator = None
843+
orchestrator._orchestrator = None
844844

845845
instance1 = get_orchestrator()
846846
instance2 = get_orchestrator()
@@ -1517,7 +1517,7 @@ async def test_get_chat_client_foundry_mode():
15171517
def test_foundry_not_available():
15181518
"""Test when Foundry SDK is not available."""
15191519
# Check that FOUNDRY_AVAILABLE is defined
1520-
assert hasattr(orch_module, 'FOUNDRY_AVAILABLE')
1520+
assert hasattr(orchestrator, 'FOUNDRY_AVAILABLE')
15211521

15221522
# Tests for workflow event handling (lines 736-799, 841-895)
15231523
# Note: These are integration-level tests that verify the workflow event

0 commit comments

Comments
 (0)