OwlCyberSecurity - MANAGER
Edit File: runner.cpython-39.pyc
a ���e�� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl Z d dl Z d dl Z d dl Z d dlmZ d dlmZ d dlmZ d dlmZ d dlZd dlZd dlmZ d dlmZ d dlm Z m Z d d l m Z m Z m!Z! d d l m"Z# d d l m$Z$ d d l m%Z& d d l m'Z' d dl(m)Z) d dl*m+Z+ d dl,m-Z- z d dl.Z/W n e0�yh d dl/Z/Y n0 z d dl1Z2W n e0�y� dZ2Y n0 G dd� de j3�Z4G dd� de j3�Z5G dd� d�Z6G dd� de j7�Z8G dd� d�Z9dd � Z:d d � Z;d a<d>d d �Z=d!d"� Z>d#d$� Z?G d%d&� d&e j@�ZAG d'd(� d(�ZBG d)d*� d*�ZCd+d,� ZDd-d.� ZEd/d0� ZFd1d2� ZGd?d4d5�ZHd@d6d7�ZId8d9� ZJd:d;� ZKd<d=� ZLdS )A� N)� defaultdict)�contextmanager)� import_module)�StringIO)� call_command)� connections)�SimpleTestCase�TestCase)�NullTimeKeeper� TimeKeeper�iter_test_cases)�setup_databases)�setup_test_environment)�teardown_databases)�teardown_test_environment)�new_hash)� OrderedSet)�RemovedInDjango50Warningc 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�streamZ descriptions� verbosity�� __class__� �S/home/gouroczh/virtualenv/pat/3.9/lib/python3.9/site-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( 5 s z DebugSQLTextTestResult.startTestc sP t � �|� | j�| j� | jrL| j�d� | j� | j� � � | j� | j � d S )Nr ) r �stopTestr � removeHandlerr& ZshowAllr �seekr �write�read�writeln� separator2r) r r" r# r+ ; s z DebugSQLTextTestResult.stopTestc sN t � �||� | jd u r d}n| j�d� | j�� }| jd |f | jd< d S )N� r ���)r �addErrorr r- r/ �errors)r r* �err�sqlr r" r# r4 C s z DebugSQLTextTestResult.addErrorc s: t � �||� | j�d� | jd | j�� f | jd< d S �Nr r3 )r � addFailurer r- �failuresr/ �r r* r6 r r" r# r9 M s z!DebugSQLTextTestResult.addFailurec s\ t � �|||� |d urX| j�d� t|d |j�r:| jn| j}|d | j�� f |d< d S r8 ) r � addSubTestr r- � issubclassZfailureExceptionr: r5 r/ )r r* �subtestr6 r5 r r" r# r< R s ��z!DebugSQLTextTestResult.addSubTestc C s| |D ]r\}}}| j �| j� | j �d|| �|�f � | j �| j� | j �|� | j �| j� | j �tj|ddd�� qd S )Nz%s: %sT�upper)ZreindentZ keyword_case)r r0 Z separator1ZgetDescriptionr1 �sqlparse�format)r �flavourr5 r* r6 Z sql_debugr" r" r# �printErrorList] s �z%DebugSQLTextTestResult.printErrorList) �__name__� __module__� __qualname__r r( r+ r4 r9 r< rC � __classcell__r" r" r r# r . s r c s@ e Zd ZdZ� fdd�Z� fdd�Z� fdd�Zdd � Z� ZS ) �PDBDebugResultz^ Custom result class that triggers a PDB session when an error or failure occurs. c s t � �||� | �|� d S r$ )r r4 �debugr; r r" r# r4 o s zPDBDebugResult.addErrorc s t � �||� | �|� d S r$ )r r9 rI r; r r" r# r9 s s zPDBDebugResult.addFailurec s& |d ur| � |� t� �|||� d S r$ )rI r r< �r r* r> r6 r r" r# r< w s zPDBDebugResult.addSubTestc C s2 | � � d| _|\}}}td| � t�|� d S )NFz Opening PDB: %r)Z_restoreStdout�buffer�print�pdbZ post_mortem)r �error�exc_type� exc_value� tracebackr" r" r# rI | s zPDBDebugResult.debug) rD rE rF �__doc__r4 r9 r< rI rG r" r" r r# rH i s rH c @ s e Zd ZdZdZdd� ZdS )� DummyListzP Dummy list class for faking storage of results in unittest.TestResult. r" c C s d S r$ r" )r �itemr" r" r# �append� s zDummyList.appendN)rD rE rF rR � __slots__rU r" r" r" r# rS � s rS c s� e Zd ZdZ� fdd�Zdd� Zedd� �Zdd � Zd d � Z d d � Z dd� Z � fdd�Z � fdd�Z � fdd�Z� fdd�Z� fdd�Z� fdd�Z� fd d �Z� fd d �Z� fd d!�Z� fd"d#�Z� fd$d%�Zd&d'� Zd(d)� Z� ZS )*�RemoteTestResultz� Extend unittest.TestResult to record events in the child processes so they can be replayed in the parent process. Events include things like which tests succeeded or failed. c sR t � j|i |�� t� }|| _|| _|| _|| _|| _td urHtj � � g | _ d S r$ ) r r rS r: r5 ZskippedZexpectedFailures�unexpectedSuccesses�tblibZpickling_support�install�events)r �args�kwargsZ dummy_listr r" r# r � s zRemoteTestResult.__init__c C s> | j �� }|�dd � |�dd � |�dd � |�dd � |S )NZ_stdout_bufferZ_stderr_bufferZ_original_stdoutZ_original_stderr)�__dict__�copy�pop)r �stater" r" r# � __getstate__� s z RemoteTestResult.__getstate__c C s | j d S )N� )ZtestsRun�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. )rL rA )r r* r> Z pickle_excr" r" r# �_print_unpicklable_subtest� s ��z+RemoteTestResult._print_unpicklable_subtestc C s� z| � |� W n� ty� } zrt|d �}tj|dddd�}t|�}tj|dddd�}td u rptd�||�� ntd�|||�� � W Y d }~n d }~0 0 d S )Nrc �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. )rj � Exception�repr�textwrap�fillrY rL rA )r r* r6 �excZoriginal_exc_txtZpickle_exc_txtr" r" r# �check_picklable� s. �� ����z RemoteTestResult.check_picklablec C sH z| � |� W n4 tyB } z | �|||� � W Y d }~n d }~0 0 d S r$ )rj ro rk )r r* r> rs r" r" r# �check_subtest_picklable s z(RemoteTestResult.check_subtest_picklablec s t � �� | j�d� d S )N)� startTestRun)r rv r[ rU rd r r" r# rv s z RemoteTestResult.startTestRunc s t � �� | j�d� d S )N)� stopTestRun)r rw r[ rU rd r r" r# rw s z RemoteTestResult.stopTestRunc s" t � �|� | j�d| jf� d S )Nr( )r r( r[ rU re r) r r" r# r( s zRemoteTestResult.startTestc s" t � �|� | j�d| jf� d S )Nr+ )r r+ r[ rU re r) r r" r# r+ s zRemoteTestResult.stopTestc s2 | � ||� | j�d| j|f� t� �||� d S )Nr4 )rt r[ rU re r r4 r; r r" r# r4 " s zRemoteTestResult.addErrorc s2 | � ||� | j�d| j|f� t� �||� d S )Nr9 )rt r[ rU re r r9 r; r r" r# r9 ' s zRemoteTestResult.addFailurec sJ |d ur6| � ||� | �||� | j�d| j||f� t� �|||� d S )Nr< )rt ru r[ rU re r r< rJ r r" r# r< , s zRemoteTestResult.addSubTestc s" | j �d| jf� t� �|� d S )N� addSuccess)r[ rU re r rx r) r r" r# rx 7 s zRemoteTestResult.addSuccessc s& | j �d| j|f� t� �||� d S )N�addSkip)r[ rU re r ry )r r* �reasonr r" r# ry ; s zRemoteTestResult.addSkipc sL t d u r|d |d d f}| �||� | j�d| j|f� t� �||� d S )Nr rc �addExpectedFailure)rY rt r[ rU re r r{ r; r r" r# r{ ? s z#RemoteTestResult.addExpectedFailurec s" | j �d| jf� t� �|� d S )N�addUnexpectedSuccess)r[ rU re r r| r) r r" r# r| J s z%RemoteTestResult.addUnexpectedSuccessc s h d�� t � fdd�| jD ��S )z/Tells whether or not this result was a success.> r< r4 r| r9 c 3 s | ]}|d � vV qdS )r Nr" )�.0�e�Z failure_typesr" r# � <genexpr>Q � z1RemoteTestResult.wasSuccessful.<locals>.<genexpr>)�allr[ rd r" r r# � wasSuccessfulN s z RemoteTestResult.wasSuccessfulc C s dS )Nr2 r" )r r6 r* r" r" r# �_exc_info_to_stringS s z$RemoteTestResult._exc_info_to_string)rD rE rF rR r rb �propertyre rj rk rt ru rv rw r( r+ r4 r9 r< rx ry r{ r| r� r� rG r" r" r r# rW � s, ; rW 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 r$ )�failfastrK � resultclass)r r� r� rK r" r" r# r c s zRemoteTestRunner.__init__c C s. | � � }t�|� | j|_| j|_||� |S r$ )r� �unittestZregisterResultr� rK )r r* �resultr" r" r# �runi s zRemoteTestRunner.run)FNF)rD rE rF rR rW r� r r� r" r" r" r# r� Z s r� c C s@ t �� dvrdS zttjd �W S ty: t �� Y S 0 dS )zP The maximum number of test processes when using the --parallel option. > �fork�spawnrc ZDJANGO_TEST_PROCESSESN)�multiprocessing�get_start_method�int�os�environ�KeyError� cpu_countr" r" r" r# �get_max_test_processesr s r� c C s> | dkr | S z t | �W S ty8 t�| �d���Y n0 dS )z,Parse value passed to the --parallel option.�autoz' is not an integer or the string 'auto'N)r� � ValueError�argparse�ArgumentTypeError)�valuer" r" r# � parallel_type� s �r� c C s� | � � �$ | jd7 _| jaW d � n1 s20 Y t�� }|dkr~|rlt|�rl|du rdd}||� t�� t|d� t D ]D}t | }|dkr�|j � || � |� |� } r�| |_ |j�t� q�dS )z� Switch to databases dedicated to this worker. This helper lives at module-level because of the multiprocessing module's requirements. rc Nr� r" �rI )Zget_lockr� � _worker_idr� r� �callable�django�setupr r � settings_dict�update�get�_test_serialized_contentsZcreationZsetup_worker_connection) �counter�initial_settings�serialized_contents� process_setup�process_setup_args� debug_modeZ start_method�alias� connectionr� r" r" r# � _init_worker� 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� rK )r� r[ )r\ � runner_class�subsuite_index�subsuiter� rK �runnerr� r" r" r# � _run_subsuite� s r� c G s dS )z-Stub method to simplify run() implementation.Nr" )r\ r" r" r# �_process_setup_stub� s r� c sN e Zd ZdZeZeZdZe Z e Z d � fdd� Z dd� Zdd � Zd d � Z� ZS ) �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. r" Fc s8 || _ || _|| _|| _|| _d | _d | _t� �� d S r$ ) � subsuites� processesr� r� rK r� r� r r )r r� r� r� r� rK r r" r# r � s zParallelTestSuite.__init__c s6 � � � t�tjd�}tj� j� jj|� j � j � j j� j � j gd�}� fdd�t� j�D �}|�� jj|�}|jr~|�� �q*z|jdd�\}}W n6 tjy� Y qlY n ty� |�� Y �q*Y n0 t� j| �}|D ]J} | d } t|| d�} | du r�q�|| d } | d d� }| | g|�R � q�ql|�� |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� Z initializerZinitargsc s$ g | ] \}}� j ||� j� jf�qS r" )r� r� rK )r} �indexr� rd r" r# � <listcomp> s �z)ParallelTestSuite.run.<locals>.<listcomp>g�������?)�timeoutNrc � )�initialize_suiter� �Value�ctypes�c_intZPoolr� � init_worker�__func__r� r� r� r� r� � enumerater� Zimap_unordered� run_subsuiteZ shouldStop� terminate�next� TimeoutError� StopIteration�close�list�getattr�join) r r� r� �poolr\ Z test_resultsr� r[ �tests�eventZ event_namer& r* r" rd r# r� � sL �� � zParallelTestSuite.runc C s t | j�S r$ )�iterr� rd r" r" r# �__iter__+ s zParallelTestSuite.__iter__c s4 t �� dkr0dd� tD �� _� fdd�tD �� _d S )Nr� c S s i | ]}|t | j�qS r" )r r� �r} r� r" r" r# � <dictcomp>0 s z6ParallelTestSuite.initialize_suite.<locals>.<dictcomp>c s" i | ]}|� j v r|t| j�qS r" )�serialized_aliasesr r� r� rd r" r# r� 3 s �)r� r� r r� r� rd r" rd r# r� . s � �z"ParallelTestSuite.initialize_suite)FFF)rD rE rF rR r� r� r� r� r� r� r� r� r� r r� r� r� rG r" r" r r# r� � s � >r� c @ sF e Zd ZdZdZedd� �Zddd�Zedd � �Z d d � Z d d � Z dS )�Shufflera� This class implements shuffling with a special consistency property. Consistency means that, for a given seed and key function, if two sets of items are shuffled, the resulting order will agree on the intersection of the two sets. For example, if items are removed from an original set, the shuffled order for the new set will be the shuffled order of the original set restricted to the smaller set. �md5c C s& t | jdd�}|�|�d�� |�� S )NF)�usedforsecurityzutf-8)r �hash_algorithmr� �encode� hexdigest)�cls�text�hr" r" r# � _hash_textG s zShuffler._hash_textNc C s. |d u rt �dd�}d}nd}|| _|| _d S )Nr l �c( � generated�given)�random�randint�seed� seed_source)r r� r� r" r" r# r M s zShuffler.__init__c C s | j �d| j� d�S )Nz (�))r� r� rd r" r" r# � seed_displayW s zShuffler.seed_displayc C s d� | j||��}| �|�S )Nz{}{})rA r� r� )r rT �keyr� r" r" r# � _hash_item[ s zShuffler._hash_itemc sZ i � |D ]:}| � ||�}|� v r:d�||� | �}t|��|� |< q� fdd�t� �D �S )ab Return a new list of the items in a shuffled order. The `key` is a function that accepts an item in `items` and returns a string unique for that item that can be viewed as a string id. The order of the return value is deterministic. It depends on the seed and key function but not on the original order. z)item {!r} has same hash {!r} as item {!r}c s g | ] }� | �qS r" r" )r} �hashed��hashesr" r# r� s r� z$Shuffler.shuffle.<locals>.<listcomp>)r� rA � RuntimeError�sorted)r �itemsr� rT r� �msgr" r� r# �shuffle_ s � zShuffler.shuffle)N) rD rE rF rR r� � classmethodr� r r� r� r� r� r" r" r" r# r� : s r� c @ s� e Zd ZdZejZeZej Z ej Z e efZd/dd�Zed d � �Zed d � �Zd0d d�Zdd� Zdd� Zedd� �Zdd� Zd1dd�Zdd� Z dd � Z d d � Z d d � Z d!d"� Z d#d$� Z!d%d&� Z"d'd(� Z#d)d*� Z$d+d,� Z%d2d-d.�Z&dS )3�DiscoverRunnerz8A Django test runner that uses unittest2 test discovery.Nrc TFr c K s || _ || _|| _|| _|| _|| _|| _|| _| | _| | _ t | pDg �| _ t | pRg �| _ t �� s�|r�zt jtj�� d� W n* ttjfy� t jtj�� d� Y n0 || _| jr�| j dkr�td��|| _d | _|r�t� nt � | _ | r�dd� | D �| _|| _ d | _ || _ d S )N)�filerc zFYou cannot use --pdb 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�reverser� � debug_sql�parallel�set�tags� exclude_tags� faulthandler� is_enabled�enable�sys�stderr�fileno�AttributeError�io�UnsupportedOperation� __stderr__rM r� rK �test_name_patternsr r � time_keeperr� � _shufflerr )r r� r� r r� r� r� r� r� r� r� r� r� r rM rK �enable_faulthandlerZtimingr� r r] r" r" r# r s@ ��zDiscoverRunner.__init__c C s |j ddddd� |j dddd d � |j d d d d� |j dddtddd� |j ddd dd� |j dd dd� |j ddd d d� |j d dd d td d!d"� |j d#d$d%d&d'� |j d(d$d)d*d'� |j d+d d,d� |j d-d.d d/d� |j d0d1d2d3d'� |j d4d d5d� |j d6d$d7d8d'� d S )9Nz-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 --shuffle�?FZSEEDzShuffles test case order.)�nargsr �type�metavarr z-rz --reversezReverses test case order.z --debug-modez Sets settings.DEBUG to True.z-dz --debug-sqlz%Prints logged SQL queries on failure.z --parallelr� r �NzqRun tests using up to N parallel processes. Use the value "auto" to run one test process for each processor core.)r �constr r r r z--tagrU 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� �parserr" r" r# � add_arguments� s� �������� �������z DiscoverRunner.add_argumentsc C s | j d u rd S | j jS r$ )r r� rd r" r" r# � shuffle_seed s zDiscoverRunner.shuffle_seedc C sV |du rt j}| jdu rD| jdks6| jdkr:|t jk r:dS t|� n| j�||� dS )a\ Log the message at the given logging level (the default is INFO). If a logger isn't set, the message is instead printed to the console, respecting the configured verbosity. A verbosity of 0 prints no output, a verbosity of 1 prints INFO and above, and a verbosity of 2 or higher prints all levels. Nr rc )r �INFOr r rL �log)r r� �levelr" r" r# r # s zDiscoverRunner.logc K s t | jd� t�� d S )Nr� )r r� r� ZinstallHandler�r r] r" r" r# r 5 s z%DiscoverRunner.setup_test_environmentc C s6 | j du rd S t| j d�}| �d|j� �� || _d S )NF)r� zUsing shuffle seed: )r� r� r r� r )r �shufflerr" r" r# �setup_shuffler9 s z DiscoverRunner.setup_shufflerc c s4 | j j}| j| j _zd V W || j _n || j _0 d S r$ )� test_loaderZtestNamePatternsr )r Zoriginal_test_name_patternsr" r" r# �load_with_patterns@ s z!DiscoverRunner.load_with_patternsc C s t j�|�}d }t j�|�s\| �� � | j�|�}W d � n1 sF0 Y |�� r\|S t|�\}}|rv|s�|S n8t j� |�s�t j�|�r�|d u s�J �t d|�d���|S |� � }t j� |�r�| j s�t |�|d< | �� �& | jjf d|i|��}W d � n1 �s 0 Y d | j_|S )Nz,One of the test labels is a path to a file: zR, which is not supported. Use a dotted module name or path to a directory instead.� top_level_dir� start_dir)r� �path�abspath�existsr# r" ZloadTestsFromNameZcountTestCases� try_importing�isdirr� r_ r� �find_top_levelZdiscoverZ_top_level_dir)r �label�discover_kwargsZ label_as_pathr� Z is_importable� is_packager] r" r" r# �load_tests_for_labelJ s2 * � 6z#DiscoverRunner.load_tests_for_labelc K s� |d urt jdtdd� |p dg}|p(g }i }| jd urB| j|d< | jd urV| j|d< | �� g }|D ] }| �||�}|�t|�� qf|�t|�� | j s�| j r�| j r�| j dd� t | j �� tjd � | j r�| j d d� t | j �� tjd � t|| j | j �}tjjg| j�R }tt||| j| jd ��}| � d t|� � | �|�} | jd k�r�t | �} t | jt| ��} | | _| d k�r�| � | | | j | j | j!�} | S )N�'The extra_tests argument is deprecated.r� �� stacklevel�.r� r$ zIncluding test tag(s): %s.�, �r zExcluding test tag(s): %s.�r r� zFound %d test(s).rc )"�warnings�warnr r� r� r! r/ �extendr r� r� r r� r� r r �filter_tests_by_tagsr� �loader� _FailedTest� reorder_byr� � reorder_testsr r� �len� test_suiter� �partition_suite_by_case�min�parallel_test_suiter� r� rK ) r � test_labels� extra_testsr] r- � all_testsr, r� Z test_types�suiter� r� r" r" r# � build_suiteo sn � ���� �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 r tS | jrtS d S r$ )r� r rM rH rd r" r" r# �get_resultclass� s z DiscoverRunner.get_resultclassc C s | j | �� | j| jd�S )N)r� r� r rK )r� rJ r rK rd 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 rM r" r" r# � run_checks� s zDiscoverRunner.run_checksc K sr | � � }| jf i |��}z0|�|�W | jd urF| jj}| �d|� �� S n$| jd url| jj}| �d|� �� 0 d S )NzUsed shuffle seed: )rK � test_runnerr� r r� r )r rG 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� t |j� S r$ )r? r: r5 rX )r rG r� r] r" r" r# � suite_result� s �zDiscoverRunner.suite_resultc sX i � t |�D ]F}t|dd �}|dkr(t}|r t|dd��� �� �fdd�|D �� q � S )NrM �__all__�serialized_rollbackFc 3 s" | ]}|�p� � |d �fV qdS )FN)r� r� �rM rU r" r# r� � s �z0DiscoverRunner._get_databases.<locals>.<genexpr>)r r� r r� )r rG r* Ztest_databasesr" rV r# �_get_databases� s � z DiscoverRunner._get_databasesc sB | � |�� � fdd�tD �}|r>| jdd�t|�� tjd� � S )Nc s g | ]}|� vr|�qS r" r" r� �rM r" r# r� � r� z0DiscoverRunner.get_databases.<locals>.<listcomp>z)Skipping setup of unused database(s): %s.r4 r5 )rW r r r� r� r r )r rG Zunused_databasesr" rX r# � get_databases� s ��z DiscoverRunner.get_databasesc K s� |durt jdtdd� | �� | �||�}| �|�}tdd� |�� D ��|_| j � d�� | j ||jd�}W d � n1 s�0 Y d }z�z| � |� | � |�}W n ty� d }� Y n0 W zB| j � d �� | �|� W d � n1 s�0 Y | �� W n t�y" |�s � Y n0 ndzD| j � d �� | �|� W d � n1 �sV0 Y | �� W n t�y� |�s�� Y n0 0 | j �� | �||�S ) z� 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. Return the number of tests that failed. Nr0 r� r1 c s s | ]\}}|r|V qd S r$ r" )r} r� � serializer" r" r# r� s z+DiscoverRunner.run_tests.<locals>.<genexpr>zTotal database setup)�aliasesr� FTzTotal database teardown)r7 r8 r r rH rY r� r� r� r Ztimedr rN rP ro r r � print_resultsrS ) r rD rE r] rG rM rR Z run_failedr� r" r" r# � run_tests sP � � �$ ( �* zDiscoverRunner.run_tests)NNrc TFFFFFr NNNFFTFFN)N)NN)N)'rD rE rF rR r� � TestSuiter@ r� rC ZTextTestRunnerrO ZdefaultTestLoaderr" r r r= r r� r r� r r r r! r r# r/ rH r rJ rK rN rP r r rS rW rY r] r" r" r" r# r� v s` � ; b % F r� c C s4 z t | �}W n ttfy$ Y dS 0 dt|d�fS )z� Try importing a test label, and return (is_importable, is_package). Relative labels like "." and ".." are seen as directories. )FFT�__path__)r � ImportError� TypeError�hasattr)r, �modr" r" r# r) 7 s r) c C s< t j�| d�}t j�|�s q8t j�| �}|| kr2q8|} q | S )Nz __init__.py)r� r& r� r( �dirname)r� Zinit_pyZtry_nextr" r" r# r+ E s r+ c C s | j � d| j� �S )Nr3 )rE rF )r� r" r" r# �_class_shuffle_key] s re c sn i � t �| t�D ]6\}}t|�}t|d �}|j|dd� d�}|� |< q|j� td�}t j� fdd�|D �� S )z� Return an iterator over the given tests in a shuffled order, keeping tests next to other tests of their class. `tests` should be an iterable of tests. r c S s | � � S r$ )�id)r* r" r" r# �<lambda>l r� z shuffle_tests.<locals>.<lambda>)r� c 3 s | ]}� | V qd S r$ r" )r} r� �Z tests_by_typer" r# r� q r� z shuffle_tests.<locals>.<genexpr>)� itertools�groupbyr r� r� re �chain)r� r �_Z class_testsZ test_type�classesr" rh r# � shuffle_testsa s rn Fc C s: |du r |rt | �S t| �S t| |�} |s.| S t t| ��S )z� Return an iterator that reorders the given tests, keeping tests next to other tests of their class. `tests` should be an iterable of tests that supports reversed(). N)�reversedr� rn r� )r� r r� r" r" r# �reorder_test_bint s rp c c s� dd� t t|�d �D �}|�^ }}| D ]<}t||�D ]\}} t|| �r6 qRq6|}|t|� �|� q(|D ],}ttj� |� � ��} t | ||d�E dH qjdS )a~ Reorder an iterable of tests, grouping by the given TestCase classes. This function also removes any duplicates and reorders so that tests of the same type are consecutive. The result is returned as an iterator. `classes` is a sequence of types. Tests that are instances of `classes[0]` are grouped first, followed by instances of `classes[1]`, etc. Tests that are not instances of any of the classes are grouped last. If `reverse` is True, the tests within each `classes` group are reversed, but without reversing the order of `classes` itself. The `shuffler` argument is an optional instance of this module's `Shuffler` class. If provided, tests will be shuffled within each `classes` group, but keeping tests with other tests of their TestCase class. Reversing is applied after shuffling to allow reversing the same random order. c S s g | ] }t t��qS r" )r r )r} �ir" r" r# r� � r� z!reorder_tests.<locals>.<listcomp>rc r6 N) �ranger? �zip� isinstancer �addr� ri rk � from_iterable�valuesrp ) r� rm r� r ZbinsZ class_binsZlast_binr* Ztest_binZ test_classr" r" r# r> � s r> c s* t | �� t| �}� fdd�t�|t �D �S )zCPartition a test suite by test case, preserving the order of tests.c s g | ]\}}� |��qS r" r" )r} rl r� �Z suite_classr" r# r� � r� z+partition_suite_by_case.<locals>.<listcomp>)r r ri rj )rG rF r" rx r# rA � s rA c C s| t | tjj�rdS tt| dg ��}t| dt| ��}t| |�r`t| |�}tt|dg ��}|� |�}|rr|� |�rrdS |� |�S )NTr� Z_testMethodNameF) rt r� r; r< r� r� �strrb r� �union� isdisjoint)r* r� r� Z test_tagsZ test_fn_nameZtest_fnZ test_fn_tagsr" r" r# �test_match_tags� s r| c s � �fdd�| D �S )z)Return the matching tests as an iterator.c 3 s | ]}t |�� �r|V qd S r$ )r| )r} r* �r� r� r" r# r� � r� z'filter_tests_by_tags.<locals>.<genexpr>r" )r� r� r� r" r} r# r: � s r: )NNNNN)NF)FN)Mr� r� r� r ri r r� r� rf r� r rq r� r7 � collectionsr � contextlibr � importlibr r r@ r� Zdjango.core.managementr Z django.dbr Z django.testr r Zdjango.test.utilsr r r r rI r r rQ r Zdjango.utils.cryptor Zdjango.utils.datastructuresr Zdjango.utils.deprecationr ZipdbrM r` Ztblib.pickling_supportrY ZTextTestResultr rH rS Z TestResultrW r� r� r� r� r� r� r� r^ r� r� r� r) r+ re rn rp r> rA r| r: r" r" r" r# �<module> s� ; L � ) p< D '