File: //lib/python3/dist-packages/samba/tests/blackbox/__pycache__/check_output.cpython-310.pyc
o
�/a
� @ sD d Z ddlZddlmZ G dd� d�Zd dd �ZG d
d � d e�ZdS )
z7
Blackbox tests for blackboxtest check output methods.
� N)�BlackboxTestCasec @ s@ e Zd ZdZG dd� de�Zdd� Zdd� Zdd � Zd
d � Z d S )
�
TimeoutHelperz�
Timeout class using alarm signal.
Raise a Timeout exception if a function timeout.
Usage:
try:
with Timeout(3):
foobar("Request 1")
except TimeoutHelper.Timeout:
print("Timeout")
c @ s e Zd ZdS )zTimeoutHelper.TimeoutN)�__name__�
__module__� __qualname__� r r �C/usr/lib/python3/dist-packages/samba/tests/blackbox/check_output.py�Timeout& s r c C s
|| _ d S �N)�sec)�selfr r r r �__init__) s
zTimeoutHelper.__init__c C s t � t j| j� t �| j� d S r
)�signal�SIGALRM�
raise_timeout�alarmr �r r r r � __enter__, s zTimeoutHelper.__enter__c G s t �d� d S )Nr )r r �r �argsr r r �__exit__0 s zTimeoutHelper.__exit__c G s t �� �r
)r r r r r r r 3 s zTimeoutHelper.raise_timeoutN)
r r r �__doc__� Exceptionr r
r r r r r r r r s
r �$� P c C s d� | ||�S )z>Build a command to call gen_output.py to generate large outputz3gen_output.py --data {0} --repeat {1} --retcode {2})�format)�data�repeat�retcoder r r �
_make_cmdline7 s �r c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �CheckOutputTestsa
Blackbox tests for check_xxx methods.
The check_xxx methods in BlackboxTestCase will deadlock
on large output from command which caused by Popen.wait().
This is a test case to show the deadlock issue,
will fix in another commit.
c C sb z t d�� | �t� � W d � W dS 1 sw Y W dS t jy0 | jdd� Y dS w )z!Call check_run with large output.�
N�Timeout!��msg)r � check_runr r �failr r r r �test_check_run_timeoutI s
&��z'CheckOutputTests.test_check_run_timeoutc C �h z"t d�� | �tdd�d� W d � W d S 1 sw Y W d S t jy3 | jdd� Y d S w )Nr! r �r r" r# �r �check_exit_coder r r& r r r r �.test_check_exit_code_with_large_output_successQ �
&��z?CheckOutputTests.test_check_exit_code_with_large_output_successc C r( )Nr! � r) r" r# r* r r r r �.test_check_exit_code_with_large_output_failureX r- z?CheckOutputTests.test_check_exit_code_with_large_output_failurec C s� d}d}|| }t ||d�}z'td�� | �|�}| �||�d�� W d � W d S 1 s.w Y W d S tjyF | jdd� Y d S w )N�@r )r r r! zutf-8r" r# )r r � check_output� assertEqual�encoder r& )r r r �expected�cmdline�actualr r r �#test_check_output_with_large_output_ s
&��z4CheckOutputTests.test_check_output_with_large_outputN)r r r r r' r, r/ r7 r r r r r > s
r )r r r )r r � samba.testsr r r r r r r r �<module> s