OwlCyberSecurity - MANAGER
Edit File: utils.cpython-39.pyc
a kH�fw� � @ s: d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZ d dl m Z d dl m Z d dl mZ d dlmZ d dlmZmZmZ d dlmZmZ d d lmZ d d lmZ d d l m Z d d l m Z m Z d d l!m"Z" d dl#m$Z$ d dl%m&Z&m'Z' d dl(m)Z)m*Z*m+Z+ d dl,m-Z- d dl.m/Z/ d dl0m1Z1 d dl2m3Z3m4Z4 d dl5m6Z6 d dl7m8Z8 z d dl9Z9W n e:�yx dZ9Y n0 dZ;e<ed�Z=G dd� d�Z>G dd � d e?�Z@d d � ZAG d d � d �ZBdad!d"�ZCd#d$� ZDdd%d%d ddd&�d'd(�ZEd)d*� ZFd+d,� ZGdbd-d.�ZHdcd/d0�ZIddd1d2�ZJG d3d4� d4�ZKG d5d6� d6eK�ZLG d7d8� d8eL�ZMG d9d:� d:eK�ZNd;d<� ZOG d=d>� d>�ZPG d?d@� d@eK�ZQee=dA�ZRedBdC� �ZSedDdE� �ZTedFdG� �ZUdHdI� ZVdJdK� ZWdLdM� ZXedNdO� �ZYdPdQ� ZZG dRdS� dSeK�Z[G dTdU� dU�Z\G dVdW� dWeK�Z]G dXdY� dY�Z^G dZd[� d[�Z_d\d]� Z`edd^�d_d`��ZadS )e� N)�contextmanager)�wraps)�StringIO)�chain)�SimpleNamespace)�TestCase�skipIf� skipUnless)�Node� parseString)�iscoroutinefunction)�apps)�Apps)�UserSettingsHolder�settings)�mail)�ImproperlyConfigured)�request_started�setting_changed)�DEFAULT_DB_ALIAS� connections� reset_queries)�Options)�Template)�template_rendered)�get_script_prefix�set_script_prefix)�RemovedInDjango50Warning)� deactivate)� Approximate� ContextList�isolate_lru_cache� get_runner�CaptureQueriesContext�ignore_warnings� isolate_apps�modify_settings�override_settings�override_system_checks�tag�requires_tz_support�setup_databases�setup_test_environment�teardown_test_environment�tzsetc @ s& e Zd Zd dd�Zdd� Zdd� ZdS ) r � c C s || _ || _d S �N)�val�places)�selfr1 r2 � r4 �S/home/gouroczh/virtualenv/test/3.9/lib/python3.9/site-packages/django/test/utils.py�__init__< s zApproximate.__init__c C s t | j�S r0 )�reprr1 �r3 r4 r4 r5 �__repr__@ s zApproximate.__repr__c C s$ | j |kp"tt| j | �| j�dkS �Nr )r1 �round�absr2 )r3 �otherr4 r4 r5 �__eq__C s zApproximate.__eq__N)r/ )�__name__� __module__� __qualname__r6 r9 r> r4 r4 r4 r5 r ; s r c s: e Zd ZdZ� fdd�Zd dd�Zdd� Zd d � Z� ZS ) r zp A wrapper that provides direct key access to context items contained in a list of context objects. c sB t |t�r2| D ]}||v r|| S qt|��n t� �|�S d S r0 )� isinstance�str�KeyError�super� __getitem__)r3 �key� subcontext�� __class__r4 r5 rF M s zContextList.__getitem__Nc C s( z | � |�W S ty" | Y S 0 d S r0 )rF rD )r3 rG �defaultr4 r4 r5 �getV s zContextList.getc C s& z | | W n t y Y dS 0 dS )NFT)rD )r3 rG r4 r4 r5 � __contains__\ s zContextList.__contains__c C s t t�dd� | D ���S )z0 Flattened keys of subcontexts. c s s | ]}|D ] }|V q qd S r0 r4 )�.0rH �dr4 r4 r5 � <genexpr>g � z#ContextList.keys.<locals>.<genexpr>)�setr � from_iterabler8 r4 r4 r5 �keysc s zContextList.keys)N) r? r@ rA �__doc__rF rL rM rT � __classcell__r4 r4 rI r5 r G s r c C s t j| | |d� | j�|�S )zt An instrumented Template render method, providing a signal that can be intercepted by the test Client. )�sender�template�context)r �sendZnodelist�render)r3 rY r4 r4 r5 �instrumented_test_renderj s r\ c @ s e Zd ZdS )� _TestStateN)r? r@ rA r4 r4 r4 r5 r] s s r] c C s~ t td�rtd��| du r tj} t� }|t_tj|_g tj�d�t_tj|_ | t_tj |_ dt_ t j |_tt _ g t_t� dS )z� Perform global pre-test setup, such as installing the instrumented template renderer and setting the email backend to the locmem email backend. � saved_datazxsetup_test_environment() was already called and can't be called again without first calling teardown_test_environment().NZ testserverz-django.core.mail.backends.locmem.EmailBackend)�hasattrr] � RuntimeErrorr �DEBUGr r^ � ALLOWED_HOSTS� allowed_hosts�debug� EMAIL_BACKEND� email_backendr �_render�template_renderr\ r �outboxr )rd r^ r4 r4 r5 r, w s$ �r, c C s2 t j} | jt_| jt_| jt_| j t _ t `t ` dS )z� Perform any global post-test teardown, such as restoring the original template renderer and restoring the email sending functions. N)r] r^ rc r rb rd ra rf re rh r rg r ri )r^ r4 r4 r5 r- � s r- F)� time_keeper�keepdb� debug_sql�parallel�aliases�serialized_aliasesc K s� |du rt � }t|�\} } g } | �� D �]F\} }d} |D �]2}t| }| �|| | du f� | du �rV|} |�d| ��j z|jd d }W n" ty� |du p�||v }Y n0 tj dt d� |j j | | ||d� W d � n1 s�0 Y |dk�rlt |�D ]P}|�d | ��* |j jt|d �| |d � W d � n1 �sF0 Y �qq8t| j �t| j� q8q&| �� D ] \}}t| j �t| j� �qx|�r�tD ]}d t| _�q�| S ) zCreate the test databases.Nz Creating '%s'�TESTZ SERIALIZEz�The SERIALIZE test database setting is deprecated as it can be inferred from the TestCase/TransactionTestCase.databases that enable the serialized_rollback feature.)�category)� verbosityZ autoclobberrk � serialize� z Cloning '%s'��suffixrr rk T)�NullTimeKeeper� get_unique_databases_and_mirrors�valuesr �append�timed� settings_dictrD �warnings�warnr �creationZcreate_test_db�rangeZ clone_test_dbrC Zset_as_test_mirror�items�force_debug_cursor)rr � interactiverj rk rl rm rn ro �kwargs�test_databases�mirrored_aliasesZ old_names�db_nameZ first_alias�alias� connectionZserialize_alias�indexZ mirror_aliasr4 r4 r5 r+ � s\ ���$ �, � �r+ c c sN | D ]D}t |t�r(td|�d| �d���t |t�r:|V qt|�E dH qdS )z� Return an iterator over a test suite's unittest.TestCase objects. The tests argument can also be an iterable of TestCase objects. zTest z< must be a test case or test suite not string (was found in z).N)rB rC � TypeErrorr �iter_test_cases)Ztests�testr4 r4 r5 r� � s �� r� c C s� g }t � }i }| D ]L\}\}}t � }|D ]} |�|�| g �� q(|�|�sVtd| ��|||< q| r�d} g } | D ]N\} \} }|| �|�r�|�|� |�| | |ff� d} qp| �| | |ff� qp| s�td��| } q`|S )zp Reorder test_databases into an order that honors the dependencies described in TEST[DEPENDENCIES]. zHCircular dependency: databases %r depend on each other, but are aliases.FTz)Circular dependency in TEST[DEPENDENCIES])rR �updaterL � isdisjointr �issubsetrz )r� � dependenciesZordered_test_databasesZresolved_databasesZdependencies_map�sig�_rn Zall_depsr� �changed�deferred� signaturer� r4 r4 r5 �dependency_ordered s6 �� r� c C s� | du r t } i }i }i }t t j�� }t D ]�}t | }|jd }|d rV|d ||< q*|| v r*|�|j�� |jd g f�}|tkr�|d �d|� n|d �|� d|v r�|d ||< q*|tkr*|j�� |kr*|�dtg�||< q*t t |� � |��}||fS )a� Figure out which databases actually need to be created. Deduplicate entries in DATABASES that correspond the same database or are configured as test mirrors. Return two values: - test_databases: ordered mapping of signatures to (name, list of aliases) where all aliases share the same underlying database. - mirrored_aliases: mapping of mirror aliases to original aliases. Nrp ZMIRROR�NAMErt r Z DEPENDENCIES) r r r Ztest_db_signaturer| � setdefault�insertrz rL �dictr� r� ) rn r� r� r� Z default_sigr� r� Z test_settings�itemr4 r4 r5 rx = s< �� �� rx c C sX | D ]N\}}}|r|dkrBt |�D ] }|jjt|d �||d� q"|j�|||� qdS )z%Destroy all the non-mirror databases.rt ru N)r� r Zdestroy_test_dbrC )Z old_configrr rm rk r� Zold_name�destroyr� r4 r4 r5 �teardown_databasesu s �r� c C sX |p| j }|�d�}t|�dkr4d�|d d� �}nd}t|i i |d �}t||d �S )N�.rt ���)Z TEST_RUNNER�split�len�join� __import__�getattr)r Ztest_runner_classZ test_pathZtest_module_nameZ test_moduler4 r4 r5 r" � s r" c @ sR e Zd ZdZddd�Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dS )�TestContextDecoratora� A base class that can either be used as a context manager during tests or as a test function or unittest.TestCase subclass decorator to perform temporary alterations. `attr_name`: attribute assigned the return value of enable() if used as a class decorator. `kwarg_name`: keyword argument passing the return value of enable() if used as a function decorator. Nc C s || _ || _d S r0 )� attr_name� kwarg_name)r3 r� r� r4 r4 r5 r6 � s z TestContextDecorator.__init__c C s t �d S r0 ��NotImplementedErrorr8 r4 r4 r5 �enable� s zTestContextDecorator.enablec C s t �d S r0 r� r8 r4 r4 r5 �disable� s z TestContextDecorator.disablec C s | � � S r0 )r� r8 r4 r4 r5 � __enter__� s z TestContextDecorator.__enter__c C s | � � d S r0 )r� �r3 �exc_type� exc_value� tracebackr4 r4 r5 �__exit__� s z TestContextDecorator.__exit__c s4 t |t�r(|j� � �fdd�}||_|S td��d S )Nc s4 �� � }| ��j� �jr(t| �j|� � | � d S r0 )r� Z addCleanupr� r� �setattr)Z inner_selfrY �Zdecorated_setUpr3 r4 r5 �setUp� s z2TestContextDecorator.decorate_class.<locals>.setUpz1Can only decorate subclasses of unittest.TestCase)� issubclassr r� r� )r3 �clsr� r4 r� r5 �decorate_class� s z#TestContextDecorator.decorate_classc s: t � �r t� �� �fdd��}nt� �� �fdd��}|S )Nc � sL ��4}�j r||�j < � | i |��I d H W d � S 1 s>0 Y d S r0 �r� ��argsr� rY ��funcr3 r4 r5 �inner� s z5TestContextDecorator.decorate_callable.<locals>.innerc sF ��.}�j r||�j < � | i |��W d � S 1 s80 Y d S r0 r� r� r� r4 r5 r� � s )r r )r3 r� r� r4 r� r5 �decorate_callable� s z&TestContextDecorator.decorate_callablec C s: t |t�r| �|�S t|�r&| �|�S tdt|� ��d S )Nz!Cannot decorate object of type %s)rB �typer� �callabler� r� )r3 Z decoratedr4 r4 r5 �__call__� s z TestContextDecorator.__call__)NN) r? r@ rA rU r6 r� r� r� r� r� r� r� r4 r4 r4 r5 r� � s r� c sD e Zd ZdZdZ� fdd�Zdd� Zdd� Zd d � Zd d � Z � Z S ) r' a4 Act as either a decorator or a context manager. If it's a decorator, take a function and return a wrapped function. If it's a contextmanager, use it with the ``with`` statement. In either event, entering/exiting are called before and after, respectively, the function/block is executed. Nc s || _ t� �� d S r0 )�optionsrE r6 �r3 r� rI r4 r5 r6 � s zoverride_settings.__init__c C s� d| j v r<zt�| j d � W n ty: t�� � Y n0 ttj�}| j �� D ]\}}t |||� qPtj| _ |t_| j �� D ]V\}}zt j tjj ||dd� W q~ ty� } z|| _| �� W Y d }~q~d }~0 0 q~d S )N�INSTALLED_APPST�rW Zsetting�valueZenter)r� r Zset_installed_apps� Exception�unset_installed_appsr r �_wrappedr� r� �wrappedr rZ rJ �enable_exceptionr� )r3 �overriderG � new_value�excr4 r4 r5 r� � s, � zoverride_settings.enablec C s� d| j v rt�� | jt_| `g }| j D ]0}tt|d �}tjtjj ||dd�}|� |� q(| j d urt| j }d | _ |�|D ]\}}t |t �rx|�qxd S )Nr� Fr� )r� r r� r� r r� r� r Z send_robustrJ �extendr� rB r� )r3 � responsesrG r� Zresponses_for_settingr� r� �responser4 r4 r5 r� s* � zoverride_settings.disablec C s* |j d u r| j|_ ni |j �| j�|_ d S r0 )Z_overridden_settingsr� �r3 � test_funcr4 r4 r5 � save_options s ��z override_settings.save_optionsc C s, ddl m} t||�s td��| �|� |S )Nr )�SimpleTestCasezPOnly subclasses of Django SimpleTestCase can be decorated with override_settings)Z django.testr� r� � ValueErrorr� )r3 r� r� r4 r4 r5 r� " s � z override_settings.decorate_class) r? r@ rA rU r� r6 r� r� r� r� rV r4 r4 rI r5 r' � s r' c s4 e Zd ZdZ� fdd�Zdd� Z� fdd�Z� ZS )r& z� Like override_settings, but makes it possible to append, prepend, or remove items instead of redefining the entire list. c s@ |r|r J �|d | _ n|r J �t|�� �| _ tt| ��� d S r: )� operations�listr� rE r' r6 )r3 r� r� rI r4 r5 r6 4 s zmodify_settings.__init__c C s* |j d u r| j|_ nt|j �| j |_ d S r0 )Z_modified_settingsr� r� r� r4 r4 r5 r� > s �z modify_settings.save_optionsc s� i | _ | jD ]�\}}z| j | �W n" tyD ttt|g ���Y n0 |�� D ]�\}� t� t�rf� g� |dkr���fdd�� D �7 �qN|dkr��fdd�� D �� �qN|dkr fdd��D ��qNt d| ��qN�| j |< q t � � � d S ) Nrz c s g | ]}|� vr|�qS r4 r4 �rN r� �r� r4 r5 � <listcomp>U rQ z*modify_settings.enable.<locals>.<listcomp>�prependc s g | ]}|� vr|�qS r4 r4 r� r� r4 r5 r� W rQ �removec s g | ]}|� vr|�qS r4 r4 r� )r� r4 r5 r� Y rQ zUnsupported action: %s) r� r� rD r� r� r r� rB rC r� rE r� )r3 �namer� �actionrI )r� r� r5 r� G s$ zmodify_settings.enable)r? r@ rA rU r6 r� r� rV r4 r4 rI r5 r& . s r& c s2 e Zd ZdZd � fdd� Zdd� Zdd� Z� ZS ) r( z� Act as a decorator. Override list of registered system checks. Useful when you override `INSTALLED_APPS`, e.g. if you exclude `auth` app, you also need to exclude its system checks. Nc s, ddl m} || _|| _|| _t� �� d S )Nr )�registry)Zdjango.core.checks.registryr� � new_checks�deployment_checksrE r6 )r3 r� r� r� rI r4 r5 r6 g s z override_system_checks.__init__c C s� | j j| _t� | j _| jD ] }| j j|gt|dd��R � q| j j| _| jd ur�t� | j _| jD ]&}| j j|gt|dd��R ddi� q`d S )N�tagsr4 ZdeployT) r� �registered_checks� old_checksrR r� �registerr� r� �old_deployment_checks)r3 �checkr4 r4 r5 r� o s z override_system_checks.enablec C s | j | j_| j| j_d S r0 )r� r� r� r� r� r8 r4 r4 r5 r� z s z override_system_checks.disable)N�r? r@ rA rU r6 r� r� rV r4 r4 rI r5 r( ` s r( c s� t �d�� � fdd��dd� �dd� ���fdd ��d d � �����fd d ��dd� }| �� �dd�} |�� �dd�}| �d�s�d}|| } || }|t| ��}|t|��}�||�S )al Try to do a 'xml-comparison' of want and got. Plain string comparison doesn't always work because, for example, attribute ordering should not be important. Ignore comment nodes, processing instructions, document type node, and leading and trailing whitespaces. Based on https://github.com/lxml/lxml/blob/master/src/lxml/doctestcompare.py z[ \t\n][ \t\n]+c s � � d| �S )N� )�sub)�v)�_norm_whitespace_rer4 r5 �norm_whitespace� s z$compare_xml.<locals>.norm_whitespacec S s d� dd� | jD ��S )N� c s s | ]}|j tjkr|jV qd S r0 )�nodeTyper Z TEXT_NODE�data�rN �cr4 r4 r5 rP � s z2compare_xml.<locals>.child_text.<locals>.<genexpr>)r� � childNodes��elementr4 r4 r5 � child_text� s �z compare_xml.<locals>.child_textc S s dd� | j D �S )Nc S s g | ]}|j tjkr|�qS r4 )r� r Z ELEMENT_NODEr� r4 r4 r5 r� � rQ z1compare_xml.<locals>.children.<locals>.<listcomp>)r� r� r4 r4 r5 �children� s z compare_xml.<locals>.childrenc s �� | ��S r0 r4 r� )r� r� r4 r5 �norm_child_text� s z$compare_xml.<locals>.norm_child_textc S s t | j�� �S r0 )r� � attributesr� r� r4 r4 r5 � attrs_dict� s z compare_xml.<locals>.attrs_dictc sx | j |j krdS �| ��|�kr$dS � | �� |�kr8dS �| �}�|�}t|�t|�kr\dS t�fdd�t||�D ��S )NFc 3 s | ]\}}� ||�V qd S r0 r4 )rN �want�got)� check_elementr4 r5 rP � s z5compare_xml.<locals>.check_element.<locals>.<genexpr>)ZtagNamer� �all�zip)Z want_elementZ got_elementZ want_childrenZ got_children)r� r� r� r� r4 r5 r� � s �z"compare_xml.<locals>.check_elementc S s. | j D ]"}|jtjtjtjfvr| S qd S r0 )r� r� r Z COMMENT_NODEZDOCUMENT_TYPE_NODEZPROCESSING_INSTRUCTION_NODE)Zdocument�noder4 r4 r5 � first_node� s �z compare_xml.<locals>.first_nodez\n� z<?xmlz<root>%s</root>)�re�compile�strip�replace� startswithr )r� r� r �wrapperZ want_rootZgot_rootr4 )r� r� r� r� r� r� r� r5 � compare_xml s" r c @ sL e Zd ZdZdd� Zdd� Zdd� Zdd � Zed d � �Z d d � Z dd� Z dS )r# zU Context manager that captures queries executed by the specified connection. c C s || _ d S r0 )r� )r3 r� r4 r4 r5 r6 � s z CaptureQueriesContext.__init__c C s t | j�S r0 )�iter�captured_queriesr8 r4 r4 r5 �__iter__� s z CaptureQueriesContext.__iter__c C s | j | S r0 )r )r3 r� r4 r4 r5 rF � s z!CaptureQueriesContext.__getitem__c C s t | j�S r0 )r� r r8 r4 r4 r5 �__len__� s z CaptureQueriesContext.__len__c C s | j j| j| j� S r0 )r� Zqueries�initial_queries� final_queriesr8 r4 r4 r5 r � s z&CaptureQueriesContext.captured_queriesc C s> | j j| _d| j _| j �� t| j j�| _d | _t�t � | S )NT) r� r� Zensure_connectionr� � queries_logr r r Z disconnectr r8 r4 r4 r5 r� � s z CaptureQueriesContext.__enter__c C s2 | j | j_ t�t� |d ur d S t| jj�| _d S r0 )r� r� r �connectr r� r r r� r4 r4 r5 r� � s z CaptureQueriesContext.__exit__N) r? r@ rA rU r6 r rF r �propertyr r� r� r4 r4 r4 r5 r# � s r# c s, e Zd Z� fdd�Zdd� Zdd� Z� ZS )r$ c s: || _ d| j v sd| j v r$tj| _ntj| _t� �� d S )N�message�module)� ignore_kwargsr} �filterwarnings� filter_func� simplefilterrE r6 r� rI r4 r5 r6 � s zignore_warnings.__init__c C s* t �� | _| j�� | jdi | j�� d S )N�ignore)r )r} �catch_warningsr� r r r8 r4 r4 r5 r� � s zignore_warnings.enablec C s | j jt�� � d S r0 )r r� �sys�exc_infor8 r4 r4 r5 r� � s zignore_warnings.disable)r? r@ rA r6 r� r� rV r4 r4 rI r5 r$ � s r$ z|This test relies on the ability to run a program in an arbitrary time zone, but your operating system isn't able to do that.c g s8 t jdd� }t j�| � zdV W |t _n|t _0 dS )z5Context manager to temporarily add paths to sys.path.N)r �pathr� )�pathsZ_orig_sys_pathr4 r4 r5 �extend_sys_path s r! c c s* | � � zdV W | � � n | � � 0 dS )z?Clear the cache of an LRU cache object on entering and exiting.N)� cache_clear)Zlru_cache_objectr4 r4 r5 r! s r! c c sH t t| �}tt| t� � z t t| �V W tt| |� ntt| |� 0 dS )a Return a context manager used by captured_stdout/stdin/stderr that temporarily replaces the sys stream *stream_name* with a StringIO. Note: This function and the following ``captured_std*`` are copied from CPython's ``test.support`` module.N)r� r r� r )Z stream_nameZ orig_stdoutr4 r4 r5 �captured_output s r# c C s t d�S )z�Capture the output of sys.stdout: with captured_stdout() as stdout: print("hello") self.assertEqual(stdout.getvalue(), "hello ") �stdout�r# r4 r4 r4 r5 �captured_stdout. s r& c C s t d�S )z�Capture the output of sys.stderr: with captured_stderr() as stderr: print("hello", file=sys.stderr) self.assertEqual(stderr.getvalue(), "hello ") �stderrr% r4 r4 r4 r5 �captured_stderr8 s r( c C s t d�S )z�Capture the input to sys.stdin: with captured_stdin() as stdin: stdin.write('hello ') stdin.seek(0) # call test code that consumes from sys.stdin captured = input() self.assertEqual(captured, "hello") �stdinr% r4 r4 r4 r5 �captured_stdinB s r* c # s2 t j }� fdd�t _ zdV W |t _ n|t _ 0 dS )aM Context manager to temporarily freeze time.time(). This temporarily modifies the time function of the time module. Modules which import the time function directly (e.g. `from time import time`) won't be affected This isn't meant as a public API, but helps reduce some repetitive code in Django's test suite. c s � S r0 r4 r4 ��tr4 r5 �<lambda>Y rQ z freeze_time.<locals>.<lambda>N)�time)r, Z _real_timer4 r+ r5 � freeze_timeO s r/ c C s6 t tdu d�| �} tddd�ddddid�gd �| �S ) z� Decorator to enable a Jinja2 template engine in addition to the regular Django template engine for a test or skip it if Jinja2 isn't available. Nzthis test requires jinja2z/django.template.backends.django.DjangoTemplatesT)�BACKEND�APP_DIRSz&django.template.backends.jinja2.Jinja2Zkeep_trailing_newline)r0 r1 �OPTIONS)Z TEMPLATES)r �jinja2r' )r� r4 r4 r5 �require_jinja2` s ���� �r4 c s0 e Zd ZdZ� fdd�Zdd� Zdd� Z� ZS )�override_script_prefixzEDecorator or context manager to temporary override the script prefix.c s || _ t� �� d S r0 )�prefixrE r6 )r3 r6 rI r4 r5 r6 x s z override_script_prefix.__init__c C s t � | _t| j� d S r0 )r � old_prefixr r6 r8 r4 r4 r5 r� | s z override_script_prefix.enablec C s t | j� d S r0 )r r7 r8 r4 r4 r5 r� � s z override_script_prefix.disabler� r4 r4 rI r5 r5 u s r5 c @ s e Zd ZdZdd� Zdd� ZdS )�LoggingCaptureMixinzn Capture the output from the 'django' logger and store it on the class's logger_output attribute. c C s8 t �d�| _| jjd j| _t� | _| j| jjd _d S )N�djangor )�logging� getLogger�logger�handlers�stream� old_streamr Z logger_outputr8 r4 r4 r5 r� � s zLoggingCaptureMixin.setUpc C s | j | jjd _d S r: )r? r< r= r> r8 r4 r4 r5 �tearDown� s z LoggingCaptureMixin.tearDownN)r? r@ rA rU r� r@ r4 r4 r4 r5 r8 � s r8 c s0 e Zd ZdZ� fdd�Zdd� Zdd� Z� ZS )r% a Act as either a decorator or a context manager to register models defined in its wrapped context to an isolated registry. The list of installed apps the isolated registry should contain must be passed as arguments. Two optional keyword arguments can be specified: `attr_name`: attribute assigned the isolated registry if used as a class decorator. `kwarg_name`: keyword argument passing the isolated registry if used as a function decorator. c s || _ t� jf i |�� d S r0 )�installed_appsrE r6 )r3 rA r� rI r4 r5 r6 � s zisolate_apps.__init__c C s" t j| _t| j�}tt d|� |S �N� default_apps)r rC �old_appsr rA r� )r3 r r4 r4 r5 r� � s zisolate_apps.enablec C s t td| j� d S rB )r� r rD r8 r4 r4 r5 r� � s zisolate_apps.disabler� r4 r4 rI r5 r% � s r% c @ s( e Zd Zdd� Zedd� �Zdd� ZdS )� TimeKeeperc C s t �t�| _d S r0 )� collections� defaultdictr� �recordsr8 r4 r4 r5 r6 � s zTimeKeeper.__init__c c s\ | j | t�� }z&d V W t�� | }| j | �|� n t�� | }| j | �|� 0 d S r0 )rH r. � perf_counterrz )r3 r� � start_time�end_timer4 r4 r5 r{ � s � zTimeKeeper.timedc C s@ | j �� D ]0\}}|D ]"}d||f }tj�|tj � qq d S )Nz %s took %.3fs)rH r� r r' �write�os�linesep)r3 r� Z end_timesZ record_time�recordr4 r4 r5 � print_results� s zTimeKeeper.print_resultsN)r? r@ rA r6 r r{ rP r4 r4 r4 r5 rE � s rE c @ s e Zd Zedd� �Zdd� ZdS )rw c c s d V d S r0 r4 )r3 r� r4 r4 r5 r{ � s zNullTimeKeeper.timedc C s d S r0 r4 r8 r4 r4 r5 rP � s z NullTimeKeeper.print_resultsN)r? r@ rA r r{ rP r4 r4 r4 r5 rw � s rw c s � fdd�}|S )z0Decorator to add tags to a test class or method.c s. t | d�r| j�� �| _nt| dt� �� | S )Nr� )r_ r� �unionr� rR )�obj�r� r4 r5 � decorator� s ztag.<locals>.decoratorr4 )r� rT r4 rS r5 r) � s r) )� lookup_namec g sT z6|D ]}| � ||� qdV W |D ]}| �||� q$n|D ]}| �||� q<0 dS )z� Context manager to temporarily register lookups on a model field using lookup_name (or the lookup's lookup_name if not provided). N)�register_lookupZ_unregister_lookup)�fieldrU �lookups�lookupr4 r4 r5 rV � s �rV )N)N)r F)N)brF r: rM r r r. r} � contextlibr � functoolsr �ior � itertoolsr �typesr Zunittestr r r Zxml.dom.minidomr r Z asgiref.syncr Z django.appsr Zdjango.apps.registryr Z django.confr r Z django.corer Zdjango.core.exceptionsr Zdjango.core.signalsr r Z django.dbr r r Zdjango.db.models.optionsr Zdjango.templater Zdjango.test.signalsr Z django.urlsr r Zdjango.utils.deprecationr Zdjango.utils.translationr r3 � ImportError�__all__r_ Z TZ_SUPPORTr r� r r\ r] r, r- r+ r� r� rx r� r" r� r' r& r( r r# r$ r* r! r! r# r& r( r* r/ r4 r5 r8 r% rE rw r) rV r4 r4 r4 r5 �<module> s� # $� S* 8 JU2 E(�