File: //lib/python3/dist-packages/samba/tests/__pycache__/messaging.cpython-310.pyc
o
�/a� � @ sh d Z ddlZddlmZ ddlmZ ddlZddlmZ ddl m
Z
ddl Z ddl Z G dd� de�Z
dS ) zTests for samba.messaging.� N)� Messaging)�TestCase)� ndr_print)� server_idc @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�MessagingTestsc O s t j�� |d< t|i |��S )N�lp_ctx)�samba�tests� env_loadparmr )�self�args�kwargs� r �7/usr/lib/python3/dist-packages/samba/tests/messaging.py� get_context! s zMessagingTests.get_contextc C s> | � � }dd� }|�|d f�}| �t|t�� |�||� d S )Nc S s d S �Nr r r r r �callback( s z.MessagingTests.test_register.<locals>.callback)r �register�
assertTrue�
isinstance�int�
deregister)r �xr �msg_typer r r �
test_register% s
z MessagingTests.test_registerc C s | � � }| �t|�� t�� d S r )r r r �irpc_all_servers�list�r r r r r �test_all_servers. � z MessagingTests.test_all_serversc C s2 | � � }|�� D ]}| �t|�|j�t�� qd S r )r r r r �irpc_servers_byname�namer )r r r! r r r � test_by_name2 s �zMessagingTests.test_by_namec C s | � � }| �t|jd� d S )Nz samba.messaging test NONEXISTING)r � assertRaises�KeyErrorr r r r r �test_unknown_name7 s �z MessagingTests.test_unknown_namec C s | � � }| �t|jtj�� d S r )r r r r r r r r �test_assign_server_id< r z$MessagingTests.test_assign_server_idc C st | � � }dt�dd� }|�|� |�|�}| �t|�d� | �t|j�t|d �� |� |� | �
t |j|� d S )Nzsamba.messaging test-%d� i@B r ) r �random�randint�
irpc_add_namer � assertEqual�lenr r �irpc_remove_namer# r$ )r r r! � name_listr r r �test_add_remove_name@ s
�
�z#MessagingTests.test_add_remove_namec s
ddi}ddi}d}d� d}| � d��� �fdd�}��||f�}dd� }| � d �}|�||f�� |�d|d
� |�d|d � t�� }|d d k sO|d d k rs|ss|�d
� ��d
� t�� | dkred}|d d k sq|d d k rs|rQ| �|d d � | �|d d � d S )N�countr F)r r' c �" | d d7 < �� |� |� d S �Nr0 r' ��send��got_pingr �src�data��msg_pong�
server_ctxr r �
ping_callbackV � z5MessagingTests.test_ping_speed.<locals>.ping_callbackc S � | d d7 < d S r2 r ��got_pongr r7 r8 r r r �
pong_callback\ � z5MessagingTests.test_ping_speed.<locals>.pong_callback)r � �testing�testing2rC 皙�����?r' T)r r r4 �time� loop_oncer+ ) r r6 r@ �timeout�msg_pingr< rA �
client_ctx�
start_timer r9 r �test_ping_speedL s,
�z MessagingTests.test_ping_speedc
s ddi}ddi}d}d� d}t �� }| �|df��� �fdd�}��||f�}dd� }| �d �}|�||f�� |�|df|d
� |�d |d � t�� } |d d
k sW|d d
k r{|s{|�d� ��d� t�� | dkrmd}|d d
k sy|d d
k r{|rY| �|d d
� | �|d d
� d S )Nr0 r Fr' c r1 r2 r3 r5 r9 r r r< � r= z9MessagingTests.test_pid_defaulting.<locals>.ping_callbackc S r> r2 r r? r r r rA � rB z9MessagingTests.test_pid_defaulting.<locals>.pong_callback)rC rD )r' rE rC rF T)�os�getpidr r r4 rG rH r+ )
r r6 r@ rI rJ �pidr< rA rK rL r r9 r �test_pid_defaultingu s.
�z"MessagingTests.test_pid_defaultingN) �__name__�
__module__� __qualname__r r r r" r% r& r/ rM rQ r r r r r s )r )�__doc__r �samba.messagingr � samba.testsr rG � samba.ndrr � samba.dcerpcr r( rN r r r r r �<module> s