�
�hC( � �b � d Z ddlZddlZddlmZ ddlmZ ddlmZ ddlZ G d� de� � Z dS )z�
Enhanced Stress Tester with Configuration Support
================================================
Builds on SimpleStressTester with file-based logging and better reporting.
� N)�datetime)�Path)�SimpleStressTesterc �H � � e Zd ZdZd
� fd� Z� fd�Zdd�Zd� Zd� Zd � Z � xZ
S )�ConfigurableStressTesterz@Stress tester that uses configuration and provides better outputNc �~ �� t � � � |pt j � � d| _ d| _ dS )zInitialize with config valuesN)�super�__init__�config�SERVER_COMMAND�results_file�start_timestamp)�self�
server_cmd� __class__s ��D/home/pshempel/src/local/time-node_tdd/tests/stress/stress_tester.pyr
z!ConfigurableStressTester.__init__ s9 �� �
������<�v�'<�=�=�=� ���#����� c � �� t � � � � � t j t j � � dS )z&Start server with configured wait timeN)r �start_server�time�sleepr �SERVER_STARTUP_TIME)r r s �r r z%ConfigurableStressTester.start_server s2 �� �
���������
�6�-�.�.�.�.�.r �standardc �
� |�t j | d }t j � � | _ | j � d� � }t j t j � |�� � z | _ t d|� d|� d�� � | j � t j � � r$| j �
t j � � n| j j }t d|� �� � t d � � |t j | d
|| j � � � dd�g g g d
�}t! j � � }| � � � }ddd|ddd�}|d � d|dd�� � d} |}
|}d}d}
t! j � � |z
|k �rk| � || � � }| � |� � }|dxx dz
cc<