|
Server : Apache System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64 User : dh_edsupp ( 6597262) PHP Version : 8.2.26 Disable Function : NONE Directory : /lib/python3/dist-packages/django/test/__pycache__/ |
Upload File : |
o
3�a�w � @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlm Z d dl
m Z d dl m
Z
d dlmZ d dlmZ d dlmZmZmZ d dlmZmZ d d lmZ d d
lm Z d d l m Z m Z d d l m!Z! d d
l"m#Z# d dl$m%Z% d dl&m'Z'm(Z(m)Z) d dl*m+Z+ d dl,m-Z- d dl.m/Z/m0Z0 d dl1m2Z2m3Z3 d dl4m5Z5 zd dl6Z6W n e7y� dZ6Y nw dZ8e9ed�Z:G dd� d�Z;G dd� de<�Z=dd � Z>G d d � d �Z?d]d d �Z@d!d"� ZAdd#d#d dd$�d%d&�ZBd'd(� ZCd]d)d*�ZDd^d+d,�ZEd]d-d.�ZFG d/d0� d0�ZGG d1d2� d2eG�ZHG d3d4� d4eH�ZIG d5d6� d6eG�ZJd7d8� ZKG d9d:� d:�ZLG d;d<� d<eG�ZMee:d=�ZNe d>d?� �ZOe d@dA� �ZPe dBdC� �ZQdDdE� ZRdFdG� ZSdHdI� ZTe dJdK� �ZUdLdM� ZVG dNdO� dOeG�ZWG dPdQ� dQ�ZXG dRdS� dSeG�ZYG dTdU� dU�ZZG dVdW� dW�Z[dXdY� Z\e ddZ�d[d\��Z]dS )_� N)�contextmanager)�wraps)�StringIO)�chain)�SimpleNamespace)�TestCase�skipIf�
skipUnless)�Node� parseString)�apps)�Apps)�UserSettingsHolder�settings)�mail)�ImproperlyConfigured)�request_started)�DEFAULT_DB_ALIAS� connections�
reset_queries)�Options)�Template)�setting_changed�template_rendered)�get_script_prefix�set_script_prefix)�
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 � || _ || _d S �N)�val�places)�selfr0 r1 � r3 �3/usr/lib/python3/dist-packages/django/test/utils.py�__init__1 �
zApproximate.__init__c C �
t | j�S r/ )�reprr0 �r2 r3 r3 r4 �__repr__5 �
zApproximate.__repr__c C s$ | j |kptt| j | �| j�dkS �Nr )r0 �round�absr1 )r2 �otherr3 r3 r4 �__eq__8 s $zApproximate.__eq__N)r- )�__name__�
__module__� __qualname__r5 r: r@ r3 r3 r3 r4 r 0 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 s< t |t�r| D ] }||v r|| S qt|��t� �|�S r/ )�
isinstance�str�KeyError�super� __getitem__)r2 �key�
subcontext�� __class__r3 r4 rH A s
� zContextList.__getitem__Nc C s$ z| � |�W S ty | Y S w r/ )rH rF )r2 rI �defaultr3 r3 r4 �getJ s
�zContextList.getc C s$ z| | W dS t y Y dS w )NFT)rF )r2 rI r3 r3 r4 � __contains__P s
��zContextList.__contains__c C s t t�dd� | D ���S )z0
Flattened keys of subcontexts.
c s s � | ]
}|D ]}|V qqd S r/ r3 )�.0rJ �dr3 r3 r4 � <genexpr>[ � � z#ContextList.keys.<locals>.<genexpr>)�setr �
from_iterabler9 r3 r3 r4 �keysW s zContextList.keysr/ ) rA rB rC �__doc__rH rN rO rV �
__classcell__r3 r3 rK r4 r < 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 �send�nodelist�render)r2 r[ r3 r3 r4 �instrumented_test_render^ s r_ c @ s e Zd ZdS )�
_TestStateN)rA rB rC r3 r3 r3 r4 r` g s r` c C s~ t td�r td��| du rtj} 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().N�
testserverz-django.core.mail.backends.locmem.EmailBackend)�hasattrr` � RuntimeErrorr �DEBUGr ra �
ALLOWED_HOSTS�
allowed_hosts�debug�
EMAIL_BACKEND�
email_backendr �_render�template_renderr_ r �outboxr )rh ra r3 r3 r4 r* k 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` ra rg r rf rh re rj ri rl r rk r rm )ra r3 r3 r4 r+ � s r+ F)� time_keeper�keepdb� debug_sql�parallel�aliasesc K sr |du rt � }t|�\}} g }
|�� D ]�\} }d} |D ]z}
t|
}|
�|| | du f� | du r�|
} |�d|
�� |jj| | ||jd � dd�d� W d � n1 sUw Y |dkr�t
|�D ]&}|�d|
�� |jj t |d �| |d � W d � n1 s�w Y qbqt|
j�
t| j� qq| �� D ]\}
}t|
j�
t| j� q�|r�tD ]}
dt|
_q�|
S )
zCreate the test databases.Nz Creating '%s'�TEST� SERIALIZET)� verbosity� autoclobberro � serialize� z Cloning '%s'��suffixru ro )�NullTimeKeeper� get_unique_databases_and_mirrors�valuesr �append�timed�creation�create_test_db�
settings_dictrN �range�
clone_test_dbrE �set_as_test_mirror�items�force_debug_cursor)ru � interactivern ro rp rq rr �kwargs�test_databases�mirrored_aliases� old_names�db_name� first_alias�alias�
connection�index� mirror_aliasr3 r3 r4 r) � sN � �
� ���
� r) c C s� g }t � }i }| D ]&\}\}}t � }|D ] } |�|�| g �� q|�|�s+td| ��|||< q | rjd}
g } | D ]'\} \}
}|| �|�rV|�|� |�| |
|ff� d}
q8| �| |
|ff� q8|
sftd��| } | s2|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])rT �updaterN �
isdisjointr �issubsetr~ )r� � dependencies�ordered_test_databases�resolved_databases�dependencies_map�sig�_rr �all_depsr� �changed�deferred� signaturer� r3 r3 r4 �dependency_ordered� s8
��
�r� c C s� | du rt } i }i }i }t t j�� }t D ][}t | }|jd }|d r+|d ||< q|| v rp|�|j�� |jd g f�}|tkrJ|d �d|� n|d �|� d|v r\|d ||< q|tkrp|j�� |krp|�dtg�||< qt 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.
Nrs �MIRROR�NAMErx r � DEPENDENCIES) r r r� �test_db_signaturer� �
setdefault�insertr~ rN �dictr� r� ) rr r� r� r� � default_sigr� r� �
test_settings�itemr3 r3 r4 r| � s4
��r| c C sX | D ]'\}}}|r)|dkr!t |�D ]}|jjt|d �||d� q|j�|||� qdS )z%Destroy all the non-mirror databases.rx ry N)r� r� �destroy_test_dbrE )�
old_configru rq ro r� �old_name�destroyr� r3 r3 r4 �teardown_databases. s
���r� c C sX |p| j }|�d�}t|�dkrd�|d d� �}nd}t|i i |d �}t||d �S )N�.rx ���)� TEST_RUNNER�split�len�join�
__import__�getattr)r �test_runner_class� test_path�test_module_name� test_moduler3 r3 r4 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 r. r/ )� attr_name�
kwarg_name)r2 r� r� r3 r3 r4 r5 T r6 z TestContextDecorator.__init__c C � t �r/ ��NotImplementedErrorr9 r3 r3 r4 �enableX � zTestContextDecorator.enablec C r� r/ r� r9 r3 r3 r4 �disable[ r� z TestContextDecorator.disablec C s | � � S r/ )r� r9 r3 r3 r4 � __enter__^ s z TestContextDecorator.__enter__c C s | � � d S r/ )r� �r2 �exc_type� exc_value� tracebackr3 r3 r4 �__exit__a � z TestContextDecorator.__exit__c s0 t |t�r|j� � �fdd�}||_|S td��)Nc s4 �� � }| ��j� �jrt| �j|� � | � d S r/ )r� �
addCleanupr� r� �setattr)�
inner_selfr[ ��decorated_setUpr2 r3 r4 �setUph s
z2TestContextDecorator.decorate_class.<locals>.setUpz1Can only decorate subclasses of unittest.TestCase)�
issubclassr r� � TypeError)r2 �clsr� r3 r� r4 �decorate_classd s
z#TestContextDecorator.decorate_classc s> t �� �rt� �� �fdd��}|S t� �� �fdd��}|S )Nc � sN ���}�j r ||�j < � | i |��I d H W d � S 1 s w Y d S r/ �r� ��argsr� r[ ��funcr2 r3 r4 �innerw s �
$�z5TestContextDecorator.decorate_callable.<locals>.innerc sF ��}�j r ||�j < � | i |��W d � S 1 s w Y d S r/ r� r� r� r3 r4 r� ~ s
$�)�asyncio�iscoroutinefunctionr )r2 r� r� r3 r� r4 �decorate_callables s
�z&TestContextDecorator.decorate_callablec C s6 t |t�r
| �|�S t|�r| �|�S tdt|� ��)Nz!Cannot decorate object of type %s)rD �typer� �callabler� r� )r2 � decoratedr3 r3 r4 �__call__� s
z TestContextDecorator.__call__)NN) rA rB rC rW r5 r� r� r� r� r� r� r� r3 r3 r3 r4 r� H 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 � || _ t� �� d S r/ )�optionsrG r5 �r2 r� rK r3 r4 r5 � � zoverride_settings.__init__c C s� d| j v r z
t�| j d � W n ty t�� � w ttj�}| j �� D ]
\}}t |||� q&tj| _
|t_| j �� D ]*\}}z
t j tjj
||dd� W q= tyg } z
|| _| �� W Y d }~q=d }~ww d S )N�INSTALLED_APPST�rY �setting�value�enter)r� r �set_installed_apps� Exception�unset_installed_appsr r �_wrappedr� r� �wrappedr r\ rL �enable_exceptionr� )r2 �overriderI � new_value�excr3 r3 r4 r� � s0
�
����zoverride_settings.enablec C s� d| j v r t�� | jt_| `g }| j D ]}tt|d �}tjtjj ||dd�}|�
|� q| j d ur:| j }d | _ |�|D ] \}}t |t
�rG|�q<d S )Nr� Fr� )r� r r� r� r r� r� r � send_robustrL �extendr� rD r� )r2 � responsesrI r� �responses_for_settingr� r� �responser3 r3 r4 r� � s*
�
��zoverride_settings.disablec C s, |j d u r | j|_ d S i |j �| j�|_ d S r/ )�_overridden_settingsr� �r2 � test_funcr3 r3 r4 � save_options� s
�
�z override_settings.save_optionsc C s, ddl m} t||�std��| �|� |S )Nr )�SimpleTestCasezPOnly subclasses of Django SimpleTestCase can be decorated with override_settings)� django.testr� r� �
ValueErrorr� )r2 r� r� r3 r3 r4 r� � s
�
z override_settings.decorate_class) rA rB rC rW r� r5 r� r� r� r� rX r3 r3 rK r4 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 |rJ �|d | _ n |rJ �t|�� �| _ tt| ��� d S r< )�
operations�listr� rG r% r5 )r2 r� r� rK r3 r4 r5 � s zmodify_settings.__init__c C s, |j d u r | j|_ d S t|j �| j |_ d S r/ )�_modified_settingsr r r� r3 r3 r4 r� � s
�
�z modify_settings.save_optionsc s� i | _ | jD ]f\}}z| j | �W n ty" ttt|g ���Y nw |�� D ]?\}� t� t�r3� g� |dkrC��fdd�� D � �q'|dkrS�fdd�� D �� �q'|dkra� fdd��D ��q't d| ���| j |< qt
� � � d S ) Nr~ c � g | ]}|� vr|�qS r3 r3 �rP r� �r� r3 r4 �
<listcomp> � z*modify_settings.enable.<locals>.<listcomp>�prependc r r3 r3 r r r3 r4 r r �removec r r3 r3 r )r� r3 r4 r r zUnsupported action: %s) r� r rF r r� r r� rD rE r rG r� )r2 �namer �actionrK )r� r� r4 r� � s& �
zmodify_settings.enable)rA rB rC rW r5 r� r� rX r3 r3 rK r4 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)�django.core.checks.registryr
�
new_checks�deployment_checksrG r5 )r2 r r r
rK r3 r4 r5 s
z override_system_checks.__init__c C s� | j j| _t� | j _| jD ]}| j j|gt|dd��R � q
| j j| _| jd urDt� | j _| jD ]}| j j|gt|dd��R ddi� q0d S d S )N�tagsr3 �deployT) r
�registered_checks�
old_checksrT r �registerr� r �old_deployment_checks)r2 �checkr3 r3 r4 r� s
$�z override_system_checks.enablec C s | j | j_| j| j_d S r/ )r r
r r r r9 r3 r3 r4 r� $ s
z override_system_checks.disabler/ �rA rB rC rW r5 r� r� rX r3 r3 rK r4 r& s
r& c s� t �d�� � fdd��dd� �dd� ���fdd ��d
d � �����fd d
��dd� }| �� �dd�} |�� �dd�}| �d�sJd}|| } || }|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_rer3 r4 �norm_whitespace4 r� z$compare_xml.<locals>.norm_whitespacec S s d� dd� | jD ��S )N� c s s" � | ] }|j tjkr|jV qd S r/ )�nodeTyper
� TEXT_NODE�data�rP �cr3 r3 r4 rR 8 s � ��z2compare_xml.<locals>.child_text.<locals>.<genexpr>)r� �
childNodes��elementr3 r3 r4 �
child_text7 s z compare_xml.<locals>.child_textc S s dd� | j D �S )Nc S s g | ]
}|j tjkr|�qS r3 )r r
� ELEMENT_NODEr" r3 r3 r4 r < s �z1compare_xml.<locals>.children.<locals>.<listcomp>)r$ r% r3 r3 r4 �children; � z compare_xml.<locals>.childrenc s �� | ��S r/ r3 r% )r' r r3 r4 �norm_child_text? r� z$compare_xml.<locals>.norm_child_textc S s t | j�� �S r/ )r� �
attributesr� r% r3 r3 r4 �
attrs_dictB � z compare_xml.<locals>.attrs_dictc sx | j |j krdS �| ��|�krdS � | �� |�kr dS �| �}�|�}t|�t|�kr.dS t�fdd�t||�D ��S )NFc 3 s � | ]
\}}� ||�V qd S r/ r3 )rP �want�got)�
check_elementr3 r4 rR P rS z5compare_xml.<locals>.check_element.<locals>.<genexpr>)�tagNamer� �all�zip)� want_element� got_element�
want_children� got_children)r- r1 r) r+ r3 r4 r1 E s z"compare_xml.<locals>.check_elementc S s. | j D ]}|jtjtjtjfvr| S qd S r/ )r$ r r
� COMMENT_NODE�DOCUMENT_TYPE_NODE�PROCESSING_INSTRUCTION_NODE)�document�noder3 r3 r4 �
first_nodeR s
���z compare_xml.<locals>.first_nodez\n�
z<?xmlz<root>%s</root>)�re�compile�strip�replace�
startswithr )r/ r0 r> �wrapper� want_root�got_rootr3 )r r- r1 r' r) r+ r r4 � compare_xml) s"
rH 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 r/ )r� )r2 r� r3 r3 r4 r5 p r; z CaptureQueriesContext.__init__c C r7 r/ )�iter�captured_queriesr9 r3 r3 r4 �__iter__s r; z CaptureQueriesContext.__iter__c C s
| j | S r/ )rJ )r2 r� r3 r3 r4 rH v r; z!CaptureQueriesContext.__getitem__c C r7 r/ )r� rJ r9 r3 r3 r4 �__len__y r; z CaptureQueriesContext.__len__c C s | j j| j| j� S r/ )r� �queries�initial_queries�
final_queriesr9 r3 r3 r4 rJ | s z&CaptureQueriesContext.captured_queriesc C s> | j j| _d| j _| j �� t| j j�| _d | _t�t � | S )NT)
r� r� �ensure_connectionr� � queries_logrN rO r �
disconnectr r9 r3 r3 r4 r� � s
z CaptureQueriesContext.__enter__c C s2 | j | j_ t�t� |d urd S t| jj�| _d S r/ )r� r� r �connectr r� rQ rO r� r3 r3 r4 r� � s
z CaptureQueriesContext.__exit__N) rA rB rC rW r5 rK rH rL �propertyrJ r� r� r3 r3 r3 r4 r! l s
r! c s, e Zd Z� fdd�Zdd� Zdd� Z� ZS )r" c s: || _ d| j v s
d| j v rtj| _ntj| _t� �� d S )N�message�module)�
ignore_kwargs�warnings�filterwarnings� filter_func� simplefilterrG r5 r� rK r3 r4 r5 � s
zignore_warnings.__init__c C s* t �� | _| j�� | jdi | j�� d S )N�ignore)r\ )rX �catch_warningsr� rZ rW r9 r3 r3 r4 r� � s
zignore_warnings.enablec C s | j jt�� � d S r/ )r] r� �sys�exc_infor9 r3 r3 r4 r� � � zignore_warnings.disable)rA rB rC r5 r� r� rX r3 r3 rK r4 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�| � z dV W |t _dS |t _w )z5Context manager to temporarily add paths to sys.path.N)r^ �pathr� )�paths�_orig_sys_pathr3 r3 r4 �extend_sys_path� s � rd c c s* � | � � z
dV W | � � dS | � � w )z?Clear the cache of an LRU cache object on entering and exiting.N)� cache_clear)�lru_cache_objectr3 r3 r4 r � s
�r c
c sH � t t| �}tt| t� � zt t| �V W tt| |� dS tt| |� w )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 )� stream_name� orig_stdoutr3 r3 r4 �captured_output� s �
ri c C � t d�S )z�Capture the output of sys.stdout:
with captured_stdout() as stdout:
print("hello")
self.assertEqual(stdout.getvalue(), "hello
")
�stdout�ri r3 r3 r3 r4 �captured_stdout� � rm c C rj )z�Capture the output of sys.stderr:
with captured_stderr() as stderr:
print("hello", file=sys.stderr)
self.assertEqual(stderr.getvalue(), "hello
")
�stderrrl r3 r3 r3 r4 �captured_stderr� rn rp c C rj )a 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")
�stdinrl r3 r3 r3 r4 �captured_stdin� s
rr c # s2 � t j }� fdd�t _ z dV W |t _ dS |t _ w )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 r/ r3 r3 ��tr3 r4 �<lambda> s z freeze_time.<locals>.<lambda>N)�time)rt �
_real_timer3 rs r4 � freeze_time� s � rx 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.Jinja2�keep_trailing_newline)ry rz �OPTIONS)� TEMPLATES)r �jinja2r% )r� r3 r3 r4 �require_jinja2 s ����r c �0 e Zd ZdZ� fdd�Zdd� Zdd� Z� ZS )�override_script_prefixzEDecorator or context manager to temporary override the script prefix.c r� r/ )�prefixrG r5 )r2 r� rK r3 r4 r5 r� z override_script_prefix.__init__c C s t � | _t| j� d S r/ )r �
old_prefixr r� r9 r3 r3 r4 r� s z override_script_prefix.enablec C s t | j� d S r/ )r r� r9 r3 r3 r4 r� ! r. z override_script_prefix.disabler r3 r3 rK r4 r� s
r� 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 �
logger_outputr9 r3 r3 r4 r� * s zLoggingCaptureMixin.setUpc C s | j | jjd _d S r< )r� r� r� r� r9 r3 r3 r4 �tearDown0 r` z LoggingCaptureMixin.tearDownN)rA rB rC rW r� r� r3 r3 r3 r4 r� % s r� c r� )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� jdi |�� d S )Nr3 )�installed_appsrG r5 )r2 r� r� rK r3 r4 r5 D s zisolate_apps.__init__c C s" t j| _t| j�}tt d|� |S �N� default_apps)r r� �old_appsr
r� r� )r2 r r3 r3 r4 r� H s
zisolate_apps.enablec C s t td| j� d S r� )r� r r� r9 r3 r3 r4 r� N s zisolate_apps.disabler r3 r3 rK r4 r# 4 s
r# c @ s( e Zd Zdd� Zedd� �Zdd� ZdS )�
TimeKeeperc C s t �t�| _d S r/ )� collections� defaultdictr �recordsr9 r3 r3 r4 r5 S r* zTimeKeeper.__init__c c s\ � | j | t�� }zd V W t�� | }| j | �|� d S t�� | }| j | �|� w r/ )r� rv � perf_counterr~ )r2 r �
start_time�end_timer3 r3 r4 r V s �
�zTimeKeeper.timedc C s@ | j �� D ]\}}|D ]}d||f }tj�|tj � q qd S )Nz
%s took %.3fs)r� r� r^ ro �write�os�linesep)r2 r � end_times� record_time�recordr3 r3 r4 �
print_results` s ��zTimeKeeper.print_resultsN)rA rB rC r5 r r r� r3 r3 r3 r4 r� R s
r� c @ s e Zd Zedd� �Zdd� ZdS )r{ c c s � d V d S r/ r3 )r2 r r3 r3 r4 r h s �
zNullTimeKeeper.timedc C s d S r/ r3 r9 r3 r3 r4 r� l r� z NullTimeKeeper.print_resultsN)rA rB rC r r r� r3 r3 r3 r4 r{ g s
r{ c s � fdd�}|S )z0Decorator to add tags to a test class or method.c s0 t | d�r| j�� �| _| S t| dt� �� | S )Nr )rc r �unionr� rT )�obj�r r3 r4 � decoratorr s
�ztag.<locals>.decoratorr3 )r r� r3 r� r4 r' p s r' )� lookup_namec g sT � z |D ]}| � ||� qdV W |D ]}| �||� qdS |D ]}| �||� q w )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_lookup�_unregister_lookup)�fieldr� �lookups�lookupr3 r3 r4 r� { s � ��r� r/ )r F)^r� r� r� r� r@ r^ rv rX �
contextlibr � functoolsr �ior � itertoolsr �typesr �unittestr r r �xml.dom.minidomr
r � django.appsr �django.apps.registryr
� django.confr r � django.corer �django.core.exceptionsr �django.core.signalsr � django.dbr r r �django.db.models.optionsr �django.templater �django.test.signalsr r � django.urlsr r �django.utils.translationr r~ � ImportError�__all__rc �
TZ_SUPPORTr r r r_ r` r* r+ r) r� r| r� r r� r% r$ r& rH r! r" r( rd r ri rm rp rr rx r r� r� r# r� r{ r' r� r3 r3 r3 r4 �<module> s� �
"
$ �2
*
3
FN0 C'�