File: //usr/lib/python3/dist-packages/django/test/__pycache__/runner.cpython-310.pyc
o
3�a�u � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl m Z d dlm
Z
d dlmZ d dlmZ d dlmZmZ d dlmZmZmZmZmZ m Z d dl m Z d d l m!Z! zd dl"Z#W n
e$y{ d dl#Z#Y nw zd dl%Z&W n e$y� dZ&Y nw G d
d � d e
j'�Z(G d d
� d
e
j'�Z)G dd� d�Z*G dd� d�Z+dd� Z,d a-dd� Z.dd� Z/G dd� de
j0�Z1G dd� d�Z2d d � Z3d'd d �Z4d'd!d"�Z5d#d$� Z6d%d&� Z7dS )(� N)�
import_module)�StringIO)� call_command)� connections)�SimpleTestCase�TestCase)�NullTimeKeeper�
TimeKeeper�setup_databases�setup_test_environment�teardown_databases�teardown_test_environment��
OrderedSet)�PY37c s` e Zd Z� fdd�Z� fdd�Z� fdd�Z� fdd�Z� fd d
�Z� fd d �Zd
d� Z � Z
S )�DebugSQLTextTestResultc s4 t �d�| _| j�t j� d | _t� �|||� d S )Nzdjango.db.backends)�logging� getLogger�logger�setLevel�DEBUG�debug_sql_stream�super�__init__)�self�stream� descriptions� verbosity�� __class__� �4/usr/lib/python3/dist-packages/django/test/runner.pyr & s z DebugSQLTextTestResult.__init__c s4 t � | _t�| j�| _| j�| j� t� �|� d S �N) r r r �
StreamHandler�handlerr �
addHandlerr � startTest�r �testr r r! r& , s z DebugSQLTextTestResult.startTestc sT t � �|� | j�| j� | jr(| j�d� | j� | j�
� � | j� | j � d S d S )Nr )
r �stopTestr �
removeHandlerr$ �showAllr �seekr �write�read�writeln�
separator2r' r r r! r) 2 s �z DebugSQLTextTestResult.stopTestc sN t � �||� | jd u rd}n | j�d� | j�� }| jd |f | jd< d S )N� r ���)r �addErrorr r, r. �errors)r r( �err�sqlr r r! r3 : s
z DebugSQLTextTestResult.addErrorc s: t � �||� | j�d� | jd | j�� f | jd< d S �Nr r2 )r �
addFailurer r, �failuresr. �r r( r5 r r r! r8 D s z!DebugSQLTextTestResult.addFailurec s` t � �|||� |d ur.| j�d� t|d |j�r | jn| j}|d | j�� f |d< d S d S r7 ) r �
addSubTestr r, �
issubclass�failureExceptionr9 r4 r. )r r( �subtestr5 r4 r r r! r; I s �z!DebugSQLTextTestResult.addSubTestc C sp |D ]3\}}}| j �| j� | j �d|| �|�f � | j �| j� | j �|� | j �| j� | j �|� qd S )Nz%s: %s)r r/ �
separator1�getDescriptionr0 )r �flavourr4 r( r5 � sql_debugr r r! �printErrorListP s �z%DebugSQLTextTestResult.printErrorList) �__name__�
__module__� __qualname__r r& r) r3 r8 r; rC �
__classcell__r r r r! r % s
r c s4 e Zd ZdZ� fdd�Z� fdd�Zdd� Z� ZS )�PDBDebugResultz^
Custom result class that triggers a PDB session when an error or failure
occurs.
c � t � �||� | �|� d S r" )r r3 �debugr: r r r! r3 ` � zPDBDebugResult.addErrorc rI r" )r r8 rJ r: r r r! r8 d rK zPDBDebugResult.addFailurec C s2 | � � d| _|\}}}td| � t�|� d S )NFz
Opening PDB: %r)�_restoreStdout�buffer�print�pdb� post_mortem)r �error�exc_type� exc_value� tracebackr r r! rJ h s
zPDBDebugResult.debug)rD rE rF �__doc__r3 r8 rJ rG r r r r! rH Z s
rH c @ s� e Zd ZdZdd� Zedd� �Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(S ))�RemoteTestResulta�
Record information about which tests have succeeded and which have failed.
The sole purpose of this class is to record events in the child processes
so they can be replayed in the master process. As a consequence it doesn't
inherit unittest.TestResult and doesn't attempt to implement all its API.
The implementation matches the unpythonic coding style of unittest2.
c C s. t d ur t j�� g | _d| _d| _d| _d S )NFr )�tblib�pickling_support�install�events�failfast�
shouldStop�testsRun�r r r r! r { s
zRemoteTestResult.__init__c C s
| j d S )N� )r] r^ r r r! �
test_index� s
zRemoteTestResult.test_indexc C s t �t �|�� dS )z�
Confirm that obj can be pickled and unpickled as multiprocessing will
need to pickle the exception in the child process and unpickle it in
the parent process. Let the exception rise, if not.
N)�pickle�loads�dumps)r �objr r r! �_confirm_picklable� s z#RemoteTestResult._confirm_picklablec C s t d�|||�� d S )Na'
Subtest failed:
test: {}
subtest: {}
Unfortunately, the subtest that failed cannot be pickled, so the parallel
test runner cannot handle it cleanly. Here is the pickling error:
> {}
You should re-run this test with --parallel=1 to reproduce the failure
with a cleaner failure message.
)rN �format)r r( r> �
pickle_excr r r! �_print_unpicklable_subtest� s
�z+RemoteTestResult._print_unpicklable_subtestc
C s� z| � |� W d S tyG } z3t|d �}tj|dddd�}t|�}tj|dddd�}td u r9td�||�� � td�|||�� � d }~ww )Nr_ �K z )�initial_indent�subsequent_indentz�
{} failed:
{}
Unfortunately, tracebacks cannot be pickled, making it impossible for the
parallel test runner to handle this exception cleanly.
In order to see the traceback, you should install tblib:
python -m pip install tblib
aL
{} failed:
{}
Unfortunately, the exception it raised cannot be pickled, making it impossible
for the parallel test runner to handle it cleanly.
Here's the error encountered while trying to pickle the exception:
{}
You should re-run this test with the --parallel=1 option to reproduce the
failure and get a correct traceback.
)re � Exception�repr�textwrap�fillrW rN rf )r r( r5 �exc�original_exc_txt�pickle_exc_txtr r r! �check_picklable� s$ � �
���z RemoteTestResult.check_picklablec
C s: z| � |� W d S ty } z| �|||� � d }~ww r" )re rl rh )r r( r> rp r r r! �check_subtest_picklable� s ��z(RemoteTestResult.check_subtest_picklablec C s | j r | �� d S d S r" )r[ �stopr^ r r r! �stop_if_failfast� s �z!RemoteTestResult.stop_if_failfastc C s
d| _ d S )NT)r\ r^ r r r! ru � �
zRemoteTestResult.stopc C � | j �d� d S )N)� startTestRun�rZ �appendr^ r r r! ry � � z RemoteTestResult.startTestRunc C rx )N)� stopTestRunrz r^ r r r! r} � r| z RemoteTestResult.stopTestRunc C s$ | j d7 _ | j�d| jf� d S )Nr_ r& )r] rZ r{ r` r' r r r! r& � s zRemoteTestResult.startTestc C � | j �d| jf� d S )Nr) �rZ r{ r` r' r r r! r) � � zRemoteTestResult.stopTestc C �, | � ||� | j�d| j|f� | �� d S )Nr3 �rs rZ r{ r` rv r: r r r! r3 � � zRemoteTestResult.addErrorc C r� )Nr8 r� r: r r r! r8 � r� zRemoteTestResult.addFailurec C sF |d ur!| � ||� | �||� | j�d| j||f� | �� d S d S )Nr; )rs rt rZ r{ r` rv )r r( r> r5 r r r! r; � s �zRemoteTestResult.addSubTestc C r~ )N�
addSuccessr r' r r r! r� � r� zRemoteTestResult.addSuccessc C s | j �d| j|f� d S )N�addSkipr )r r( �reasonr r r! r� s zRemoteTestResult.addSkipc C s> t d u r
|d |d d f}| �||� | j�d| j|f� d S )Nr r_ �addExpectedFailure)rW rs rZ r{ r` r: r r r! r� s z#RemoteTestResult.addExpectedFailurec C s | j �d| jf� | �� d S )N�addUnexpectedSuccess)rZ r{ r` rv r' r r r! r� s z%RemoteTestResult.addUnexpectedSuccessN)rD rE rF rU r �propertyr` re rh rs rt rv ru ry r} r& r) r3 r8 r; r� r� r� r� r r r r! rV p s,
/
rV c @ s& e Zd ZdZeZddd�Zdd� ZdS ) �RemoteTestRunnerz�
Run tests and record everything but don't display anything.
The implementation matches the unpythonic coding style of unittest2.
FNc C s || _ |d ur || _d S d S r" )r[ � resultclass)r r[ r� r r r! r s
�zRemoteTestRunner.__init__c C s& | � � }t�|� | j|_||� |S r" )r� �unittest�registerResultr[ )r r( �resultr r r! �run" s
zRemoteTestRunner.run)FN)rD rE rF rU rV r� r r� r r r r! r� s
r� c C s< t �� dkrdS zttjd �W S ty t �� Y S w )zBDefault number of test processes when using the --parallel option.�forkr_ �DJANGO_TEST_PROCESSES)�multiprocessing�get_start_method�int�os�environ�KeyError� cpu_countr r r r! �default_test_processes* s �r� c C sv | � � � | jd7 _| jaW d � n1 sw Y tD ]}t| }|j�tt��}|j�|� |� � q dS )z�
Switch to databases dedicated to this worker.
This helper lives at module-level because of the multiprocessing module's
requirements.
r_ N)
�get_lock�value�
_worker_idr �creation�get_test_db_clone_settings�str�
settings_dict�update�close)�counter�alias�
connectionr� r r r! � _init_worker9 s
�
�r� c C s* | \}}}}||d�}|� |�}||jfS )z�
Run a suite of tests with a RemoteTestRunner and return a RemoteTestResult.
This helper lives at module-level and its arguments are wrapped in a tuple
because of the multiprocessing module's requirements.
)r[ )r� rZ )�args� runner_class�subsuite_index�subsuiter[ �runnerr� r r r! �
_run_subsuiteR s
r� c s> e Zd ZdZeZeZeZ d � fdd� Z
dd� Z dd� Z � Z
S )
�ParallelTestSuitea�
Run a series of tests in parallel in several processes.
While the unittest module's documentation implies that orchestrating the
execution of tests is the responsibility of the test runner, in practice,
it appears that TestRunner classes are more concerned with formatting and
displaying test results.
Since there are fewer use cases for customizing TestSuite than TestRunner,
implementing parallelization at the level of the TestSuite improves
interoperability with existing custom test runners. A single instance of a
test runner can still collect results from all tests without being aware
that they have been run in parallel.
Fc s$ t |�| _|| _|| _t� �� d S r" )�partition_suite_by_case� subsuites� processesr[ r r )r �suiter� r[ r r r! r t s
zParallelTestSuite.__init__c
s t �tjd�}t j� j� jj|gd�}� fdd�t� j �D �}|�
� j j|�} |j r0|�
� nQz
|jdd�\}}W n t jyD Y q' tyP |�� Y n1w t� j | �}|D ]%} | d }
t||
d�} | du rkqZ|| d } | d
d� }| | g|�R � qZq(|�� |S ) a
Distribute test cases across workers.
Return an identifier of each test case with its result in order to use
imap_unordered to show results as soon as they're available.
To minimize pickling errors when getting results from workers:
- pass back numeric indexes in self.subsuites instead of tests
- make tracebacks picklable with tblib, if available
Even with tblib, errors may still occur for dynamically created
exception classes which cannot be unpickled.
r )r� � initializer�initargsc s g | ] \}}� j ||� jf�qS r )r� r[ )�.0�indexr� r^ r r! �
<listcomp>� s ��z)ParallelTestSuite.run.<locals>.<listcomp>Tg�������?)�timeoutNr_ � )r� �Value�ctypes�c_int�Poolr� � init_worker�__func__� enumerater� �imap_unordered� run_subsuiter\ � terminate�next� TimeoutError�
StopIterationr� �list�getattr�join)
r r� r� �poolr� � test_resultsr� rZ �tests�event�
event_namer$ r( r r^ r! r� z sD �
� � �zParallelTestSuite.runc C s
t | j�S r" )�iterr� r^ r r r! �__iter__� rw zParallelTestSuite.__iter__�F)rD rE rF rU r� r� r� r� r� r� r r� r� rG r r r r! r� _ s 6r� c @ s� e Zd ZdZejZeZej Z
ej Z e
efZ d%dd�Zed d
� �Zd d � Zd&d
d�Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Z d d � Z d!d"� Z d'd#d$�Z dS )(�DiscoverRunnerz8A Django test runner that uses unittest2 test discovery.Nr_ TFr c K s || _ || _|| _|| _|| _|| _|| _|| _| | _|
| _ t
| p"g �| _ t
| p)g �| _ t
�� sS|rSz t
jtj�� d� W n ttjfyR t
jtj�� d� Y nw || _| jrb| j dkrbtd��|| _| jrq| j dkrqtd��d | _|ryt� nt � | _ |
r�dd� |
D �| _d S d S )N)�filer_ zFYou cannot use --pdb with parallel tests; pass --parallel=1 to use it.zLYou cannot use -b/--buffer with parallel tests; pass --parallel=1 to use it.c S s h | ] }d |v r
|nd| �qS )�*z*%s*r )r� �patternr r r! � <setcomp>� s ��z*DiscoverRunner.__init__.<locals>.<setcomp>) r� � top_levelr � interactiver[ �keepdb�reverse�
debug_mode� debug_sql�parallel�set�tags� exclude_tags� faulthandler�
is_enabled�enable�sys�stderr�fileno�AttributeError�io�UnsupportedOperation�
__stderr__rO �
ValueErrorrM �test_name_patternsr r � time_keeper)r r� r� r r� r[ r� r� r� r� r� r� r� r� rO rM �enable_faulthandler�timing�kwargsr r r! r � sB �� ��zDiscoverRunner.__init__c C s |j ddddd� |j dddd d
� |j d d d
d� |j ddd dd� |j dd dd� |j ddd dd� |j dddtt� ddd � |j d d d d d!� |j d"d d#d$d!� |j d%d d&d� |j d'd(d d)d� |j d*d+d,d-d!� |j d.d d/d� tr�|j d0d d1d2d!� d S d S )3Nz-tz--top-level-directoryr� z,Top level of project for unittest discovery.)�dest�helpz-pz --patternztest*.pyz0The test matching pattern. Defaults to test*.py.)�defaultr� z--keepdb�
store_truez#Preserves the test DB between runs.)�actionr� z-rz --reversezReverses test cases order.z --debug-modez Sets settings.DEBUG to True.z-dz --debug-sqlz%Prints logged SQL queries on failure.z
--parallel�?r_ �Nz+Run tests using up to N parallel processes.)�nargsr� �type�const�metavarr� z--tagr{ r� zBRun only tests with the specified tag. Can be used multiple times.)r� r� r� z
--exclude-tagr� zDDo not run tests with the specified tag. Can be used multiple times.z--pdbz@Runs a debugger (pdb, or ipdb if installed) on error or failure.z-bz--bufferz"Discard output from passing tests.z--no-faulthandler� store_falser� z5Disables the Python faulthandler module during tests.z--timingz=Output timings, including database set up and total run time.z-kr� z~Only run test methods and classes that match the pattern or substring. Can be used multiple times. Same as unittest -k option.)� add_argumentr� r� r )�cls�parserr r r! �
add_arguments� sv �������������
��z DiscoverRunner.add_argumentsc K s t | jd� t�� d S )N)rJ )r r� r� �installHandler�r r� r r r! r + s z%DiscoverRunner.setup_test_environmentc K s� | � � }|pdg}|p g }| j| j_i }| jd ur | j|d< | jd ur(| j|d< |D ]e}|�� }d }tj� |�}tj�
|�sE| j� |�}n,tj� |�rq| jsq|} tj�
| d�}
tj�
|
�rltj�| �} | | krin| } qP | |d< |rw|�� s�t|�r�| jjd d|i|��}d | j_|�|� q*|D ]} |�| � q�| js�| jr�| jdkr�| jr�tdd �
t| j�� � | jr�td
d �
t| j�� � t|| j| j�}t|| j | j �}| j d kr�| � || j | j �}
t!|
j"�}t#| j |�| _ | j d kr�|
}|S )
N�.r� �
top_level_dirTz __init__.py� start_dirr� zIncluding test tag(s): %s.�, zExcluding test tag(s): %s.r_ r )$�
test_suiter� � test_loader�testNamePatternsr� r� �copyr� �path�abspath�exists�loadTestsFromName�isdirr� �dirname�countTestCases�is_discoverable�discover�_top_level_dir�addTests�addTestr� r� r rN �sorted�filter_tests_by_tags�
reorder_suite�
reorder_byr� r� �parallel_test_suiter[ �lenr� �min)r � test_labels� extra_testsr� r� �discover_kwargs�labelr� �
label_as_pathr� �init_py�try_nextr( �parallel_suite�parallel_unitsr r r! � build_suite/ s`
zDiscoverRunner.build_suitec K s( t | j| jf| j| j| j| jd�|��S )N)r� r� r� r� )�_setup_databasesr r� r� r� r� r� r r r r! r
� s ���z DiscoverRunner.setup_databasesc C s | j rtS | jr
tS d S r" )r� r rO rH r^ r r r! �get_resultclass� s
�z DiscoverRunner.get_resultclassc C s | j | �� | j| jd�S )N)r[ r� r rM )r[ r( r rM r^ r r r! �get_test_runner_kwargs� s
�z%DiscoverRunner.get_test_runner_kwargsc C s t d| j|d� d S )N�check)r � databases)r r )r r+ r r r! �
run_checks� s zDiscoverRunner.run_checksc K s" | � � }| jdi |��}|�|�S )Nr )r) � test_runnerr� )r r� r� r� r r r! � run_suite� s
zDiscoverRunner.run_suitec K s t || j| j| jd� dS )z%Destroy all the non-mirror databases.)r r� r� N)�_teardown_databasesr r� r� )r �
old_configr� r r r! r � s
�z!DiscoverRunner.teardown_databasesc K s t �� t� d S r" )r� r* r
r r r r! r
� s
z(DiscoverRunner.teardown_test_environmentc K s t |j�t |j� S r" )r r9 r4 )r r� r� r� r r r! � suite_result� s zDiscoverRunner.suite_resultc C s` t � }|D ](}t|tj�r%t|dd �}|dkr t t� S |r$|�|� q|�| �|�� q|S )Nr+ �__all__)r� �
isinstancer� r r� r r� �_get_databases)r r� r+ r( �test_databasesr r r! r4 � s
�z DiscoverRunner._get_databasesc sD | � |�� | jdkr � fdd�tD �}|r tdd�t|�� � � S )Nr� c s g | ]}|� vr|�qS r r )r� r� �r+ r r! r� � s z0DiscoverRunner.get_databases.<locals>.<listcomp>z)Skipping setup of unused database(s): %s.r )r4 r r rN r� r )r r� �unused_databasesr r6 r! �
get_databases� s
z DiscoverRunner.get_databasesc K sZ | � � | �||�}| �|�}| j�d�� | j|d�}W d � n1 s&w Y d}zFz | �|� | �|�}W n
tyD d}� w W z!| j�d��
| � |� W d � n1 s]w Y | �
� W n: tys |sq� Y n/w z!| j�d��
| � |� W d � n1 s�w Y | �
� W w ty� |s�� Y w w | j� � | � ||�S )aZ
Run the unit tests for all the test labels in the provided list.
Test labels should be dotted Python paths to test modules, test
classes, or test methods.
A list of 'extra' tests may also be provided; these tests
will be added to the test suite.
Return the number of tests that failed.
zTotal database setup)�aliasesNFTzTotal database teardown)
r r& r8 r� �timedr
r, r. rl r r
�
print_resultsr1 ) r r r r� r� r+ r0 �
run_failedr� r r r! � run_tests� sJ
�
�� � ��� � ��
zDiscoverRunner.run_tests)NNr_ TFFFFFr NNNFFTF)NNr" ) rD rE rF rU r� � TestSuiter r� r �TextTestRunnerr- �defaultTestLoaderr r r r r � classmethodr� r r& r
r( r) r, r. r r
r1 r4 r8 r= r r r r! r� � s:
�+
B
S
r� c C sB zt | �}W n ttfy Y nw t|d�S tj�tj�| ��S )z�
Check if a test label points to a Python package or file directory.
Relative labels like "." and ".." are seen as directories.
�__path__)r � ImportError� TypeError�hasattrr� r
r r )r �modr r r! r � s �
r Fc C s` t |�}t| �}dd� t|d �D �}t| |||d� |� }t|d �D ] }|�|| � q$|S )aI
Reorder a test suite by test type.
`classes` is a sequence of types
All tests of type classes[0] are placed first, then tests of type
classes[1], etc. Tests with no match in classes are placed last.
If `reverse` is True, sort tests within classes in opposite order but
don't reverse test classes.
c S s g | ]}t � �qS r r )r� �ir r r! r� s z!reorder_suite.<locals>.<listcomp>r_ �r� )r r� �range�partition_suite_by_typer )r� �classesr� � class_count� suite_class�bins�reordered_suiterG r r r! r � s r c C s� t | �}|r tt| ��} | D ]0}t||�r t||||d� qtt|��D ]}t||| �r6|| �|� nq$|d �|� qdS )aV
Partition a test suite by test type. Also prevent duplicated tests.
classes is a sequence of types
bins is a sequence of TestSuites, one more than classes
reverse changes the ordering of tests within bins
Tests of type classes[i] are added to bins[i],
tests with no match found in classes are place in bins[-1]
rH r2 N)r� �reversed�tupler3 rJ rI r �add)r� rK rN r� rM r( rG r r r! rJ s
���rJ c C sZ g }t | �}t�| t �D ] \}}t|tj�r |�||�� q |D ] }|�t|�� q q |S )zCPartition a test suite by test case, preserving the order of tests.) r� � itertools�groupbyr< r� r r{ �extendr� )r� �groupsrM � test_type�
test_group�itemr r r! r� + s �r� c C s� t | �}|� }| D ]I}t||�r|�t|||�� q tt|dt� ��}t|dt|��}t|||�}tt|dt� ��} |�| �}
|
�|�} | sH|sR|
�|�sR|� |� q |S )Nr� �_testMethodName)
r� r3 r r r� r� r� �union� intersectionr ) r� r� r� rM �filtered_suiter( � test_tags� test_fn_name�test_fn� test_fn_tags�all_tags� matched_tagsr r r! r 8 s
�r r� )8r� r� r� rS r r� r� ra r� rn r� � importlibr r �django.core.managementr � django.dbr � django.testr r �django.test.utilsr r r
r' r r r/ r
�django.utils.datastructuresr �django.utils.versionr �ipdbrO rC �tblib.pickling_supportrW �TextTestResultr rH rV r� r� r� r� r� r> r� r� r r rJ r� r r r r r! �<module> s^ � �5 %
U 9