MCP Hub

� PiEh���>�dZd
dl
Zd
dlmc
mZ
d
dlZd
dlZd
dl Z d
dl mZmZm Z
d
dlmZmZmZ
ddddd d d�Zej'�D]\ZZee j,e<�ej.d��Zej.d ��Zej.d��Zej.d��Zej.d��Zej.d��Zej.d��Zej.d��ZGd�d�Z ej.d��Z!ej.d��Z"y)z'Common test fixtures and configuration.�N)�Mock� MagicMock�patch)�Dict�Any� Generatorztvly-test-key-12345�test-nebius-keyztest-openai-keyztest-anthropic-keyztest-hf-key�nebius)�TAVILY_API_KEY�NEBIUS_API_KEY�OPENAI_API_KEY�ANTHROPIC_API_KEY�HUGGINGFACE_API_KEY�LLM_PROVIDERc�Z�t
�}d
dddd�dddd d�gd d�|j_|S)z(Mock Tavily client for web search tests.z Test Result 1zhttps://example.com/1zTest content 1g�������?��title�url�content�scorez Test Result 2zhttps://example.com/2zTest content 2g�������?zTest search summary)�results�answer)r�search�return_value)
�mock_clients
�>C:\Users\User\PythonProjects\mcp_hub_project\tests\conftest.py�mock_tavily_clientrsS���&�K�)�.�+��  �)�.�+��  �  �(�'�K���#�"��c��y
)zMock LLM completion response.z@{"sub_questions": ["Question 1?", "Question 2?", "Question 3?"]}�r rr�mock_llm_responser!.s��N
rc�T�t
�}t
d
dd��|j_|S)z,Mock Modal sandbox for code execution tests.zTest output�r)�stdout�stderr� returncode)r�execr)
�mock_sandboxs
r�mock_modal_sandboxr)3s)���6�L�%)��r�VW�%X�L���"��rc��y
)z Sample user request for testing.z>Create a Python script to analyze CSV data and generate chartsr r rr�sample_user_requestr+:s��L
rc��d
dddd�dddd d�gS) z"Sample search results for testing.zPython Data Analysis Tutorialz#https://example.com/pandas-tutorialz;Learn how to analyze CSV data with pandas and matplotlib...gffffff�?rzChart Generation with Pythonzhttps://example.com/chartsz,Create stunning charts and visualizations...gףp= ��?r r rr�sample_search_resultsr-?s2�� 5�8�T��  �4�/�E��  � � rc��y
)zSample Python code for testing.z� import pandas as pd import matplotlib.pyplot as plt # Load data df = pd.read_csv('data.csv') # Generate chart df.plot(kind='bar') plt.show() r r rr�sample_coder/Qs�� rc�~�t
�}d
|_
d|_d|_t
�}
d|
j_||
fS)zMock configuration objects.z tvly-test-keyr r z meta-llama/llama-3.1-8b-instruct)r�tavily_api_key�llm_provider�nebius_api_key�get_model_for_providerr)� api_config�model_configs r�mock_configr7`sC����J� /�J��&�J�� 1�J���6�L�7Y�L�'�'�4��|�#�#rc#�\K
�t
j�}|�
�

|j�
y
�w
)z%Create an event loop for async tests.N)�asyncio�new_event_loop�close)
�loops
r� event_loopr=ms$�����!�!�#�D� �J��J�J�L�s�*,
c�"�eZ
dZd
Zdefd�Zd�Zy)� MockAgentz"Base mock agent class for testing.�namec� �|
|_d
|_
y)Nr)r@� call_count)�selfr@s r�__init__zMockAgent.__init__vs���� ���rc
�`�|x
jd
z c_d|j|jd�S)N�
T)�success�agent�calls)rBr@)rC�args�kwargss r�__call__zMockAgent.__call__zs%�����1����$�)�)�d�o�o�N�NrN)�__name__� __module__�__qualname__�__doc__�strrDrLr rrr?r?ts��,��S��O
rr?c��t
d
�
t
d�
t
d�
t
d�
t
d�
t
d�
d�S)z.Mock agent instances for orchestrator testing.�question_enhancer� web_search� llm_processor�citation_formatter�code_generator�code_runner)rSrTrUrVrWrX)
r?r rr�mock_agentsrY~sA��'�':�;���-�"�?�3�'�(<�=�#�$4�5� ��/� �rc#�VK
�t
d
d�5
d��

ddd�
y#1s
w
Y

yxYw
�w
)z,Disable advanced features for basic testing.zapp.ADVANCED_FEATURES_AVAILABLEFN)
rr rr�disable_advanced_featuresr[�s*���� �0�%� 8�
� �
�
�
�s� )
�� )
�&�)
)#rP�builtins�@py_builtins�_pytest.assertion.rewrite� assertion�rewrite� @pytest_ar�pytestr9�os� unittest.mockrrr�typingrrr� TEST_ENV_VARS�items�key�value�environ�fixturerr!r)r+r-r/r7r=r?rYr[r rr�<module>rl
sg
�


�-��� �� �0�0�'�'�,�'�'�-�(�� � � �%�%�'�

�J�C���B�J�J�s�O�

����
��
�,���
N
��
N
����
��
����
L
��
L
����
��
�"���
��
����
$��
$����
��
�
O
�
O
����
��
����
��
r