o
    uvXjIt  ã                   @   sü  d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dlm	Z	 d dl
mZ d dlmZ d dlmZ d dlmZmZmZ d dlmZmZ d d	lmZ d d
lmZ d dlmZmZ d dlmZ d dl m!Z! d dl"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-m.Z. d dl/m0Z0m1Z1 d dl2m3Z3 zd dl4Z4W n e5y«   dZ4Y nw dZ6e7edƒZ8G dd„ dƒZ9G dd„ de:ƒZ;dd„ Z<G dd„ dƒZ=d\dd „Z>d!d"„ Z?d]d$d%„Z@d&d'„ ZAd\d(d)„ZBd^d*d+„ZCd\d,d-„ZDG d.d/„ d/ƒZEG d0d1„ d1eEƒZFG d2d3„ d3eFƒZGG d4d5„ d5eEƒZHd6d7„ ZId8d9„ ZJG d:d;„ d;ƒZKG d<d=„ d=eEƒZLed_d>d?„ƒZMee8d@ƒZNedAdB„ ƒZOedCdD„ ƒZPedEdF„ ƒZQdGdH„ ZRdIdJ„ ZSdKdL„ ZTedMdN„ ƒZUdOdP„ ZVG dQdR„ dReEƒZWG dSdT„ dTƒZXG dUdV„ dVeEƒZYdWdX„ ZZeddYœdZd[„ƒZ[dS )`é    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Úmodify_settingsÚoverride_settingsÚrequires_tz_supportÚsetup_test_environmentÚteardown_test_environmentÚtzsetc                   @   s&   e Zd Zd	dd„Zdd„ Zdd„ ZdS )
r   é   c                 C   ó   || _ || _d S ©N)ÚvalÚplaces)Úselfr+   r,   © r.   úN/var/www/html/myproject/venv/lib/python3.10/site-packages/django/test/utils.pyÚ__init__-   ó   
zApproximate.__init__c                 C   ó
   t | jƒS r*   )Úreprr+   ©r-   r.   r.   r/   Ú__repr__1   ó   
zApproximate.__repr__c                 C   s$   | j |kptt| j | ƒ| jƒdkS ©Nr   )r+   ÚroundÚabsr,   )r-   Úotherr.   r.   r/   Ú__eq__4   s   $zApproximate.__eq__N)r(   )Ú__name__Ú
__module__Ú__qualname__r0   r5   r;   r.   r.   r.   r/   r   ,   s    
r   c                       s:   e Zd ZdZ‡ fdd„Zddd„Zdd„ Zd	d
„ Z‡  ZS )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 qt|ƒ‚tƒ  |¡S r*   )Ú
isinstanceÚstrÚKeyErrorÚsuperÚ__getitem__)r-   ÚkeyÚ
subcontext©Ú	__class__r.   r/   rC   =   s   
ÿzContextList.__getitem__Nc                 C   s$   z|   |¡W S  ty   | Y S w r*   )rC   rA   )r-   rD   Údefaultr.   r.   r/   ÚgetF   s
   ÿzContextList.getc                 C   s$   z| |  W dS  t y   Y dS w )NFT)rA   )r-   rD   r.   r.   r/   Ú__contains__L   s   
þÿzContextList.__contains__c                 C   s   t t dd„ | D ƒ¡ƒS )z0
        Flattened keys of subcontexts.
        c                 s   s   � | ]
}|D ]}|V  qqd S r*   r.   )Ú.0rE   Údr.   r.   r/   Ú	<genexpr>W   ó   € z#ContextList.keys.<locals>.<genexpr>)Úsetr   Úfrom_iterabler4   r.   r.   r/   ÚkeysS   s   zContextList.keysr*   )	r<   r=   r>   Ú__doc__rC   rI   rJ   rQ   Ú__classcell__r.   r.   rF   r/   r   8   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)r-   rV   r.   r.   r/   Úinstrumented_test_renderZ   s   rZ   c                   @   s   e Zd ZdS )Ú
_TestStateN)r<   r=   r>   r.   r.   r.   r/   r[   c   s    r[   c                 C   s~   t tdƒr	tdƒ‚| du rtj} tƒ }|t_tj|_g tj¢d‘t_tj|_	| t_tj
|_dt_
tj|_tt_g t_tƒ  dS )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   r\   ÚALLOWED_HOSTSÚallowed_hostsÚdebugÚEMAIL_BACKENDÚemail_backendr   Ú_renderÚtemplate_renderrZ   r   Úoutboxr   )rc   r\   r.   r.   r/   r%   g   s$   
ÿ
r%   c                  C   s2   t j} | jt_| jt_| jt_| j	t
_t `t`dS )z�
    Perform any global post-test teardown, such as restoring the original
    template renderer and restoring the email sending functions.
    N)r[   r\   rb   r   ra   rc   r`   re   rd   rg   r   rf   r   rh   )r\   r.   r.   r/   r&   ‹   s   r&   Fc              
   K   s  t |ƒ\}}g }	| ¡ D ]W\}
}d}|D ]N}t| }|	 ||
|du f¡ |du rW|}|jj| | ||j di ¡ dd¡d� |dkrVt|ƒD ]}|jj	t
|d ƒ| |d� qFqt| j t| j¡ qq| ¡ D ]\}}t| j t| j¡ qh|r„tD ]}dt| _q||	S )zCreate the test databases.NÚTESTÚ	SERIALIZET)Ú	verbosityÚautoclobberÚkeepdbÚ	serializeé   ©Úsuffixrk   rm   )Ú get_unique_databases_and_mirrorsÚvaluesr   ÚappendÚcreationÚcreate_test_dbÚsettings_dictrI   ÚrangeÚclone_test_dbr@   Úset_as_test_mirrorÚitemsÚforce_debug_cursor)rk   Úinteractiverm   Ú	debug_sqlÚparallelÚaliasesÚkwargsÚtest_databasesÚmirrored_aliasesÚ	old_namesÚdb_nameÚfirst_aliasÚaliasÚ
connectionÚindexÚmirror_aliasr.   r.   r/   Úsetup_databases›   sB   ü
ý€ê
ÿr‹   c                 C   sØ   g }t ƒ }i }| D ]&\}\}}t ƒ }|D ]}	| | |	g ¡¡ q| |¡s+td| ƒ‚|||< q	| rjd}
g }| D ]'\}\}}||  |¡rV| |¡ | |||ff¡ d}
q8| |||ff¡ q8|
sftdƒ‚|} | 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])rO   ÚupdaterI   Ú
isdisjointr   Úissubsetrt   )r‚   ÚdependenciesÚordered_test_databasesÚresolved_databasesÚdependencies_mapÚsigÚ_r€   Úall_depsr‡   ÚchangedÚdeferredÚ	signaturer…   r.   r.   r/   Údependency_orderedÇ   s8   
ÿÿ

ñr™   c           	      C   sê   | du rt } i }i }i }t t j ¡ }t D ]O}t | }|jd }|d r+|d ||< q|| v rd| |j ¡ |jd tƒ f¡}|d  |¡ d|v rP|d ||< q|tkrd|j ¡ |krd| dtg¡||< qt	| 
¡ |ƒ}t |¡}||fS )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.
    Nri   ÚMIRRORÚNAMEro   ÚDEPENDENCIES)r   r   ru   Útest_db_signaturerw   Ú
setdefaultrO   ÚaddrI   r™   r{   ÚcollectionsÚOrderedDict)	r€   rƒ   r‚   r�   Údefault_sigr‡   rˆ   Útest_settingsÚitemr.   r.   r/   rr   ñ   s2   
þ€
rr   c                 C   sX   | D ]'\}}}|r)|dkr!t |ƒD ]}|jjt|d ƒ||d� q|j |||¡ qdS )z%Destroy all the non-mirror databases.ro   rp   N)rx   ru   Údestroy_test_dbr@   )Ú
old_configrk   r   rm   rˆ   Úold_nameÚdestroyr‰   r.   r.   r/   Úteardown_databases   s   
ý€÷r©   c                 C   s\   |p| j }| d¡}t|ƒdkrd |d d… ¡}nd}t|i i |d ƒ}t||d ƒ}|S )NÚ.ro   éÿÿÿÿ)ÚTEST_RUNNERÚsplitÚlenÚjoinÚ
__import__Úgetattr)r   Útest_runner_classÚ	test_pathÚtest_module_nameÚtest_moduleÚtest_runnerr.   r.   r/   r!   .  s   

r!   c                   @   sR   e Zd ZdZddd„Zdd„ Zdd„ Zd	d
„ Zdd„ Zdd„ Z	dd„ Z
dd„ ZdS )Ú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)r-   r¸   r¹   r.   r.   r/   r0   G  r1   zTestContextDecorator.__init__c                 C   ó   t ‚r*   ©ÚNotImplementedErrorr4   r.   r.   r/   ÚenableK  ó   zTestContextDecorator.enablec                 C   rº   r*   r»   r4   r.   r.   r/   ÚdisableN  r¾   zTestContextDecorator.disablec                 C   s   |   ¡ S r*   )r½   r4   r.   r.   r/   Ú	__enter__Q  s   zTestContextDecorator.__enter__c                 C   s   |   ¡  d S r*   ©r¿   ©r-   Úexc_typeÚ	exc_valueÚ	tracebackr.   r.   r/   Ú__exit__T  ó   zTestContextDecorator.__exit__c                    sJ   t |tƒr!|j‰ |j‰‡ ‡fdd„}‡‡fdd„}||_||_|S tdƒ‚)Nc                    sD   ˆ  ¡ }ˆjrt| ˆj|ƒ zˆ | ƒ W d S  ty!   ˆ ¡  ‚ w r*   )r½   r¸   ÚsetattrÚ	Exceptionr¿   )Ú
inner_selfrV   )Údecorated_setUpr-   r.   r/   ÚsetUp\  s   þz2TestContextDecorator.decorate_class.<locals>.setUpc                    s   ˆ | ƒ ˆ  ¡  d S r*   rÁ   )rÊ   )Údecorated_tearDownr-   r.   r/   ÚtearDownf  s   z5TestContextDecorator.decorate_class.<locals>.tearDownz1Can only decorate subclasses of unittest.TestCase)Ú
issubclassr   rÌ   rÎ   Ú	TypeError)r-   ÚclsrÌ   rÎ   r.   )rË   rÍ   r-   r/   Údecorate_classW  s   

z#TestContextDecorator.decorate_classc                    s   t ˆ ƒ‡ ‡fdd„ƒ}|S )Nc                     sF   ˆ�}ˆj r||ˆj < ˆ | i |¤ŽW  d   ƒ S 1 sw   Y  d S r*   )r¹   )Úargsr�   rV   ©Úfuncr-   r.   r/   Úinnerp  s
   
$ýz5TestContextDecorator.decorate_callable.<locals>.innerr   )r-   rÕ   rÖ   r.   rÔ   r/   Údecorate_callableo  s   z&TestContextDecorator.decorate_callablec                 C   s6   t |tƒr
|  |¡S t|ƒr|  |¡S tdt|ƒ ƒ‚)Nz!Cannot decorate object of type %s)r?   ÚtyperÒ   Úcallabler×   rÐ   )r-   Ú	decoratedr.   r.   r/   Ú__call__x  s
   


zTestContextDecorator.__call__)NN)r<   r=   r>   rR   r0   r½   r¿   rÀ   rÆ   rÒ   r×   rÛ   r.   r.   r.   r/   r·   ;  s    
	r·   c                       sD   e Zd ZdZdZ‡ fdd„Zdd„ Zdd„ Zd	d
„ Zd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*   )ÚoptionsrB   r0   ©r-   r�   rF   r.   r/   r0   ‰  ó   zoverride_settings.__init__c                 C   sÔ   d| j v rz
t | j d ¡ W n ty   t ¡  ‚ w ttjƒ}| j  ¡ D ]
\}}t	|||ƒ q&tj| _
|t_| j  ¡ D ]*\}}ztjtjj||dd� W q= tyg } z|| _|  ¡  W Y d }~q=d }~ww d S )NÚINSTALLED_APPST©rT   ÚsettingÚvalueÚenter)rÝ   r   Úset_installed_appsrÉ   Úunset_installed_appsr   r   Ú_wrappedr{   rÈ   Úwrappedr   rW   rG   Úenable_exceptionr¿   )r-   ÚoverriderD   Ú	new_valueÚexcr.   r.   r/   r½   �  s0   
þ

þ€þúzoverride_settings.enablec                 C   s”   d| j v r	t ¡  | jt_| `g }| j D ]}tt|d ƒ}tjtjj	||dd�}| 
|¡ q| jd ur:| j}d | _|‚|D ]\}}t|tƒrG|‚q<d S )Nrà   Frá   )rÝ   r   ræ   rè   r   rç   r±   r   Úsend_robustrG   Úextendré   r?   rÉ   )r-   Ú	responsesrD   rë   Úresponses_for_settingrì   r”   Úresponser.   r.   r/   r¿   ¥  s*   

þ

ÿÿzoverride_settings.disablec                 C   s,   |j d u r| j|_ d S i |j ¥| j¥|_ d S r*   )Ú_overridden_settingsrÝ   ©r-   Ú	test_funcr.   r.   r/   Úsave_optionsº  s   
ÿ
þzoverride_settings.save_optionsc                 C   s,   ddl m} t||ƒstdƒ‚|  |¡ |S )Nr   )ÚSimpleTestCasezPOnly subclasses of Django SimpleTestCase can be decorated with override_settings)Údjango.teströ   rÏ   Ú
ValueErrorrõ   )r-   rÑ   rö   r.   r.   r/   rÒ   Ä  s   
ÿ
z override_settings.decorate_class)r<   r=   r>   rR   ré   r0   r½   r¿   rõ   rÒ   rS   r.   r.   rF   r/   r#   €  s    
r#   c                       s4   e Zd ZdZ‡ fdd„Zdd„ Z‡ fdd„Z‡  ZS )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| ¡ ƒ| _ tt| ƒ ¡  d S r7   )Ú
operationsÚlistr{   rB   r#   r0   )r-   rÓ   r�   rF   r.   r/   r0   Ó  s   zmodify_settings.__init__c                 C   s,   |j d u r| j|_ d S t|j ƒ| j |_ d S r*   )Ú_modified_settingsrù   rú   ró   r.   r.   r/   rõ   Ý  s   
ÿ
ÿzmodify_settings.save_optionsc              	      sè   i | _ | jD ]f\}}z| j | ‰W n ty"   ttt|g ƒƒ‰Y nw | ¡ D ]?\}‰ tˆ tƒr3ˆ g‰ |dkrCˆ‡fdd„ˆ D ƒ ‰q'|dkrS‡fdd„ˆ D ƒˆ ‰q'|dkra‡ fdd„ˆD ƒ‰q't	d| ƒ‚ˆ| j |< qt
ƒ  ¡  d S )	Nrt   c                    ó   g | ]}|ˆ vr|‘qS r.   r.   ©rK   r¤   ©rã   r.   r/   Ú
<listcomp>ó  ó    z*modify_settings.enable.<locals>.<listcomp>Úprependc                    rü   r.   r.   rý   rþ   r.   r/   rÿ   õ  r   Úremovec                    rü   r.   r.   rý   )r{   r.   r/   rÿ   ÷  r   zUnsupported action: %s)rÝ   rù   rA   rú   r±   r   r{   r?   r@   rø   rB   r½   )r-   Únamerù   ÚactionrF   )r{   rã   r/   r½   å  s&   ÿ
zmodify_settings.enable)r<   r=   r>   rR   r0   rõ   r½   rS   r.   r.   rF   r/   r"   Î  s
    
r"   c                       s2   e Zd ZdZd	‡ fdd„	Zdd„ Zdd„ Z‡  ZS )
Úoverride_system_checkszÇ
    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,   ddl m} || _|| _|| _tƒ  ¡  d S )Nr   )Úregistry)Údjango.core.checks.registryr  Ú
new_checksÚdeployment_checksrB   r0   )r-   r  r	  r  rF   r.   r/   r0     s
   zoverride_system_checks.__init__c                 C   s�   | j j| _tƒ | j _| jD ]}| j j|gt|ddƒ¢R Ž  q| j j| _| jd urDtƒ | j _| jD ]}| j j|gt|ddƒ¢R ddiŽ q0d S d S )NÚtagsr.   ÚdeployT)	r  Úregistered_checksÚ
old_checksrO   r  Úregisterr±   r	  Úold_deployment_checks)r-   Úcheckr.   r.   r/   r½     s   






$ýzoverride_system_checks.enablec                 C   s   | j | j_| j| j_d S r*   )r  r  r  r  r	  r4   r.   r.   r/   r¿     s   
zoverride_system_checks.disabler*   ©r<   r=   r>   rR   r0   r½   r¿   rS   r.   r.   rF   r/   r  þ  s
    r  c                    s¶   t  d¡‰ ‡ fdd„‰dd„ ‰dd„ ‰‡‡fdd	„‰d
d„ ‰‡‡‡‡fdd„‰dd„ }|  ¡  dd¡} | ¡  dd¡}|  d¡sJd}||  } || }|t| ƒƒ}|t|ƒƒ}ˆ||ƒS )a9  
    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 and leading and trailing whitespace.

    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_rer.   r/   Únorm_whitespace%  rÇ   z$compare_xml.<locals>.norm_whitespacec                 S   s   d  dd„ | jD ƒ¡S )NÚ c                 s   s"   � | ]}|j tjkr|jV  qd S r*   )ÚnodeTyper   Ú	TEXT_NODEÚdata©rK   Úcr.   r.   r/   rM   )  s   € ÿÿz2compare_xml.<locals>.child_text.<locals>.<genexpr>)r¯   Ú
childNodes©Úelementr.   r.   r/   Ú
child_text(  s   zcompare_xml.<locals>.child_textc                 S   s   dd„ | j D ƒS )Nc                 S   s   g | ]
}|j tjkr|‘qS r.   )r  r   ÚELEMENT_NODEr  r.   r.   r/   rÿ   -  s    ÿz1compare_xml.<locals>.children.<locals>.<listcomp>)r  r  r.   r.   r/   Úchildren,  s   zcompare_xml.<locals>.childrenc                    s   ˆˆ | ƒƒS r*   r.   r  )r   r  r.   r/   Únorm_child_text0  rÇ   z$compare_xml.<locals>.norm_child_textc                 S   s   t | j ¡ ƒS r*   )ÚdictÚ
attributesr{   r  r.   r.   r/   Ú
attrs_dict3  ó   zcompare_xml.<locals>.attrs_dictc                    sx   | j |j krdS ˆ| ƒˆ|ƒkrdS ˆ | ƒˆ |ƒkrdS ˆ| ƒ}ˆ|ƒ}t|ƒt|ƒkr.dS t‡fdd„t||ƒD ƒƒS )NFc                 3   s   � | ]
\}}ˆ ||ƒV  qd S r*   r.   )rK   ÚwantÚgot)Úcheck_elementr.   r/   rM   A  rN   z5compare_xml.<locals>.check_element.<locals>.<genexpr>)ÚtagNamer®   ÚallÚzip)Úwant_elementÚgot_elementÚwant_childrenÚgot_children)r&  r*  r"  r#  r.   r/   r*  6  s   z"compare_xml.<locals>.check_elementc                 S   s$   | j D ]}|jtjkr|  S qd S r*   )r  r  r   ÚCOMMENT_NODE)ÚdocumentÚnoder.   r.   r/   Ú
first_nodeC  s
   
ÿÿzcompare_xml.<locals>.first_nodez\nÚ
z<?xmlz<root>%s</root>)ÚreÚcompileÚstripÚreplaceÚ
startswithr   )r(  r)  r5  ÚwrapperÚ	want_rootÚgot_rootr.   )r  r&  r*  r   r"  r#  r  r/   Úcompare_xml  s"   


r?  c                 C   s   | ddi S )Nr”   r  r.   )Úsr.   r.   r/   Ú
str_prefixY  rÇ   rA  c                   @   sL   e Zd ZdZdd„ Zdd„ Zdd„ Zdd	„ Zed
d„ ƒZ	dd„ Z
dd„ ZdS )ÚCaptureQueriesContextzU
    Context manager that captures queries executed by the specified connection.
    c                 C   s
   || _ d S r*   )rˆ   )r-   rˆ   r.   r.   r/   r0   a  r6   zCaptureQueriesContext.__init__c                 C   r2   r*   )ÚiterÚcaptured_queriesr4   r.   r.   r/   Ú__iter__d  r6   zCaptureQueriesContext.__iter__c                 C   s
   | j | S r*   )rD  )r-   r‰   r.   r.   r/   rC   g  r6   z!CaptureQueriesContext.__getitem__c                 C   r2   r*   )r®   rD  r4   r.   r.   r/   Ú__len__j  r6   zCaptureQueriesContext.__len__c                 C   s   | j j| j| j… S r*   )rˆ   ÚqueriesÚinitial_queriesÚfinal_queriesr4   r.   r.   r/   rD  m  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_logrH  rI  r   Ú
disconnectr   r4   r.   r.   r/   rÀ   q  s   


zCaptureQueriesContext.__enter__c                 C   s2   | j | j_ t t¡ |d urd S t| jjƒ| _d S r*   )r|   rˆ   r   Úconnectr   r®   rK  rI  rÂ   r.   r.   r/   rÆ   |  s
   

zCaptureQueriesContext.__exit__N)r<   r=   r>   rR   r0   rE  rC   rF  ÚpropertyrD  rÀ   rÆ   r.   r.   r.   r/   rB  ]  s    
rB  c                       s,   e Zd Z‡ fdd„Zdd„ Zdd„ Z‡  ZS )Úignore_warningsc                    s:   || _ d| j v sd| j v rtj| _ntj| _tƒ  ¡  d S )NÚmessageÚmodule)Úignore_kwargsÚwarningsÚfilterwarningsÚfilter_funcÚsimplefilterrB   r0   rÞ   rF   r.   r/   r0   …  s
   
zignore_warnings.__init__c                 C   s*   t  ¡ | _| j ¡  | jdi | j¤Ž d S )NÚignore)rW  )rS  Úcatch_warningsrÀ   rU  rR  r4   r.   r.   r/   r½   �  s   

zignore_warnings.enablec                 C   s   | j jt ¡ Ž  d S r*   )rX  rÆ   ÚsysÚexc_infor4   r.   r.   r/   r¿   ’  ó   zignore_warnings.disable)r<   r=   r>   r0   r½   r¿   rS   r.   r.   rF   r/   rO  „  s    rO  c              
   #   s\   � g ‰ ‡ ‡fdd„}t  | ¡}t||ƒ}t|||ƒ zˆ V  W t|||ƒ dS t|||ƒ w )a  
    Context manager that takes a named logger and the logging level
    and provides a simple mock-like list of messages received.

    Use unittest.assertLogs() if you only need Python 3 support. This
    private API will be removed after Python 2 EOL in 2020 (#27753).
    c                    s(   | | }ˆ   ˆr||f¡ d S |¡ d S r*   )rt   )ÚmsgrÓ   r�   Úcall©ÚcallsÚ
log_kwargsr.   r/   Úreplacement¡  s    z!patch_logger.<locals>.replacementN)ÚloggingÚ	getLoggerr±   rÈ   )Úlogger_nameÚ	log_levelr`  ra  ÚloggerÚorigr.   r^  r/   Úpatch_logger–  s   €	

rh  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 jdd… }t j | ¡ z	dV  W |t _dS |t _w )z5Context manager to temporarily add paths to sys.path.N)rY  Úpathrî   )ÚpathsÚ_orig_sys_pathr.   r.   r/   Úextend_sys_path¹  s   €rl  c                 c   s*   � |   ¡  z
dV  W |   ¡  dS |   ¡  w )z?Clear the cache of an LRU cache object on entering and exiting.N)Úcache_clear)Úlru_cache_objectr.   r.   r/   r    Ä  s
   €r    c              
   c   sH   � t t| ƒ}tt| tƒ ƒ zt t| ƒV  W tt| |ƒ dS tt| |ƒ 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±   rY  rÈ   r   )Ústream_nameÚorig_stdoutr.   r.   r/   Úcaptured_outputÎ  s   €
rq  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©rq  r.   r.   r.   r/   Úcaptured_stdoutÝ  ó   ru  c                   C   rr  )z°Capture the output of sys.stderr:

       with captured_stderr() as stderr:
           print("hello", file=sys.stderr)
       self.assertEqual(stderr.getvalue(), "hello
")
    Ústderrrt  r.   r.   r.   r/   Úcaptured_stderrç  rv  rx  c                   C   rr  )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")
    Ústdinrt  r.   r.   r.   r/   Úcaptured_stdinñ  s   
rz  c                 #   s2   � t j }‡ fdd„t _ z	dV  W |t _ dS |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*   r.   r.   ©Útr.   r/   Ú<lambda>  s    zfreeze_time.<locals>.<lambda>N)Útime)r|  Ú
_real_timer.   r{  r/   Úfreeze_timeþ  s   €	r€  c                 C   s:   t tdu dƒ| ƒ} tdddœddddidœgd	�| ƒ} | 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.
    Nzthis test requires jinja2z/django.template.backends.django.DjangoTemplatesT)ÚBACKENDÚAPP_DIRSz&django.template.backends.jinja2.Jinja2Úkeep_trailing_newline)r�  r‚  ÚOPTIONS)Ú	TEMPLATES)r	   Újinja2r#   )rô   r.   r.   r/   Úrequire_jinja2  s   þýýùr‡  c                       ó0   e Zd ZdZ‡ fdd„Zdd„ Zdd„ Z‡  ZS )Úoverride_script_prefixzEDecorator or context manager to temporary override the script prefix.c                    rÜ   r*   )ÚprefixrB   r0   )r-   rŠ  rF   r.   r/   r0   "  rß   zoverride_script_prefix.__init__c                 C   s   t ƒ | _t| jƒ d S r*   )r   Ú
old_prefixr   rŠ  r4   r.   r.   r/   r½   &  s   zoverride_script_prefix.enablec                 C   s   t | jƒ d S r*   )r   r‹  r4   r.   r.   r/   r¿   *  r'  zoverride_script_prefix.disabler  r.   r.   rF   r/   r‰     s
    r‰  c                   @   s    e Zd ZdZdd„ Zdd„ ZdS )ÚLoggingCaptureMixinzn
    Capture the output from the 'django' logger and store it on the class's
    logger_output attribute.
    c                 C   s8   t  d¡| _| jjd j| _tƒ | _| j| jjd _d S )NÚdjangor   )rb  rc  rf  ÚhandlersÚstreamÚ
old_streamr   Úlogger_outputr4   r.   r.   r/   rÌ   3  s   zLoggingCaptureMixin.setUpc                 C   s   | j | jjd _d S r7   )r�  rf  rŽ  r�  r4   r.   r.   r/   rÎ   9  r[  zLoggingCaptureMixin.tearDownN)r<   r=   r>   rR   rÌ   rÎ   r.   r.   r.   r/   rŒ  .  s    rŒ  c                       rˆ  )Úisolate_appsa  
    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ƒ jdi |¤Ž d S )Nr.   )Úinstalled_appsrB   r0   )r-   r“  r�   rF   r.   r/   r0   M  s   zisolate_apps.__init__c                 C   s"   t j| _t| jƒ}tt d|ƒ |S ©NÚdefault_apps)r   r•  Úold_appsr   r“  rÈ   )r-   r   r.   r.   r/   r½   Q  s   
zisolate_apps.enablec                 C   s   t td| jƒ d S r”  )rÈ   r   r–  r4   r.   r.   r/   r¿   W  s   zisolate_apps.disabler  r.   r.   rF   r/   r’  =  s
    r’  c                     s   ‡ fdd„}|S )z0Decorator to add tags to a test class or method.c                    s0   t | dƒr| j ˆ ¡| _| S t| dtˆ ƒƒ | S )Nr
  )r^   r
  ÚunionrÈ   rO   )Úobj©r
  r.   r/   Ú	decorator]  s
   
ÿztag.<locals>.decoratorr.   )r
  rš  r.   r™  r/   Útag[  s   r›  )Úlookup_namec                g   sT   � z|D ]}|   ||¡ qdV  W |D ]}|  ||¡ qdS |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Úlookupr.   r.   r/   r�  f  s   €ÿÿr�  r*   )FFr   N)r   F)F)\r    rb  r7  rY  r~  rS  Ú
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__r^   Ú
TZ_SUPPORTr   rú   r   rZ   r[   r%   r&   r‹   r™   rr   r©   r!   r·   r#   r"   r  r?  rA  rB  rO  rh  r$   rl  r    rq  ru  rx  rz  r€  r‡  r‰  rŒ  r’  r›  r�  r.   r.   r.   r/   Ú<module>   s–    ÿ
"	
$
,
*
/
EN0>'þ


	



