�
K�h� � �F � d Z ddlZddlZddlZej � ej � ej � ej � e� � � � � � � � ddl m
Z
G d� dej � � Z e
dk r ej � � dS dS )z+
TDD Tests for rate limiting stress tester
� N)�RateLimitStressTesterc �< � e Zd ZdZd� Zd� Zd� Zd� Zd� Zd� Z d� Z
d S )
�TestRateLimitStressTesterz.Test rate limiting stress tester functionalityc �< � | � t d� � dS )z,Test that RateLimitStressTester class existsz(RateLimitStressTester class should existN)�assertIsNotNoner )�selfs �I/home/pshempel/src/local/time-node_tdd/tests/stress/test_rate_limiting.py�$test_rate_limit_stress_tester_existsz>TestRateLimitStressTester.test_rate_limit_stress_tester_exists s � � � � � 2�4^�_�_�_�_�_� c � � t dd�� � }| � |� � | � |j d� � | � |j d� � dS )z8Test that we can create a RateLimitStressTester instance�
�� ��
rate_limit� window_msN)r r � assertEqualr r )r �testers r � test_can_create_tester_instancez9TestRateLimitStressTester.test_can_create_tester_instance s` � �&�"��E�E�E�� � � �V�$�$�$� ����*�B�/�/�/� ����)�4�0�0�0�0�0r c �~ � t dd�� � }| � t |d� � � � |� � � }| � d|� � | � d|� � | � d|� � | � d|� � | � |d d � � d
S ) z Test burst at limit scenario� r r �test_burst_at_limit�scenario�successful_requests�rate_limited_requests�errors�burst_at_limitN)r �
assertTrue�hasattrr �assertInr �r r �results r � test_burst_at_limit_scenarioz6TestRateLimitStressTester.test_burst_at_limit_scenario s� � �&�!�t�D�D�D��
�����(=�>�>�?�?�?� �+�+�-�-��
�
�
�j�&�)�)�)� �
�
�+�V�4�4�4� �
�
�-�v�6�6�6� �
�
�h��'�'�'� ����
�+�-=�>�>�>�>�>r c �� � t dd�� � }| � t |d� � � � |� d�� � }| � d|� � | � d|� � | � d |� � | � d
|� � | � |d d � � | � t
|d � � d� � d S )
z"Test parallel connections scenarior r r �test_parallel_connections� )�num_connectionsr � connections�total_requests�bypass_detected�parallel_connectionsN)r r r r$ r r �lenr s r �"test_parallel_connections_scenarioz<TestRateLimitStressTester.test_parallel_connections_scenario- s� � �&�!�t�D�D�D��
�����(C�D�D�E�E�E� �1�1�!�1�D�D��
�
�
�j�&�)�)�)� �
�
�m�V�,�,�,� �
�
�&��/�/�/� �
�
�'��0�0�0� ����
�+�-C�D�D�D� ����V�M� 2�3�3�Q�7�7�7�7�7r c � � t dd�� � }| � t |d� � � � |� d�� � }| � d|� � | � d|� � | � d |� � | � d
|� � | � d |� � | � |d d � � d
S )z!Test memory exhaustion prevention�d i`� r �test_memory_exhaustionr )�duration_secondsr �memory_start_mb�
memory_end_mb�memory_growth_mb�
requests_made�memory_exhaustionN)r r r r/ r r r s r � test_memory_exhaustion_scenarioz9TestRateLimitStressTester.test_memory_exhaustion_scenario? s� � �&�#��G�G�G��
�����(@�A�A�B�B�B� �.�.��.�B�B��
�
�
�j�&�)�)�)� �
�
�'��0�0�0� �
�
�o�v�.�.�.� �
�
�(�&�1�1�1� �
�
�o�v�.�.�.� ����
�+�-@�A�A�A�A�Ar c � � t dd�� � }| � t |d� � � � |� d�� � }| � d|� � | � d|� � | � d |� � | � d
|� � | � d |� � | � |d d � � | � t
|d � � d� � d
S )z(Test rapid disconnect/reconnect scenarior r r �test_rapid_reconnectr% )�
num_cyclesr �cyclesr( �expected_limited�actual_limited�rapid_reconnectN)r r r r8 r r r+ r s r � test_rapid_reconnect_scenarioz7TestRateLimitStressTester.test_rapid_reconnect_scenarioQ s� � �&�!�t�D�D�D��
�����(>�?�?�@�@�@� �,�,��,�:�:��
�
�
�j�&�)�)�)� �
�
�h��'�'�'� �
�
�&��/�/�/� �
�
�(�&�1�1�1� �
�
�&��/�/�/� ����
�+�->�?�?�?� ����V�H� -�.�.��2�2�2�2�2r c �� � t dd�� � }| � t |d� � � � dddd�}|� |� � }| � |� d � � � � d
S ) z&Test that results can be saved to filer r r � save_results�testz2025-01-01T00:00:00T)r � timestamp�successz.jsonN)r r r r@ �endswith)r r �results�filenames r �test_can_save_resultsz/TestRateLimitStressTester.test_can_save_resultsd s� � �&�!�t�D�D�D��
������7�7�8�8�8� �.��
�
�� �&�&�w�/�/�� ����)�)�'�2�2�3�3�3�3�3r N) �__name__�
__module__� __qualname__�__doc__r
r r" r, r6 r> rG � r r r r s� � � � � � �8�8�`� `� `�1� 1� 1�?� ?� ?�"8� 8� 8�$B� B� B�$3� 3� 3�&4� 4� 4� 4� 4r r �__main__)rK �unittest�sys�os�path�append�dirname�abspath�__file__�stress.rate_limit_stressr �TestCaser rH �mainrL r r �<module>rY s� ��� � ����
�
�
�
� � � � � �����������������0I�0I� J� J�K�K� L� L� L� :� :� :� :� :� :�f4� f4� f4� f4� f4�� 1� f4� f4� f4�R �z����H�M�O�O�O�O�O� �r
