File: //lib/python3/dist-packages/twisted/trial/test/__pycache__/test_warning.cpython-38.pyc
U
W[�A � @ s� d Z ddlmZmZ ddlZddlZddlmZ ddlm Z
ddl m Z ddl
mZ ddlmZmZ G d d
� d
e�ZG d d � d e�ZG d
d� de�ZG dd� de�ZdS )z?
Tests for Trial's interaction with the Python warning system.
� )�division�absolute_importN)�
TestResult)�NativeStringIO)�FilePath)�SynchronousTestCase)�_collectWarnings�_setWarningRegistryToNonec @ s e Zd ZdZG dd� de�ZdS )�MaskzQ
Hide a test case definition from trial's automatic discovery mechanism.
c @ s( e Zd ZdZdZeZdd� Zdd� ZdS )zMask.MockTestsz�
A test case which is used by L{FlushWarningsTests} to verify behavior
which cannot be verified by code inside a single test method.
�some warning textc C s t �| j| j� dS )z@
Generate a warning and don't flush it.
N)�warnings�warn�message�category��self� r �A/usr/lib/python3/dist-packages/twisted/trial/test/test_warning.py�test_unflushed s z Mask.MockTests.test_unflushedc C s( t �| j| j� | �t| �� �d� dS )z:
Generate a warning and flush it.
� N)r r
r r � assertEqual�len�
flushWarningsr r r r � test_flushed' s zMask.MockTests.test_flushedN) �__name__�
__module__� __qualname__�__doc__r � UserWarningr r r r r r r � MockTests s
r N)r r r r r r r r r r r
s r
c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"S )#�FlushWarningsTestsze
Tests for C{flushWarnings}, an API for examining the warnings
emitted so far in a test.
c C s& |� � D ]\}}| �|| |� qdS )z�
Assert that all the keys present in C{subset} are also present in
C{set} and that the corresponding values are equal.
N)�itemsr )r �setZsubset�k�vr r r �assertDictSubset6 s z#FlushWarningsTests.assertDictSubsetc C s8 | � t|�t|�� t||�D ]\}}| �||� q dS )z�
For each pair of corresponding elements in C{sets} and C{subsets},
assert that the element from C{subsets} is a subset of the element from
C{sets}.
N)r r �zipr% )r ZsetsZsubsets�a�br r r �assertDictSubsets? s z$FlushWarningsTests.assertDictSubsetsc C s | � | �� g � dS )zg
If no warnings are emitted by a test, C{flushWarnings} returns an empty
list.
N)r r r r r r � test_noneJ s z FlushWarningsTests.test_nonec C sP d}t }tj||d� d}t}tj||d� | �| �� ||d�||d�g� dS )z|
If several warnings are emitted by a test, C{flushWarnings} returns a
list containing all of them.
zfirst warning message�r r zsecond warning message�r r N)r r r
�RuntimeWarningr) r )r � firstMessage�
firstCategory�
secondMessage�secondCategoryr r r � test_severalR s ��z FlushWarningsTests.test_severalc C sD d}t }td�D ]}tj||d� q| �| �� ||d�gd � dS )z�
The same warning triggered twice from the same place is included twice
in the list returned by C{flushWarnings}.
� the message� r+ r, N)r- �ranger r
r) r )r r r �ir r r �
test_repeatede s �z FlushWarningsTests.test_repeatedc C sB d}t }tj||d� | �| �� ||d�g� | �| �� g � dS )z�
After a particular warning event has been returned by C{flushWarnings},
it is not returned by subsequent calls.
r3 r+ r, N)r- r r
r) r r )r r r r r r � test_clearedt s
�z FlushWarningsTests.test_clearedc C s� t � }t�d�}|�|� | �tjjg�}| �|d d d� | �|d d t� t |�jj
}|j }|j d }| �|d d |� | �|d d |� | �t
|�d � d
S ) zx
Any warnings emitted by a test which are not flushed are emitted to the
Python warning system.
r r r r r � �filename�linenor N)r r
r �runr r r �assertIdenticalr �type�__code__� co_filename�co_firstlinenor )r �result�caseZ
warningsShown�wherer: r; r r r r � s
z!FlushWarningsTests.test_unflushedc C sJ t � }t�d�}t� }| �td|�}|�|� |�� | �|� � d� dS )zx
Any warnings emitted by a test which are flushed are not emitted to the
Python warning system.
r �stdout� N)
r r
r �StringIO�patch�sysr< Zrestorer �getvalue)r rB rC �outputZmonkeyr r r r � s
z FlushWarningsTests.test_flushedc
C s� G dd� dt �}t� }t�d�}||_tjdd� }zbt�d� |�|� | � t
|j �d� | � |j d d |� | �
|j d d �� d �d �� W 5 |tjdd�< X dS )
z�
If a warnings filter has been installed which turns warnings into
exceptions, tests have an error added to the reporter for them for each
unflushed warning.
c @ s e Zd ZdS )zIFlushWarningsTests.test_warningsConfiguredAsErrors.<locals>.CustomWarningN�r r r r r r r �
CustomWarning� s rM r N�errorr r ���z CustomWarning: some warning text)�Warningr r
r r r �filters� simplefilterr< r r �errorsr= Z
assertTrue�
splitlines�endswith�r rM rB rC ZoriginalWarningsr r r � test_warningsConfiguredAsErrors� s
��z2FlushWarningsTests.test_warningsConfiguredAsErrorsc
C sp G dd� dt �}t� }t�d�}||_tjdd� }z&t�d� |�|� | � |j
g � W 5 |tjdd�< X dS )z�
If a warnings filter has been installed which turns warnings into
exceptions, tests which emit those warnings but flush them do not have
an error added to the reporter.
c @ s e Zd ZdS )zPFlushWarningsTests.test_flushedWarningsConfiguredAsErrors.<locals>.CustomWarningNrL r r r r rM � s rM r NrN ) rP r r
r r r rQ rR r< r rS rV r r r �&test_flushedWarningsConfiguredAsErrors� s
z9FlushWarningsTests.test_flushedWarningsConfiguredAsErrorsc C s@ t �d� | �t| �� �d� t �d� | �t| �� �d� dS )z�
Any warnings emitted after a call to C{flushWarnings} can be flushed by
another call to C{flushWarnings}.
z
first messager zsecond messageN)r r
r r r r r r r �test_multipleFlushes� s
z'FlushWarningsTests.test_multipleFlushesc sx d�t � � �fdd�}d�t���fdd�}|� |� | �| j|gd�� �d�g� | �| j|gd���d�g� d S )
z�
The list returned by C{flushWarnings} includes only those
warnings which refer to the source of the function passed as the value
for C{offendingFunction}, if a value is passed for that parameter.
zfirst warning textc s t j�� dd� d S �Nr )�
stacklevel�r r
r )r/ r. r r �one� s z>FlushWarningsTests.test_filterOnOffendingFunction.<locals>.onez some textc s t j�� dd� d S rZ r\ r )r1 r0 r r �two� s z>FlushWarningsTests.test_filterOnOffendingFunction.<locals>.two�ZoffendingFunctionsr, N)r r- r) r )r r] r^ r )r/ r. r1 r0 r � test_filterOnOffendingFunction� s
�
�z1FlushWarningsTests.test_filterOnOffendingFunctionc C s, dd� }|� | � t| j|gd��d� dS )z�
Verify that warnings emitted at the very edges of a function are still
determined to be emitted from that function.
c S s" t �d� t �d� t �d� d S )Nzfirst line warningzinternal line warningzlast line warningr\ r r r r �warner s
z:FlushWarningsTests.test_functionBoundaries.<locals>.warnerr_ � N)r r r )r ra r r r �test_functionBoundaries s �z*FlushWarningsTests.test_functionBoundariesc C s4 t �d� | �t| jdg� | �t| �� �d� dS )z�
If an object which is neither a function nor a method is included in the
C{offendingFunctions} list, C{flushWarnings} raises L{ValueError}. Such
a call flushes no warnings.
zoh noNr )r r
Z assertRaises�
ValueErrorr r r r r r r �test_invalidFilter s
z%FlushWarningsTests.test_invalidFilterc C s� t | �� �d���d�}|�� |�d��d� |�d��d� |�� j�d�}t j�
d|� | � t jj |� ddl
m} | � t jjd � | � t jj|j� |�d�� � |�� | �t| �|jg��d
� d S ) zx
Warnings emitted by a function the source code of which is not
available can still be flushed.
�utf-8� twisted_private_helper� __init__.py� s missingsourcefile.py�7
import warnings
def foo():
warnings.warn("oh no")
r )�missingsourcefile�twisted_private_helperr N)r �mktemp�encode�child�makedirs�
setContent�parent�path�decoderI �insert�
addCleanup�removerl rk �modules�popr �foor r r )r �package� pathEntryrk r r r �test_missingSource s z%FlushWarningsTests.test_missingSourcec C s t | �� �d���d�}|�� |�d��d� |�d��d� |�� j�d�}t j�
d|� | � t jj |� ddl
m} t jd = t j|j= zdd
lm} W n tk
r� Y nX |� |�|�d �� ddlm} | � t jjd � | � t jj|j� |�� | �t| �|jg��d
� dS )a
Warnings emitted by a function defined in a file which has been renamed
since it was initially compiled can still be flushed.
This is testing the code which specifically supports working around the
unfortunate behavior of CPython to write a .py source file name into
the .pyc files it generates and then trust that it is correct in
various places. If source files are renamed, .pyc files may not be
regenerated, but they will contain incorrect filenames.
rf rg rh ri s module.pyrj r )�modulerl )�invalidate_cachess twisted_renamed_helper�twisted_renamed_helperr N) r rm rn ro rp rq rr rs rt rI ru rv rw rl r~ rx r � importlibr � ImportErrorZmoveToZsiblingr� ry rz r r r )r r{ r| r~ r r r r �test_renamedSource3 s*
z%FlushWarningsTests.test_renamedSourceN)r r r r r% r) r* r2 r7 r8 r r rW rX rY r` rc re r} r� r r r r r 0 s" r c @ s e Zd ZdS )� FakeWarningNrL r r r r r� l s r� c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�CollectWarningsTestsz(
Tests for L{_collectWarnings}.
c s� d��ddd� �t ddd��g � � ���fdd�}t� j|� | �� d d
� | �� d j�� | �� d j�� | �� d jt��� | �� d d � | �t� �d
� dS )zS
L{_collectWarnings} calls the observer with each emitted warning.
z dummy calls observer warningNrO r r4 rb c s6 � � d� t��� t��� t��� � � d� d S )N�call� returning)�appendr r
r ��eventsr. r0 Z thirdMessager r �f} s
z2CollectWarningsTests.test_callsObserver.<locals>.fr r� r9 r� � )rP r r� r r �strr )r r� r r� r �test_callsObserveru s z'CollectWarningsTests.test_callsObserverc C s: t � }| �td|� tdd� tjd� | �|�� d� dS )z�
Any warnings emitted by a call to a function passed to
L{_collectWarnings} are not actually emitted to the warning system.
rE c S s d S �Nr ��xr r r �<lambda>� ri z6CollectWarningsTests.test_suppresses.<locals>.<lambda>�textrF N)rG rH rI r r r
r rJ )r rK r r r �test_suppresses� s z$CollectWarningsTests.test_suppressesc sX g � t � �� �fdd�}tdd� |ddddd �}| �� d
ddd �fg� | �|�� d S ) z�
L{_collectWarnings} returns the result of calling the callable passed to
it with the parameters given.
c s � � | |f� �S r� )r� )�args�kwargs�Z arguments�valuer r r� � s z2CollectWarningsTests.test_callsFunction.<locals>.fc S s d S r� r r� r r r r� � ri z9CollectWarningsTests.test_callsFunction.<locals>.<lambda>r r' r4 �d)r( �c)r r' N)�objectr r r= )r r� rB r r� r �test_callsFunction� s z'CollectWarningsTests.test_callsFunctionc C sd b dd� }t�d� |� g }t|j|� | �t|�d� | �|d jd� | �t| �� �d� dS )z�
Subsequent emissions of a warning from a particular source site can be
collected by L{_collectWarnings}. In particular, the per-module
emitted-warning cache should be bypassed (I{__warningregistry__}).
c S s t �d� d S )Nrz r\ r r r r r� � s z>CollectWarningsTests.test_duplicateWarningCollected.<locals>.f�defaultr r rz N) Z__warningregistry__r rR r r� r r r r )r r� r� r r r � test_duplicateWarningCollected� s
z3CollectWarningsTests.test_duplicateWarningCollectedc C s, t � }|tj|< | �tjj|� | �� dS )z�
L{_collectWarnings}'s behavior is not altered by the presence of an
object which cannot have attributes set on it as a value in
C{sys.modules}.
N)r� rI rx rv ry r� �r �keyr r r �test_immutableObject� s
z)CollectWarningsTests.test_immutableObjectc st i � G � fdd�dt �}t � }t � }||�� |< t � }t � }||�� |< t� � | �t||||g�t� �� �� dS )a�
If the dictionary passed to L{_setWarningRegistryToNone} changes size
partway through the process, C{_setWarningRegistryToNone} continues to
set C{__warningregistry__} to L{None} on the rest of the values anyway.
This might be caused by C{sys.modules} containing something that's not
really a module and imports things on setattr. py.test does this, as
does L{twisted.python.deprecate.deprecatedModuleAttribute}.
c s e Zd Zdd� Z� fdd�ZdS )zKCollectWarningsTests.test_setWarningRegistryChangeWhileIterating.<locals>.Ac S s || j d< d S )N�_key)�__dict__r� r r r �__init__� s zTCollectWarningsTests.test_setWarningRegistryChangeWhileIterating.<locals>.A.__init__c s d � | j < d S r� )r� )r r� �item�r� r r � __setattr__� s zWCollectWarningsTests.test_setWarningRegistryChangeWhileIterating.<locals>.A.__setattr__N)r r r r� r� r r� r r �A� s r� N)r� r r r" �keys)r r� Zkey1Zkey2Zkey3Zkey4r r� r �+test_setWarningRegistryChangeWhileIterating� s z@CollectWarningsTests.test_setWarningRegistryChangeWhileIteratingN)
r r r r r� r� r� r� r� r� r r r r r� q s r� )r Z
__future__r r rI r Zunittestr Ztwisted.python.compatr rG Ztwisted.python.filepathr Ztwisted.trial.unittestr Ztwisted.trial._synctestr r r� r
r rP r� r� r r r r �<module> s >