File: //lib/python3/dist-packages/samba/tests/blackbox/__pycache__/traffic_replay.cpython-310.pyc
o
�/a� � @ s� d Z ddlZddlZddlmZ dZdZdZejd Z ejd Z
ejd Z d
e e
f Z ej
�ed �Zej
�ed �ZG d
d� de�ZdS )z!Blackbox tests for traffic_replay� N)�BlackboxTestCasez$python/samba/tests/blackbox/testdatazscript/traffic_replayz!--fixed-password=trafficreplay01%�SERVER�PASSWORD�USERNAMEz-U%s%%%sz traffic-sample-very-short.modelztraffic_replay-%s.expectedc @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�TrafficLearnerTestsc C s" d}dt |ttf }| �|� d S )Nz
--clean-up� %s %s %s %s)�SCRIPT�CREDSr � check_run��self�options�command� r �E/usr/lib/python3/dist-packages/samba/tests/blackbox/traffic_replay.py�tearDown& s z TrafficLearnerTests.tearDownc C sB d}dt |tttf }| �|� dt t|tttf }| �|� dS )zEEnsure the generate users only option functions correctly
�[--generate-users-only --number-of-users 20 --number-of-groups 5 --average-groups-per-user 2�%s %s %s %s %sz%s %s %s %s %s %sN)r �FIXEDr r r
�MODELr r r r �test_generate_users_only+ s
�
�z,TrafficLearnerTests.test_generate_users_onlyc C s� t dgdgddgg d�f�D ]@\}}| �� �0}ttd|ddg| tttg }| �|� tt | ��
� }t|��
� }| � ||� W d � n1 sIw Y qdS ) z?Ensure a summary file is generated and the contents are correct�--random-seed=3z--random-seed=4z --conversation-persistence=0.5)r z --old-scalez --conversation-persistence=0.95z--traffic-summaryz-D1�-S0.1N) � enumerate�mktempr r r r r r
�open�EXPECTED_OUTPUT�read�assertStringsEqual)r �i�opts�outputr �expected�actualr r r �test_summary_generation7 s. ��
���
���z+TrafficLearnerTests.test_summary_generationc C s t tttg}| �|d� dS )z>Ensure a summary file with no fixed password fails
� N)r r r r �check_exit_code�r r r r r � test_summary_replay_no_fixedM s z0TrafficLearnerTests.test_summary_replay_no_fixedc C s t ttddttg}| �|� dS )z7Ensure a model can be replayed against a DC
z-D2r N)r r r r r r
r' r r r �test_model_replayS s �z%TrafficLearnerTests.test_model_replayc C sB d}dt |ttf }| �|d� dt t|ttf }| �|d� dS )zOEnsure the generate users only fails if no fixed_password supplied"
r r r% r N)r r r r& r r r r r �$test_generate_users_only_no_password\ s
z8TrafficLearnerTests.test_generate_users_only_no_passwordN) �__name__�
__module__� __qualname__r r r$ r( r) r* r r r r r $ s r )�__doc__�os�tempfile� samba.testsr �DATA_DIRr r �environr r �USERr �path�joinr r r r r r r �<module> s