�
m��h$ � �D � d Z ddlZddlZddlZddlZ G d� d� � ZdS )zY
Simple MCP Stress Tester - KISS Implementation
Starting with minimal code to pass tests
� Nc �@ � e Zd ZdZd� Zd� Zd� Zd� Zd d�Zd� Z d
d
�Z
d S )�SimpleStressTesterz$Simple stress tester for MCP serversc �" � || _ d | _ d S )N)�
server_cmd�process)�selfr s �B/home/pshempel/src/local/time-node_tdd/tests/stress/stress_test.py�__init__zSimpleStressTester.__init__ s � �$����� � � � c � � t j | j t j t j t j dd�� � | _ t j d� � dS )zStart the MCP serverTr )�stdin�stdout�stderr�text�bufsize� N)�
subprocess�Popenr �PIPEr �time�sleep�r s r � start_serverz SimpleStressTester.start_server sH � �!�'� �O� �/� �?� �?���
�
�
�� �
�
�1�
�
�
�
�
r c � � | j r| j � � � �dS t j | j j � � }|� � � j dz dz S # t j t j f$ r Y dS w xY w)z Check server memory usage in MBNg i ) r �poll�psutil�Process�pid� memory_info�rss�
NoSuchProcess� AccessDenied)r �procs r � check_memoryz SimpleStressTester.check_memory s� � ��|� �t�|� 0� 0� 2� 2� >��3� ��>�$�,�"2�3�3�D��#�#�%�%�)�D�0�4�7� 7���$�f�&9�:� � � ��3�3� ���s �<A! �!B �?B c �x � | j r�| j j r | j j � � � | j j r | j j � � � | j j r | j j � � � | j � � � | j � � � dS dS )zStop the serverN)r r
�closer r � terminate�waitr s r � stop_serverz SimpleStressTester.stop_server* s� � � �<� ��|�!�
+�� �"�(�(�*�*�*��|�"�
,�� �#�)�)�+�+�+��|�"�
,�� �#�)�)�+�+�+�
�L� "� "� $� $� $� �L� � � � � � � � � r � �?c � � ||z
|z }||k S )a< Check if memory growth indicates a leak
Args:
initial_memory: Starting memory in MB
final_memory: Ending memory in MB
threshold: Growth factor threshold (default 0.5 = 50%)
Returns:
True if memory growth exceeds threshold
� )r �initial_memory� final_memory� threshold�
growth_factors r �is_memory_leakz!SimpleStressTester.is_memory_leak9 s � � &��6�.�H�
� �y�(�(r c �z � | j r| j � � � �dS | j j � t j |� � dz � � | j j � � � | j j � � � }|rt j |� � S dS # t $ r Y dS w xY w)z*Send a single MCP request and get responseN�
) r r r
�write�json�dumps�flushr �readline�loads� Exception)r �request�
response_lines r � send_requestz SimpleStressTester.send_requestG s� � ��|� �t�|� 0� 0� 2� 2� >��4� � �L� � $� $�T�Z��%8�%8�4�%?� @� @� @� �L� � $� $� &� &� &� !�L� /� 8� 8� :� :�M� �
1��z�-�0�0�0��4��� � � ��4�4� ���s �BB, �,
B:�9B:�< c �t � | j r| j � � � �dS t j � � }| � � � }ddd|ddd�}d}t j � � |z
|k rzdd|dd d
id �d �}| � |� � }|d
xx dz
cc< |�d|v r|dxx dz
cc< |dz
}t j d� � t j � � |z
|k �zt j � � |z
|d< | � � � |d< | � ||d � � |d< |S )z6Hammer the server with requests for specified durationNr F)�total_requests�errors�duration� memory_start�
memory_end�memory_leak_detected� z2.0z
tools/call�get_current_time�timezone�UTC)�name� arguments)�jsonrpc�method�id�paramsr@ �errorrA g{�G�z�?rB rD rE )r r r r$ r= r r1 )r �seconds�
start_time� start_memory�results�
request_idr; �responses r �
hammer_serverz SimpleStressTester.hammer_serverY s� � ��|� �t�|� 0� 0� 2� 2� >��4��Y�[�[�
��(�(�*�*� � ��� (��$)�
�
�� �
��i�k�k�J�&��0�0� �&� � .�",�e�!4�� � � �G� �(�(��1�1�H� �$� %� %� %�� *� %� %� %�� �7�h�#6�#6���!�!�!�Q�&�!�!�!� �!�O�J� �J�t� � � �' �i�k�k�J�&��0�0�, #�i�k�k�J� 6��
�� $� 1� 1� 3� 3�� � �*.�*=�*=� �'�,�/�+
� +
��&�'� �r N)r* )r> ) �__name__�
__module__� __qualname__�__doc__r
r r$ r) r1 r= rW r, r r r r
s� � � � � � �.�.� � � �
�
�
�� � �
�
�
� )� )� )� )� � � �$/� /� /� /� /� /r r )r[ r r r r5 r r, r r �<module>r\ st ��� �
� � � � � � � �
�
�
�
� � � � �{� {� {� {� {� {� {� {� {� {r
