Files
Mutual_Fund/tests/test_customer_agent_runtime.py
T

26 lines
874 B
Python
Raw Normal View History

import unittest
from service.customer_agent.runtime import build_anonymous_runtime
from agent.customer_agent.session import AnonymousSessionService
from agent.customer_agent.context import RedisConversationContext
class RuntimeTests(unittest.TestCase):
def test_runtime_wires_session_context_and_agent_components(self):
runtime = build_anonymous_runtime(
redis=object(),
milvus_client=object(),
llm_client=object(),
config_getter={}.get,
audit_writer=lambda **kwargs: None,
)
self.assertIsInstance(runtime.session_service, AnonymousSessionService)
self.assertIsInstance(runtime.context, RedisConversationContext)
self.assertIs(runtime.redis, runtime.session_service.redis)
self.assertIsNotNone(runtime.agent)
if __name__ == "__main__":
unittest.main()