68 lines
2.2 KiB
Python
68 lines
2.2 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
"""
|
||
|
|
Basic tests for the LangChain + LangGraph scaffolding
|
||
|
|
"""
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
import sys
|
||
|
|
import os
|
||
|
|
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||
|
|
|
||
|
|
from workflows.workflow_manager import WorkflowManager, WorkflowType
|
||
|
|
|
||
|
|
|
||
|
|
class TestWorkflowManager(unittest.TestCase):
|
||
|
|
"""Test WorkflowManager functionality"""
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
"""Set up test fixtures"""
|
||
|
|
self.manager = WorkflowManager()
|
||
|
|
|
||
|
|
def test_get_available_workflows(self):
|
||
|
|
"""Test that available workflows are returned"""
|
||
|
|
workflows = self.manager.get_available_workflows()
|
||
|
|
self.assertIsInstance(workflows, list)
|
||
|
|
self.assertGreater(len(workflows), 0)
|
||
|
|
self.assertIn("conversation", workflows)
|
||
|
|
self.assertIn("tool_using", workflows)
|
||
|
|
|
||
|
|
def test_get_workflow(self):
|
||
|
|
"""Test getting workflow instances"""
|
||
|
|
conversation_workflow = self.manager.get_workflow(WorkflowType.CONVERSATION)
|
||
|
|
self.assertIsNotNone(conversation_workflow)
|
||
|
|
|
||
|
|
tool_workflow = self.manager.get_workflow(WorkflowType.TOOL_USING)
|
||
|
|
self.assertIsNotNone(tool_workflow)
|
||
|
|
|
||
|
|
def test_session_management(self):
|
||
|
|
"""Test session creation and retrieval"""
|
||
|
|
# Execute a workflow to create a session
|
||
|
|
result = self.manager.execute_workflow(
|
||
|
|
WorkflowType.CONVERSATION,
|
||
|
|
"Hello, test session"
|
||
|
|
)
|
||
|
|
|
||
|
|
session_id = result["session_id"]
|
||
|
|
self.assertIsNotNone(session_id)
|
||
|
|
|
||
|
|
# Test session info retrieval
|
||
|
|
session_info = self.manager.get_session_info(session_id)
|
||
|
|
self.assertIsNotNone(session_info)
|
||
|
|
self.assertEqual(session_info["workflow_type"], WorkflowType.CONVERSATION)
|
||
|
|
|
||
|
|
|
||
|
|
class TestConfiguration(unittest.TestCase):
|
||
|
|
"""Test configuration validation"""
|
||
|
|
|
||
|
|
def test_config_import(self):
|
||
|
|
"""Test that configuration can be imported"""
|
||
|
|
try:
|
||
|
|
from config import Config
|
||
|
|
# This should not raise an exception if .env file exists with valid API key
|
||
|
|
self.assertTrue(hasattr(Config, 'OPENAI_API_KEY'))
|
||
|
|
except ImportError:
|
||
|
|
self.fail("Could not import config module")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|