File: //lib/python3/dist-packages/samba/subunit/__pycache__/run.cpython-310.pyc
o
�/ad` � @ s& d Z ddlZddlZddlZddlZG dd� dej�Zd dd�Zd dd �Zd
d � Z ej
Z
ej Z e
e
d d�du rRzddlZe�� Z
ejZ d
ZW n
eyQ dZY nw d
ZdZdZdZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�Zedkr�edej ej d � dS dS ) znRun a unittest testcase reporting results as Subunit.
$ python -m samba.subunit.run mylib.tests.test_suite
� Nc sf e Zd ZdZdd� Zddd�Zdd� Z� fd d
�Z� fd d �Zd
d� Z dd� Z
dd� Z dd� Z � Z
S )�TestProtocolClienta� A TestResult which generates a subunit stream for a test run.
# Get a TestSuite or TestCase to run
suite = make_suite()
# Create a stream (any object with a 'write' method). This should accept
# bytes not strings: subunit is a byte orientated protocol.
stream = open('tests.log', 'wb')
# Create a subunit result object which will output to the stream
result = subunit.TestProtocolClient(stream)
# Optionally, to get timing data for performance analysis, wrap the
# serialiser with a timing decorator
result = subunit.test_results.AutoTimingTestResultDecorator(result)
# Run the test suite reporting to the subunit result object
suite.run(result)
# Close the stream.
stream.close()
c C s t j�| � || _g | _d S �N)�unittest�
TestResult�__init__�_stream� successes)�self�stream� r �3/usr/lib/python3/dist-packages/samba/subunit/run.pyr 4 s
zTestProtocolClient.__init__Nc C sp | j �d| |�� � |r0| j �d� |D ]}| j �|� |�d�s)| j �d� q| j �d� | j �d� dS )z�Report an outcome of test test.
:param outcome: A string describing the outcome - used as the
event name in the subunit stream.
:param errors: A list of strings describing the errors.
z%s: z [
�
�]N)r �write�id�endswith)r �outcome�test�errors�errorr r r � _addOutcome9 s
� z TestProtocolClient._addOutcomec C s | j �|� dS )zReport a success in a test.N)r �append�r r r r r �
addSuccessJ s z TestProtocolClient.addSuccessc s6 t t| ��|� | j�d|�� d � | j�� dS )z%Mark a test as starting its test run.ztest: r
N)�superr � startTestr r r �flushr �� __class__r r r N s z TestProtocolClient.startTestc s t t| ��|� | �|� dS )z,Mark a test as having finished its test run.N)r r �stopTest� writeOutcomer r r r r T s zTestProtocolClient.stopTestc C s� | � || j�\}| _| � || j�\}| _| � || j�\}| _| � || j�\}| _| �|| j�\}| _| �|| j�\}| _|rAd}n|rFd}n|rKd}n|rPd}n |rUd}n|rZd}nd}|rl| j|||| | | d� | j �
� dS ) z)Output the overall outcome for test test.r �failure�skip� uxsuccess�xfail�
successfulN)r ) �
_filterErrorsr �failures�expectedFailures�skipped�_filterSuccessesr �unexpectedSuccessesr r r ) r r �err�failr$ r" �successr# r r r r r Y s, z TestProtocolClient.writeOutcomec C s@ g }g }|D ]}|d |u r|� |d � q|� |� q||fS )a^ Filter a list of errors by test test.
:param test: The test to filter by.
:param errors: A list of <test, error> pairs to filter.
:return: A pair whose first element is a list of strings containing
errors that apply to test test, and whose second element is a list
of the remaining elements.
r � �r )r r r �filtered�
unfilteredr r r r r& v s
z TestProtocolClient._filterErrorsc C s2 d}g }|D ]}||u rd}q|� |� q||fS )ab Filter a list of successes by test test.
:param test: The test to filter by.
:param successes: A list of tests to filter.
:return: A tuple whose first element is a boolean stating whether test
test was found in the list of successes, and whose second element is
a list of the remaining elements.
FTr0 )r r r r1 r2 r. r r r r* � s
z#TestProtocolClient._filterSuccessesc
C s> |� tjj�}| j�d|j|j|j|j |j
|j |j f � dS )z^Inform the client of the time.
":param datetime: A datetime.datetime object.
z*time: %04d-%02d-%02d %02d:%02d:%02d.%06dZ
N)
�
astimezone�datetime�timezone�utcr r �year�month�day�hour�minute�second� microsecond)r �
a_datetime�timer r r r? � s
�zTestProtocolClient.timer )�__name__�
__module__� __qualname__�__doc__r r r r r r r& r* r? �
__classcell__r r r r r ! s
r Fc C s� zt | �}W n ty | �� | fg Y S w t| �tjfv s!|r1g }|D ] }|�t|�� q%|S d }t| �}|D ]}|�� } t | dd �d urL| �
� || fgS )N�
sort_tests) �iter� TypeErrorr �typer � TestSuite�extend�_flatten_tests�
iterate_tests�getattrrE )�
suite_or_case� unpack_outer�tests�resultr �suite_idr r r rK � s( ��
rK c C s( t | |d�}|�� t�dd� |D ��S )z;Sort suite_or_case while preserving non-vanilla TestSuites.)rO c S s g | ]\}}|�qS r r )�.0�sort_keyr r r r �
<listcomp>� s z sorted_tests.<locals>.<listcomp>)rK �sortr rI )rN rO rP r r r � sorted_tests� s rW c c sL � zt | �}W n
ty | V Y dS w |D ] }t|�D ]}|V q qdS )z>Iterate through all of the test cases in 'test_suite_or_case'.N)rF rG rL )�test_suite_or_case�suiter �subtestr r r rL � s � � ��rL �discoverTz7 -c, --catch Catch control-C and display results
z= -b, --buffer Buffer stdout and stderr during test runs
a/ Usage: %(progName)s [options] [tests]
Options:
-h, --help Show this message
-v, --verbose Verbose output
-q, --quiet Minimal output
-l, --list List tests rather than executing them.
--load-list Specifies a file containing test ids, only tests matching
those ids are executed.
%(catchbreak)s%(buffer)s
Examples:
%(progName)s test_module - run tests from test_module
%(progName)s module.TestClass - run tests from module.TestClass
%(progName)s module.Class.test_method - run specified test method
All options must come before [tests]. [tests] can be a list of any number of
test modules, classes and test methods.
Alternative Usage: %(progName)s discover [options]
Options:
-v, --verbose Verbose output
s%(catchbreak)s%(buffer)s -s directory Directory to start discovery ('.' default)
-p pattern Pattern to match test files ('test*.py' default)
-t directory Top level directory of project (default to
start directory)
-l, --list List tests rather than executing them.
--load-list Specifies a file containing test ids, only tests matching
those ids are executed.
For test discovery all test modules must be importable from the top
level directory of the project.
c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd#d
d�Z d#dd�Z
dd� Z d#dd�Z d#dd�Z
dd� Zdd� Zedd � �Zd d � Zed d � �Zd!d"� Zd S )$�TestResultDecoratora General pass-through decorator.
This provides a base that other TestResults can inherit from to
gain basic forwarding functionality. It also takes care of
handling the case where the target doesn't support newer methods
or features by degrading them.
c C s
|| _ dS )z5Create a TestResultDecorator forwarding to decorated.N)� decorated�r r] r r r r 1 s
z TestResultDecorator.__init__c C � | j �|�S r )r] r r r r r r 6 � z TestResultDecorator.startTestc C �
| j �� S r )r] � startTestRun�r r r r rb 9 �
z TestResultDecorator.startTestRunc C r_ r )r] r r r r r r < r` z TestResultDecorator.stopTestc C ra r )r] � stopTestRunrc r r r re ? rd z TestResultDecorator.stopTestRunNc C � | j �||�S r )r] �addError�r r r, r r r rg B � z TestResultDecorator.addErrorc C rf r )r] �
addFailurerh r r r rj E ri z TestResultDecorator.addFailurec C r_ r )r] r r r r r r H r` z TestResultDecorator.addSuccessc C rf r )r] �addSkip�r r �reasonr r r rk K ri zTestResultDecorator.addSkipc C rf r )r] �addExpectedFailurerh r r r rn N ri z&TestResultDecorator.addExpectedFailurec C r_ r )r] �addUnexpectedSuccessr r r r ro Q r` z(TestResultDecorator.addUnexpectedSuccessc C ra r )r] �
wasSuccessfulrc r r r rp T rd z!TestResultDecorator.wasSuccessfulc C � | j jS r �r] �
shouldStoprc r r r rs W � z TestResultDecorator.shouldStopc C ra r )r] �stoprc r r r ru [ rd zTestResultDecorator.stopc C rq r )r] �testsRunrc r r r rv ^ rt z TestResultDecorator.testsRunc C r_ r )r] r? �r r> r r r r? b r` zTestResultDecorator.timer )r@ rA rB rC r r rb r re rg rj r rk rn ro rp �propertyrs ru rv r? r r r r r\ ( s(
r\ c s� e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd
d � Zd!d
d�Z d!dd�Z
dd� Z d!dd�Z d!dd�Z
dd� Zdd� Zedd � �Zd d � Zd d � Z� ZS )"�HookedTestResultDecoratorz/A TestResult which calls a hook on every event.c s t t| �| _ | j �|� d S r )r ry r r^ r r r r i s z"HookedTestResultDecorator.__init__c C � | � � | j�|�S r )�
_before_eventr r r r r r r m � z#HookedTestResultDecorator.startTestc C � | � � | j�� S r )r{ r rb rc r r r rb q �
z&HookedTestResultDecorator.startTestRunc C rz r )r{ r r r r r r r u r| z"HookedTestResultDecorator.stopTestc C r} r )r{ r re rc r r r re y r~ z%HookedTestResultDecorator.stopTestRunNc C � | � � | j�||�S r )r{ r rg rh r r r rg } � z"HookedTestResultDecorator.addErrorc C r r )r{ r rj rh r r r rj � r� z$HookedTestResultDecorator.addFailurec C rz r )r{ r r r r r r r � r| z$HookedTestResultDecorator.addSuccessc C r r )r{ r rk rl r r r rk � r� z!HookedTestResultDecorator.addSkipc C r r )r{ r rn rh r r r rn � r� z,HookedTestResultDecorator.addExpectedFailurec C rz r )r{ r ro r r r r ro � r| z.HookedTestResultDecorator.addUnexpectedSuccessc C r} r )r{ r rp rc r r r rp � r~ z'HookedTestResultDecorator.wasSuccessfulc C s | � � | jjS r )r{ r rs rc r r r rs � s z$HookedTestResultDecorator.shouldStopc C r} r )r{ r ru rc r r r ru � r~ z HookedTestResultDecorator.stopc C rz r )r{ r r? rw r r r r? � r| z HookedTestResultDecorator.timer )r@ rA rB rC r r rb r re rg rj r rk rn ro rp rx rs ru r? rD r r r r ry f s$
ry c s< e Zd ZdZ� fdd�Zdd� Zedd� �Zdd � Z� Z S )
� AutoTimingTestResultDecoratora Decorate a TestResult to add time events to a test run.
By default this will cause a time event before every test event,
but if explicit time data is being provided by the test run, then
this decorator will turn itself off to prevent causing confusion.
c s d | _ tt| ��|� d S r )�_timer r� r r^ r r r r � s z&AutoTimingTestResultDecorator.__init__c C s8 | j }|d ur d S tj�� jtjjd�}| j�|� d S )N)�tzinfo)r� r4 �utcnow�replacer5 r6 r] r? )r r? r r r r{ � s
z+AutoTimingTestResultDecorator._before_eventc C rq r rr rc r r r rs � rt z(AutoTimingTestResultDecorator.shouldStopc C s || _ | j�|�S )a\ Provide a timestamp for the current test activity.
:param a_datetime: If None, automatically add timestamps before every
event (this is the default behaviour if time() is not called at
all). If not None, pass the provided time onto the decorated
result object and disable automatic timestamps.
)r� r] r? rw r r r r? � s z"AutoTimingTestResultDecorator.time)
r@ rA rB rC r r{ rx rs r? rD r r r r r� � s
r� c @ s e Zd Zddd�Zdd� ZdS )�SubunitTestRunnerNc C s |pt j| _dS )zhCreate a SubunitTestRunner.
:param verbosity: Ignored.
:param buffer: Ignored.
N)�sys�stdoutr
)r � verbosity�bufferr
r r r r � s zSubunitTestRunner.__init__c C s t | j�}t|�}||� |S )z&Run the given test case or test suite.)r r
r� )r r rQ r r r �run� s
zSubunitTestRunner.run)NNN)r@ rA rB r r� r r r r r� � s
r� c
@ sl e Zd ZdZeZd Z ZZe ddde dddddf
dd�Z
dd� Z d d
� Z e
fd d �Zd
d� Zddd�ZdS )� TestProgramzA command-line program that runs a set of tests; this is primarily
for making test modules conveniently executable.
NTr/ c C s� |t krd | _n!t|t�r&t|�| _|�d�dd � D ] } t| j| �| _qn|| _|d u r0tj}|
d u r7tj }
|d u r>t
� }|| _ || _ || _
| | _|| _d| _d | _|| _|| _|d } | �dtjj �rt| �tjj�}
d|
d } ntj�|d �} | | _| �|� | jr�t| jd�}z
|� � }W |� � n|� � w t d d
� |D ��}t � � }t!| j"�D ]
}|�#� |v r�|�$|� q�|| _"| js�| �%� d S t!| j"�D ] }|
�&d |�#� � q�d S ) N�.r/ Fr z%srun.pyz%s.run����rbc s s � | ]
}|� � �d �V qdS )zutf-8N)�strip�decode)rS �liner r r � <genexpr> s � z'TestProgram.__init__.<locals>.<genexpr>z%s
)'r@ �module�
isinstance�str�
__import__�splitrM r� �argvr� r� �exit�
catchbreakr� r� � defaultTest� listtests� load_list�
testRunner�
testLoaderr �os�path�sep�basename�progName� parseArgs�open� readlines�close�setr rI rL r r �addTest�runTestsr )r r� r� r� r� r� r� r� r� r� r� �partr� �elements�source�lines�test_idsr1 r r r r r � s^
�
� �zTestProgram.__init__c C sV t |�dkr|d �� dkr| �|dd � � d S dd l}g d�}zu|�|dd � d|�\}}|D ]@\}}|dv r=| �� |dv rDd| _|d v rKd| _|d
v rW| jd u rWd | _|d v rc| jd u rcd | _|d
v rjd | _|dkrq|| _ q1t |�dkr�| j
d u r�d | _ nt |�dkr�|| _ n| j
f| _ | � � W d S |j
y� | �t�� d � Y d S w )Nr/ r[ � r )�help�verbose�quiet�catchr� �listz
load-list=�hHvqfcbl)z-hz-Hz--help)z-qz--quiet)�-v� --verbose)�-c�--catchT)�-b�--buffer)�-l�--list� --load-list)�len�lower�
_do_discovery�getopt� usageExitr� r� r� r� r� r� � testNames� createTestsr r� �exc_info)r r� r� � long_opts�options�args�opt�valuer r r r� sD
�
�zTestProgram.parseArgsc C s6 | j d u r| j�| j�| _d S | j�| j | j�| _d S r )r� r� �loadTestsFromModuler� r �loadTestsFromNamesrc r r r r� G s
�zTestProgram.createTestsc C s� t std��d| j | _dd l}|�� }| j|_|jdddddd d
� | jdkr3|jd d d
ddd d
� | jdkrC|jdddddd d
� |jdddddd� |jdddd d d� |jd d d d d!d� |jd"d#d$dd d%d&� |jd'd(d d)d� |� |�\}}t
|�d*kr�| � � t d+|�D ]
\}}t
|||� q�| jd u r�|j| _| jd u r�|j| _|j| _|j| _|jr�d,| _|j} |j}
|j} |� } | �| |
| �}
t|
�| _d S )-NzYUnable to use discovery, must use python 2.7 or greater, or install the discover package.z %s discoverr r� r� r� FzVerbose output�
store_true)�dest�defaultr� �actionr� r� r� z'Catch ctrl-C and display results so farr� r� r� z%Buffer stdout and stderr during testsz-sz--start-directory�startr� z*Directory to start discovery ('.' default))r� r� r� z-pz --pattern�patternztest*.pyz+Pattern to match tests ('test*.py' default)z-tz--top-level-directory�topz<Top level directory of project (defaults to start directory)r� r� r� z$List tests rather than running them.)r� r� r� r� r� r� z2Specify a filename containing the test ids to use.� )r� r� r� r� )�
have_discover�AssertionErrorr� �optparse� OptionParser�prog�
add_optionr� r� �
parse_argsr� r� �zip�setattrr� r� r� r� r� r� r� r[ rW r )r r� �Loaderr� �parserr� r� �namer� � start_dirr� �
top_level_dir�loader�loadedr r r r� N sj �
�
� � � ��
�
zTestProgram._do_discoveryc C sN | j rttdd �d urt�� | j�| j�| _| jr%t �| j�
� � d S d S )N�installHandler) r� rM r r� r� r� r rQ r� r� rp rc r r r r� � s �zTestProgram.runTestsc C s� |rt |� | jddd�}| jdkrt|d< | jdkr t|d< | j| }|�d�}|�dd� |�d d� t d� |�� t
� d� d S )
N� )r� r� r� Fr� r� r
r� z)Run a test suite with a subunit reporter.r� ) �printr� r� �
CATCHBREAKr� � BUFFEROUTPUT�USAGEr� �insert�joinr� r� )r �msg�usage�
usage_text� usage_linesr r r r� � s �
zTestProgram.usageExitr )r@ rA rB rC �
USAGE_AS_MAINr� r� r� r� �defaultTestLoaderr r� r� �defaultTestLoaderClsr� r� r� r r r r r� � s
�:( Br� �__main__)r� r� r� )F) rC r4 r� r� r r r rK rW rL r� �
TestLoaderr� rM r[ �DiscoveringTestLoaderr� � ImportErrorr� r� r� �objectr\ ry r� r� r� r@ r� r� r r r r �<module> sD
�&>A# N�