File: //lib/python3/dist-packages/samba/tests/emulate/__pycache__/traffic.cpython-310.pyc
o
�/a� � @ s< d dl mZ d dlZd dlmZ dZG dd� dejj�Z dS )� )�StringION)�trafficz&testdata/traffic-sample-very-short.txtc @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�TrafficEmulatorTestsc C s t �� | _d S �N)r � TrafficModel�model��self� r
�=/usr/lib/python3/dist-packages/samba/tests/emulate/traffic.py�setUp s zTrafficEmulatorTests.setUpc C s | ` d S r )r r r
r
r �tearDown! s z TrafficEmulatorTests.tearDownc
C sR t �� }tt�}t j|gdd�\}}}}|�� |�|� g d�g d�dgddgdgg d�dgd gd
gd gdgd gd
� }g d�g d�dgdgddgdgd�}d| _dd� |j� � D �} dd� |j
� � D �}
| � || � | � ||
� t � }|�
|� t �� } |�d� | �|� dd� | j� � D �} d d� | j
� � D �}
| � || � | � ||
� d S ) N�include)�dns_mode)�dns:0r r �ldap:3)r r r �wait:0�cldap:3�rpc_netlogon:29)r r r r �-�ldap:2r � kerberos:) �r r )r r �r r �r r �r r )r r )r r �r r �r r �r r )r r �r r ��� r" r" �Netlogonr" r" r" r! r! ) r
r
r
r
r
r
r
r
r
�r" �r" r" r" r" r" r" r" �r" r" r" a subschemaSubentry,dsServiceName,namingContexts,defaultNamingContext,schemaNamingContext,configurationNamingContext,rootDomainNamingContext,supportedControl,supportedLDAPVersion,supportedLDAPPolicies,supportedSASLMechanisms,dnsHostName,ldapServiceName,serverName,supportedCapabilitiesr" r" r" ��2zDC,DCr" �cnr" r" r" r
)r r r r r r � c S � i | ] \}}|t |��qS r
��sorted��.0�k�vr
r
r �
<dictcomp>S � zGTrafficEmulatorTests.test_parse_ngrams_dns_included.<locals>.<dictcomp>c S r+ r
r, r. r
r
r r2 T r3 r c S r+ r
r, r. r
r
r r2 ` r3 c S r+ r
r, r. r
r
r r2 a r3 )r r �open� TEST_FILE�ingest_summaries�close�learn�maxDiff�ngrams�items�
query_details� assertEqualr �save�seek�load) r r �f�
conversations�interval�duration�
dns_counts�expected_ngrams�expected_query_detailsr: �details�model2r
r
r � test_parse_ngrams_dns_included$ sZ �
�
��
z3TrafficEmulatorTests.test_parse_ngrams_dns_includedc
C s: t t�}t�|g�\}}}}|�� | j�||� ddgdgdgddgdgdgdgdgdgd� }g d �d
gd gd d
gdgd�}d| _dd� | jj� � D �}dd� | jj
� � D �} | � ||� | � || � t � }| j�
|� t�� }
|�d� |
�|� dd� |
j� � D �}dd� |
j
� � D �} | � ||� | � || � d S )Nr r r r r r r ) r )r r r r r r r r r r r$ r% r& r' r
)r r r r r r* c S r+ r
r, r. r
r
r r2 � r3 z:TrafficEmulatorTests.test_parse_ngrams.<locals>.<dictcomp>c S r+ r
r, r. r
r
r r2 � r3 r c S r+ r
r, r. r
r
r r2 � r3 c S r+ r
r, r. r
r
r r2 � r3 )r4 r5 r r6 r7 r r8 r9 r: r; r< r= r r> r r? r@ ) r rA rB rC rD rE rF rG r: rH rI r
r
r �test_parse_ngramse sP
��
��
z&TrafficEmulatorTests.test_parse_ngramsN)�__name__�
__module__� __qualname__r r
rJ rK r
r
r
r r s
Ar )
�ior � samba.tests�samba�
samba.emulater r5 �tests�TestCaser r
r
r
r �<module> s