Spaces:
Running
Running
| import pytest | |
| from unittest.mock import MagicMock, patch | |
| from unittest.mock import AsyncMock | |
| def mock_chat_openai(): | |
| """Fixture to mock langchain_openai.ChatOpenAI.""" | |
| mock = MagicMock() | |
| # Simulate the behavior of the ChatOpenAI constructor if needed | |
| mock.return_value = MagicMock() | |
| return mock | |
| def mock_cl(): | |
| """Fixture to mock the chainlit library.""" | |
| mock = MagicMock() # Removed spec=True | |
| mock.__name__ = "chainlit" # Act more like a module | |
| # mock.__file__ = "mocked_chainlit.py" # Another common module attribute | |
| # Mock specific chainlit attributes or methods if tests rely on them | |
| mock.user_session = MagicMock() | |
| mock.user_session.set = MagicMock() | |
| mock.Message = MagicMock() | |
| # Ensure send() is an AsyncMock if it needs to be awaited and its result checked | |
| mock.Message.return_value.send = AsyncMock() | |
| # Mock cl.Action as it will be instantiated by the app | |
| mock.Action = MagicMock() # Allows app to call cl.Action(...) | |
| # Mock cl.Select as it will be instantiated by the app for persona choices | |
| mock.Select = MagicMock() # Allows app to call cl.Select(...) | |
| # Mock cl.Switch for UI toggles | |
| mock.Switch = MagicMock() | |
| # Mock cl.ChatSettings for UI settings panel | |
| mock.ChatSettings = MagicMock() # Mock the ChatSettings class | |
| # Ensure the instance returned by ChatSettings() has an async send method | |
| mock.ChatSettings.return_value.send = AsyncMock() | |
| # Mock decorators like on_chat_start and on_message | |
| # They need to be callable and return the function they decorate. | |
| mock.on_chat_start = MagicMock(side_effect=lambda func: func) | |
| mock.on_message = MagicMock(side_effect=lambda func: func) | |
| return mock | |
| def mock_dotenv_load(): | |
| """Fixture to mock dotenv.load_dotenv.""" | |
| mock = MagicMock() | |
| return mock | |
| # It's good practice to also provide the mock StateGraph classes here | |
| # if they are used across multiple test files for langgraph setup. | |
| def mock_state_graph_instance(): | |
| """Returns a mock instance of StateGraph, typically the one returned by the class mock.""" | |
| instance = MagicMock(spec=['add_node', 'set_entry_point', 'add_conditional_edges', 'add_edge', 'compile']) | |
| # If add_node returns self or another mock, configure here if needed for chaining | |
| return instance | |
| def mock_state_graph_class(mock_state_graph_instance): | |
| """Fixture to mock the StateGraph class itself.""" | |
| mock = MagicMock(return_value=mock_state_graph_instance) | |
| return mock | |
| def mock_chainlit_context(mock_cl): # Depends on mock_cl to use its user_session mock | |
| """Mocks chainlit.context.context_var.get() to return a mock context with a session.""" | |
| mock_context_obj = MagicMock(name="MockChainlitContextObject") | |
| # Use the same user_session mock from the mock_cl fixture for consistency | |
| mock_context_obj.session = mock_cl.user_session | |
| with patch('chainlit.context.context_var.get', return_value=mock_context_obj) as mock_context_var_get: | |
| yield mock_context_var_get |