Warning: Cannot modify header information - headers already sent by (output started at /var/www/jonas-eriksen.dk/pic/private/index.php:1) in /var/www/jonas-eriksen.dk/pic/private/index.php on line 215
PK ! &% util.cpython-36.pycnu [ 3
\9 @ s d Z ddlmZmZ ddlmZ dZdZdZdZ dZ
dZ ee e e e e
Z e dks\t
dd Zd
d Zd d
dZdd Zdd Zdd Zdd ZeddZdd Zdd Zd d Zd S )!zVarious utility functions. )
namedtuple OrderedDict) commonprefixTP c C sB t | | | }|tkr>d| d | || t | | d f } | S )Nz%s[%d chars]%s)len_PLACEHOLDER_LEN)s prefixlenZ suffixlenskip r
%/usr/lib64/python3.6/unittest/util.py_shorten s &r c s t tt| } ttt| }|tkr(| S t| t t| t t }|t krtt t | tk spt
t t| t fdd| D S t tt t fdd| D S )Nc 3 s | ]} |d V qd S )Nr
).0r
)prefixr r
r ' s z'_common_shorten_repr..c 3 s& | ] } t |d tt V qd S )N)r
_MIN_DIFF_LEN _MIN_END_LEN)r r
)r r r
r r * s ) tuplemap safe_reprmaxr _MAX_LENGTHr _MIN_BEGIN_LENr _MIN_COMMON_LENAssertionErrorr )argsmaxlenZ
common_lenr
)r r r _common_shorten_repr s
r Fc C sR y t | }W n tk
r* tj| }Y nX | s>t|tk rB|S |d t d S )Nz [truncated]...)repr Exceptionobject__repr__r r )objZshortresultr
r
r r - s r c C s d| j | jf S )Nz%s.%s)
__module__ __qualname__)clsr
r
r strclass6 s r) c C s* d }}g }g }xy| | }|| }||k r\|j | |d7 }x| | |krX|d7 }qBW n||kr|j | |d7 }xf|| |kr|d7 }qxW nL|d7 }z x| | |kr|d7 }qW W d|d7 }x|| |kr|d7 }qW X W q tk
r |j| |d |j||d P Y qX qW ||fS )ar Finds elements in only one or the other of two, sorted input lists.
Returns a two-element tuple of lists. The first list contains those
elements in the "expected" list but not in the "actual" list, and the
second contains those elements in the "actual" list but not in the
"expected" list. Duplicate elements in either input list are ignored.
r N)append
IndexErrorextend)expectedactualijmissingZ
unexpectedear
r
r sorted_list_difference9 s:
r5 c C sL g }x>| rB| j }y|j| W q tk
r> |j| Y qX qW ||fS )zSame behavior as sorted_list_difference but
for lists of unorderable items (like dicts).
As it does a linear search per item (remove) it
has O(n*n) performance.)popremove
ValueErrorr+ )r. r/ r2 itemr
r
r unorderable_list_differenceb s r: c C s | |k| |k S )z.Return -1 if x < y, 0 if x == y and 1 if x > yr
)xyr
r
r
three_way_cmps s r= ZMismatchzactual expected valuec C sD t | t | }}t|t| }}t }g }xt|D ]\}} | |krJq8d }
} x.t||D ] } || | kr^|
d7 }
||| < q^W x,t|D ] \} }
|
| kr| d7 } ||| < qW |
| kr8t|
| | }|j| q8W xlt|D ]`\}} | |krqd} x2t||D ]$} || | k r| d7 } ||| < qW td| | }|j| qW |S )zHReturns list of (cnt_act, cnt_exp, elem) triples where the counts differr r* )listr r" enumeraterange _Mismatchr+ )r/ r. r
tmnZNULLr% r0 elemcnt_scnt_tr1 Z
other_elemdiffr
r
r _count_diff_all_purposey s< rI c C s, t }x | D ]}|j|dd ||< q W |S )z@Return dict of element counts, in the order they were first seenr r* )r get)iterablecrE r
r
r _ordered_count s
rM c C s t | t | }}g }x>|j D ]2\}}|j|d}||kr t|||}|j| q W x2|j D ]&\}}||kr`td||}|j| q`W |S )zHReturns list of (cnt_act, cnt_exp, elem) triples where the counts differr )rM itemsrJ rA r+ ) r/ r. r
rB r% rE rF rG rH r
r
r _count_diff_hashable s rO N)F)__doc__ collectionsr r Zos.pathr Z
__unittestr r r r r r r r r r r) r5 r: r= rA rI rM rO r
r
r
r s,
)
#PK ! u __init__.cpython-36.opt-1.pycnu [ 3
_Vjk @ s d Z ddddddddd d
d d d
dddddgZejdddg dZddlmZ ddlmZmZm Z m
Z
m Z m Z m
Z
mZ ddlmZmZ ddlmZmZmZmZmZ dd lmZmZ dd lmZm Z dd l m Z m Z m Z m!Z! e Z"d d Z#d!S )"a
Python unit testing framework, based on Erich Gamma's JUnit and Kent Beck's
Smalltalk testing framework (used with permission).
This module contains the core framework classes that form the basis of
specific test cases and suites (TestCase, TestSuite etc.), and also a
text-based utility class for running the tests and reporting the results
(TextTestRunner).
Simple usage:
import unittest
class IntegerArithmeticTestCase(unittest.TestCase):
def testAdd(self): # test method names begin with 'test'
self.assertEqual((1 + 2), 3)
self.assertEqual(0 + 1, 1)
def testMultiply(self):
self.assertEqual((0 * 10), 0)
self.assertEqual((5 * 8), 40)
if __name__ == '__main__':
unittest.main()
Further information is available in the bundled documentation, and from
http://docs.python.org/library/unittest.html
Copyright (c) 1999-2003 Steve Purcell
Copyright (c) 2003-2010 Python Software Foundation
This module is free software, and you may redistribute it and/or modify
it under the same terms as Python itself, so long as this copyright message
and disclaimer are retained in their original form.
IN NO EVENT SHALL THE AUTHOR BE LIABLE TO ANY PARTY FOR DIRECT, INDIRECT,
SPECIAL, INCIDENTAL, OR CONSEQUENTIAL DAMAGES ARISING OUT OF THE USE OF
THIS CODE, EVEN IF THE AUTHOR HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH
DAMAGE.
THE AUTHOR SPECIFICALLY DISCLAIMS ANY WARRANTIES, INCLUDING, BUT NOT
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
PARTICULAR PURPOSE. THE CODE PROVIDED HEREUNDER IS ON AN "AS IS" BASIS,
AND THERE IS NO OBLIGATION WHATSOEVER TO PROVIDE MAINTENANCE,
SUPPORT, UPDATES, ENHANCEMENTS, OR MODIFICATIONS.
TestResultTestCase TestSuiteTextTestRunner
TestLoaderFunctionTestCasemaindefaultTestLoaderSkipTestskipskipIf
skipUnlessexpectedFailureTextTestResultinstallHandlerregisterResult removeResult
removeHandlergetTestCaseNames makeSuite
findTestCasesT )r )r r r r
r r r
_skipInRpmBuild)
BaseTestSuiter )r r r r r ) TestProgramr )r r )r r r r c C s" dd l }|jjt}| j||dS )N )Z start_dirpattern)Zos.pathpathdirname__file__Zdiscover)loaderZtestsr osZthis_dir r! )/usr/lib64/python3.6/unittest/__init__.py
load_testsK s r# N)$__doc____all__extendZ
__unittestresultr Zcaser r r r
r r r
r Zsuiter r r r r r r r r r Zrunnerr r Zsignalsr r r r Z_TextTestResultr# r! r! r! r" - s
( PK ! kR __init__.cpython-36.opt-2.pycnu [ 3
_Vjk @ s d ddddddddd d
d d d
ddddgZ e jdddg dZddlmZ ddlmZmZmZm Z m
Z
m Z m Z m
Z
ddlmZmZ ddlmZmZmZmZmZ ddlmZmZ dd lmZmZ dd l m Z m Z m Z m Z eZ!d d Z"d S )!
TestResultTestCase TestSuiteTextTestRunner
TestLoaderFunctionTestCasemaindefaultTestLoaderSkipTestskipskipIf
skipUnlessexpectedFailureTextTestResultinstallHandlerregisterResult removeResult
removeHandlergetTestCaseNames makeSuite
findTestCasesT )r )r r r r
r r r
_skipInRpmBuild)
BaseTestSuiter )r r r r r ) TestProgramr )r r )r r r r c C s" dd l }|jjt}| j||dS )N )Z start_dirpattern)Zos.pathpathdirname__file__Zdiscover)loaderZtestsr osZthis_dir r! )/usr/lib64/python3.6/unittest/__init__.py
load_testsK s r# N)#__all__extendZ
__unittestresultr Zcaser r r r
r r r
r Zsuiter r r r r r r r r r Zrunnerr r Zsignalsr r r r Z_TextTestResultr# r! r! r! r" / s
( PK ! u __init__.cpython-36.pycnu [ 3
_Vjk @ s d Z ddddddddd d
d d d
dddddgZejdddg dZddlmZ ddlmZmZm Z m
Z
m Z m Z m
Z
mZ ddlmZmZ ddlmZmZmZmZmZ dd lmZmZ dd lmZm Z dd l m Z m Z m Z m!Z! e Z"d d Z#d!S )"a
Python unit testing framework, based on Erich Gamma's JUnit and Kent Beck's
Smalltalk testing framework (used with permission).
This module contains the core framework classes that form the basis of
specific test cases and suites (TestCase, TestSuite etc.), and also a
text-based utility class for running the tests and reporting the results
(TextTestRunner).
Simple usage:
import unittest
class IntegerArithmeticTestCase(unittest.TestCase):
def testAdd(self): # test method names begin with 'test'
self.assertEqual((1 + 2), 3)
self.assertEqual(0 + 1, 1)
def testMultiply(self):
self.assertEqual((0 * 10), 0)
self.assertEqual((5 * 8), 40)
if __name__ == '__main__':
unittest.main()
Further information is available in the bundled documentation, and from
http://docs.python.org/library/unittest.html
Copyright (c) 1999-2003 Steve Purcell
Copyright (c) 2003-2010 Python Software Foundation
This module is free software, and you may redistribute it and/or modify
it under the same terms as Python itself, so long as this copyright message
and disclaimer are retained in their original form.
IN NO EVENT SHALL THE AUTHOR BE LIABLE TO ANY PARTY FOR DIRECT, INDIRECT,
SPECIAL, INCIDENTAL, OR CONSEQUENTIAL DAMAGES ARISING OUT OF THE USE OF
THIS CODE, EVEN IF THE AUTHOR HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH
DAMAGE.
THE AUTHOR SPECIFICALLY DISCLAIMS ANY WARRANTIES, INCLUDING, BUT NOT
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
PARTICULAR PURPOSE. THE CODE PROVIDED HEREUNDER IS ON AN "AS IS" BASIS,
AND THERE IS NO OBLIGATION WHATSOEVER TO PROVIDE MAINTENANCE,
SUPPORT, UPDATES, ENHANCEMENTS, OR MODIFICATIONS.
TestResultTestCase TestSuiteTextTestRunner
TestLoaderFunctionTestCasemaindefaultTestLoaderSkipTestskipskipIf
skipUnlessexpectedFailureTextTestResultinstallHandlerregisterResult removeResult
removeHandlergetTestCaseNames makeSuite
findTestCasesT )r )r r r r
r r r
_skipInRpmBuild)
BaseTestSuiter )r r r r r ) TestProgramr )r r )r r r r c C s" dd l }|jjt}| j||dS )N )Z start_dirpattern)Zos.pathpathdirname__file__Zdiscover)loaderZtestsr osZthis_dir r! )/usr/lib64/python3.6/unittest/__init__.py
load_testsK s r# N)$__doc____all__extendZ
__unittestresultr Zcaser r r r
r r r
r Zsuiter r r r r r r r r r Zrunnerr r Zsignalsr r r r Z_TextTestResultr# r! r! r! r" - s
( PK ! v? __main__.cpython-36.opt-1.pycnu [ 3
\ @ sd d Z ddlZejd jdrBddlZejjejZed ejd< [dZ ddl
m
Z
m Z e
dd dS ) zMain entry point Nz __main__.pyz -m unittestT )main TestProgram)module) __doc__sysargvendswithZos.pathospathbasename
executableZ
__unittestr r r r )/usr/lib64/python3.6/unittest/__main__.py s PK ! ݥ~ ~ __main__.cpython-36.opt-2.pycnu [ 3
\ @ s` d dl Z e jd jdr>d dlZejje jZed e jd < [dZddl m Z m
Z
e dd dS ) Nz __main__.pyz -m unittestT )main TestProgram)module) sysargvendswithZos.pathospathbasename
executableZ
__unittestr r r
r
)/usr/lib64/python3.6/unittest/__main__.py s PK ! v? __main__.cpython-36.pycnu [ 3
\ @ sd d Z ddlZejd jdrBddlZejjejZed ejd< [dZ ddl
m
Z
m Z e
dd dS ) zMain entry point Nz __main__.pyz -m unittestT )main TestProgram)module) __doc__sysargvendswithZos.pathospathbasename
executableZ
__unittestr r r r )/usr/lib64/python3.6/unittest/__main__.py s PK ! 0Ļ Ļ case.cpython-36.opt-1.pycnu [ 3
_Vj @ s d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl m
Z
ddlmZmZmZmZmZ dZe ZdZG dd d eZG d
d d eZG d d
d
eZG dd deZ dd Z dd Z dd Z dd Z dd Z!dd Z"G d d d Z#d d Z$G d d! d!e#Z%G d"d# d#e%Z&G d$d% d%e%Z'e j(d&d'd(gZ)G d)d* d*ej*Z+G d+d, d,e#Z,G d-d. d.eZ-G d/d0 d0e-Z.G d1d2 d2e-Z/dS )3zTest case implementation N )result)strclass safe_repr_count_diff_all_purpose_count_diff_hashable_common_shorten_reprTz@
Diff is %s characters long. Set self.maxDiff to None to see it.c @ s e Zd ZdZdS )SkipTestz
Raise this exception in a test to skip it.
Usually you can use TestCase.skipTest() or one of the skipping decorators
instead of raising this directly.
N)__name__
__module__ __qualname____doc__ r r %/usr/lib64/python3.6/unittest/case.pyr s r c @ s e Zd ZdZdS ) _ShouldStopz
The test should stop.
N)r
r r r
r r r r r " s r c @ s e Zd ZdZdS )_UnexpectedSuccessz7
The test was supposed to fail, but it didn't!
N)r
r r r
r r r r r ' s r c @ s& e Zd ZdddZejdddZdS ) _OutcomeNc C s4 d| _ || _t|d| _d| _g | _d | _g | _d S )NF
addSubTestT)expecting_failurer hasattrresult_supports_subtestssuccessskippedexpectedFailureerrors)selfr r r r __init__. s z_Outcome.__init__Fc c s | j }d| _ zy
d V W n tk
r. Y n tk
rh } z d| _ | jj|t|f W Y d d }~X nj tk
rz Y nX tj }| j r|| _
nd| _ | j j||f d }Y n X | j r| j r| j j|d f W d | j o|| _ X d S )NTF)
r KeyboardInterruptr r appendstrr sysexc_infor r r r )r test_caseisTestZ old_successer! r r r testPartExecutor7 s*
$
z_Outcome.testPartExecutor)N)F)r
r r r
contextlibcontextmanagerr% r r r r r - s
r c C s | S )Nr )objr r r _idU s r) c s fdd}|S )z&
Unconditionally skip a test.
c s4 t | ts$tj| fdd}|} d| _ | _| S )Nc s t d S )N)r )argskwargs)reasonr r skip_wrapper^ s z-skip..decorator..skip_wrapperT)
isinstancetype functoolswraps__unittest_skip____unittest_skip_why__) test_itemr- )r, r r decorator\ s
zskip..decoratorr )r, r5 r )r, r skipX s
r6 c C s | r t |S tS )z/
Skip a test if the condition is true.
)r6 r) ) conditionr, r r r skipIfh s r8 c C s | s t |S tS )z3
Skip a test unless the condition is true.
)r6 r) )r7 r, r r r
skipUnlessp s r9 c C s
d| _ | S )NT) __unittest_expecting_failure__)r4 r r r r x s r c s4 t | tr t fdd| D S t | to2t| S )Nc 3 s | ]}t | V qd S )N) _is_subtype).0r$ )basetyper r ~ s z _is_subtype..)r. tupleallr/
issubclass)expectedr= r )r= r r; | s
r; c @ s e Zd Zdd Zdd ZdS )_BaseTestCaseContextc C s
|| _ d S )N)r" )r r" r r r r s z _BaseTestCaseContext.__init__c C s | j j| j|}| j j|d S )N)r" _formatMessagemsgfailureException)r standardMsgrE r r r
_raiseFailure s z"_BaseTestCaseContext._raiseFailureN)r
r r r rH r r r r rC s rC c C s dt jkrt| S tS dS )a
Non-standard/downstream-only decorator for marking a specific unit test
to be skipped when run within the %check of an rpmbuild.
Specifically, this takes effect when WITHIN_PYTHON_RPM_BUILD is set within
the environment, and has no effect otherwise.
ZWITHIN_PYTHON_RPM_BUILDN)osenvironr6 r) )r, r r r _skipInRpmBuild s
rK c @ s e Zd ZdddZdd ZdS )_AssertRaisesBaseContextNc C s@ t j| | || _|| _|d k r*tj|}|| _d | _d | _d S )N) rC r rB r" recompileexpected_regexobj_namerE )r rB r" rO r r r r s
z!_AssertRaisesBaseContext.__init__c C s zt | j| js"td|| jf |rD|d dkrDtjdtd f }|sx|jdd| _ |rttjdt
t | td | S |^}}y |j | _
W n tk
r t|| _
Y nX | ||| W dQ R X W dd} X dS )z
If args is empty, assertRaises/Warns is being used as a
context manager, so check for a 'msg' kwarg and return self.
If args is not empty, call a callable passing positional and keyword
arguments.
z%s() arg 1 must be %sr Nzcallable is None rE z3%r is an invalid keyword argument for this function)r; rB
_base_type TypeError_base_type_strwarningswarnDeprecationWarningpoprE nextiterr
rP AttributeErrorr )r namer* r+ Z callable_objr r r handle s. z _AssertRaisesBaseContext.handle)N)r
r r r r] r r r r rL s
rL c @ s( e Zd ZdZeZdZdd Zdd ZdS )_AssertRaisesContextzCA context manager used to implement TestCase.assertRaises* methods.z-an exception type or tuple of exception typesc C s | S )Nr )r r r r __enter__ s z _AssertRaisesContext.__enter__c
C s |d krby | j j}W n tk
r2 t| j }Y nX | jrP| jdj|| j ql| jdj| n
tj| t || j s|dS |j
d | _ | j d krdS | j }|j
t|s| jdj|jt| dS )Nz{} not raised by {}z
{} not raisedFTz"{}" does not match "{}")rB r
r[ r rP rH format traceback clear_framesrA with_tracebackZ exceptionrO searchpattern)r exc_type exc_valuetbexc_namerO r r r __exit__ s(
z _AssertRaisesContext.__exit__N) r
r r r
BaseExceptionrR rT r_ rj r r r r r^ s
r^ c @ s( e Zd ZdZeZdZdd Zdd ZdS )_AssertWarnsContextzBA context manager used to implement TestCase.assertWarns* methods.z(a warning type or tuple of warning typesc C sR x$t jj D ]}t|dd r i |_q W tjdd| _| jj | _tj d| j
| S )N__warningregistry__T)recordalways) r modulesvaluesgetattrrm rU catch_warningswarnings_managerr_ simplefilterrB )r vr r r r_ s
z _AssertWarnsContext.__enter__c
C s | j j||| |d k r d S y | jj}W n tk
rF t| j}Y nX d }xd| jD ]Z}|j}t|| jslqT|d krx|}| j d k r| j j
t| rqT|| _ |j | _ |j
| _
d S W |d k r| jdj| j jt| | jr| jdj|| j n| jdj| d S )Nz"{}" does not match "{}"z{} not triggered by {}z{} not triggered)rt rj rB r
r[ r rU messager. rO rd ZwarningfilenamelinenorH r` re rP )r rf rg rh ri Zfirst_matchingmwr r r rj s8
z _AssertWarnsContext.__exit__N) r
r r r
WarningrR rT r_ rj r r r r rl s
rl _LoggingWatcherrecordsoutputc @ s( e Zd ZdZdd Zdd Zdd ZdS ) _CapturingHandlerzM
A logging handler capturing all (raw and formatted) logging output.
c C s t jj| tg g | _d S )N)loggingHandlerr r} watcher)r r r r r , s z_CapturingHandler.__init__c C s d S )Nr )r r r r flush0 s z_CapturingHandler.flushc C s* | j jj| | j|}| j jj| d S )N)r r~ r r` r )r rn rE r r r emit3 s
z_CapturingHandler.emitN)r
r r r
r r r r r r r r ' s r c @ s, e Zd ZdZdZdd Zdd Zdd Zd S )
_AssertLogsContextz:A context manager used to implement TestCase.assertLogs().z"%(levelname)s:%(name)s:%(message)sc C s: t j| | || _|r(tjj||| _ntj| _d | _d S )N) rC r logger_namer Z _nameToLevelgetlevelINFOrE )r r" r r r r r r ? s z_AssertLogsContext.__init__c C s t | jtjr | j }| _ntj| j }| _tj| j}t }|j | |j
| _
|j d d | _ |j
| _|j| _|g|_ |j| j
d|_|j
S )NF)r. r r ZLoggerloggerZ getLoggerZ FormatterLOGGING_FORMATr Z setFormatterr handlers old_handlersr old_level propagate
old_propagatesetLevel)r r Z formatterZhandlerr r r r_ H s
z _AssertLogsContext.__enter__c C s` | j | j_| j| j_| jj| j |d k r.dS t| jj dkr\| j
dj t j
| j| jj d S )NFr z-no logs of level {} or higher triggered on {})r r r r r r r lenr r~ rH r` r Z getLevelNamer r\ )r rf rg rh r r r rj Y s
z_AssertLogsContext.__exit__N)r
r r r
r r r_ rj r r r r r : s
r c @ s e Zd ZdZeZdZdZdZdZ dd d
Z
d d Z d
d Z dd Z
dd Zedd Zedd Zdd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zej e fd)d*Z d+d, Z d-d. Z d/d0 Z!dd2d3Z"d4d5 Z#d6d7 Z$d8d9 Z%d:d; Z&ddd?Z(dd@dAZ)dBdC Z*dDdE Z+dFdG Z,ddHdIZ-dJdK Z.ddLdMZ/ddNdOZ0ddPdQZ1ddRdSZ2ddTdUZ3ddVdWZ4dXdY Z5ddZd[Z6dd\d]Z7dd^d_Z8dd`daZ9ddbdcZ:ddddeZ;ddfdgZddldmZ?ddndoZ@ddpdqZAddrdsZBddtduZCddvdwZDddxdyZEddzd{ZFdd|d}ZGdd~dZHdd ZIdd ZJdddZKdddZLdd ZMeMe0 ZNZOeMe1 ZPZQeMe2 ZRZSeMe3 ZTZUeMe) ZVZWeMe+ZXeMe(ZYeMeIZZeMeKZ[eMeLZ\d1S )TestCaseaW A class whose instances are single test cases.
By default, the test code itself should be placed in a method named
'runTest'.
If the fixture may be used for many test cases, create as
many test methods as are needed. When instantiating such a TestCase
subclass, specify in the constructor arguments the name of the test method
that the instance is to execute.
Test authors should subclass TestCase for their own tests. Construction
and deconstruction of the test's environment ('fixture') can be
implemented by overriding the 'setUp' and 'tearDown' methods respectively.
If it is necessary to override the __init__ method, the base class
__init__ method must always be called. It is important that subclasses
should not change the signature of their __init__ method, since instances
of the classes are instantiated automatically by parts of the framework
in order to be run.
When subclassing TestCase, you can set these attributes:
* failureException: determines which exception will be raised when
the instance's assertion methods fail; test methods raising this
exception will be deemed to have 'failed' rather than 'errored'.
* longMessage: determines whether long messages (including repr of
objects used in assert methods) will be printed on failure in *addition*
to any explicit message passed.
* maxDiff: sets the maximum length of a diff in failure messages
by assert methods using difflib. It is looked up as an instance
attribute so can be configured by individual tests if required.
TP FrunTestc C s || _ d| _d| _yt| |}W n. tk
rN |dkrJtd| j|f Y n
X |j| _g | _d| _ i | _
| j t d | j t
d | j td | j td | j td | j td dS )
zCreate an instance of the class that will use the named test
method when executed. Raises a ValueError if the instance does
not have a method with the specified name.
NzNo testr z no such test method in %s: %sassertDictEqualassertListEqualassertTupleEqualassertSetEqualassertMultiLineEqual)_testMethodName_outcome_testMethodDocrr r[
ValueError __class__r
_cleanups_subtest_type_equality_funcsaddTypeEqualityFuncdictlistr? set frozensetr )r Z
methodName
testMethodr r r r s& zTestCase.__init__c C s || j |< dS )a[ Add a type specific assertEqual style function to compare a type.
This method is for use by TestCase subclasses that need to register
their own type equality functions to provide nicer error messages.
Args:
typeobj: The data type to call this function on when both values
are of the same type in assertEqual().
function: The callable taking two arguments and an optional
msg= argument that raises self.failureException with a
useful error message when the two arguments are not equal.
N)r )r Ztypeobjfunctionr r r r s
z TestCase.addTypeEqualityFuncc O s | j j|||f dS )a Add a function, with arguments, to be called when the test is
completed. Functions added are called on a LIFO basis and are
called after tearDown on test failure or success.
Cleanup items are called even if setUp fails (unlike tearDown).N)r r )r r r* r+ r r r
addCleanup s zTestCase.addCleanupc C s dS )zAHook method for setting up the test fixture before exercising it.Nr )r r r r setUp s zTestCase.setUpc C s dS )zAHook method for deconstructing the test fixture after testing it.Nr )r r r r tearDown s zTestCase.tearDownc C s dS )zKHook method for setting up class fixture before running tests in the class.Nr )clsr r r
setUpClass s zTestCase.setUpClassc C s dS )zVHook method for deconstructing the class fixture after running all tests in the class.Nr )r r r r
tearDownClass s zTestCase.tearDownClassc C s dS )Nr r )r r r r countTestCases s zTestCase.countTestCasesc C s t j S )N)r Z
TestResult)r r r r defaultTestResult s zTestCase.defaultTestResultc C s | j }|r |jdd j p dS )zReturns a one-line description of the test, or None if no
description has been provided.
The default implementation of this method returns the first line of
the specified test method's docstring.
r N)r splitstrip)r docr r r shortDescription s zTestCase.shortDescriptionc C s dt | j| jf S )Nz%s.%s)r r r )r r r r id s z TestCase.idc C s t | t |k rtS | j|jkS )N)r/ NotImplementedr )r otherr r r __eq__ s zTestCase.__eq__c C s t t| | jfS )N)hashr/ r )r r r r __hash__ s zTestCase.__hash__c C s d| j t| jf S )Nz%s (%s))r r r )r r r r __str__ s zTestCase.__str__c C s dt | j| jf S )Nz<%s testMethod=%s>)r r r )r r r r __repr__ s zTestCase.__repr__c C s< t |dd }|d k r ||| ntjdtd |j| d S )NaddSkipz4TestResult has no addSkip method, skips not reportedr )rr rU rV RuntimeWarning
addSuccess)r r r" r, r r r r _addSkip s zTestCase._addSkipc k s | j dks| j j r dV dS | j}|dkr8tj|}n |jj|}t| ||| _zX| j j| jdd dV W dQ R X | j j s| j j
}|dk r|j rt n | j j
rt W d|| _X dS )aP Return a context manager that will return the enclosed block
of code in a subtest identified by the optional message and
keyword parameters. A failure in the subtest marks the test
case as failed but resumes execution at the end of the enclosed
block, allowing further test code to be executed.
NT)r# )r r r collectionsChainMapparams new_child_SubTestr% r r Zfailfastr r )r rE r parentZ
params_mapr r r r subTest s$ zTestCase.subTestc C sd x^|D ]V\}}t |tr*|j|j|| q|d k rt|d | jrP|j|| q|j|| qW d S )Nr )r. r r r" rA rF
addFailureZaddError)r r r testr! r r r _feedErrorsToResult( s
z TestCase._feedErrorsToResultc C sD y
|j }W n* tk
r4 tjdt |j| Y n X || | d S )Nz@TestResult has no addExpectedFailure method, reporting as passes)addExpectedFailurer[ rU rV r r )r r r! r r r r _addExpectedFailure2 s
z TestCase._addExpectedFailurec C sh y
|j }W nP tk
rZ tjdt y
td W n$ tk
rT |j| tj Y nX Y n
X || d S )NzCTestResult has no addUnexpectedSuccess method, reporting as failure) addUnexpectedSuccessr[ rU rV r r r r r! )r r r r r r _addUnexpectedSuccess< s
z TestCase._addUnexpectedSuccessNc
( C s |}|d kr.| j }t|dd }|d k r.| |j| t| | j}t| jdds^t|ddrz,t| jddpxt|dd}| j|| | W d |j| X d S t|dd}t| dd}|p|}t|} z| | _| j | | j
W d Q R X | j r<|| _ | j | dd | W d Q R X d| _ | j | | j
W d Q R X | j x"| jD ]\}
} | j||
| qLW | j|| j | j r|r| jr| j|| j n
| j| n
|j| |S |j| |d krt|d d } | d k r| | jj d | _d | _X d S )
N startTestRunr2 Fr3 r: T)r# stopTestRun)r rr Z startTestr r r ZstopTestr r r% r r r r
doCleanupsr r r r r r r clear)
r r Z orig_resultr r Zskip_whyZexpecting_failure_methodZexpecting_failure_classr outcomer r, r r r r runK sh
z TestCase.runc
C sN | j p
t }x:| jrF| jj \}}}|j| ||| W dQ R X qW |jS )zNExecute all cleanup functions. Normally called for you after
tearDown.N)r r r rX r% r )r r r r* r+ r r r r s zTestCase.doCleanupsc O s | j ||S )N)r )r r* kwdsr r r __call__ s zTestCase.__call__c C sJ | j t| | j | j x&| jrD| jjd\}}}||| q W dS )z6Run the test without collecting errors in a TestResultr N)r rr r r r rX )r r r* r+ r r r debug s zTestCase.debugc C s t |dS )zSkip this test.N)r )r r, r r r skipTest s zTestCase.skipTestc C s | j |dS )z)Fail immediately, with the given message.N)rF )r rE r r r fail s z
TestCase.failc C s& |r"| j |dt| }| j|dS )z#Check that the expression is false.z%s is not falseN)rD r rF )r exprrE r r r assertFalse s zTestCase.assertFalsec C s& |s"| j |dt| }| j|dS )z"Check that the expression is true.z%s is not trueN)rD r rF )r r rE r r r
assertTrue s zTestCase.assertTruec C sP | j s|p |S |dkr|S y d||f S tk
rJ dt|t|f S X dS )a Honour the longMessage attribute when generating failure messages.
If longMessage is False this means:
* Use only an explicit message if it is provided
* Otherwise use the standard message for the assert
If longMessage is True:
* Use the standard message
* If an explicit message is provided, plus ' : ' and the explicit message
Nz%s : %s) longMessageUnicodeDecodeErrorr )r rE rG r r r rD s
zTestCase._formatMessagec
O s$ t || }z|jd||S d}X dS )a= Fail unless an exception of class expected_exception is raised
by the callable when invoked with specified positional and
keyword arguments. If a different type of exception is
raised, it will not be caught, and the test case will be
deemed to have suffered an error, exactly as for an
unexpected exception.
If called with the callable and arguments omitted, will return a
context object used like this::
with self.assertRaises(SomeException):
do_something()
An optional keyword argument 'msg' can be provided when assertRaises
is used as a context object.
The context manager keeps a reference to the exception as
the 'exception' attribute. This allows you to inspect the
exception after the assertion::
with self.assertRaises(SomeException) as cm:
do_something()
the_exception = cm.exception
self.assertEqual(the_exception.error_code, 3)
assertRaisesN)r^ r] )r expected_exceptionr* r+ contextr r r r s
zTestCase.assertRaisesc O s t || }|jd||S )a Fail unless a warning of class warnClass is triggered
by the callable when invoked with specified positional and
keyword arguments. If a different type of warning is
triggered, it will not be handled: depending on the other
warning filtering rules in effect, it might be silenced, printed
out, or raised as an exception.
If called with the callable and arguments omitted, will return a
context object used like this::
with self.assertWarns(SomeWarning):
do_something()
An optional keyword argument 'msg' can be provided when assertWarns
is used as a context object.
The context manager keeps a reference to the first matching
warning as the 'warning' attribute; similarly, the 'filename'
and 'lineno' attributes give you information about the line
of Python code from which the warning was triggered.
This allows you to inspect the warning after the assertion::
with self.assertWarns(SomeWarning) as cm:
do_something()
the_warning = cm.warning
self.assertEqual(the_warning.some_attribute, 147)
assertWarns)rl r] )r expected_warningr* r+ r r r r r s
zTestCase.assertWarnsc C s t | ||S )a Fail unless a log message of level *level* or higher is emitted
on *logger_name* or its children. If omitted, *level* defaults to
INFO and *logger* defaults to the root logger.
This method must be used as a context manager, and will yield
a recording object with two attributes: `output` and `records`.
At the end of the context manager, the `output` attribute will
be a list of the matching formatted log messages and the
`records` attribute will be a list of the corresponding LogRecord
objects.
Example::
with self.assertLogs('foo', level='INFO') as cm:
logging.getLogger('foo').info('first message')
logging.getLogger('foo.bar').error('second message')
self.assertEqual(cm.output, ['INFO:foo:first message',
'ERROR:foo.bar:second message'])
)r )r r r r r r
assertLogs s zTestCase.assertLogsc C sF t |t |kr@| jjt |}|dk r@t|trt|| |kr>dS dt|t|t|f }nF|dkrfd}||k rtt|| |dkrdS dt|t||f }| j||}| j|dS )a Fail if the two objects are equal as determined by their
difference rounded to the given number of decimal places
(default 7) and comparing to zero, or by comparing that the
difference between the two objects is less than the given delta.
Note that decimal places (from zero) are usually not the same
as significant digits (measured from the most significant digit).
Objects that are equal automatically fail.
Nz specify delta or places not bothz%s == %s within %s deltar r z%s == %s within %r places)rS r r r rD rF )r r r r rE r rG r r r assertNotAlmostEqual s" z TestCase.assertNotAlmostEqualc ' C s |dk rP|j }t||s.| jd|t|f t||sT| jd|t|f nd}d}y t|}W n ttfk
r d| }Y nX |dkry t|}W n ttfk
r d| }Y nX |dkr||krdS d|j ft|| }xt t
||D ]} y || }
W n. tt tfk
r8 |d| |f 7 }P Y nX y || } W n. tt tfk
rt |d | |f 7 }P Y nX |
| kr|d
| ft|
| 7 }P qW ||kr|dkrt |t |krdS ||kr0|d ||| f 7 }y |d |t|| f 7 }W n, tt tfk
r, |d
||f 7 }Y nX nh||k r|d||| f 7 }y |d |t|| f 7 }W n, tt tfk
r |d||f 7 }Y nX |} ddj
tjtj|j tj|j }
| j| |
} | j|| }| j| dS )aA An equality assertion for ordered sequences (like lists and tuples).
For the purposes of this function, a valid ordered sequence type is one
which can be indexed, has a length, and has an equality operator.
Args:
seq1: The first sequence to compare.
seq2: The second sequence to compare.
seq_type: The expected datatype of the sequences, or None if no
datatype should be enforced.
msg: Optional message to use on failure instead of a list of
differences.
Nz First sequence is not a %s: %sz Second sequence is not a %s: %sZsequencez(First %s has no length. Non-sequence?z)Second %s has no length. Non-sequence?z%ss differ: %s != %s
z(
Unable to index element %d of first %s
z)
Unable to index element %d of second %s
z#
First differing element %d:
%s
%s
z+
First %s contains %d additional elements.
zFirst extra element %d:
%s
z'Unable to index element %d of first %s
z,
Second %s contains %d additional elements.
z(Unable to index element %d of second %s
r )r
r. rF r r rS NotImplementedError
capitalizer rangemin
IndexErrorr/ joindifflibndiffpprintpformat
splitlines_truncateMessagerD r )r Zseq1Zseq2rE seq_typeZ
seq_type_nameZ differingZlen1Zlen2iZitem1Zitem2rG diffMsgr r r assertSequenceEqual s
z TestCase.assertSequenceEqualc C s2 | j }|d kst||kr"|| S |tt| S )N)maxDiffr DIFF_OMITTED)r rw diffZmax_diffr r r r s zTestCase._truncateMessagec C s | j |||td dS )a A list-specific equality assertion.
Args:
list1: The first list to compare.
list2: The second list to compare.
msg: Optional message to use on failure instead of a list of
differences.
)r N)r r )r Zlist1Zlist2rE r r r r s
zTestCase.assertListEqualc C s | j |||td dS )a A tuple-specific equality assertion.
Args:
tuple1: The first tuple to compare.
tuple2: The second tuple to compare.
msg: Optional message to use on failure instead of a list of
differences.
)r N)r r? )r Ztuple1Ztuple2rE r r r r s zTestCase.assertTupleEqualc
C sh y|j |}W n^ tk
r> } z| jd| W Y dd}~X n0 tk
rl } z| jd| W Y dd}~X nX y|j |}W n^ tk
r } z| jd| W Y dd}~X n0 tk
r } z| jd| W Y dd}~X nX |p|sdS g }|r|jd x |D ]}|jt| qW |rH|jd x |D ]}|jt| q0W dj|} | j| j|| dS )a A set-specific equality assertion.
Args:
set1: The first set to compare.
set2: The second set to compare.
msg: Optional message to use on failure instead of a list of
differences.
assertSetEqual uses ducktyping to support different types of sets, and
is optimized for sets specifically (parameters must support a
difference method).
z/invalid type when attempting set difference: %sNz2first argument does not support set difference: %sz3second argument does not support set difference: %sz*Items in the first set but not the second:z*Items in the second set but not the first:r )
differencerS r r[ r reprr rD )
r Zset1Zset2rE Z difference1r$ Z difference2linesitemrG r r r r " s2
zTestCase.assertSetEqualc C s2 ||kr.dt |t |f }| j| j|| dS )zDJust like self.assertTrue(a in b), but with a nicer default message.z%s not found in %sN)r r rD )r member containerrE rG r r r assertInM s zTestCase.assertInc C s2 ||kr.dt |t |f }| j| j|| dS )zHJust like self.assertTrue(a not in b), but with a nicer default message.z%s unexpectedly found in %sN)r r rD )r r r rE rG r r r assertNotInT s zTestCase.assertNotInc C s2 ||k r.dt |t |f }| j| j|| dS )zDJust like self.assertTrue(a is b), but with a nicer default message.z %s is not %sN)r r rD )r expr1expr2rE rG r r r assertIs[ s zTestCase.assertIsc C s, ||kr(dt |f }| j| j|| dS )zHJust like self.assertTrue(a is not b), but with a nicer default message.zunexpectedly identical: %sN)r r rD )r r r rE rG r r r assertIsNotb s zTestCase.assertIsNotc C s~ | j |td | j |td ||krzdt|| }ddjtjtj|j tj|j }| j ||}| j
| j || d S )Nz"First argument is not a dictionaryz#Second argument is not a dictionaryz%s != %sr ) assertIsInstancer r r r r r r r r r rD )r Zd1Zd2rE rG r r r r r h s
zTestCase.assertDictEqualc C s t jdt g }g }xX|j D ]L\}}||kr:|j| q ||| kr |jdt|t|t|| f q W |pt|szdS d}|rddjdd |D }|r|r|d 7 }|d
dj| 7 }| j| j|| dS ) z2Checks whether dictionary is a superset of subset.z&assertDictContainsSubset is deprecatedz %s, expected: %s, actual: %sNr z Missing: %s,c s s | ]}t |V qd S )N)r )r< rz r r r r> s z4TestCase.assertDictContainsSubset..z; zMismatched values: %s) rU rV rW itemsr r r r rD ) r ZsubsetZ
dictionaryrE ZmissingZ
mismatchedkeyvaluerG r r r assertDictContainsSubsett s, z!TestCase.assertDictContainsSubsetc
C s t |t | }}ytj|}tj|}W n tk
rH t||}Y nX ||krVdS t||}|rd}dd |D }dj|} | j|| }| j||}| j | dS )a An unordered sequence comparison asserting that the same elements,
regardless of order. If the same element occurs more than once,
it verifies that the elements occur the same number of times.
self.assertEqual(Counter(list(first)),
Counter(list(second)))
Example:
- [0, 1, 1] and [1, 0, 1] compare equal.
- [0, 0, 1] and [0, 1] compare unequal.
Nz Element counts were not equal:
c S s g | ] }d | qS )z First has %d, Second has %d: %rr )r< r r r r
s z-TestCase.assertCountEqual..r )
r r CounterrS r r r r rD r )
r r r rE Z first_seqZ
second_seqZ differencesrG r r r r r assertCountEqual s
zTestCase.assertCountEqualc C s | j |td | j |td ||krt|| jks@t|| jkrN| j||| |jdd}|jdd}t|dkr|jd|kr|d g}|d g}dt|| }dd jt j
|| }| j ||}| j | j
|| d
S ) z-Assert that two multi-line strings are equal.z First argument is not a stringz Second argument is not a stringT)keependsr z
r z%s != %sr N)r r r _diffThresholdr r r r r r r r r rD )r r r rE Z
firstlinesZ secondlinesrG r r r r r s
z TestCase.assertMultiLineEqualc C s2 ||k s.dt |t |f }| j| j|| dS )zCJust like self.assertTrue(a < b), but with a nicer default message.z%s not less than %sN)r r rD )r abrE rG r r r
assertLess s zTestCase.assertLessc C s2 ||ks.dt |t |f }| j| j|| dS )zDJust like self.assertTrue(a <= b), but with a nicer default message.z %s not less than or equal to %sN)r r rD )r r$ r% rE rG r r r assertLessEqual s zTestCase.assertLessEqualc C s2 ||ks.dt |t |f }| j| j|| dS )zCJust like self.assertTrue(a > b), but with a nicer default message.z%s not greater than %sN)r r rD )r r$ r% rE rG r r r
assertGreater s zTestCase.assertGreaterc C s2 ||ks.dt |t |f }| j| j|| dS )zDJust like self.assertTrue(a >= b), but with a nicer default message.z"%s not greater than or equal to %sN)r r rD )r r$ r% rE rG r r r assertGreaterEqual s zTestCase.assertGreaterEqualc C s, |dk r(dt |f }| j| j|| dS )zCSame as self.assertTrue(obj is None), with a nicer default message.Nz%s is not None)r r rD )r r( rE rG r r r assertIsNone s zTestCase.assertIsNonec C s" |dkr d}| j | j|| dS )z(Included for symmetry with assertIsNone.Nzunexpectedly None)r rD )r r( rE rG r r r assertIsNotNone s zTestCase.assertIsNotNonec C s0 t ||s,dt||f }| j| j|| dS )zTSame as self.assertTrue(isinstance(obj, cls)), with a nicer
default message.z%s is not an instance of %rN)r. r r rD )r r( r rE rG r r r r s
zTestCase.assertIsInstancec C s0 t ||r,dt||f }| j| j|| dS )z,Included for symmetry with assertIsInstance.z%s is an instance of %rN)r. r r rD )r r( r rE rG r r r assertNotIsInstance s
z TestCase.assertNotIsInstancec O s t || |}|jd||S )a Asserts that the message in a raised exception matches a regex.
Args:
expected_exception: Exception class expected to be raised.
expected_regex: Regex (re pattern object or string) expected
to be found in error message.
args: Function to be called and extra positional args.
kwargs: Extra kwargs.
msg: Optional message used in case of failure. Can only be used
when assertRaisesRegex is used as a context manager.
assertRaisesRegex)r^ r] )r r rO r* r+ r r r r r- s
zTestCase.assertRaisesRegexc O s t || |}|jd||S )a Asserts that the message in a triggered warning matches a regexp.
Basic functioning is similar to assertWarns() with the addition
that only warnings whose messages also match the regular expression
are considered successful matches.
Args:
expected_warning: Warning class expected to be triggered.
expected_regex: Regex (re pattern object or string) expected
to be found in error message.
args: Function to be called and extra positional args.
kwargs: Extra kwargs.
msg: Optional message used in case of failure. Can only be used
when assertWarnsRegex is used as a context manager.
assertWarnsRegex)rl r] )r r rO r* r+ r r r r r. s zTestCase.assertWarnsRegexc C sJ t |ttfrtj|}|j|sFd|j|f }| j||}| j|dS )z=Fail the test unless the text matches the regular expression.z&Regex didn't match: %r not found in %rN) r. r bytesrM rN rd re rD rF )r textrO rE rG r r r assertRegex s
zTestCase.assertRegexc C s` t |ttfrtj|}|j|}|r\d||j |j |j|f }| j ||}| j
|dS )z9Fail the test if the text matches the regular expression.z"Regex matched: %r matches %r in %rN) r. r r/ rM rN rd startendre rD rF )r r0 Zunexpected_regexrE matchrG r r r assertNotRegex& s
zTestCase.assertNotRegexc s fdd}|S )Nc s t jdj jtd | |S )NzPlease use {0} instead.r )rU rV r` r
rW )r* r+ )
original_funcr r deprecated_func6 s
z,TestCase._deprecate..deprecated_funcr )r6 r7 r )r6 r
_deprecate5 s zTestCase._deprecatei i )r )N)N)N)N)NN)N)N)N)NNN)NNN)NN)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)]r
r r r
AssertionErrorrF r r
r# Z_classSetupFailedr r r r r classmethodr r r r r r r r r r r r& r' _subtest_msg_sentinelr r r r r r r r r r r r rD r r r r r r r r r r r r r r r r r r r r r! r r& r' r( r) r* r+ r r, r- r. r1 r5 r8 ZfailUnlessEqualZ assertEqualsZ failIfEqualZassertNotEqualsZfailUnlessAlmostEqualZassertAlmostEqualsZfailIfAlmostEqualZassertNotAlmostEqualsZ
failUnlessZassert_ZfailUnlessRaisesZfailIfZassertRaisesRegexpZassertRegexpMatchesZassertNotRegexpMatchesr r r r r f s
E
!
'
c
+
!
r c sj e Zd ZdZd fdd Zdd Zdd Zd d
Zd d Zd
d Z dd Z
dd Z dd Z dd Z
ZS )FunctionTestCaseaI A test case that wraps a test function.
This is useful for slipping pre-existing test functions into the
unittest framework. Optionally, set-up and tidy-up functions can be
supplied. As with TestCase, the tidy-up ('tearDown') function will
always be called if the set-up ('setUp') function ran successfully.
Nc s* t t| j || _|| _|| _|| _d S )N)superr< r
_setUpFunc
_tearDownFunc _testFunc _description)r ZtestFuncr r Z description)r r r r T s
zFunctionTestCase.__init__c C s | j d k r| j d S )N)r> )r r r r r [ s
zFunctionTestCase.setUpc C s | j d k r| j d S )N)r? )r r r r r _ s
zFunctionTestCase.tearDownc C s | j d S )N)r@ )r r r r r c s zFunctionTestCase.runTestc C s | j jS )N)r@ r
)r r r r r f s zFunctionTestCase.idc C s@ t || jstS | j|jko>| j|jko>| j|jko>| j|jkS )N)r. r r r> r? r@ rA )r r r r r r i s zFunctionTestCase.__eq__c C s t t| | j| j| j| jfS )N)r r/ r> r? r@ rA )r r r r r r s zFunctionTestCase.__hash__c C s dt | j| jjf S )Nz%s (%s))r r r@ r
)r r r r r v s
zFunctionTestCase.__str__c C s dt | j| jf S )Nz <%s tec=%s>)r r r@ )r r r r r z s
zFunctionTestCase.__repr__c C s2 | j d k r| j S | jj}|r.|jdd j p0d S )Nr r )rA r@ r
r r )r r r r r r ~ s
z!FunctionTestCase.shortDescription)NNN)r
r r r
r r r r r r r r r r
__classcell__r r )r r r< K s r< c sD e Zd Z fddZdd Zdd Zdd Zd d
Zd d Z Z S )
r c s( t j || _|| _|| _|j| _d S )N)r= r _messager" r rF )r r" rw r )r r r r s
z_SubTest.__init__c C s t dd S )Nz subtests cannot be run directly)r )r r r r r s z_SubTest.runTestc C sb g }| j tk r |jdj| j | jrTdjdd t| jj D }|jdj| dj|p`dS )Nz[{}]z, c s s | ]\}}d j ||V qdS )z{}={!r}N)r` )r< krv r r r r> s z+_SubTest._subDescription..z({}) z ())rC r; r r` r r sortedr )r partsZ params_descr r r _subDescription s
z_SubTest._subDescriptionc C s dj | jj | j S )Nz{} {})r` r" r rH )r r r r r s z _SubTest.idc C s
| j j S )zlReturns a one-line description of the subtest, or None if no
description has been provided.
)r" r )r r r r r s z_SubTest.shortDescriptionc C s dj | j| j S )Nz{} {})r` r" rH )r r r r r s z_SubTest.__str__)
r
r r r r rH r r r rB r r )r r r s r )0r
r r0 r rI r r rM rU r r& ra r r utilr r r r r Z
__unittestobjectr; r Exceptionr r r r r) r6 r8 r9 r r; rC rK rL r^ rl
namedtupler} r r r r r< r r r r r sZ (
/%5
, l:PK ! +! case.cpython-36.opt-2.pycnu [ 3
_Vj @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
ddl m Z ddl
mZmZmZmZmZ dZe ZdZG dd deZG d d
d
eZG d d d eZG d
d deZdd Z dd Z dd Z dd Z dd Z dd Z!G dd d Z"d d Z#G d d d e"Z$G d!d" d"e$Z%G d#d$ d$e$Z&ej'd%d&d'gZ(G d(d) d)ej)Z*G d*d+ d+e"Z+G d,d- d-eZ,G d.d/ d/e,Z-G d0d1 d1e,Z.dS )2 N )result)strclass safe_repr_count_diff_all_purpose_count_diff_hashable_common_shorten_reprTz@
Diff is %s characters long. Set self.maxDiff to None to see it.c @ s e Zd ZdS )SkipTestN)__name__
__module__ __qualname__ r
r
%/usr/lib64/python3.6/unittest/case.pyr s r c @ s e Zd ZdS ) _ShouldStopN)r
r r r
r
r
r r " s r c @ s e Zd ZdS )_UnexpectedSuccessN)r
r r r
r
r
r r ' s r c @ s& e Zd ZdddZejdddZdS ) _OutcomeNc C s4 d| _ || _t|d| _d| _g | _d | _g | _d S )NF
addSubTestT)expecting_failurer hasattrresult_supports_subtestssuccessskippedexpectedFailureerrors)selfr r
r
r __init__. s z_Outcome.__init__Fc c s | j }d| _ zy
d V W n tk
r. Y n tk
rh } z d| _ | jj|t|f W Y d d }~X nj tk
rz Y nX tj }| j r|| _
nd| _ | j j||f d }Y n X | j r| j r| j j|d f W d | j o|| _ X d S )NTF)
r KeyboardInterruptr r appendstrr sysexc_infor r r r )r test_caseisTestZ old_successer r
r
r testPartExecutor7 s*
$
z_Outcome.testPartExecutor)N)F)r
r r r
contextlibcontextmanagerr$ r
r
r
r r - s
r c C s | S )Nr
)objr
r
r _idU s r( c s fdd}|S )Nc s4 t | ts$tj| fdd}|} d| _ | _| S )Nc s t d S )N)r )argskwargs)reasonr
r skip_wrapper^ s z-skip..decorator..skip_wrapperT)
isinstancetype functoolswraps__unittest_skip____unittest_skip_why__) test_itemr, )r+ r
r decorator\ s
zskip..decoratorr
)r+ r4 r
)r+ r skipX s
r5 c C s | r t |S tS )N)r5 r( ) conditionr+ r
r
r skipIfh s r7 c C s | s t |S tS )N)r5 r( )r6 r+ r
r
r
skipUnlessp s r8 c C s
d| _ | S )NT) __unittest_expecting_failure__)r3 r
r
r r x s r c s4 t | tr t fdd| D S t | to2t| S )Nc 3 s | ]}t | V qd S )N) _is_subtype).0r# )basetyper
r ~ s z _is_subtype..)r- tupleallr.
issubclass)expectedr< r
)r< r r: | s
r: c @ s e Zd Zdd Zdd ZdS )_BaseTestCaseContextc C s
|| _ d S )N)r! )r r! r
r
r r s z _BaseTestCaseContext.__init__c C s | j j| j|}| j j|d S )N)r! _formatMessagemsgfailureException)r standardMsgrD r
r
r
_raiseFailure s z"_BaseTestCaseContext._raiseFailureN)r
r r r rG r
r
r
r rB s rB c C s dt jkrt| S tS d S )NZWITHIN_PYTHON_RPM_BUILD)osenvironr5 r( )r+ r
r
r _skipInRpmBuild s
rJ c @ s e Zd ZdddZdd ZdS )_AssertRaisesBaseContextNc C s@ t j| | || _|| _|d k r*tj|}|| _d | _d | _d S )N) rB r rA r! recompileexpected_regexobj_namerD )r rA r! rN r
r
r r s
z!_AssertRaisesBaseContext.__init__c C s zt | j| js"td|| jf |rD|d d krDtjdtd f }|sx|jdd | _ |rttjdt
t | td | S |^}}y |j | _
W n tk
r t|| _
Y nX | ||| W d Q R X W d d } X d S )Nz%s() arg 1 must be %sr zcallable is None rD z3%r is an invalid keyword argument for this function)r: rA
_base_type TypeError_base_type_strwarningswarnDeprecationWarningpoprD nextiterr
rO AttributeErrorr )r namer) r* Z callable_objr
r
r handle s. z _AssertRaisesBaseContext.handle)N)r
r r r r\ r
r
r
r rK s
rK c @ s$ e Zd ZeZdZdd Zdd ZdS )_AssertRaisesContextz-an exception type or tuple of exception typesc C s | S )Nr
)r r
r
r __enter__ s z _AssertRaisesContext.__enter__c
C s |d krby | j j}W n tk
r2 t| j }Y nX | jrP| jdj|| j ql| jdj| n
tj| t || j s|dS |j
d | _ | j d krdS | j }|j
t|s| jdj|jt| dS )Nz{} not raised by {}z
{} not raisedFTz"{}" does not match "{}")rA r
rZ r rO rG format traceback clear_framesr@ with_tracebackZ exceptionrN searchpattern)r exc_type exc_valuetbexc_namerN r
r
r __exit__ s(
z _AssertRaisesContext.__exit__N)r
r r
BaseExceptionrQ rS r^ ri r
r
r
r r] s r] c @ s$ e Zd ZeZdZdd Zdd ZdS )_AssertWarnsContextz(a warning type or tuple of warning typesc C sR x$t jj D ]}t|dd r i |_q W tjdd| _| jj | _tj d| j
| S )N__warningregistry__T)recordalways) r modulesvaluesgetattrrl rT catch_warningswarnings_managerr^ simplefilterrA )r vr
r
r r^ s
z _AssertWarnsContext.__enter__c
C s | j j||| |d k r d S y | jj}W n tk
rF t| j}Y nX d }xd| jD ]Z}|j}t|| jslqT|d krx|}| j d k r| j j
t| rqT|| _ |j | _ |j
| _
d S W |d k r| jdj| j jt| | jr| jdj|| j n| jdj| d S )Nz"{}" does not match "{}"z{} not triggered by {}z{} not triggered)rs ri rA r
rZ r rT messager- rN rc ZwarningfilenamelinenorG r_ rd rO )r re rf rg rh Zfirst_matchingmwr
r
r ri s8
z _AssertWarnsContext.__exit__N)r
r r WarningrQ rS r^ ri r
r
r
r rk s rk _LoggingWatcherrecordsoutputc @ s$ e Zd Zdd Zdd Zdd ZdS )_CapturingHandlerc C s t jj| tg g | _d S )N)loggingHandlerr r| watcher)r r
r
r r , s z_CapturingHandler.__init__c C s d S )Nr
)r r
r
r flush0 s z_CapturingHandler.flushc C s* | j jj| | j|}| j jj| d S )N)r r} r r_ r~ )r rm rD r
r
r emit3 s
z_CapturingHandler.emitN)r
r r r r r r
r
r
r r ' s r c @ s( e Zd ZdZdd Zdd Zdd ZdS ) _AssertLogsContextz"%(levelname)s:%(name)s:%(message)sc C s: t j| | || _|r(tjj||| _ntj| _d | _d S )N) rB r logger_namer Z _nameToLevelgetlevelINFOrD )r r! r r r
r
r r ? s z_AssertLogsContext.__init__c C s t | jtjr | j }| _ntj| j }| _tj| j}t }|j | |j
| _
|j d d | _ |j
| _|j| _|g|_ |j| j
d|_|j
S )NF)r- r r ZLoggerloggerZ getLoggerZ FormatterLOGGING_FORMATr Z setFormatterr handlers old_handlersr old_level propagate
old_propagatesetLevel)r r Z formatterZhandlerr
r
r r^ H s
z _AssertLogsContext.__enter__c C s` | j | j_| j| j_| jj| j |d k r.dS t| jj dkr\| j
dj t j
| j| jj d S )NFr z-no logs of level {} or higher triggered on {})r r r r r r r lenr r} rG r_ r Z getLevelNamer r[ )r re rf rg r
r
r ri Y s
z_AssertLogsContext.__exit__N)r
r r r r r^ ri r
r
r
r r : s r c @ s e Zd ZeZdZdZdZdZddd Z d
d Z
d d
Z dd Z dd Z
edd Zedd Zdd Zdd Zdd Zd d Zd d Zd d! Zd"d# Zd$d% Zd&d' Zeje fd(d)Z d*d+ Z d,d- Z d.d/ Z dd1d2Z!d3d4 Z"d5d6 Z#d7d8 Z$d9d: Z%dd;d<Z&dd=d>Z'dd?d@Z(dAdB Z)dCdD Z*dEdF Z+ddGdHZ,dIdJ Z-ddKdLZ.ddMdNZ/ddOdPZ0ddQdRZ1ddSdTZ2ddUdVZ3dWdX Z4ddYdZZ5dd[d\Z6dd]d^Z7dd_d`Z8ddadbZ9ddcddZ:ddedfZ;ddgdhZddmdnZ?ddodpZ@ddqdrZAddsdtZBddudvZCddwdxZDddydzZEdd{d|ZFdd}d~ZGdd ZHdd ZIdddZJdddZKdd ZLeLe/ ZMZNeLe0 ZOZPeLe1 ZQZReLe2 ZSZTeLe( ZUZVeLe*ZWeLe'ZXeLeHZYeLeJZZeLeKZ[d0S )TestCaseTP FrunTestc C s || _ d | _d| _yt| |}W n. tk
rN |dkrJtd| j|f Y n
X |j| _g | _d | _ i | _
| j t d | j t
d | j td | j td | j td | j td d S ) NzNo testr z no such test method in %s: %sassertDictEqualassertListEqualassertTupleEqualassertSetEqualassertMultiLineEqual)_testMethodName_outcome_testMethodDocrq rZ
ValueError __class____doc__ _cleanups_subtest_type_equality_funcsaddTypeEqualityFuncdictlistr> set frozensetr )r Z
methodName
testMethodr
r
r r s& zTestCase.__init__c C s || j |< d S )N)r )r Ztypeobjfunctionr
r
r r s
z TestCase.addTypeEqualityFuncc O s | j j|||f d S )N)r r )r r r) r* r
r
r
addCleanup s zTestCase.addCleanupc C s d S )Nr
)r r
r
r setUp s zTestCase.setUpc C s d S )Nr
)r r
r
r tearDown s zTestCase.tearDownc C s d S )Nr
)clsr
r
r
setUpClass s zTestCase.setUpClassc C s d S )Nr
)r r
r
r
tearDownClass s zTestCase.tearDownClassc C s dS )Nr r
)r r
r
r countTestCases s zTestCase.countTestCasesc C s t j S )N)r Z
TestResult)r r
r
r defaultTestResult s zTestCase.defaultTestResultc C s | j }|r |jdd j p d S )N
r )r splitstrip)r docr
r
r shortDescription s zTestCase.shortDescriptionc C s dt | j| jf S )Nz%s.%s)r r r )r r
r
r id s z TestCase.idc C s t | t |k rtS | j|jkS )N)r. NotImplementedr )r otherr
r
r __eq__ s zTestCase.__eq__c C s t t| | jfS )N)hashr. r )r r
r
r __hash__ s zTestCase.__hash__c C s d| j t| jf S )Nz%s (%s))r r r )r r
r
r __str__ s zTestCase.__str__c C s dt | j| jf S )Nz<%s testMethod=%s>)r r r )r r
r
r __repr__ s zTestCase.__repr__c C s< t |dd }|d k r ||| ntjdtd |j| d S )NaddSkipz4TestResult has no addSkip method, skips not reportedr )rq rT rU RuntimeWarning
addSuccess)r r r! r+ r r
r
r _addSkip s zTestCase._addSkipc k s | j d ks| j j r d V d S | j}|d kr8tj|}n |jj|}t| ||| _zX| j j| jdd d V W d Q R X | j j s| j j
}|d k r|j rt n | j j
rt W d || _X d S )NT)r" )r r r collectionsChainMapparams new_child_SubTestr$ r r Zfailfastr r )r rD r parentZ
params_mapr r
r
r subTest s$ zTestCase.subTestc C sd x^|D ]V\}}t |tr*|j|j|| q|d k rt|d | jrP|j|| q|j|| qW d S )Nr )r- r r r! r@ rE
addFailureZaddError)r r r testr r
r
r _feedErrorsToResult( s
z TestCase._feedErrorsToResultc C sD y
|j }W n* tk
r4 tjdt |j| Y n X || | d S )Nz@TestResult has no addExpectedFailure method, reporting as passes)addExpectedFailurerZ rT rU r r )r r r r r
r
r _addExpectedFailure2 s
z TestCase._addExpectedFailurec C sh y
|j }W nP tk
rZ tjdt y
td W n$ tk
rT |j| tj Y nX Y n
X || d S )NzCTestResult has no addUnexpectedSuccess method, reporting as failure) addUnexpectedSuccessrZ rT rU r r r r r )r r r r
r
r _addUnexpectedSuccess< s
z TestCase._addUnexpectedSuccessNc
( C s |}|d kr.| j }t|dd }|d k r.| |j| t| | j}t| jdds^t|ddrz,t| jddpxt|dd}| j|| | W d |j| X d S t|dd}t| dd}|p|}t|} z| | _| j | | j
W d Q R X | j r<|| _ | j | dd | W d Q R X d| _ | j | | j
W d Q R X | j x"| jD ]\}
} | j||
| qLW | j|| j | j r|r| jr| j|| j n
| j| n
|j| |S |j| |d krt|d d } | d k r| | jj d | _d | _X d S )
N startTestRunr1 Fr2 r9 T)r" stopTestRun)r rq Z startTestr r r ZstopTestr r r$ r r r r
doCleanupsr r r r r r r clear)
r r Z orig_resultr r Zskip_whyZexpecting_failure_methodZexpecting_failure_classr outcomer r+ r r
r
r runK sh
z TestCase.runc
C sN | j p
t }x:| jrF| jj \}}}|j| ||| W d Q R X qW |jS )N)r r r rW r$ r )r r r r) r* r
r
r r s zTestCase.doCleanupsc O s | j ||S )N)r )r r) kwdsr
r
r __call__ s zTestCase.__call__c C sJ | j t| | j | j x&| jrD| jjd\}}}||| q W d S )Nr )r rq r r r rW )r r r) r* r
r
r debug s zTestCase.debugc C s t |d S )N)r )r r+ r
r
r skipTest s zTestCase.skipTestc C s | j |d S )N)rE )r rD r
r
r fail s z
TestCase.failc C s& |r"| j |dt| }| j|d S )Nz%s is not false)rC r rE )r exprrD r
r
r assertFalse s zTestCase.assertFalsec C s& |s"| j |dt| }| j|d S )Nz%s is not true)rC r rE )r r rD r
r
r
assertTrue s zTestCase.assertTruec C sP | j s|p |S |d kr|S y d||f S tk
rJ dt|t|f S X d S )Nz%s : %s) longMessageUnicodeDecodeErrorr )r rD rF r
r
r rC s
zTestCase._formatMessagec
O s$ t || }z|jd||S d }X d S )N assertRaises)r] r\ )r expected_exceptionr) r* contextr
r
r r s
zTestCase.assertRaisesc O s t || }|jd||S )N assertWarns)rk r\ )r expected_warningr) r* r r
r
r r s
zTestCase.assertWarnsc C s t | ||S )N)r )r r r r
r
r
assertLogs s zTestCase.assertLogsc C sF t |t |kr@| jjt |}|d k r@t|trt|| |kr>d S dt|t|t|f }nF|d krfd}||k rtt|| |dkrd S dt|t||f }| j||}| j|d S )Nz specify delta or places not bothz%s == %s within %s deltar r z%s == %s within %r places)rR r r r rC rE )r r r r rD r rF r
r
r assertNotAlmostEqual s" z TestCase.assertNotAlmostEqualc ' C s |d k rP|j }t||s.| jd|t|f t||sT| jd|t|f nd}d }y t|}W n ttfk
r d| }Y nX |d kry t|}W n ttfk
r d| }Y nX |d kr||krd S d|j ft|| }xt t
||D ]} y || }
W n. tt tfk
r8 |d| |f 7 }P Y nX y || } W n. tt tfk
rt |d| |f 7 }P Y nX |
| kr|d | ft|
| 7 }P qW ||kr|d krt |t |krd S ||kr0|d
||| f 7 }y |d |t|| f 7 }W n, tt tfk
r, |d ||f 7 }Y nX nh||k r|d
||| f 7 }y |d |t|| f 7 }W n, tt tfk
r |d||f 7 }Y nX |} ddj
tjtj|j tj|j }
| j| |
} | j|| }| j| d S )Nz First sequence is not a %s: %sz Second sequence is not a %s: %sZsequencez(First %s has no length. Non-sequence?z)Second %s has no length. Non-sequence?z%ss differ: %s != %s
z(
Unable to index element %d of first %s
z)
Unable to index element %d of second %s
z#
First differing element %d:
%s
%s
z+
First %s contains %d additional elements.
zFirst extra element %d:
%s
z'Unable to index element %d of first %s
z,
Second %s contains %d additional elements.
z(Unable to index element %d of second %s
r )r
r- rE r r rR NotImplementedError
capitalizer rangemin
IndexErrorr. joindifflibndiffpprintpformat
splitlines_truncateMessagerC r )r Zseq1Zseq2rD seq_typeZ
seq_type_nameZ differingZlen1Zlen2iZitem1Zitem2rF diffMsgr
r
r assertSequenceEqual s
z TestCase.assertSequenceEqualc C s2 | j }|d kst||kr"|| S |tt| S )N)maxDiffr DIFF_OMITTED)r rv diffZmax_diffr
r
r r s zTestCase._truncateMessagec C s | j |||td d S )N)r )r r )r Zlist1Zlist2rD r
r
r r s
zTestCase.assertListEqualc C s | j |||td d S )N)r )r r> )r Ztuple1Ztuple2rD r
r
r r s zTestCase.assertTupleEqualc
C sh y|j |}W n^ tk
r> } z| jd| W Y d d }~X n0 tk
rl } z| jd| W Y d d }~X nX y|j |}W n^ tk
r } z| jd| W Y d d }~X n0 tk
r } z| jd| W Y d d }~X nX |p|sd S g }|r|jd x |D ]}|jt| qW |rH|jd x |D ]}|jt| q0W dj|} | j| j|| d S )Nz/invalid type when attempting set difference: %sz2first argument does not support set difference: %sz3second argument does not support set difference: %sz*Items in the first set but not the second:z*Items in the second set but not the first:r )
differencerR r rZ r reprr rC )
r Zset1Zset2rD Z difference1r# Z difference2linesitemrF r
r
r r " s2
zTestCase.assertSetEqualc C s2 ||kr.dt |t |f }| j| j|| d S )Nz%s not found in %s)r r rC )r member containerrD rF r
r
r assertInM s zTestCase.assertInc C s2 ||kr.dt |t |f }| j| j|| d S )Nz%s unexpectedly found in %s)r r rC )r r r rD rF r
r
r assertNotInT s zTestCase.assertNotInc C s2 ||k r.dt |t |f }| j| j|| d S )Nz %s is not %s)r r rC )r expr1expr2rD rF r
r
r assertIs[ s zTestCase.assertIsc C s, ||kr(dt |f }| j| j|| d S )Nzunexpectedly identical: %s)r r rC )r r r rD rF r
r
r assertIsNotb s zTestCase.assertIsNotc C s~ | j |td | j |td ||krzdt|| }ddjtjtj|j tj|j }| j ||}| j
| j || d S )Nz"First argument is not a dictionaryz#Second argument is not a dictionaryz%s != %sr ) assertIsInstancer r r r r r r r r r rC )r Zd1Zd2rD rF r r
r
r r h s
zTestCase.assertDictEqualc C s t jdt g }g }xX|j D ]L\}}||kr:|j| q ||| kr |jdt|t|t|| f q W |pt|szd S d}|rddjdd |D }|r|r|d7 }|d dj| 7 }| j| j|| d S )
Nz&assertDictContainsSubset is deprecatedz %s, expected: %s, actual: %sr z Missing: %s,c s s | ]}t |V qd S )N)r )r; ry r
r
r r= s z4TestCase.assertDictContainsSubset..z; zMismatched values: %s) rT rU rV itemsr r r r rC ) r ZsubsetZ
dictionaryrD ZmissingZ
mismatchedkeyvaluerF r
r
r assertDictContainsSubsett s, z!TestCase.assertDictContainsSubsetc
C s t |t | }}ytj|}tj|}W n tk
rH t||}Y nX ||krVd S t||}|rd}dd |D }dj|} | j|| }| j||}| j | d S )Nz Element counts were not equal:
c S s g | ] }d | qS )z First has %d, Second has %d: %rr
)r; r r
r
r
s z-TestCase.assertCountEqual..r )
r r CounterrR r r r r rC r )
r r r rD Z first_seqZ
second_seqZ differencesrF r r r
r
r assertCountEqual s
zTestCase.assertCountEqualc C s | j |td | j |td ||krt|| jks@t|| jkrN| j||| |jdd}|jdd}t|dkr|jd|kr|d g}|d g}dt|| }dd jt j
|| }| j ||}| j | j
|| d S )
Nz First argument is not a stringz Second argument is not a stringT)keependsr z
r z%s != %sr )r r r _diffThresholdr r r r r r r r r rC )r r r rD Z
firstlinesZ secondlinesrF r r
r
r r s
z TestCase.assertMultiLineEqualc C s2 ||k s.dt |t |f }| j| j|| d S )Nz%s not less than %s)r r rC )r abrD rF r
r
r
assertLess s zTestCase.assertLessc C s2 ||ks.dt |t |f }| j| j|| d S )Nz %s not less than or equal to %s)r r rC )r r$ r% rD rF r
r
r assertLessEqual s zTestCase.assertLessEqualc C s2 ||ks.dt |t |f }| j| j|| d S )Nz%s not greater than %s)r r rC )r r$ r% rD rF r
r
r
assertGreater s zTestCase.assertGreaterc C s2 ||ks.dt |t |f }| j| j|| d S )Nz"%s not greater than or equal to %s)r r rC )r r$ r% rD rF r
r
r assertGreaterEqual s zTestCase.assertGreaterEqualc C s, |d k r(dt |f }| j| j|| d S )Nz%s is not None)r r rC )r r' rD rF r
r
r assertIsNone s zTestCase.assertIsNonec C s" |d kr d}| j | j|| d S )Nzunexpectedly None)r rC )r r' rD rF r
r
r assertIsNotNone s zTestCase.assertIsNotNonec C s0 t ||s,dt||f }| j| j|| d S )Nz%s is not an instance of %r)r- r r rC )r r' r rD rF r
r
r r s
zTestCase.assertIsInstancec C s0 t ||r,dt||f }| j| j|| d S )Nz%s is an instance of %r)r- r r rC )r r' r rD rF r
r
r assertNotIsInstance s
z TestCase.assertNotIsInstancec O s t || |}|jd||S )NassertRaisesRegex)r] r\ )r r rN r) r* r r
r
r r- s
zTestCase.assertRaisesRegexc O s t || |}|jd||S )NassertWarnsRegex)rk r\ )r r rN r) r* r r
r
r r. s zTestCase.assertWarnsRegexc C sJ t |ttfrtj|}|j|sFd|j|f }| j||}| j|d S )Nz&Regex didn't match: %r not found in %r) r- r bytesrL rM rc rd rC rE )r textrN rD rF r
r
r assertRegex s
zTestCase.assertRegexc C s` t |ttfrtj|}|j|}|r\d||j |j |j|f }| j ||}| j
|d S )Nz"Regex matched: %r matches %r in %r) r- r r/ rL rM rc startendrd rC rE )r r0 Zunexpected_regexrD matchrF r
r
r assertNotRegex& s
zTestCase.assertNotRegexc s fdd}|S )Nc s t jdj jtd | |S )NzPlease use {0} instead.r )rT rU r_ r
rV )r) r* )
original_funcr
r deprecated_func6 s
z,TestCase._deprecate..deprecated_funcr
)r6 r7 r
)r6 r
_deprecate5 s zTestCase._deprecatei i )r )N)N)N)N)NN)N)N)N)NNN)NNN)NN)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)\r
r r AssertionErrorrE r r
r# Z_classSetupFailedr r r r r classmethodr r r r r r r r r r r r% r& _subtest_msg_sentinelr r r r r r r r r r r r rC r r r r r r r r r r r r r r r r r r r r r! r r& r' r( r) r* r+ r r, r- r. r1 r5 r8 ZfailUnlessEqualZ assertEqualsZ failIfEqualZassertNotEqualsZfailUnlessAlmostEqualZassertAlmostEqualsZfailIfAlmostEqualZassertNotAlmostEqualsZ
failUnlessZassert_ZfailUnlessRaisesZfailIfZassertRaisesRegexpZassertRegexpMatchesZassertNotRegexpMatchesr
r
r
r r f s !
E
!
'
c
+
!
r c sf e Zd Zd fdd Zdd Zdd Zdd Zd
d Zd d
Zdd Z dd Z
dd Z dd Z Z
S )FunctionTestCaseNc s* t t| j || _|| _|| _|| _d S )N)superr< r
_setUpFunc
_tearDownFunc _testFunc _description)r ZtestFuncr r Z description)r r
r r T s
zFunctionTestCase.__init__c C s | j d k r| j d S )N)r> )r r
r
r r [ s
zFunctionTestCase.setUpc C s | j d k r| j d S )N)r? )r r
r
r r _ s
zFunctionTestCase.tearDownc C s | j d S )N)r@ )r r
r
r r c s zFunctionTestCase.runTestc C s | j jS )N)r@ r
)r r
r
r r f s zFunctionTestCase.idc C s@ t || jstS | j|jko>| j|jko>| j|jko>| j|jkS )N)r- r r r> r? r@ rA )r r r
r
r r i s zFunctionTestCase.__eq__c C s t t| | j| j| j| jfS )N)r r. r> r? r@ rA )r r
r
r r r s zFunctionTestCase.__hash__c C s dt | j| jjf S )Nz%s (%s))r r r@ r
)r r
r
r r v s
zFunctionTestCase.__str__c C s dt | j| jf S )Nz <%s tec=%s>)r r r@ )r r
r
r r z s
zFunctionTestCase.__repr__c C s2 | j d k r| j S | jj}|r.|jdd j p0d S )Nr r )rA r@ r r r )r r r
r
r r ~ s
z!FunctionTestCase.shortDescription)NNN)r
r r r r r r r r r r r r
__classcell__r
r
)r r r< K s r< c sD e Zd Z fddZdd Zdd Zdd Zd d
Zd d Z Z S )
r c s( t j || _|| _|| _|j| _d S )N)r= r _messager! r rE )r r! rv r )r r
r r s
z_SubTest.__init__c C s t dd S )Nz subtests cannot be run directly)r )r r
r
r r s z_SubTest.runTestc C sb g }| j tk r |jdj| j | jrTdjdd t| jj D }|jdj| dj|p`dS )Nz[{}]z, c s s | ]\}}d j ||V qdS )z{}={!r}N)r_ )r; kru r
r
r r= s z+_SubTest._subDescription..z({}) z ())rC r; r r_ r r sortedr )r partsZ params_descr
r
r _subDescription s
z_SubTest._subDescriptionc C s dj | jj | j S )Nz{} {})r_ r! r rH )r r
r
r r s z _SubTest.idc C s
| j j S )N)r! r )r r
r
r r s z_SubTest.shortDescriptionc C s dj | j| j S )Nz{} {})r_ r! rH )r r
r
r r s z_SubTest.__str__)
r
r r r r rH r r r rB r
r
)r r r s r )/r r/ r rH r r rL rT r r% r` r r utilr r r r r Z
__unittestobjectr; r Exceptionr r r r r( r5 r7 r8 r r: rB rJ rK r] rk
namedtupler| r r r r r< r r
r
r
r sX (
/%5
, l:PK ! ( case.cpython-36.pycnu [ 3
_Vj @ s d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl m
Z
ddlmZmZmZmZmZ dZe ZdZG dd d eZG d
d d eZG d d
d
eZG dd deZ dd Z dd Z dd Z dd Z dd Z!dd Z"G d d d Z#d d Z$G d d! d!e#Z%G d"d# d#e%Z&G d$d% d%e%Z'e j(d&d'd(gZ)G d)d* d*ej*Z+G d+d, d,e#Z,G d-d. d.eZ-G d/d0 d0e-Z.G d1d2 d2e-Z/dS )3zTest case implementation N )result)strclass safe_repr_count_diff_all_purpose_count_diff_hashable_common_shorten_reprTz@
Diff is %s characters long. Set self.maxDiff to None to see it.c @ s e Zd ZdZdS )SkipTestz
Raise this exception in a test to skip it.
Usually you can use TestCase.skipTest() or one of the skipping decorators
instead of raising this directly.
N)__name__
__module__ __qualname____doc__ r r %/usr/lib64/python3.6/unittest/case.pyr s r c @ s e Zd ZdZdS ) _ShouldStopz
The test should stop.
N)r
r r r
r r r r r " s r c @ s e Zd ZdZdS )_UnexpectedSuccessz7
The test was supposed to fail, but it didn't!
N)r
r r r
r r r r r ' s r c @ s& e Zd ZdddZejdddZdS ) _OutcomeNc C s4 d| _ || _t|d| _d| _g | _d | _g | _d S )NF
addSubTestT)expecting_failurer hasattrresult_supports_subtestssuccessskippedexpectedFailureerrors)selfr r r r __init__. s z_Outcome.__init__Fc c s | j }d| _ zy
d V W n tk
r. Y n tk
rh } z d| _ | jj|t|f W Y d d }~X nj tk
rz Y nX tj }| j r|| _
nd| _ | j j||f d }Y n X | j r| j r| j j|d f W d | j o|| _ X d S )NTF)
r KeyboardInterruptr r appendstrr sysexc_infor r r r )r test_caseisTestZ old_successer! r r r testPartExecutor7 s*
$
z_Outcome.testPartExecutor)N)F)r
r r r
contextlibcontextmanagerr% r r r r r - s
r c C s | S )Nr )objr r r _idU s r) c s fdd}|S )z&
Unconditionally skip a test.
c s4 t | ts$tj| fdd}|} d| _ | _| S )Nc s t d S )N)r )argskwargs)reasonr r skip_wrapper^ s z-skip..decorator..skip_wrapperT)
isinstancetype functoolswraps__unittest_skip____unittest_skip_why__) test_itemr- )r, r r decorator\ s
zskip..decoratorr )r, r5 r )r, r skipX s
r6 c C s | r t |S tS )z/
Skip a test if the condition is true.
)r6 r) ) conditionr, r r r skipIfh s r8 c C s | s t |S tS )z3
Skip a test unless the condition is true.
)r6 r) )r7 r, r r r
skipUnlessp s r9 c C s
d| _ | S )NT) __unittest_expecting_failure__)r4 r r r r x s r c s4 t | tr t fdd| D S t | to2t| S )Nc 3 s | ]}t | V qd S )N) _is_subtype).0r$ )basetyper r ~ s z _is_subtype..)r. tupleallr/
issubclass)expectedr= r )r= r r; | s
r; c @ s e Zd Zdd Zdd ZdS )_BaseTestCaseContextc C s
|| _ d S )N)r" )r r" r r r r s z _BaseTestCaseContext.__init__c C s | j j| j|}| j j|d S )N)r" _formatMessagemsgfailureException)r standardMsgrE r r r
_raiseFailure s z"_BaseTestCaseContext._raiseFailureN)r
r r r rH r r r r rC s rC c C s dt jkrt| S tS dS )a
Non-standard/downstream-only decorator for marking a specific unit test
to be skipped when run within the %check of an rpmbuild.
Specifically, this takes effect when WITHIN_PYTHON_RPM_BUILD is set within
the environment, and has no effect otherwise.
ZWITHIN_PYTHON_RPM_BUILDN)osenvironr6 r) )r, r r r _skipInRpmBuild s
rK c @ s e Zd ZdddZdd ZdS )_AssertRaisesBaseContextNc C s@ t j| | || _|| _|d k r*tj|}|| _d | _d | _d S )N) rC r rB r" recompileexpected_regexobj_namerE )r rB r" rO r r r r s
z!_AssertRaisesBaseContext.__init__c C s zt | j| js"td|| jf |rD|d dkrDtjdtd f }|sx|jdd| _ |rttjdt
t | td | S |^}}y |j | _
W n tk
r t|| _
Y nX | ||| W dQ R X W dd} X dS )z
If args is empty, assertRaises/Warns is being used as a
context manager, so check for a 'msg' kwarg and return self.
If args is not empty, call a callable passing positional and keyword
arguments.
z%s() arg 1 must be %sr Nzcallable is None rE z3%r is an invalid keyword argument for this function)r; rB
_base_type TypeError_base_type_strwarningswarnDeprecationWarningpoprE nextiterr
rP AttributeErrorr )r namer* r+ Z callable_objr r r handle s. z _AssertRaisesBaseContext.handle)N)r
r r r r] r r r r rL s
rL c @ s( e Zd ZdZeZdZdd Zdd ZdS )_AssertRaisesContextzCA context manager used to implement TestCase.assertRaises* methods.z-an exception type or tuple of exception typesc C s | S )Nr )r r r r __enter__ s z _AssertRaisesContext.__enter__c
C s |d krby | j j}W n tk
r2 t| j }Y nX | jrP| jdj|| j ql| jdj| n
tj| t || j s|dS |j
d | _ | j d krdS | j }|j
t|s| jdj|jt| dS )Nz{} not raised by {}z
{} not raisedFTz"{}" does not match "{}")rB r
r[ r rP rH format traceback clear_framesrA with_tracebackZ exceptionrO searchpattern)r exc_type exc_valuetbexc_namerO r r r __exit__ s(
z _AssertRaisesContext.__exit__N) r
r r r
BaseExceptionrR rT r_ rj r r r r r^ s
r^ c @ s( e Zd ZdZeZdZdd Zdd ZdS )_AssertWarnsContextzBA context manager used to implement TestCase.assertWarns* methods.z(a warning type or tuple of warning typesc C sR x$t jj D ]}t|dd r i |_q W tjdd| _| jj | _tj d| j
| S )N__warningregistry__T)recordalways) r modulesvaluesgetattrrm rU catch_warningswarnings_managerr_ simplefilterrB )r vr r r r_ s
z _AssertWarnsContext.__enter__c
C s | j j||| |d k r d S y | jj}W n tk
rF t| j}Y nX d }xd| jD ]Z}|j}t|| jslqT|d krx|}| j d k r| j j
t| rqT|| _ |j | _ |j
| _
d S W |d k r| jdj| j jt| | jr| jdj|| j n| jdj| d S )Nz"{}" does not match "{}"z{} not triggered by {}z{} not triggered)rt rj rB r
r[ r rU messager. rO rd ZwarningfilenamelinenorH r` re rP )r rf rg rh ri Zfirst_matchingmwr r r rj s8
z _AssertWarnsContext.__exit__N) r
r r r
WarningrR rT r_ rj r r r r rl s
rl _LoggingWatcherrecordsoutputc @ s( e Zd ZdZdd Zdd Zdd ZdS ) _CapturingHandlerzM
A logging handler capturing all (raw and formatted) logging output.
c C s t jj| tg g | _d S )N)loggingHandlerr r} watcher)r r r r r , s z_CapturingHandler.__init__c C s d S )Nr )r r r r flush0 s z_CapturingHandler.flushc C s* | j jj| | j|}| j jj| d S )N)r r~ r r` r )r rn rE r r r emit3 s
z_CapturingHandler.emitN)r
r r r
r r r r r r r r ' s r c @ s, e Zd ZdZdZdd Zdd Zdd Zd S )
_AssertLogsContextz:A context manager used to implement TestCase.assertLogs().z"%(levelname)s:%(name)s:%(message)sc C s: t j| | || _|r(tjj||| _ntj| _d | _d S )N) rC r logger_namer Z _nameToLevelgetlevelINFOrE )r r" r r r r r r ? s z_AssertLogsContext.__init__c C s t | jtjr | j }| _ntj| j }| _tj| j}t }|j | |j
| _
|j d d | _ |j
| _|j| _|g|_ |j| j
d|_|j
S )NF)r. r r ZLoggerloggerZ getLoggerZ FormatterLOGGING_FORMATr Z setFormatterr handlers old_handlersr old_level propagate
old_propagatesetLevel)r r Z formatterZhandlerr r r r_ H s
z _AssertLogsContext.__enter__c C s` | j | j_| j| j_| jj| j |d k r.dS t| jj dkr\| j
dj t j
| j| jj d S )NFr z-no logs of level {} or higher triggered on {})r r r r r r r lenr r~ rH r` r Z getLevelNamer r\ )r rf rg rh r r r rj Y s
z_AssertLogsContext.__exit__N)r
r r r
r r r_ rj r r r r r : s
r c @ s e Zd ZdZeZdZdZdZdZ dd d
Z
d d Z d
d Z dd Z
dd Zedd Zedd Zdd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zej e fd)d*Z d+d, Z d-d. Z d/d0 Z!dd2d3Z"d4d5 Z#d6d7 Z$d8d9 Z%d:d; Z&ddd?Z(dd@dAZ)dBdC Z*dDdE Z+dFdG Z,ddHdIZ-dJdK Z.ddLdMZ/ddNdOZ0ddPdQZ1ddRdSZ2ddTdUZ3ddVdWZ4dXdY Z5ddZd[Z6dd\d]Z7dd^d_Z8dd`daZ9ddbdcZ:ddddeZ;ddfdgZddldmZ?ddndoZ@ddpdqZAddrdsZBddtduZCddvdwZDddxdyZEddzd{ZFdd|d}ZGdd~dZHdd ZIdd ZJdddZKdddZLdd ZMeMe0 ZNZOeMe1 ZPZQeMe2 ZRZSeMe3 ZTZUeMe) ZVZWeMe+ZXeMe(ZYeMeIZZeMeKZ[eMeLZ\d1S )TestCaseaW A class whose instances are single test cases.
By default, the test code itself should be placed in a method named
'runTest'.
If the fixture may be used for many test cases, create as
many test methods as are needed. When instantiating such a TestCase
subclass, specify in the constructor arguments the name of the test method
that the instance is to execute.
Test authors should subclass TestCase for their own tests. Construction
and deconstruction of the test's environment ('fixture') can be
implemented by overriding the 'setUp' and 'tearDown' methods respectively.
If it is necessary to override the __init__ method, the base class
__init__ method must always be called. It is important that subclasses
should not change the signature of their __init__ method, since instances
of the classes are instantiated automatically by parts of the framework
in order to be run.
When subclassing TestCase, you can set these attributes:
* failureException: determines which exception will be raised when
the instance's assertion methods fail; test methods raising this
exception will be deemed to have 'failed' rather than 'errored'.
* longMessage: determines whether long messages (including repr of
objects used in assert methods) will be printed on failure in *addition*
to any explicit message passed.
* maxDiff: sets the maximum length of a diff in failure messages
by assert methods using difflib. It is looked up as an instance
attribute so can be configured by individual tests if required.
TP FrunTestc C s || _ d| _d| _yt| |}W n. tk
rN |dkrJtd| j|f Y n
X |j| _g | _d| _ i | _
| j t d | j t
d | j td | j td | j td | j td dS )
zCreate an instance of the class that will use the named test
method when executed. Raises a ValueError if the instance does
not have a method with the specified name.
NzNo testr z no such test method in %s: %sassertDictEqualassertListEqualassertTupleEqualassertSetEqualassertMultiLineEqual)_testMethodName_outcome_testMethodDocrr r[
ValueError __class__r
_cleanups_subtest_type_equality_funcsaddTypeEqualityFuncdictlistr? set frozensetr )r Z
methodName
testMethodr r r r s& zTestCase.__init__c C s || j |< dS )a[ Add a type specific assertEqual style function to compare a type.
This method is for use by TestCase subclasses that need to register
their own type equality functions to provide nicer error messages.
Args:
typeobj: The data type to call this function on when both values
are of the same type in assertEqual().
function: The callable taking two arguments and an optional
msg= argument that raises self.failureException with a
useful error message when the two arguments are not equal.
N)r )r Ztypeobjfunctionr r r r s
z TestCase.addTypeEqualityFuncc O s | j j|||f dS )a Add a function, with arguments, to be called when the test is
completed. Functions added are called on a LIFO basis and are
called after tearDown on test failure or success.
Cleanup items are called even if setUp fails (unlike tearDown).N)r r )r r r* r+ r r r
addCleanup s zTestCase.addCleanupc C s dS )zAHook method for setting up the test fixture before exercising it.Nr )r r r r setUp s zTestCase.setUpc C s dS )zAHook method for deconstructing the test fixture after testing it.Nr )r r r r tearDown s zTestCase.tearDownc C s dS )zKHook method for setting up class fixture before running tests in the class.Nr )clsr r r
setUpClass s zTestCase.setUpClassc C s dS )zVHook method for deconstructing the class fixture after running all tests in the class.Nr )r r r r
tearDownClass s zTestCase.tearDownClassc C s dS )Nr r )r r r r countTestCases s zTestCase.countTestCasesc C s t j S )N)r Z
TestResult)r r r r defaultTestResult s zTestCase.defaultTestResultc C s | j }|r |jdd j p dS )zReturns a one-line description of the test, or None if no
description has been provided.
The default implementation of this method returns the first line of
the specified test method's docstring.
r N)r splitstrip)r docr r r shortDescription s zTestCase.shortDescriptionc C s dt | j| jf S )Nz%s.%s)r r r )r r r r id s z TestCase.idc C s t | t |k rtS | j|jkS )N)r/ NotImplementedr )r otherr r r __eq__ s zTestCase.__eq__c C s t t| | jfS )N)hashr/ r )r r r r __hash__ s zTestCase.__hash__c C s d| j t| jf S )Nz%s (%s))r r r )r r r r __str__ s zTestCase.__str__c C s dt | j| jf S )Nz<%s testMethod=%s>)r r r )r r r r __repr__ s zTestCase.__repr__c C s< t |dd }|d k r ||| ntjdtd |j| d S )NaddSkipz4TestResult has no addSkip method, skips not reportedr )rr rU rV RuntimeWarning
addSuccess)r r r" r, r r r r _addSkip s zTestCase._addSkipc k s | j dks| j j r dV dS | j}|dkr8tj|}n |jj|}t| ||| _zX| j j| jdd dV W dQ R X | j j s| j j
}|dk r|j rt n | j j
rt W d|| _X dS )aP Return a context manager that will return the enclosed block
of code in a subtest identified by the optional message and
keyword parameters. A failure in the subtest marks the test
case as failed but resumes execution at the end of the enclosed
block, allowing further test code to be executed.
NT)r# )r r r collectionsChainMapparams new_child_SubTestr% r r Zfailfastr r )r rE r parentZ
params_mapr r r r subTest s$ zTestCase.subTestc C sd x^|D ]V\}}t |tr*|j|j|| q|d k rt|d | jrP|j|| q|j|| qW d S )Nr )r. r r r" rA rF
addFailureZaddError)r r r testr! r r r _feedErrorsToResult( s
z TestCase._feedErrorsToResultc C sD y
|j }W n* tk
r4 tjdt |j| Y n X || | d S )Nz@TestResult has no addExpectedFailure method, reporting as passes)addExpectedFailurer[ rU rV r r )r r r! r r r r _addExpectedFailure2 s
z TestCase._addExpectedFailurec C sh y
|j }W nP tk
rZ tjdt y
td W n$ tk
rT |j| tj Y nX Y n
X || d S )NzCTestResult has no addUnexpectedSuccess method, reporting as failure) addUnexpectedSuccessr[ rU rV r r r r r! )r r r r r r _addUnexpectedSuccess< s
z TestCase._addUnexpectedSuccessNc
( C s |}|d kr.| j }t|dd }|d k r.| |j| t| | j}t| jdds^t|ddrz,t| jddpxt|dd}| j|| | W d |j| X d S t|dd}t| dd}|p|}t|} z| | _| j | | j
W d Q R X | j r<|| _ | j | dd | W d Q R X d| _ | j | | j
W d Q R X | j x"| jD ]\}
} | j||
| qLW | j|| j | j r|r| jr| j|| j n
| j| n
|j| |S |j| |d krt|d d } | d k r| | jj d | _d | _X d S )
N startTestRunr2 Fr3 r: T)r# stopTestRun)r rr Z startTestr r r ZstopTestr r r% r r r r
doCleanupsr r r r r r r clear)
r r Z orig_resultr r Zskip_whyZexpecting_failure_methodZexpecting_failure_classr outcomer r, r r r r runK sh
z TestCase.runc
C sN | j p
t }x:| jrF| jj \}}}|j| ||| W dQ R X qW |jS )zNExecute all cleanup functions. Normally called for you after
tearDown.N)r r r rX r% r )r r r r* r+ r r r r s zTestCase.doCleanupsc O s | j ||S )N)r )r r* kwdsr r r __call__ s zTestCase.__call__c C sJ | j t| | j | j x&| jrD| jjd\}}}||| q W dS )z6Run the test without collecting errors in a TestResultr N)r rr r r r rX )r r r* r+ r r r debug s zTestCase.debugc C s t |dS )zSkip this test.N)r )r r, r r r skipTest s zTestCase.skipTestc C s | j |dS )z)Fail immediately, with the given message.N)rF )r rE r r r fail s z
TestCase.failc C s& |r"| j |dt| }| j|dS )z#Check that the expression is false.z%s is not falseN)rD r rF )r exprrE r r r assertFalse s zTestCase.assertFalsec C s& |s"| j |dt| }| j|dS )z"Check that the expression is true.z%s is not trueN)rD r rF )r r rE r r r
assertTrue s zTestCase.assertTruec C sP | j s|p |S |dkr|S y d||f S tk
rJ dt|t|f S X dS )a Honour the longMessage attribute when generating failure messages.
If longMessage is False this means:
* Use only an explicit message if it is provided
* Otherwise use the standard message for the assert
If longMessage is True:
* Use the standard message
* If an explicit message is provided, plus ' : ' and the explicit message
Nz%s : %s) longMessageUnicodeDecodeErrorr )r rE rG r r r rD s
zTestCase._formatMessagec
O s$ t || }z|jd||S d}X dS )a= Fail unless an exception of class expected_exception is raised
by the callable when invoked with specified positional and
keyword arguments. If a different type of exception is
raised, it will not be caught, and the test case will be
deemed to have suffered an error, exactly as for an
unexpected exception.
If called with the callable and arguments omitted, will return a
context object used like this::
with self.assertRaises(SomeException):
do_something()
An optional keyword argument 'msg' can be provided when assertRaises
is used as a context object.
The context manager keeps a reference to the exception as
the 'exception' attribute. This allows you to inspect the
exception after the assertion::
with self.assertRaises(SomeException) as cm:
do_something()
the_exception = cm.exception
self.assertEqual(the_exception.error_code, 3)
assertRaisesN)r^ r] )r expected_exceptionr* r+ contextr r r r s
zTestCase.assertRaisesc O s t || }|jd||S )a Fail unless a warning of class warnClass is triggered
by the callable when invoked with specified positional and
keyword arguments. If a different type of warning is
triggered, it will not be handled: depending on the other
warning filtering rules in effect, it might be silenced, printed
out, or raised as an exception.
If called with the callable and arguments omitted, will return a
context object used like this::
with self.assertWarns(SomeWarning):
do_something()
An optional keyword argument 'msg' can be provided when assertWarns
is used as a context object.
The context manager keeps a reference to the first matching
warning as the 'warning' attribute; similarly, the 'filename'
and 'lineno' attributes give you information about the line
of Python code from which the warning was triggered.
This allows you to inspect the warning after the assertion::
with self.assertWarns(SomeWarning) as cm:
do_something()
the_warning = cm.warning
self.assertEqual(the_warning.some_attribute, 147)
assertWarns)rl r] )r expected_warningr* r+ r r r r r s
zTestCase.assertWarnsc C s t | ||S )a Fail unless a log message of level *level* or higher is emitted
on *logger_name* or its children. If omitted, *level* defaults to
INFO and *logger* defaults to the root logger.
This method must be used as a context manager, and will yield
a recording object with two attributes: `output` and `records`.
At the end of the context manager, the `output` attribute will
be a list of the matching formatted log messages and the
`records` attribute will be a list of the corresponding LogRecord
objects.
Example::
with self.assertLogs('foo', level='INFO') as cm:
logging.getLogger('foo').info('first message')
logging.getLogger('foo.bar').error('second message')
self.assertEqual(cm.output, ['INFO:foo:first message',
'ERROR:foo.bar:second message'])
)r )r r r r r r
assertLogs s zTestCase.assertLogsc C sF t |t |kr@| jjt |}|dk r@t|trt|| |kr>dS dt|t|t|f }nF|dkrfd}||k rtt|| |dkrdS dt|t||f }| j||}| j|dS )a Fail if the two objects are equal as determined by their
difference rounded to the given number of decimal places
(default 7) and comparing to zero, or by comparing that the
difference between the two objects is less than the given delta.
Note that decimal places (from zero) are usually not the same
as significant digits (measured from the most significant digit).
Objects that are equal automatically fail.
Nz specify delta or places not bothz%s == %s within %s deltar r z%s == %s within %r places)rS r r r rD rF )r r r r rE r rG r r r assertNotAlmostEqual s" z TestCase.assertNotAlmostEqualc ' C s |dk rP|j }t||s.| jd|t|f t||sT| jd|t|f nd}d}y t|}W n ttfk
r d| }Y nX |dkry t|}W n ttfk
r d| }Y nX |dkr||krdS d|j ft|| }xt t
||D ]} y || }
W n. tt tfk
r8 |d| |f 7 }P Y nX y || } W n. tt tfk
rt |d | |f 7 }P Y nX |
| kr|d
| ft|
| 7 }P qW ||kr|dkrt |t |krdS ||kr0|d ||| f 7 }y |d |t|| f 7 }W n, tt tfk
r, |d
||f 7 }Y nX nh||k r|d||| f 7 }y |d |t|| f 7 }W n, tt tfk
r |d||f 7 }Y nX |} ddj
tjtj|j tj|j }
| j| |
} | j|| }| j| dS )aA An equality assertion for ordered sequences (like lists and tuples).
For the purposes of this function, a valid ordered sequence type is one
which can be indexed, has a length, and has an equality operator.
Args:
seq1: The first sequence to compare.
seq2: The second sequence to compare.
seq_type: The expected datatype of the sequences, or None if no
datatype should be enforced.
msg: Optional message to use on failure instead of a list of
differences.
Nz First sequence is not a %s: %sz Second sequence is not a %s: %sZsequencez(First %s has no length. Non-sequence?z)Second %s has no length. Non-sequence?z%ss differ: %s != %s
z(
Unable to index element %d of first %s
z)
Unable to index element %d of second %s
z#
First differing element %d:
%s
%s
z+
First %s contains %d additional elements.
zFirst extra element %d:
%s
z'Unable to index element %d of first %s
z,
Second %s contains %d additional elements.
z(Unable to index element %d of second %s
r )r
r. rF r r rS NotImplementedError
capitalizer rangemin
IndexErrorr/ joindifflibndiffpprintpformat
splitlines_truncateMessagerD r )r Zseq1Zseq2rE seq_typeZ
seq_type_nameZ differingZlen1Zlen2iZitem1Zitem2rG diffMsgr r r assertSequenceEqual s
z TestCase.assertSequenceEqualc C s2 | j }|d kst||kr"|| S |tt| S )N)maxDiffr DIFF_OMITTED)r rw diffZmax_diffr r r r s zTestCase._truncateMessagec C s | j |||td dS )a A list-specific equality assertion.
Args:
list1: The first list to compare.
list2: The second list to compare.
msg: Optional message to use on failure instead of a list of
differences.
)r N)r r )r Zlist1Zlist2rE r r r r s
zTestCase.assertListEqualc C s | j |||td dS )a A tuple-specific equality assertion.
Args:
tuple1: The first tuple to compare.
tuple2: The second tuple to compare.
msg: Optional message to use on failure instead of a list of
differences.
)r N)r r? )r Ztuple1Ztuple2rE r r r r s zTestCase.assertTupleEqualc
C sh y|j |}W n^ tk
r> } z| jd| W Y dd}~X n0 tk
rl } z| jd| W Y dd}~X nX y|j |}W n^ tk
r } z| jd| W Y dd}~X n0 tk
r } z| jd| W Y dd}~X nX |p|sdS g }|r|jd x |D ]}|jt| qW |rH|jd x |D ]}|jt| q0W dj|} | j| j|| dS )a A set-specific equality assertion.
Args:
set1: The first set to compare.
set2: The second set to compare.
msg: Optional message to use on failure instead of a list of
differences.
assertSetEqual uses ducktyping to support different types of sets, and
is optimized for sets specifically (parameters must support a
difference method).
z/invalid type when attempting set difference: %sNz2first argument does not support set difference: %sz3second argument does not support set difference: %sz*Items in the first set but not the second:z*Items in the second set but not the first:r )
differencerS r r[ r reprr rD )
r Zset1Zset2rE Z difference1r$ Z difference2linesitemrG r r r r " s2
zTestCase.assertSetEqualc C s2 ||kr.dt |t |f }| j| j|| dS )zDJust like self.assertTrue(a in b), but with a nicer default message.z%s not found in %sN)r r rD )r member containerrE rG r r r assertInM s zTestCase.assertInc C s2 ||kr.dt |t |f }| j| j|| dS )zHJust like self.assertTrue(a not in b), but with a nicer default message.z%s unexpectedly found in %sN)r r rD )r r r rE rG r r r assertNotInT s zTestCase.assertNotInc C s2 ||k r.dt |t |f }| j| j|| dS )zDJust like self.assertTrue(a is b), but with a nicer default message.z %s is not %sN)r r rD )r expr1expr2rE rG r r r assertIs[ s zTestCase.assertIsc C s, ||kr(dt |f }| j| j|| dS )zHJust like self.assertTrue(a is not b), but with a nicer default message.zunexpectedly identical: %sN)r r rD )r r r rE rG r r r assertIsNotb s zTestCase.assertIsNotc C s~ | j |td | j |td ||krzdt|| }ddjtjtj|j tj|j }| j ||}| j
| j || d S )Nz"First argument is not a dictionaryz#Second argument is not a dictionaryz%s != %sr ) assertIsInstancer r r r r r r r r r rD )r Zd1Zd2rE rG r r r r r h s
zTestCase.assertDictEqualc C s t jdt g }g }xX|j D ]L\}}||kr:|j| q ||| kr |jdt|t|t|| f q W |pt|szdS d}|rddjdd |D }|r|r|d 7 }|d
dj| 7 }| j| j|| dS ) z2Checks whether dictionary is a superset of subset.z&assertDictContainsSubset is deprecatedz %s, expected: %s, actual: %sNr z Missing: %s,c s s | ]}t |V qd S )N)r )r< rz r r r r> s z4TestCase.assertDictContainsSubset..z; zMismatched values: %s) rU rV rW itemsr r r r rD ) r ZsubsetZ
dictionaryrE ZmissingZ
mismatchedkeyvaluerG r r r assertDictContainsSubsett s, z!TestCase.assertDictContainsSubsetc
C s t |t | }}ytj|}tj|}W n tk
rH t||}Y nX ||krVdS t||}|rd}dd |D }dj|} | j|| }| j||}| j | dS )a An unordered sequence comparison asserting that the same elements,
regardless of order. If the same element occurs more than once,
it verifies that the elements occur the same number of times.
self.assertEqual(Counter(list(first)),
Counter(list(second)))
Example:
- [0, 1, 1] and [1, 0, 1] compare equal.
- [0, 0, 1] and [0, 1] compare unequal.
Nz Element counts were not equal:
c S s g | ] }d | qS )z First has %d, Second has %d: %rr )r< r r r r
s z-TestCase.assertCountEqual..r )
r r CounterrS r r r r rD r )
r r r rE Z first_seqZ
second_seqZ differencesrG r r r r r assertCountEqual s
zTestCase.assertCountEqualc C s | j |td | j |td ||krt|| jks@t|| jkrN| j||| |jdd}|jdd}t|dkr|jd|kr|d g}|d g}dt|| }dd jt j
|| }| j ||}| j | j
|| d
S ) z-Assert that two multi-line strings are equal.z First argument is not a stringz Second argument is not a stringT)keependsr z
r z%s != %sr N)r r r _diffThresholdr r r r r r r r r rD )r r r rE Z
firstlinesZ secondlinesrG r r r r r s
z TestCase.assertMultiLineEqualc C s2 ||k s.dt |t |f }| j| j|| dS )zCJust like self.assertTrue(a < b), but with a nicer default message.z%s not less than %sN)r r rD )r abrE rG r r r
assertLess s zTestCase.assertLessc C s2 ||ks.dt |t |f }| j| j|| dS )zDJust like self.assertTrue(a <= b), but with a nicer default message.z %s not less than or equal to %sN)r r rD )r r$ r% rE rG r r r assertLessEqual s zTestCase.assertLessEqualc C s2 ||ks.dt |t |f }| j| j|| dS )zCJust like self.assertTrue(a > b), but with a nicer default message.z%s not greater than %sN)r r rD )r r$ r% rE rG r r r
assertGreater s zTestCase.assertGreaterc C s2 ||ks.dt |t |f }| j| j|| dS )zDJust like self.assertTrue(a >= b), but with a nicer default message.z"%s not greater than or equal to %sN)r r rD )r r$ r% rE rG r r r assertGreaterEqual s zTestCase.assertGreaterEqualc C s, |dk r(dt |f }| j| j|| dS )zCSame as self.assertTrue(obj is None), with a nicer default message.Nz%s is not None)r r rD )r r( rE rG r r r assertIsNone s zTestCase.assertIsNonec C s" |dkr d}| j | j|| dS )z(Included for symmetry with assertIsNone.Nzunexpectedly None)r rD )r r( rE rG r r r assertIsNotNone s zTestCase.assertIsNotNonec C s0 t ||s,dt||f }| j| j|| dS )zTSame as self.assertTrue(isinstance(obj, cls)), with a nicer
default message.z%s is not an instance of %rN)r. r r rD )r r( r rE rG r r r r s
zTestCase.assertIsInstancec C s0 t ||r,dt||f }| j| j|| dS )z,Included for symmetry with assertIsInstance.z%s is an instance of %rN)r. r r rD )r r( r rE rG r r r assertNotIsInstance s
z TestCase.assertNotIsInstancec O s t || |}|jd||S )a Asserts that the message in a raised exception matches a regex.
Args:
expected_exception: Exception class expected to be raised.
expected_regex: Regex (re pattern object or string) expected
to be found in error message.
args: Function to be called and extra positional args.
kwargs: Extra kwargs.
msg: Optional message used in case of failure. Can only be used
when assertRaisesRegex is used as a context manager.
assertRaisesRegex)r^ r] )r r rO r* r+ r r r r r- s
zTestCase.assertRaisesRegexc O s t || |}|jd||S )a Asserts that the message in a triggered warning matches a regexp.
Basic functioning is similar to assertWarns() with the addition
that only warnings whose messages also match the regular expression
are considered successful matches.
Args:
expected_warning: Warning class expected to be triggered.
expected_regex: Regex (re pattern object or string) expected
to be found in error message.
args: Function to be called and extra positional args.
kwargs: Extra kwargs.
msg: Optional message used in case of failure. Can only be used
when assertWarnsRegex is used as a context manager.
assertWarnsRegex)rl r] )r r rO r* r+ r r r r r. s zTestCase.assertWarnsRegexc C sV t |ttfr$|stdtj|}|j|sRd|j|f }| j||}| j |dS )z=Fail the test unless the text matches the regular expression.z!expected_regex must not be empty.z&Regex didn't match: %r not found in %rN)
r. r bytesAssertionErrorrM rN rd re rD rF )r textrO rE rG r r r assertRegex s
zTestCase.assertRegexc C s` t |ttfrtj|}|j|}|r\d||j |j |j|f }| j ||}| j
|dS )z9Fail the test if the text matches the regular expression.z"Regex matched: %r matches %r in %rN) r. r r/ rM rN rd startendre rD rF )r r1 Zunexpected_regexrE matchrG r r r assertNotRegex& s
zTestCase.assertNotRegexc s fdd}|S )Nc s t jdj jtd | |S )NzPlease use {0} instead.r )rU rV r` r
rW )r* r+ )
original_funcr r deprecated_func6 s
z,TestCase._deprecate..deprecated_funcr )r7 r8 r )r7 r
_deprecate5 s zTestCase._deprecatei i )r )N)N)N)N)NN)N)N)N)NNN)NNN)NN)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)]r
r r r
r0 rF r r
r# Z_classSetupFailedr r r r r classmethodr r r r r r r r r r r r& r' _subtest_msg_sentinelr r r r r r r r r r r r rD r r r r r r r r r r r r r r r r r r r r r! r r& r' r( r) r* r+ r r, r- r. r2 r6 r9 ZfailUnlessEqualZ assertEqualsZ failIfEqualZassertNotEqualsZfailUnlessAlmostEqualZassertAlmostEqualsZfailIfAlmostEqualZassertNotAlmostEqualsZ
failUnlessZassert_ZfailUnlessRaisesZfailIfZassertRaisesRegexpZassertRegexpMatchesZassertNotRegexpMatchesr r r r r f s
E
!
'
c
+
!
r c sj e Zd ZdZd fdd Zdd Zdd Zd d
Zd d Zd
d Z dd Z
dd Z dd Z dd Z
ZS )FunctionTestCaseaI A test case that wraps a test function.
This is useful for slipping pre-existing test functions into the
unittest framework. Optionally, set-up and tidy-up functions can be
supplied. As with TestCase, the tidy-up ('tearDown') function will
always be called if the set-up ('setUp') function ran successfully.
Nc s* t t| j || _|| _|| _|| _d S )N)superr< r
_setUpFunc
_tearDownFunc _testFunc _description)r ZtestFuncr r Z description)r r r r T s
zFunctionTestCase.__init__c C s | j d k r| j d S )N)r> )r r r r r [ s
zFunctionTestCase.setUpc C s | j d k r| j d S )N)r? )r r r r r _ s
zFunctionTestCase.tearDownc C s | j d S )N)r@ )r r r r r c s zFunctionTestCase.runTestc C s | j jS )N)r@ r
)r r r r r f s zFunctionTestCase.idc C s@ t || jstS | j|jko>| j|jko>| j|jko>| j|jkS )N)r. r r r> r? r@ rA )r r r r r r i s zFunctionTestCase.__eq__c C s t t| | j| j| j| jfS )N)r r/ r> r? r@ rA )r r r r r r s zFunctionTestCase.__hash__c C s dt | j| jjf S )Nz%s (%s))r r r@ r
)r r r r r v s
zFunctionTestCase.__str__c C s dt | j| jf S )Nz <%s tec=%s>)r r r@ )r r r r r z s
zFunctionTestCase.__repr__c C s2 | j d k r| j S | jj}|r.|jdd j p0d S )Nr r )rA r@ r
r r )r r r r r r ~ s
z!FunctionTestCase.shortDescription)NNN)r
r r r
r r r r r r r r r r
__classcell__r r )r r r< K s r< c sD e Zd Z fddZdd Zdd Zdd Zd d
Zd d Z Z S )
r c s( t j || _|| _|| _|j| _d S )N)r= r _messager" r rF )r r" rw r )r r r r s
z_SubTest.__init__c C s t dd S )Nz subtests cannot be run directly)r )r r r r r s z_SubTest.runTestc C sb g }| j tk r |jdj| j | jrTdjdd t| jj D }|jdj| dj|p`dS )Nz[{}]z, c s s | ]\}}d j ||V qdS )z{}={!r}N)r` )r< krv r r r r> s z+_SubTest._subDescription..z({}) z ())rC r; r r` r r sortedr )r partsZ params_descr r r _subDescription s
z_SubTest._subDescriptionc C s dj | jj | j S )Nz{} {})r` r" r rH )r r r r r s z _SubTest.idc C s
| j j S )zlReturns a one-line description of the subtest, or None if no
description has been provided.
)r" r )r r r r r s z_SubTest.shortDescriptionc C s dj | j| j S )Nz{} {})r` r" rH )r r r r r s z_SubTest.__str__)
r
r r r r rH r r r rB r r )r r r s r )0r
r r0 r rI r r rM rU r r& ra r r utilr r r r r Z
__unittestobjectr; r Exceptionr r r r r) r6 r8 r9 r r; rC rK rL r^ rl
namedtupler} r r r r r< r r r r r sZ (
/%5
, l:PK ! Fk95 5 loader.cpython-36.opt-1.pycnu [ 3
\V @ s d Z ddlZddlZddlZddlZddlZddlZddlZddlmZ ddl m
Z
m Z m Z dZ
ejdejZG dd d e
jZd
d Zd d
Zdd Zdd Zdd ZG dd deZe Zd ddZe j fddZ de j e j fdd Z de j e j fd d Z dS ) zLoading unittests. N)fnmatch )casesuiteutilTz[_a-z]\w*\.py$c s, e Zd ZdZ fddZ fddZ ZS ) _FailedTestNc s || _ tt| j| d S )N)
_exceptionsuperr __init__)selfZ method_name exception) __class__ '/usr/lib64/python3.6/unittest/loader.pyr
s z_FailedTest.__init__c s* | j krtt j|S fdd}|S )Nc s
j d S )N)r r )r r r testFailure! s z,_FailedTest.__getattr__..testFailure)_testMethodNamer r __getattr__)r namer )r
)r r r s
z_FailedTest.__getattr__)__name__
__module__ __qualname__r r
r
__classcell__r r )r
r r s r c C s" d| t j f }t| t|||S )Nz#Failed to import test module: %s
%s) traceback
format_exc_make_failed_test ImportError)r
suiteClassmessager r r _make_failed_import_test& s r c C s dt j f }t| |||S )Nz Failed to call load_tests:
%s)r r r )r r r r r r r _make_failed_load_tests+ s r c C s t | |}||f|fS )N)r )
methodnamer r r testr r r r 0 s
r c C s< t jt|dd }| |i}tdt jf|}||| fS )Nc S s d S )Nr )r r r r testSkipped5 s z'_make_skipped_test..testSkippedZ
ModuleSkipped)r skipstrtypeTestCase)r r r r" ZattrsZ TestClassr r r _make_skipped_test4 s r' c C s* | j jdr| d d S tjj| d S )Nz $py.class r i)lowerendswithospathsplitext)r, r r r _jython_aware_splitext< s r. c s e Zd ZdZdZeejZe j
Z dZ fddZ
dd Zddd d
Zd!d d Zd"d
dZdd Zd#ddZdd Zdd Zdd Zdd Zd$d d Zd%d d Z ZS )&
TestLoaderz
This class is responsible for loading tests according to various criteria
and returning them wrapped in a TestSuite
r! Nc s t t| j g | _t | _d S )N)r r/ r
errorsset_loading_packages)r )r
r r r
L s zTestLoader.__init__c C sH t |tjrtd| j|}| r4t|dr4dg}| jt||}|S )z;Return a suite of all test cases contained in testCaseClasszYTest cases should not be derived from TestSuite. Maybe you meant to derive from TestCase?ZrunTest)
issubclassr TestSuite TypeErrorgetTestCaseNameshasattrr map)r
testCaseClassZ
testCaseNamesZ loaded_suiter r r loadTestsFromTestCaseS s
z TestLoader.loadTestsFromTestCase)patternc
O s2 t |dksd|kr,tjdt |jdd t |dkrRt |d }tdj|t |dkrxt|d }tdj|g }x@t|D ]4}t ||}t
|t rt |t
jr|j| j| qW t |dd} | j|}| dk r.y | | ||S tk
r, }
z$t|j|
| j\} } | jj| | S d}
~
X nX |S ) z>Return a suite of all test cases contained in the given moduler Zuse_load_testsz(use_load_tests is deprecated and ignoredNr zCloadTestsFromModule() takes 1 positional argument but {} were givenz=loadTestsFromModule() got an unexpected keyword argument '{}'
load_tests)lenwarningswarnDeprecationWarningpopr5 formatsorteddirgetattr
isinstancer% r3 r r& appendr: r Exceptionr r r0 )
r moduler; argsZkwsZ complainttestsr objr< e
error_case
error_messager r r loadTestsFromModulea s4
z TestLoader.loadTestsFromModulec C s> |j d}d\}}|dkr|dd }xb|rydj|}t|}P W q( tk
r |j }t|| j\}}|s| jj| |S Y q(X q(W |dd }|} x|D ]}
y| t | |
} } W q t
k
r2 } z\t | dddk r|dk r| jj| |S t |
| | jdt j
f \}}| jj| |S W Y dd} ~ X qX qW t| tjrP| j| S t| trtt| tjrt| j| S t| tjrt| trt| tjr|d }| |}
tt |
|tjs| j|
gS nt| tjr| S t| r.| }t|tjr|S t|tjr | j|gS td| |f n td| dS )
aS Return a suite of all test cases given a string specifier.
The name may resolve either to a module, a test case class, a
test method within a test case class, or a callable object which
returns a TestCase or TestSuite instance.
The method optionally resolves the names relative to a given module.
.Nr __path__z Failed to access attribute:
%sz"calling %s returned %s, not a testz$don't know how to make test from: %s)NN) splitjoin
__import__r rA r r r0 rG rE AttributeErrorr r r rF types
ModuleTyperP r% r3 r r& r: FunctionTyper r4 callabler5 )r r rI partsrN rO Z
parts_copy module_nameZnext_attributerL partparentrM Zinstr! r r r loadTestsFromName sl
z TestLoader.loadTestsFromNamec s fdd|D }j |S )zReturn a suite of all test cases found using the given sequence
of string specifiers. See 'loadTestsFromName()'.
c s g | ]}j | qS r )r` ).0r )rI r r r
s z1TestLoader.loadTestsFromNames..)r )r namesrI Zsuitesr )rI r r loadTestsFromNames s z TestLoader.loadTestsFromNamesc C s@ || j fdd}tt|t|}| jr<|jtj| jd |S )zLReturn a sorted sequence of method names found within testCaseClass
c S s | j |ott|| S )N)
startswithr[ rE )Zattrnamer9 prefixr r r isTestMethod s
z1TestLoader.getTestCaseNames..isTestMethod)key)testMethodPrefixlistfilterrD sortTestMethodsUsingsort functools
cmp_to_key)r r9 rg Z testFnNamesr r r r6 s zTestLoader.getTestCaseNamestest*.pyc C sR d}|dkr | j dk r | j }n|dkr.d}|}tjj|}|tjkrRtjjd| || _ d}d}g }tjjtjj|rtjj|}||krtjjtjj|d }nxy t | W n t
k
r d}Y nRX tj | }|j dd } ytjjtjj
|j}W n tk
r y
|j}
W n tk
r8 d}
Y nX |
r|
jdkr|
jdk rd}x|jD ]T} | r| j| rqd| j |jjdtjjd | _ |j| j| |dd qdW n*|jtjkrtddntd j |dY nX |r |s| j | | _ tjj | n tjj | |r2t
d
| |sHt | j||}| j |S ) a% Find and return all test modules from the specified start
directory, recursing into subdirectories to find them and return all
tests found within them. Only test files that match the pattern will
be loaded. (Using shell style pattern matching.)
All test modules must be importable from the top level of the project.
If the start directory is not the top level directory then the top
level directory must be specified separately.
If a test package name (directory with '__init__.py') matches the
pattern then the package will be checked for a 'load_tests' function. If
this exists then it will be called with (loader, tests, pattern) unless
the package has already had load_tests called from the same discovery
invocation, in which case the package module object is not scanned for
tests - this ensures that when a package uses discover to further
discover child tests that infinite recursion does not happen.
If load_tests exists then discovery does *not* recurse into the package,
load_tests is responsible for loading all tests in the package.
The pattern is deliberately not stored as a loader attribute so that
packages can continue discovery themselves. top_level_dir is stored so
load_tests does not need to pass this argument in to loader.discover().
Paths are sorted before being imported to ensure reproducible execution
order even on filesystems with non-alphabetical ordering like ext3/4.
FNTr z __init__.pyrQ ) namespacez2Can not use builtin modules as dotted module namesz$don't know how to discover from {!r}z%Start directory is not importable: %r)!_top_level_dirr+ r, abspathsysinsertisdirisfilerU rV r modulesrT dirname__file__rW __spec__loadersubmodule_search_locationsrR re r replacesepextend _find_testsbuiltin_module_namesr5 rB _get_directory_containing_moduleremoverj r ) r start_dirr; Z
top_level_dirZset_implicit_topZis_not_importable is_namespacerK Z
the_moduleZtop_partspecr, r r r discover sv
zTestLoader.discoverc C sR t j| }tjj|j}tjj|j jdrBtjj tjj |S tjj |S d S )Nz __init__.py)
rt rx r+ r, rs rz basenamer) re ry )r r] rI full_pathr r r r X s
z+TestLoader._get_directory_containing_modulec C sB || j krdS ttjj|}tjj|| j }|jtjjd}|S )NrQ )rr r. r+ r, normpathrelpathr~ r )r r, Z_relpathr r r r _get_name_from_pathd s
z TestLoader._get_name_from_pathc C s t | tj| S )N)rV rt rx )r r r r r _get_module_from_namep s z TestLoader._get_module_from_namec C s
t ||S )N)r )r r, r r; r r r _match_patht s zTestLoader._match_pathFc
c s | j |}|dkrD|| jkrD| j|||\}}|dk r<|V |sDdS ttj|}x||D ]t}tjj||} | j| ||\}}|dk r|V |rX| j | }| jj| z| j | ||E dH W d| jj
| X qXW dS )z/Used by discovery. Yields test suites it loads.rQ N) r r2 _find_test_pathrC r+ listdirr, rU addr discard)
r r r; rq r rK Zshould_recursepathsr, r r r r r x s*
zTestLoader._find_testsc C sP t jj|}t jj|rFtj|s(d S | j|||s:d
S | j|}y| j|}W nX t j
k
r } zt ||| j dfS d}~X n t
|| j \}} | jj| |dfS t jjt|d|}
tt jj|
} tt jj|} | j | j kr0t jj| }
tt jj|}t jj|}d}t|||
|f | j||ddfS nt jj|rH| rzt jjt jj|d rzd S d}d}| j|}y| j|}W nZ t j
k
r } zt ||| j dfS d}~X nz t
|| j \}} | jj| |dfS t|dd}| jj| z(| j||d}|dk r0|dfS |dfS | jj | X nd S dS )
zUsed by discovery.
Loads tests from a single file, or a directories' __init__.py when
passed the directory.
Returns a tuple (None_or_tests_from_file, should_recurse).
NFrz zW%r module incorrectly imported from %r. Expected %r. Is this module globally installed?)r; z __init__.pyr< T)NF)NF)NF)NF) r+ r, r rw VALID_MODULE_NAMEmatchr r r r ZSkipTestr' r r r0 rG rs rE r. realpathr) ry r rP rv rU r2 r r )r r r; rq r r rI rM rN rO Zmod_filer Zfullpath_noextZ
module_dirZmod_nameZ expected_dirmsgr< rK packager r r r sl
zTestLoader._find_test_path)N)N)rp N)F)F)r r r __doc__ri staticmethodr
three_way_cmprl r r4 r rr r
r: rP r` rd r6 r r r r r r r r r r )r
r r/ B s$
(
N
n
"r/ c C s t }||_| |_|r ||_|S )N)r/ rl ri r )rf sortUsingr r| r r r _makeLoader s r c C s t ||j| S )N)r r6 )r9 rf r r r r r6 s r6 r! c C s t |||j| S )N)r r: )r9 rf r r r r r makeSuite s r c C s t |||j| S )N)r rP )rI rf r r r r r
findTestCases s r )N)!r r+ rert r rX rn r> r r r r Z
__unittestcompile
IGNORECASEr r& r r r r r' r. objectr/ ZdefaultTestLoaderr r r6 r4 r r r r r r s8 '
PK ! bC+ C+ loader.cpython-36.opt-2.pycnu [ 3
\V @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZ ddlm Z m
Z
m Z dZ ej
dejZG dd de jZd d
Zd d Zd
d Zdd Zdd ZG dd deZe Zd ddZe jfddZ de je
j fddZ de je
j fd d Z dS ) N)fnmatch )casesuiteutilTz[_a-z]\w*\.py$c s, e Zd ZdZ fddZ fddZ ZS ) _FailedTestNc s || _ tt| j| d S )N)
_exceptionsuperr __init__)selfZ method_name exception) __class__ '/usr/lib64/python3.6/unittest/loader.pyr
s z_FailedTest.__init__c s* | j krtt j|S fdd}|S )Nc s
j d S )N)r r )r r r testFailure! s z,_FailedTest.__getattr__..testFailure)_testMethodNamer r __getattr__)r namer )r
)r r r s
z_FailedTest.__getattr__)__name__
__module__ __qualname__r r
r
__classcell__r r )r
r r s r c C s" d| t j f }t| t|||S )Nz#Failed to import test module: %s
%s) traceback
format_exc_make_failed_test ImportError)r
suiteClassmessager r r _make_failed_import_test& s r c C s dt j f }t| |||S )Nz Failed to call load_tests:
%s)r r r )r r r r r r r _make_failed_load_tests+ s r c C s t | |}||f|fS )N)r )
methodnamer r r testr r r r 0 s
r c C s< t jt|dd }| |i}tdt jf|}||| fS )Nc S s d S )Nr )r r r r testSkipped5 s z'_make_skipped_test..testSkippedZ
ModuleSkipped)r skipstrtypeTestCase)r r r r" ZattrsZ TestClassr r r _make_skipped_test4 s r' c C s* | j jdr| d d S tjj| d S )Nz $py.class r i)lowerendswithospathsplitext)r, r r r _jython_aware_splitext< s r. c s e Zd ZdZeejZej Z
dZ fddZ dd Z
dddd Zd d
d Zd!d d
Zdd Zd"ddZdd Zdd Zdd Zdd Zd#d d Zd$d d Z ZS )%
TestLoaderr! Nc s t t| j g | _t | _d S )N)r r/ r
errorsset_loading_packages)r )r
r r r
L s zTestLoader.__init__c C sH t |tjrtd| j|}| r4t|dr4dg}| jt||}|S )NzYTest cases should not be derived from TestSuite. Maybe you meant to derive from TestCase?ZrunTest)
issubclassr TestSuite TypeErrorgetTestCaseNameshasattrr map)r
testCaseClassZ
testCaseNamesZ loaded_suiter r r loadTestsFromTestCaseS s
z TestLoader.loadTestsFromTestCase)patternc
O s2 t |dksd|kr,tjdt |jdd t |dkrRt |d }tdj|t |dkrxt|d }tdj|g }x@t|D ]4}t ||}t
|t rt |t
jr|j| j| qW t |dd } | j|}| d k r.y | | ||S tk
r, }
z$t|j|
| j\} } | jj| | S d }
~
X nX |S )Nr Zuse_load_testsz(use_load_tests is deprecated and ignoredr zCloadTestsFromModule() takes 1 positional argument but {} were givenz=loadTestsFromModule() got an unexpected keyword argument '{}'
load_tests)lenwarningswarnDeprecationWarningpopr5 formatsorteddirgetattr
isinstancer% r3 r r& appendr: r Exceptionr r r0 )
r moduler; argsZkwsZ complainttestsr objr< e
error_case
error_messager r r loadTestsFromModulea s4
z TestLoader.loadTestsFromModulec C s> |j d}d\}}|d kr|d d }xb|rydj|}t|}P W q( tk
r |j }t|| j\}}|s| jj| |S Y q(X q(W |dd }|} x|D ]}
y| t | |
} } W q t
k
r2 } z\t | dd d k r|d k r| jj| |S t |
| | jdt j
f \}}| jj| |S W Y d d } ~ X qX qW t| tjrP| j| S t| trtt| tjrt| j| S t| tjrt| trt| tjr|d }| |}
tt |
|tjs| j|
gS nt| tjr| S t| r.| }t|tjr|S t|tjr | j|gS td| |f n td| d S ) N.r __path__z Failed to access attribute:
%sz"calling %s returned %s, not a testz$don't know how to make test from: %s)NN) splitjoin
__import__r rA r r r0 rG rE AttributeErrorr r r rF types
ModuleTyperP r% r3 r r& r: FunctionTyper r4 callabler5 )r r rI partsrN rO Z
parts_copy module_nameZnext_attributerL partparentrM Zinstr! r r r loadTestsFromName sl
z TestLoader.loadTestsFromNamec s fdd|D }j |S )Nc s g | ]}j | qS r )r` ).0r )rI r r r
s z1TestLoader.loadTestsFromNames..)r )r namesrI Zsuitesr )rI r r loadTestsFromNames s z TestLoader.loadTestsFromNamesc C s@ || j fdd}tt|t|}| jr<|jtj| jd |S )Nc S s | j |ott|| S )N)
startswithr[ rE )Zattrnamer9 prefixr r r isTestMethod s
z1TestLoader.getTestCaseNames..isTestMethod)key)testMethodPrefixlistfilterrD sortTestMethodsUsingsort functools
cmp_to_key)r r9 rg Z testFnNamesr r r r6 s zTestLoader.getTestCaseNamestest*.pyc C sR d}|d kr | j d k r | j }n|d kr.d}|}tjj|}|tjkrRtjjd| || _ d}d}g }tjjtjj|rtjj|}||krtjjtjj|d }nxy t | W n t
k
r d}Y nRX tj | }|j dd } ytjjtjj
|j}W n tk
r y
|j}
W n tk
r8 d }
Y nX |
r|
jd kr|
jd k rd}x|jD ]T} | r| j| rqd| j |jjdtjjd | _ |j| j| |dd qdW n*|jtjkrtdd ntdj |d Y nX |r |s| j | | _ tjj | n tjj | |r2t
d | |sHt | j||}| j |S )
NFTr z __init__.pyrQ ) namespacez2Can not use builtin modules as dotted module namesz$don't know how to discover from {!r}z%Start directory is not importable: %r)!_top_level_dirr+ r, abspathsysinsertisdirisfilerU rV r modulesrT dirname__file__rW __spec__loadersubmodule_search_locationsrR re r replacesepextend _find_testsbuiltin_module_namesr5 rB _get_directory_containing_moduleremoverj r ) r start_dirr; Z
top_level_dirZset_implicit_topZis_not_importable is_namespacerK Z
the_moduleZtop_partspecr, r r r discover sv
zTestLoader.discoverc C sR t j| }tjj|j}tjj|j jdrBtjj tjj |S tjj |S d S )Nz __init__.py)
rt rx r+ r, rs rz basenamer) re ry )r r] rI full_pathr r r r X s
z+TestLoader._get_directory_containing_modulec C sB || j krdS ttjj|}tjj|| j }|jtjjd}|S )NrQ )rr r. r+ r, normpathrelpathr~ r )r r, Z_relpathr r r r _get_name_from_pathd s
z TestLoader._get_name_from_pathc C s t | tj| S )N)rV rt rx )r r r r r _get_module_from_namep s z TestLoader._get_module_from_namec C s
t ||S )N)r )r r, r r; r r r _match_patht s zTestLoader._match_pathFc
c s | j |}|dkrD|| jkrD| j|||\}}|d k r<|V |sDd S ttj|}x||D ]t}tjj||} | j| ||\}}|d k r|V |rX| j | }| jj| z| j | ||E d H W d | jj
| X qXW d S )NrQ ) r r2 _find_test_pathrC r+ listdirr, rU addr discard)
r r r; rq r rK Zshould_recursepathsr, r r r r r x s*
zTestLoader._find_testsc C sP t jj|}t jj|rFtj|s(dS | j|||s:d S | j|}y| j|}W nX t j
k
r } zt ||| j dfS d }~X n t
|| j \}} | jj| |dfS t jjt|d|}
tt jj|
} tt jj|} | j | j kr0t jj| }
tt jj|}t jj|}d}t|||
|f | j||ddfS nt jj|rH| rzt jjt jj|d rzd
S d }d }| j|}y| j|}W nZ t j
k
r } zt ||| j dfS d }~X nz t
|| j \}} | jj| |dfS t|dd }| jj| z(| j||d}|d k r0|dfS |dfS | jj | X nd S d S ) NFrz zW%r module incorrectly imported from %r. Expected %r. Is this module globally installed?)r; z __init__.pyr< T)NF)NF)NF)NF) r+ r, r rw VALID_MODULE_NAMEmatchr r r r ZSkipTestr' r r r0 rG rs rE r. realpathr) ry r rP rv rU r2 r r )r r r; rq r r rI rM rN rO Zmod_filer Zfullpath_noextZ
module_dirZmod_nameZ expected_dirmsgr< rK packager r r r sl
zTestLoader._find_test_path)N)N)rp N)F)F)r r r ri staticmethodr
three_way_cmprl r r4 r rr r
r: rP r` rd r6 r r r r r r r r r r )r
r r/ B s"
(
N
n
"r/ c C s t }||_| |_|r ||_|S )N)r/ rl ri r )rf sortUsingr r| r r r _makeLoader s r c C s t ||j| S )N)r r6 )r9 rf r r r r r6 s r6 r! c C s t |||j| S )N)r r: )r9 rf r r r r r makeSuite s r c C s t |||j| S )N)r rP )rI rf r r r r r
findTestCases s r )N) r+ rert r rX rn r> r r r r Z
__unittestcompile
IGNORECASEr r& r r r r r' r. objectr/ ZdefaultTestLoaderr r r6 r4 r r r r r r s6 '
PK ! ,6 ,6 loader.cpython-36.pycnu [ 3
\V @ s d Z ddlZddlZddlZddlZddlZddlZddlZddlmZ ddl m
Z
m Z m Z dZ
ejdejZG dd d e
jZd
d Zd d
Zdd Zdd Zdd ZG dd deZe Zd ddZe j fddZ de j e j fdd Z de j e j fd d Z dS ) zLoading unittests. N)fnmatch )casesuiteutilTz[_a-z]\w*\.py$c s, e Zd ZdZ fddZ fddZ ZS ) _FailedTestNc s || _ tt| j| d S )N)
_exceptionsuperr __init__)selfZ method_name exception) __class__ '/usr/lib64/python3.6/unittest/loader.pyr
s z_FailedTest.__init__c s* | j krtt j|S fdd}|S )Nc s
j d S )N)r r )r r r testFailure! s z,_FailedTest.__getattr__..testFailure)_testMethodNamer r __getattr__)r namer )r
)r r r s
z_FailedTest.__getattr__)__name__
__module__ __qualname__r r
r
__classcell__r r )r
r r s r c C s" d| t j f }t| t|||S )Nz#Failed to import test module: %s
%s) traceback
format_exc_make_failed_test ImportError)r
suiteClassmessager r r _make_failed_import_test& s r c C s dt j f }t| |||S )Nz Failed to call load_tests:
%s)r r r )r r r r r r r _make_failed_load_tests+ s r c C s t | |}||f|fS )N)r )
methodnamer r r testr r r r 0 s
r c C s< t jt|dd }| |i}tdt jf|}||| fS )Nc S s d S )Nr )r r r r testSkipped5 s z'_make_skipped_test..testSkippedZ
ModuleSkipped)r skipstrtypeTestCase)r r r r" ZattrsZ TestClassr r r _make_skipped_test4 s r' c C s* | j jdr| d d S tjj| d S )Nz $py.class r i)lowerendswithospathsplitext)r, r r r _jython_aware_splitext< s r. c s e Zd ZdZdZeejZe j
Z dZ fddZ
dd Zddd d
Zd!d d Zd"d
dZdd Zd#ddZdd Zdd Zdd Zdd Zd$d d Zd%d d Z ZS )&
TestLoaderz
This class is responsible for loading tests according to various criteria
and returning them wrapped in a TestSuite
r! Nc s t t| j g | _t | _d S )N)r r/ r
errorsset_loading_packages)r )r
r r r
L s zTestLoader.__init__c C sH t |tjrtd| j|}| r4t|dr4dg}| jt||}|S )z;Return a suite of all test cases contained in testCaseClasszYTest cases should not be derived from TestSuite. Maybe you meant to derive from TestCase?ZrunTest)
issubclassr TestSuite TypeErrorgetTestCaseNameshasattrr map)r
testCaseClassZ
testCaseNamesZ loaded_suiter r r loadTestsFromTestCaseS s
z TestLoader.loadTestsFromTestCase)patternc
O s2 t |dksd|kr,tjdt |jdd t |dkrRt |d }tdj|t |dkrxt|d }tdj|g }x@t|D ]4}t ||}t
|t rt |t
jr|j| j| qW t |dd} | j|}| dk r.y | | ||S tk
r, }
z$t|j|
| j\} } | jj| | S d}
~
X nX |S ) z>Return a suite of all test cases contained in the given moduler Zuse_load_testsz(use_load_tests is deprecated and ignoredNr zCloadTestsFromModule() takes 1 positional argument but {} were givenz=loadTestsFromModule() got an unexpected keyword argument '{}'
load_tests)lenwarningswarnDeprecationWarningpopr5 formatsorteddirgetattr
isinstancer% r3 r r& appendr: r Exceptionr r r0 )
r moduler; argsZkwsZ complainttestsr objr< e
error_case
error_messager r r loadTestsFromModulea s4
z TestLoader.loadTestsFromModulec C s> |j d}d\}}|dkr|dd }xb|rydj|}t|}P W q( tk
r |j }t|| j\}}|s| jj| |S Y q(X q(W |dd }|} x|D ]}
y| t | |
} } W q t
k
r2 } z\t | dddk r|dk r| jj| |S t |
| | jdt j
f \}}| jj| |S W Y dd} ~ X qX qW t| tjrP| j| S t| trtt| tjrt| j| S t| tjrt| trt| tjr|d }| |}
tt |
|tjs| j|
gS nt| tjr| S t| r.| }t|tjr|S t|tjr | j|gS td| |f n td| dS )
aS Return a suite of all test cases given a string specifier.
The name may resolve either to a module, a test case class, a
test method within a test case class, or a callable object which
returns a TestCase or TestSuite instance.
The method optionally resolves the names relative to a given module.
.Nr __path__z Failed to access attribute:
%sz"calling %s returned %s, not a testz$don't know how to make test from: %s)NN) splitjoin
__import__r rA r r r0 rG rE AttributeErrorr r r rF types
ModuleTyperP r% r3 r r& r: FunctionTyper r4 callabler5 )r r rI partsrN rO Z
parts_copy module_nameZnext_attributerL partparentrM Zinstr! r r r loadTestsFromName sl
z TestLoader.loadTestsFromNamec s fdd|D }j |S )zReturn a suite of all test cases found using the given sequence
of string specifiers. See 'loadTestsFromName()'.
c s g | ]}j | qS r )r` ).0r )rI r r r
s z1TestLoader.loadTestsFromNames..)r )r namesrI Zsuitesr )rI r r loadTestsFromNames s z TestLoader.loadTestsFromNamesc C s@ || j fdd}tt|t|}| jr<|jtj| jd |S )zLReturn a sorted sequence of method names found within testCaseClass
c S s | j |ott|| S )N)
startswithr[ rE )Zattrnamer9 prefixr r r isTestMethod s
z1TestLoader.getTestCaseNames..isTestMethod)key)testMethodPrefixlistfilterrD sortTestMethodsUsingsort functools
cmp_to_key)r r9 rg Z testFnNamesr r r r6 s zTestLoader.getTestCaseNamestest*.pyc C sR d}|dkr | j dk r | j }n|dkr.d}|}tjj|}|tjkrRtjjd| || _ d}d}g }tjjtjj|rtjj|}||krtjjtjj|d }nxy t | W n t
k
r d}Y nRX tj | }|j dd } ytjjtjj
|j}W n tk
r y
|j}
W n tk
r8 d}
Y nX |
r|
jdkr|
jdk rd}x|jD ]T} | r| j| rqd| j |jjdtjjd | _ |j| j| |dd qdW n*|jtjkrtddntd j |dY nX |r |s| j | | _ tjj | n tjj | |r2t
d
| |sHt | j||}| j |S ) a% Find and return all test modules from the specified start
directory, recursing into subdirectories to find them and return all
tests found within them. Only test files that match the pattern will
be loaded. (Using shell style pattern matching.)
All test modules must be importable from the top level of the project.
If the start directory is not the top level directory then the top
level directory must be specified separately.
If a test package name (directory with '__init__.py') matches the
pattern then the package will be checked for a 'load_tests' function. If
this exists then it will be called with (loader, tests, pattern) unless
the package has already had load_tests called from the same discovery
invocation, in which case the package module object is not scanned for
tests - this ensures that when a package uses discover to further
discover child tests that infinite recursion does not happen.
If load_tests exists then discovery does *not* recurse into the package,
load_tests is responsible for loading all tests in the package.
The pattern is deliberately not stored as a loader attribute so that
packages can continue discovery themselves. top_level_dir is stored so
load_tests does not need to pass this argument in to loader.discover().
Paths are sorted before being imported to ensure reproducible execution
order even on filesystems with non-alphabetical ordering like ext3/4.
FNTr z __init__.pyrQ ) namespacez2Can not use builtin modules as dotted module namesz$don't know how to discover from {!r}z%Start directory is not importable: %r)!_top_level_dirr+ r, abspathsysinsertisdirisfilerU rV r modulesrT dirname__file__rW __spec__loadersubmodule_search_locationsrR re r replacesepextend _find_testsbuiltin_module_namesr5 rB _get_directory_containing_moduleremoverj r ) r start_dirr; Z
top_level_dirZset_implicit_topZis_not_importable is_namespacerK Z
the_moduleZtop_partspecr, r r r discover sv
zTestLoader.discoverc C sR t j| }tjj|j}tjj|j jdrBtjj tjj |S tjj |S d S )Nz __init__.py)
rt rx r+ r, rs rz basenamer) re ry )r r] rI full_pathr r r r X s
z+TestLoader._get_directory_containing_modulec C sl || j krdS ttjj|}tjj|| j }tjj| sDtd|jd sXtd|j tjj
d}|S )NrQ z Path must be within the projectz..) rr r. r+ r, normpathrelpathisabsAssertionErrorre r~ r )r r, Z_relpathr r r r _get_name_from_pathd s
z TestLoader._get_name_from_pathc C s t | tj| S )N)rV rt rx )r r r r r _get_module_from_namep s z TestLoader._get_module_from_namec C s
t ||S )N)r )r r, r r; r r r _match_patht s zTestLoader._match_pathFc
c s | j |}|dkrD|| jkrD| j|||\}}|dk r<|V |sDdS ttj|}x||D ]t}tjj||} | j| ||\}}|dk r|V |rX| j | }| jj| z| j | ||E dH W d| jj
| X qXW dS )z/Used by discovery. Yields test suites it loads.rQ N) r r2 _find_test_pathrC r+ listdirr, rU addr discard)
r r r; rq r rK Zshould_recursepathsr, r r r r r x s*
zTestLoader._find_testsc C sP t jj|}t jj|rFtj|s(d S | j|||s:d
S | j|}y| j|}W nX t j
k
r } zt ||| j dfS d}~X n t
|| j \}} | jj| |dfS t jjt|d|}
tt jj|
} tt jj|} | j | j kr0t jj| }
tt jj|}t jj|}d}t|||
|f | j||ddfS nt jj|rH| rzt jjt jj|d rzd S d}d}| j|}y| j|}W nZ t j
k
r } zt ||| j dfS d}~X nz t
|| j \}} | jj| |dfS t|dd}| jj| z(| j||d}|dk r0|dfS |dfS | jj | X nd S dS )
zUsed by discovery.
Loads tests from a single file, or a directories' __init__.py when
passed the directory.
Returns a tuple (None_or_tests_from_file, should_recurse).
NFrz zW%r module incorrectly imported from %r. Expected %r. Is this module globally installed?)r; z __init__.pyr< T)NF)NF)NF)NF) r+ r, r rw VALID_MODULE_NAMEmatchr r r r ZSkipTestr' r r r0 rG rs rE r. realpathr) ry r rP rv rU r2 r r )r r r; rq r r rI rM rN rO Zmod_filer Zfullpath_noextZ
module_dirZmod_nameZ expected_dirmsgr< rK packager r r r sl
zTestLoader._find_test_path)N)N)rp N)F)F)r r r __doc__ri staticmethodr
three_way_cmprl r r4 r rr r
r: rP r` rd r6 r r r r r r r r r r )r
r r/ B s$
(
N
n
"r/ c C s t }||_| |_|r ||_|S )N)r/ rl ri r )rf sortUsingr r| r r r _makeLoader s r c C s t ||j| S )N)r r6 )r9 rf r r r r r6 s r6 r! c C s t |||j| S )N)r r: )r9 rf r r r r r makeSuite s r c C s t |||j| S )N)r rP )rI rf r r r r r
findTestCases s r )N)!r r+ rert r rX rn r> r r r r Z
__unittestcompile
IGNORECASEr r& r r r r r' r. objectr/ ZdefaultTestLoaderr r r6 r4 r r r r r r s8 '
PK ! vf f main.cpython-36.opt-1.pycnu [ 3
\8) @ sl d Z ddlZddlZddlZddlmZmZ ddlmZ dZ dZ
dZ d d
Z d d Z
G d
d deZeZdS )zUnittest main program N )loaderrunner)installHandlerTa Examples:
%(prog)s test_module - run tests from test_module
%(prog)s module.TestClass - run tests from module.TestClass
%(prog)s module.Class.test_method - run specified test method
%(prog)s path/to/test_file.py - run tests from test_file.py
aF Examples:
%(prog)s - run default set of tests
%(prog)s MyTestSuite - run suite 'MyTestSuite'
%(prog)s MyTestCase.testSomething - run MyTestCase.testSomething
%(prog)s MyTestCase - run all 'test*' test methods
in MyTestCase
c C sx t jj| rt| j jdrtt jj| rXt jj| t j }t jj|sP|jt j rT| S |} | d d j
ddj
ddS | S )Nz.py \./) ospathisfilelowerendswithisabsrelpathgetcwd
startswithpardirreplace)nameZrel_path r %/usr/lib64/python3.6/unittest/main.py
_convert_name s r c C s dd | D S )Nc S s g | ] }t |qS r )r ).0r r r r
. s z"_convert_names..r )namesr r r _convert_names- s r c @ s e Zd ZdZdZdZd Z Z Z Z Z
dZ dddde j
ddddddf dddd Zd d
d Zd d
Zdd Zdd Zdd Zdd Zdd Zdd Zd ddZd d ZdS ) TestProgramzA command-line program that runs a set of tests; this is primarily
for making test modules conveniently executable.
Nr __main__TF) tb_localsc C s t |tr@t|| _x0|jddd D ]}
t| j|
| _q(W n|| _|d krTtj}|| _|| _ | | _
|| _ |
| _ | | _
| d krtj rd| _n| | _|| _|| _|| _tjj|d | _| j| | j d S )Nr r defaultr )
isinstancestr
__import__modulesplitgetattrsysargvexitfailfast
catchbreak verbositybufferr warnoptionswarnings defaultTest
testRunner
testLoaderr r basenameprogName parseArgsrunTests)selfr% r1 r) r2 r3 r* r- r+ r, r. r0 r partr r r __init__; s,
zTestProgram.__init__c C s4 |r t | | jd kr | j | j tjd d S )N )print_discovery_parser_initArgParsers _print_helpr( r* )r8 msgr r r usageExita s
zTestProgram.usageExitc O sZ | j d kr6t| jj ttd| ji | jj n t| jj ttd| ji d S )Nprog) r% r< _main_parserZ format_help
MAIN_EXAMPLESr5 r=
print_helpMODULE_EXAMPLES)r8 argskwargsr r r r? i s
zTestProgram._print_helpc C s | j | jd krpt|dkrD|d j dkrD| j|dd d S | jj|dd | | js| jg d S n| jj|dd | | jrt| j| _ t
dkrd | _n6| j d krd | _ n$t | j t
r| j f| _ n t| j | _ | j d S )Nr discoverr; r )r> r% lenr
_do_discoveryrC
parse_argstestsr testNames__name__r1 r" r# list createTests)r8 r) r r r r6 r s(
zTestProgram.parseArgsc C s4 | j d kr | jj| j| _n| jj| j | j| _d S )N)rN r3 ZloadTestsFromModuler% testZloadTestsFromNames)r8 r r r rQ s
zTestProgram.createTestsc C s$ | j }| j|| _| j|| _d S )N)_getParentArgParser_getMainArgParserrC _getDiscoveryArgParserr= )r8 Z
parent_parserr r r r> s zTestProgram._initArgParsersc C s t jdd}|jddddddd |jd
d ddd d
d |jddddd | jd krn|jdddddd d| _| jd kr|jdddddd d| _| jd kr|jdd d dd d d| _|S ) NF)Zadd_helpz-vz --verboser- Z store_constr; zVerbose output)destactionZconsthelpz-qz--quietr z Quiet outputz--localsr
store_truez"Show local variables in tracebacks)rV rW rX z-fz
--failfastr+ zStop on first fail or errorz-cz--catchr, z'Catch Ctrl-C and display results so farz-bz--bufferr. z%Buffer stdout and stderr during tests)argparseArgumentParser add_argumentr+ r, r. )r8 parserr r r rS s4
z TestProgram._getParentArgParserc C s2 t j|gd}| j|_| j|_|jdddd |S )N)parentsrM *z?a list of any number of test modules, classes and test methods.)nargsrX )rZ r[ r5 rB r? rE r\ )r8 parentr] r r r rT s z TestProgram._getMainArgParserc C s~ t j|gd}d| j |_d|_|jddddd |jd d
d d d |jd
dddd x"dD ]}|j|dt jt jd q\W |S )N)r^ z %s discoverzcFor test discovery all test modules must be importable from the top level directory of the project.z-sz--start-directorystartz*Directory to start discovery ('.' default))rV rX z-pz --patternpatternz+Pattern to match tests ('test*.py' default)z-tz--top-level-directorytopz rL r3 rI rR )r8 r) Loaderr r r r rK s
zTestProgram._do_discoveryc C s | j r t | jd kr tj| _t| jtryVy"| j| j| j| j | j
| j d}W n. t k
r| | j| j| j| j | j
d}Y nX W q t k
r | j }Y qX n| j}|j
| j| _| jrtj| jj d S )N)r- r+ r. r0 r )r- r+ r. r0 )r, r r2 r ZTextTestRunnerr" typer- r+ r. r0 r TypeErrorZrunrR resultr* r( Z
wasSuccessful)r8 r2 r r r r7 s.
zTestProgram.runTests)N)N)rO
__module__ __qualname____doc__r% r- r+ r, r. r5 r0 r= r ZdefaultTestLoaderr: rA r? r6 rQ r> rS rT rU rK r7 r r r r r 1 s& #
r )rm r( rZ r r r Zsignalsr Z
__unittestrD rF r r objectr mainr r r r s TPK ! eg main.cpython-36.opt-2.pycnu [ 3
\8) @ sh d dl Z d dlZd dlZddlmZmZ ddlmZ dZdZ dZ
dd Z d
d Z G d d
d
e
ZeZdS ) N )loaderrunner)installHandlerTa Examples:
%(prog)s test_module - run tests from test_module
%(prog)s module.TestClass - run tests from module.TestClass
%(prog)s module.Class.test_method - run specified test method
%(prog)s path/to/test_file.py - run tests from test_file.py
aF Examples:
%(prog)s - run default set of tests
%(prog)s MyTestSuite - run suite 'MyTestSuite'
%(prog)s MyTestCase.testSomething - run MyTestCase.testSomething
%(prog)s MyTestCase - run all 'test*' test methods
in MyTestCase
c C sx t jj| rt| j jdrtt jj| rXt jj| t j }t jj|sP|jt j rT| S |} | d d j
ddj
ddS | S )Nz.py \./) ospathisfilelowerendswithisabsrelpathgetcwd
startswithpardirreplace)nameZrel_path r %/usr/lib64/python3.6/unittest/main.py
_convert_name s r c C s dd | D S )Nc S s g | ] }t |qS r )r ).0r r r r
. s z"_convert_names..r )namesr r r _convert_names- s r c @ s e Zd ZdZdZd Z Z Z ZZ dZ
dddde j ddddddf ddddZ
d d d
Zd d Zd
d Zdd Zdd Zdd Zdd Zdd Zd ddZdd ZdS ) TestProgramNr __main__TF) tb_localsc C s t |tr@t|| _x0|jddd D ]}
t| j|
| _q(W n|| _|d krTtj}|| _|| _ | | _
|| _ |
| _ | | _
| d krtj rd| _n| | _|| _|| _|| _tjj|d | _| j| | j d S )Nr r defaultr )
isinstancestr
__import__modulesplitgetattrsysargvexitfailfast
catchbreak verbositybufferr warnoptionswarnings defaultTest
testRunner
testLoaderr r basenameprogName parseArgsrunTests)selfr% r1 r) r2 r3 r* r- r+ r, r. r0 r partr r r __init__; s,
zTestProgram.__init__c C s4 |r t | | jd kr | j | j tjd d S )N )print_discovery_parser_initArgParsers _print_helpr( r* )r8 msgr r r usageExita s
zTestProgram.usageExitc O sZ | j d kr6t| jj ttd| ji | jj n t| jj ttd| ji d S )Nprog) r% r< _main_parserZ format_help
MAIN_EXAMPLESr5 r=
print_helpMODULE_EXAMPLES)r8 argskwargsr r r r? i s
zTestProgram._print_helpc C s | j | jd krpt|dkrD|d j dkrD| j|dd d S | jj|dd | | js| jg d S n| jj|dd | | jrt| j| _ t
dkrd | _n6| j d krd | _ n$t | j t
r| j f| _ n t| j | _ | j d S )Nr discoverr; r )r> r% lenr
_do_discoveryrC
parse_argstestsr testNames__name__r1 r" r# list createTests)r8 r) r r r r6 r s(
zTestProgram.parseArgsc C s4 | j d kr | jj| j| _n| jj| j | j| _d S )N)rN r3 ZloadTestsFromModuler% testZloadTestsFromNames)r8 r r r rQ s
zTestProgram.createTestsc C s$ | j }| j|| _| j|| _d S )N)_getParentArgParser_getMainArgParserrC _getDiscoveryArgParserr= )r8 Z
parent_parserr r r r> s zTestProgram._initArgParsersc C s t jdd}|jddddddd |jd
d ddd d
d |jddddd | jd krn|jdddddd d| _| jd kr|jdddddd d| _| jd kr|jdd d dd d d| _|S ) NF)Zadd_helpz-vz --verboser- Z store_constr; zVerbose output)destactionZconsthelpz-qz--quietr z Quiet outputz--localsr
store_truez"Show local variables in tracebacks)rV rW rX z-fz
--failfastr+ zStop on first fail or errorz-cz--catchr, z'Catch Ctrl-C and display results so farz-bz--bufferr. z%Buffer stdout and stderr during tests)argparseArgumentParser add_argumentr+ r, r. )r8 parserr r r rS s4
z TestProgram._getParentArgParserc C s2 t j|gd}| j|_| j|_|jdddd |S )N)parentsrM *z?a list of any number of test modules, classes and test methods.)nargsrX )rZ r[ r5 rB r? rE r\ )r8 parentr] r r r rT s z TestProgram._getMainArgParserc C s~ t j|gd}d| j |_d|_|jddddd |jd d
d d d |jd
dddd x"dD ]}|j|dt jt jd q\W |S )N)r^ z %s discoverzcFor test discovery all test modules must be importable from the top level directory of the project.z-sz--start-directorystartz*Directory to start discovery ('.' default))rV rX z-pz --patternpatternz+Pattern to match tests ('test*.py' default)z-tz--top-level-directorytopz rL r3 rI rR )r8 r) Loaderr r r r rK s
zTestProgram._do_discoveryc C s | j r t | jd kr tj| _t| jtryVy"| j| j| j| j | j
| j d}W n. t k
r| | j| j| j| j | j
d}Y nX W q t k
r | j }Y qX n| j}|j
| j| _| jrtj| jj d S )N)r- r+ r. r0 r )r- r+ r. r0 )r, r r2 r ZTextTestRunnerr" typer- r+ r. r0 r TypeErrorZrunrR resultr* r( Z
wasSuccessful)r8 r2 r r r r7 s.
zTestProgram.runTests)N)N)rO
__module__ __qualname__r% r- r+ r, r. r5 r0 r= r ZdefaultTestLoaderr: rA r? r6 rQ r> rS rT rU rK r7 r r r r r 1 s$ #
r )r( rZ r r r Zsignalsr Z
__unittestrD rF r r objectr mainr r r r s TPK ! vf f main.cpython-36.pycnu [ 3
\8) @ sl d Z ddlZddlZddlZddlmZmZ ddlmZ dZ dZ
dZ d d
Z d d Z
G d
d deZeZdS )zUnittest main program N )loaderrunner)installHandlerTa Examples:
%(prog)s test_module - run tests from test_module
%(prog)s module.TestClass - run tests from module.TestClass
%(prog)s module.Class.test_method - run specified test method
%(prog)s path/to/test_file.py - run tests from test_file.py
aF Examples:
%(prog)s - run default set of tests
%(prog)s MyTestSuite - run suite 'MyTestSuite'
%(prog)s MyTestCase.testSomething - run MyTestCase.testSomething
%(prog)s MyTestCase - run all 'test*' test methods
in MyTestCase
c C sx t jj| rt| j jdrtt jj| rXt jj| t j }t jj|sP|jt j rT| S |} | d d j
ddj
ddS | S )Nz.py \./) ospathisfilelowerendswithisabsrelpathgetcwd
startswithpardirreplace)nameZrel_path r %/usr/lib64/python3.6/unittest/main.py
_convert_name s r c C s dd | D S )Nc S s g | ] }t |qS r )r ).0r r r r
. s z"_convert_names..r )namesr r r _convert_names- s r c @ s e Zd ZdZdZdZd Z Z Z Z Z
dZ dddde j
ddddddf dddd Zd d
d Zd d
Zdd Zdd Zdd Zdd Zdd Zdd Zd ddZd d ZdS ) TestProgramzA command-line program that runs a set of tests; this is primarily
for making test modules conveniently executable.
Nr __main__TF) tb_localsc C s t |tr@t|| _x0|jddd D ]}
t| j|
| _q(W n|| _|d krTtj}|| _|| _ | | _
|| _ |
| _ | | _
| d krtj rd| _n| | _|| _|| _|| _tjj|d | _| j| | j d S )Nr r defaultr )
isinstancestr
__import__modulesplitgetattrsysargvexitfailfast
catchbreak verbositybufferr warnoptionswarnings defaultTest
testRunner
testLoaderr r basenameprogName parseArgsrunTests)selfr% r1 r) r2 r3 r* r- r+ r, r. r0 r partr r r __init__; s,
zTestProgram.__init__c C s4 |r t | | jd kr | j | j tjd d S )N )print_discovery_parser_initArgParsers _print_helpr( r* )r8 msgr r r usageExita s
zTestProgram.usageExitc O sZ | j d kr6t| jj ttd| ji | jj n t| jj ttd| ji d S )Nprog) r% r< _main_parserZ format_help
MAIN_EXAMPLESr5 r=
print_helpMODULE_EXAMPLES)r8 argskwargsr r r r? i s
zTestProgram._print_helpc C s | j | jd krpt|dkrD|d j dkrD| j|dd d S | jj|dd | | js| jg d S n| jj|dd | | jrt| j| _ t
dkrd | _n6| j d krd | _ n$t | j t
r| j f| _ n t| j | _ | j d S )Nr discoverr; r )r> r% lenr
_do_discoveryrC
parse_argstestsr testNames__name__r1 r" r# list createTests)r8 r) r r r r6 r s(
zTestProgram.parseArgsc C s4 | j d kr | jj| j| _n| jj| j | j| _d S )N)rN r3 ZloadTestsFromModuler% testZloadTestsFromNames)r8 r r r rQ s
zTestProgram.createTestsc C s$ | j }| j|| _| j|| _d S )N)_getParentArgParser_getMainArgParserrC _getDiscoveryArgParserr= )r8 Z
parent_parserr r r r> s zTestProgram._initArgParsersc C s t jdd}|jddddddd |jd
d ddd d
d |jddddd | jd krn|jdddddd d| _| jd kr|jdddddd d| _| jd kr|jdd d dd d d| _|S ) NF)Zadd_helpz-vz --verboser- Z store_constr; zVerbose output)destactionZconsthelpz-qz--quietr z Quiet outputz--localsr
store_truez"Show local variables in tracebacks)rV rW rX z-fz
--failfastr+ zStop on first fail or errorz-cz--catchr, z'Catch Ctrl-C and display results so farz-bz--bufferr. z%Buffer stdout and stderr during tests)argparseArgumentParser add_argumentr+ r, r. )r8 parserr r r rS s4
z TestProgram._getParentArgParserc C s2 t j|gd}| j|_| j|_|jdddd |S )N)parentsrM *z?a list of any number of test modules, classes and test methods.)nargsrX )rZ r[ r5 rB r? rE r\ )r8 parentr] r r r rT s z TestProgram._getMainArgParserc C s~ t j|gd}d| j |_d|_|jddddd |jd d
d d d |jd
dddd x"dD ]}|j|dt jt jd q\W |S )N)r^ z %s discoverzcFor test discovery all test modules must be importable from the top level directory of the project.z-sz--start-directorystartz*Directory to start discovery ('.' default))rV rX z-pz --patternpatternz+Pattern to match tests ('test*.py' default)z-tz--top-level-directorytopz rL r3 rI rR )r8 r) Loaderr r r r rK s
zTestProgram._do_discoveryc C s | j r t | jd kr tj| _t| jtryVy"| j| j| j| j | j
| j d}W n. t k
r| | j| j| j| j | j
d}Y nX W q t k
r | j }Y qX n| j}|j
| j| _| jrtj| jj d S )N)r- r+ r. r0 r )r- r+ r. r0 )r, r r2 r ZTextTestRunnerr" typer- r+ r. r0 r TypeErrorZrunrR resultr* r( Z
wasSuccessful)r8 r2 r r r r7 s.
zTestProgram.runTests)N)N)rO
__module__ __qualname____doc__r% r- r+ r, r. r5 r0 r= r ZdefaultTestLoaderr: rA r? r6 rQ r> rS rT rU rK r7 r r r r r 1 s& #
r )rm r( rZ r r r Zsignalsr Z
__unittestrD rF r r objectr mainr r r r s TPK ! g"\ \ mock.cpython-36.opt-1.pycnu [ 3
\7 @ sF dZ d
ZddlZddlZddlZddlZddlmZ ddlm Z m
Z
dd e eD Z e
fZdejkrzddlZe
ejjfZdZeZdd Zdd Zdd Zdd d Zd d Zd!d" Zd#d$ Z d%d& Z dd'd(Z d)d* Z d+d, Z G d-d. d.e!Z"G d/d0 d0e!Z#e# Z$e$j%Z%e$j&Z'e$j(Z)d1d2 Z*d3d4d5d6d7d8d9d:hZ+d;d< Z,G d=d> d>e-Z.d?d@ Z/G dAdB dBe!Z0G dCdD dDe!Z1G dEd d e1Z2dFdG Z3G dHdI dIe1Z4G dJd d e4e2Z5dKdL Z6dMdN Z7dOdP Z8G dQdR dRe!Z9dSdT Z:e%dd dddfdUdVZ;ddWdXZd\d] Z?d^d_ Z@e;e=_!e>e=_Ae# s r javaTc C s t t| tS )N)
issubclasstyper
)objr r r _is_instance_mock2 s r c C s t | tp t | to t| tS )N)
isinstanceBaseExceptionsr r )r r r r
_is_exception8 s
r c C s t | tr6| r6y
| j} W n tk
r. dS X d}n*t | ts`y
| j} W n tk
r^ dS X |rpt| d}n| }y| tj|fS t k
r dS X dS )z
Given an arbitrary, possibly callable object, try to create a suitable
signature object.
Return a (reduced func, signature) tuple, or None.
NT)
r r __init__AttributeError
FunctionTypes__call__r inspectZ signature
ValueError)funcZ as_instanceZeat_selfZsig_funcr r r _get_signature_object? s$
r( Fc sD t | || d krd S \} fdd}t| | |t|_d S )Nc s j || d S )N)bind)
_mock_selfargskwargs)sigr r checksigd s z"_check_signature..checksig)r( _copy_func_detailsr _mock_check_sig)r' mock skipfirstinstancer. r )r- r _check_signature_ s
r4 c # C s | j |_ | j|_y | j|_W n tk
r0 Y nX y | j|_W n tk
rR Y nX y | j|_W n tk
rt Y nX y | j|_W n tk
r Y nX d S )N)__name____doc____text_signature__r"
__module__ __defaults____kwdefaults__)r' funcopyr r r r/ j s$ r/ c C s& t | trdS t| dd d k r"dS dS )NTr$ F)r r getattr)r r r r _callable s
r= c C s t | ttfkS )N)r listtuple)r r r r _is_list s r@ c C sH t | tst| dddk S x(| f| j D ]}|jjddk r(dS q(W dS )ztGiven an object, return True if the object is callable.
For classes, return True if instances would be callable.r$ NTF)r r r< __mro____dict__get)r baser r r _instance_callable s
rE c s t |s d S t|t}t|||}|d kr.| S |\} fdd}t|| |j}|j s^d}|| d}d| } t| | || }
t|
| |
S )Nc s j | | d S )N)r) )r+ r, )r- r r r. s z _set_signature..checksigr; )Z
_checksig_r1 zYdef %s(*args, **kwargs):
_checksig_(*args, **kwargs)
return mock(*args, **kwargs)) r= r r r( r/ r5 isidentifierexec _setup_func) r1 originalr3 r2 resultr' r. r contextsrcr; r )r- r _set_signature s$
rM c
s _ tsd S fdd}fdd}fdd}fdd}fd d
}fd d }fd
d} fdd} d _d _d _t _t _t _j _ j
_
j _ | _ | _
| _| _| _| _| _| _ _d S )Nc s j | |S )N)assert_called_with)r+ r, )r1 r r rN s z'_setup_func..assert_called_withc s j | |S )N)
assert_called)r+ r, )r1 r r rO s z"_setup_func..assert_calledc s j | |S )N)assert_not_called)r+ r, )r1 r r rP s z&_setup_func..assert_not_calledc s j | |S )N)assert_called_once)r+ r, )r1 r r rQ s z'_setup_func..assert_called_oncec s j | |S )N)assert_called_once_with)r+ r, )r1 r r rR s z,_setup_func..assert_called_once_withc s j | |S )N)assert_has_calls)r+ r, )r1 r r rS s z%_setup_func..assert_has_callsc s j | |S )N)assert_any_call)r+ r, )r1 r r rT s z$_setup_func..assert_any_callc s: t _t _j j} t| r6| k r6| j d S )N) _CallList method_calls
mock_calls
reset_mock return_valuer )ret)r; r1 r r rX s z _setup_func..reset_mockFr )r1 r called
call_count call_argsrU call_args_listrV rW rY side_effect_mock_childrenrN rR rS rT rX rO rP rQ _mock_delegate)
r; r1 rN rO rP rQ rR rS rT rX r )r; r1 r rH s: rH c C s d| dd | kS )Nz__%s__ r )r r r r _is_magic s rd c @ s e Zd ZdZdd Zdd ZdS )_SentinelObjectz!A unique, named, sentinel object.c C s
|| _ d S )N)r )selfr r r r r! s z_SentinelObject.__init__c C s
d| j S )Nz sentinel.%s)r )rf r r r __repr__ s z_SentinelObject.__repr__N)r5 r8 __qualname__r6 r! rg r r r r re s re c @ s e Zd ZdZdd Zdd ZdS ) _SentinelzAAccess attributes to return a named object, usable as a sentinel.c C s
i | _ d S )N)
_sentinels)rf r r r r! s z_Sentinel.__init__c C s |dkr t | jj|t|S )N __bases__)r" rj
setdefaultre )rf r r r r __getattr__ s z_Sentinel.__getattr__N)r5 r8 rh r6 r! rm r r r r ri s ri c C s$ t | ttttfkr t | | S | S )N)r dictr> r? set)valuer r r _copy s rq rY _mock_return_valuer_ _mock_side_effect _mock_parent_mock_new_parent
_mock_name_mock_new_namec C s8 t j| d| }| |fdd}| |fdd}t||S )NZ_mock_c S s" | j }|d krt| |S t||S )N)ra r< )rf r _the_namer- r r r _get s
z"_delegating_property.._getc S s* | j }|d kr|| j|< n t||| d S )N)ra rB setattr)rf rp r rx r- r r r _set$ s z"_delegating_property.._set)_allowed_namesaddproperty)r rx ry r{ r r r _delegating_property s
r c @ s e Zd Zdd Zdd ZdS )rU c C sn t |tstj| |S t|}t| }||kr2dS x6td|| d D ] }| ||| }||krFdS qFW dS )NFr T)r r> __contains__lenrange)rf rp Z len_valueZlen_selfiZsub_listr r r r 1 s
z_CallList.__contains__c C s t jt| S )N)pprintZpformatr> )rf r r r rg ? s z_CallList.__repr__N)r5 r8 rh r rg r r r r rU / s rU c C sx t |s dS |js,|js,|jd k s,|jd k r0dS | }x |d k rR||krJdS |j}q6W |rd| |_||_|rt| |_||_dS )NFT)r rv rw rt ru )parentrp r new_name_parentr r r _check_and_set_parentC s$
r c @ s$ e Zd Zdd Zdd Zdd ZdS ) _MockIterc C s t || _d S )N)iterr )rf r r r r r! ] s z_MockIter.__init__c C s | S )Nr )rf r r r __iter___ s z_MockIter.__iter__c C s
t | jS )N)nextr )rf r r r __next__a s z_MockIter.__next__N)r5 r8 rh r! r r r r r r r \ s r c @ s e Zd ZeZdZdd ZdS )BaseNc O s d S )Nr )rf r+ r, r r r r! g s z
Base.__init__)r5 r8 rh r rr rs r! r r r r r d s r c @ sH e Zd ZdZdd ZdDddZd d
ZdEd d ZdFd
dZdd Z dd Z
dZ e e e
e Z
e dd ZedZedZedZedZedZdd Zd d Ze eeZdGddd d d!Zd"d# Zd$d% Zd&d' Zd(d) Z d*d+ Z d,d- Z d.d/ Z d0d1 Z d2d3 Z!d4d5 Z"d6d7 Z#d8d9 Z$d:d; Z%dd?Z'd@dA Z(dBdC Z)dS )Ir
z A non-callable version of `Mock`c O s$ t | j| fd| ji}tj|}|S )Nr6 )r r5 r6 object__new__)clsr+ kwnewr3 r r r r o s
zNonCallableMock.__new__N Fc K s |d kr |}| j }
||
d< ||
d< ||
d< ||
d< |d k rB|}d}|
d krR|d k }
| j||| |
i |
d< ||
d< d |
d< d |
d
< d |
d < d |
d
< t |
d< t |
d< t |
d< | |
d< | r| jf | tt| j|||||| d S )Nrt rv rw ru Tr` _mock_wrapsra FZ _mock_calledZ_mock_call_argsr Z_mock_call_countZ_mock_call_args_listZ_mock_mock_callsrV _mock_unsafe)rB _mock_add_specrU configure_mock _safe_superr
r! )rf specr r spec_setr _spec_state _new_name _new_parent_spec_as_instance _eat_selfZunsafer, rB r r r r! x s8
zNonCallableMock.__init__c C s( d|_ d|_d|_d|_t| || dS )z
Attach a mock as an attribute of this one, replacing its name and
parent. Calls to the attached mock will be recorded in the
`method_calls` and `mock_calls` attributes of this one.Nr )rt ru rv rw rz )rf r1 attributer r r attach_mock s
zNonCallableMock.attach_mockc C s | j || dS )zAdd a spec to a mock. `spec` can either be an object or a
list of strings. Only attributes on the `spec` can be fetched as
attributes from the mock.
If `spec_set` is True then only attributes on the spec can be set.N)r )rf r r r r r
mock_add_spec s z NonCallableMock.mock_add_specc C s| d }d }|d k rRt | rRt|tr*|}nt|}t|||}|oH|d }t|}| j}||d< ||d< ||d< ||d< d S )Nr _spec_class _spec_set_spec_signature
_mock_methods)r@ r r
_get_classr( dirrB ) rf r r r r r r resrB r r r r s
z NonCallableMock._mock_add_specc C s8 | j }| jd k r| jj}|tkr4| j| dd}|| _|S )Nz())r r )rr ra rY r _get_child_mock)rf rZ r r r Z__get_return_value s
z"NonCallableMock.__get_return_valuec C s, | j d k r|| j _n|| _t| |d d d S )Nz())ra rY rr r )rf rp r r r Z__set_return_value s
z"NonCallableMock.__set_return_valuez1The value to be returned when the mock is called.c C s | j d krt| S | j S )N)r r )rf r r r __class__ s
zNonCallableMock.__class__r[ r\ r] r^ rW c C sT | j }|d kr| jS |j}|d k rPt| rPt|t rPt| rPt|}||_|S )N)ra rs r_ callabler r r )rf delegatedZsfr r r Z__get_side_effect s z!NonCallableMock.__get_side_effectc C s( t |}| j}|d kr || _n||_d S )N) _try_iterra rs r_ )rf rp r r r r Z__set_side_effect s
z!NonCallableMock.__set_side_effect)rY r_ c C s |dkr g }t | |kr dS |jt | d| _d| _d| _t | _t | _t | _|r^t | _
|rhd| _ x0| j j
D ]"}t|tst|tkrqt|j| qtW | j
}t|r|| k r|j| dS )z-Restore the mock object to its initial state.NFr )idappendr[ r] r\ rU rW r^ rV r rr rs r` valuesr
_SpecState_deletedrX r )rf ZvisitedrY r_ ZchildrZ r r r rX
s, zNonCallableMock.reset_mockc K s` xZt |j dd dD ]B\}}|jd}|j }| }x|D ]}t||}q:W t||| qW dS )aZ Set attributes on the mock through keyword arguments.
Attributes plus return values and side effects can be set on child
mocks using standard dot notation and unpacking a dictionary in the
method call:
>>> attrs = {'method.return_value': 3, 'other.side_effect': KeyError}
>>> mock.configure_mock(**attrs)c S s | d j dS )Nr .)count)entryr r r 8 s z0NonCallableMock.configure_mock..)keyr N)sorteditemssplitpopr< rz )rf r, argvalr+ finalr r r r r r + s
z NonCallableMock.configure_mockc C s |dkrt |n:| jd k r<|| jks.|tkrLt d| nt|rLt || jsd|jdrdt || jj|}|tkrt |np|d krd }| j d k rt
| j |}| j | |||| d}|| j|< n.t |t
rt|j|j|j|j|j}|| j|< |S ) Nr r z Mock object has no attribute %rassertassret)r r r r r > r r )r r )r" r _all_magicsrd r r r` rC r r r< r r r r r r r3 r r )rf r rJ r r r r rm A s6
zNonCallableMock.__getattr__c
C s" | j g}| j}| }d}|dgkr$d}t }xT|d k r~|}|j|j | d}|j dkrZd}|j}t||krnP |jt| q,W tt|}|jpd}t |dkr|d d
kr|d7 }||d< dj
|}d}|dkrd | }d} | j d k rd
} | j rd } | | j j
} d t| j
|| t| f S )Nr ()r r1 r ().r mock.z name=%rz spec=%rz spec_set=%rz<%s%s%s id='%s'>)r r )r1 r )rw ru ro r r r} r> reversedrv r joinr r r5 r )
rf Z
_name_listr ZlastdotseenZ_firstr Z name_stringZ spec_stringr r r rg g sL
zNonCallableMock.__repr__c C sh t stj| S | jpg }tt| }t| j}dd |D }dd |D }tt || | t| j
S )z8Filter the output of `dir(mock)` to only useful members.c S s g | ]}|j d s|qS )r )r )r er r r
s z+NonCallableMock.__dir__..c S s$ g | ] }|j d s t|r|qS )r )r rd )r r r r r r s ) r r __dir__r r r r> rB r ro r` )rf ZextrasZ from_typeZ from_dictr r r r s
zNonCallableMock.__dir__c s" |t krtj||S jrHjd k rH|jkrH|jkrHtd| n|tkrbd| }t|n|tkrjd k r|jkrtd| t |st
t |t || | fdd}n(t
|d | t
t || |j|< n.|dk r|_d S t
|||r|j|< tj||S )Nz!Mock object has no attribute '%s'z.Attempting to set unsupported magic method %r.c s f| |S )Nr )r+ r )rI rf r r r s z-NonCallableMock.__setattr__..r )r| r __setattr__r r rB r" _unsupported_magicsr r rz r _get_methodr r` r )rf r rp msgr )rI rf r r s2
zNonCallableMock.__setattr__c C s |t kr2|t| jkr2tt| | || jkr2d S || jkrHtj| | | jj|t}|t krft
||tk rv| j|= t | j|< d S )N) r r rB delattrr __delattr__r` rC _missingr r" )rf r r r r r r s
zNonCallableMock.__delattr__c C s | j pd}t|||S )Nr1 )rv _format_call_signature)rf r+ r, r r r r _format_mock_call_signature s
z+NonCallableMock._format_mock_call_signaturec C sD d}| j ||}| j}t|dkr.|dd }| j | }|||f S )Nz!Expected call: %s
Actual call: %s r )r r] r )rf r+ r, messageexpected_stringr] Z
actual_stringr r r _format_mock_failure_message s
z,NonCallableMock._format_mock_failure_messagec C st | j }|dk rlt|dkr(d}|\}}n
|\}}}y||j||fS tk
rh } z
|jdS d}~X qpX n|S dS )a
Given a call (or simply an (args, kwargs) tuple), return a
comparison key suitable for matching with other calls.
This is a best effort method which relies on the spec's signature,
if available, or falls back on the arguments themselves.
Nrb r )r r r) TypeErrorwith_traceback)rf _callr- r r+ r, r r r r
_call_matcher s
z NonCallableMock._call_matcherc C s. | }|j dkr*d|jpd|j f }t|dS )z/assert that the mock was never called.
r z7Expected '%s' to not have been called. Called %s times.r1 N)r\ rv AssertionError)r* rf r r r r rP s
z!NonCallableMock.assert_not_calledc C s( | }|j dkr$d|j pd}t|dS )z6assert that the mock was called at least once
r z"Expected '%s' to have been called.r1 N)r\ rv r )r* rf r r r r rO s
z NonCallableMock.assert_calledc C s. | }|j dks*d|jpd|j f }t|dS )z3assert that the mock was called only once.
r z8Expected '%s' to have been called once. Called %s times.r1 N)r\ rv r )r* rf r r r r rQ s
z"NonCallableMock.assert_called_oncec s| | j dkr(j }td|f fdd}j f}jj }||krxt|trh|nd}t| |dS )zassert that the mock was called with the specified arguments.
Raises an AssertionError if the args and keyword args passed in are
different to the last call to the mock.Nz Expected call: %s
Not calledc s j } | S )N)r )r )r+ r, rf r r _error_message' s z:NonCallableMock.assert_called_with.._error_message)r] r r r r Exception)r* r+ r, expectedr actualcauser )r+ r, rf r rN s
z"NonCallableMock.assert_called_withc O s6 | }|j dks*d|jpd|j f }t||j||S )ziassert that the mock was called exactly once and that that call was
with the specified arguments.r z1Expected '%s' to be called once. Called %s times.r1 )r\ rv r rN )r* r+ r, rf r r r r rR 1 s
z'NonCallableMock.assert_called_once_withc s fdd|D }t |tr |nd}t fdd jD }|sd||kr`tdt| jf |dS t|}g }x:|D ]2}y|j| W qv tk
r |j| Y qvX qvW |rtdt |f |dS )a assert the mock has been called with the specified calls.
The `mock_calls` list is checked for the calls.
If `any_order` is False (the default) then the calls must be
sequential. There can be extra calls before or after the
specified calls.
If `any_order` is True then the calls can be in any order, but
they must all appear in `mock_calls`.c s g | ]} j |qS r )r )r c)rf r r r F s z4NonCallableMock.assert_has_calls..Nc 3 s | ]} j |V qd S )N)r )r r )rf r r H s z3NonCallableMock.assert_has_calls..z(Calls not found.
Expected: %r
Actual: %rz %r not all found in call list)
r r rU rW r r> remover& r r? )rf ZcallsZ any_orderr r Z all_callsZ not_foundZkallr )rf r rS < s*
z NonCallableMock.assert_has_callsc sZ j ||f} fdd jD }||krVt|tr8|nd} j||}td| |dS )zassert the mock has been called with the specified arguments.
The assert passes if the mock has *ever* been called, unlike
`assert_called_with` and `assert_called_once_with` that only pass if
the call is the most recent one.c s g | ]} j |qS r )r )r r )rf r r r f s z3NonCallableMock.assert_any_call..Nz%s call not found)r r^ r r r r )rf r+ r, r r r r r )rf r rT _ s z NonCallableMock.assert_any_callc K sF t | }t|ts2t|tr"t}q.) r r r r r r r r r r, attribute_namer )rf r r r r r i s z _patch.copyc C s t |tr| j|S | j|S )N)r r decorate_classdecorate_callable)rf r' r r r r$ v s
z_patch.__call__c C sR xLt |D ]@}|jtjs q
t||}t|ds2q
| j }t|||| q
W |S )Nr$ )r r r TEST_PREFIXr< r r rz )rf r attr
attr_valuer r r r r | s
z_patch.decorate_classc s< t dr jj| S t fdd| g_S )N patchingsc s g }g }t }zybxJjD ]@}|j }|j| |jd k rF|j| q|jtkr|j| qW | t |7 } | |S ||krt|r|j| t j
} Y nX W d xt |D ]}|j | qW X d S )N)
r? r __enter__r r updater r r sysexc_infor __exit__)r+ Zkeywargs
extra_argsZentered_patchersr patchingr )r' patchedr r r s*
z)_patch.decorate_callable..patched)r r r r )rf r' r )r' r r r s
z_patch.decorate_callablec C s | j }| j}t}d}y|j| }W n$ ttfk
rH t||t}Y nX d}|tkrft|t rfd| _
| j
r|tkrtd||f ||fS )NFTz!%s does not have the attribute %r) r r r rB r" KeyErrorr< _builtinsr r r )rf r r rI localr r r get_original s z_patch.get_originalc C s> | j | j| j }}}| j| j }}| j}| j | _|dkr@d}|dkrLd}|dkrXd}|dk rp|dk rptd|dk s|dk r|dkrtd| j \}}|t
ko|dkrTd} |dkr|}|dkr|}d}n&|dk r|dkr|}d}n |dkr|}|dk s
|dk r,|t
kr tdt |t r,d} t
}
i } |dk rD|}
nN|dk sX|dk r|} |dk rj|} t| r~d| k}
n
t| }
|
rt}
|dk r|| d< |dk r|| d < t |
t rt|
tr| jr| j| d
< | j| |
f | }| rt|r|} |dk r|} t| r2t| r2t}
| jd
|
f |d d | |_nl|dk r|t
k rptd
|t
krtdt|}|dkr|}t|f|| jd|}n|rtd|}|| _|| _ t | j| j| | j dk r:i }| j t
kr||| j < x.| j D ]$}|j }|j t
kr|j| qW |S |S )zPerform the patch.FNz Can't specify spec and autospecTz6Can't provide explicit spec_set *and* spec or autospecz!Can't use 'spec' with create=Truer$ r r r z())r r zBautospec creates the mock for you. Can't specify autospec and new.z%Can't use 'autospec' with create=True)r _namez.Can't pass kwargs to a mock we aren't creating)TN)!r r r r r, r r r r r r r r r r@ r r r r
r r
r rE r rY boolr
temp_originalr rz r r r )rf r r r r r, r rI r ZinheritKlass_kwargsZ this_specZ not_callableZnew_attrr r r r r r r s
z_patch.__enter__c G s t | std| jr4| jtk r4t| j| j| j nBt| j| j | j rvt
| j| j sd| jdkrvt| j| j| j | `| `| `x$t | j D ]}t |r|j
| qW dS ) zUndo the patch.z stop called on unstarted patcherr6 r8 r9 __annotations__r: N)r6 r8 r9 r r: )r RuntimeErrorr r r rz r r r r r r r r
)rf r r r r r r
H s z_patch.__exit__c C s | j }| jj| |S )z-Activate a patch, returning any created mock.)r _active_patchesr )rf rJ r r r start` s z _patch.startc C s. y| j j| W n tk
r$ Y nX | j S )zStop an active patch.)r r r& r
)rf r r r stopg s
z _patch.stop)r5 r8 rh r r r! r r$ r r r r r
r r r r r r r I s
(~r c sP y j dd\ }W n& ttfk
r: td f Y nX fdd}||fS )Nr r z.Need a valid target to patch. You supplied: %rc s t S )N)r r )r r r r y s z _get_target..)rsplitr r& )r r r r )r r _get_targets s r c
s$ fdd} t | |||||||| S )a
patch the named member (`attribute`) on an object (`target`) with a mock
object.
`patch.object` can be used as a decorator, class decorator or a context
manager. Arguments `new`, `spec`, `create`, `spec_set`,
`autospec` and `new_callable` have the same meaning as for `patch`. Like
`patch`, `patch.object` takes arbitrary keyword arguments for configuring
the mock object it creates.
When used as a class decorator `patch.object` honours `patch.TEST_PREFIX`
for choosing which methods to wrap.
c s S )Nr r )r r r r s z _patch_object..)r )
r r r r r r r r r, r r )r r
_patch_object} s
r! c
s t tkr fdd}n fdd}|s2tdt|j }|d \} }
t|| |
|||||i } | | _xB|dd D ]2\} }
t|| |
|||||i } | | _| jj| qvW | S )a Perform multiple patches in a single call. It takes the object to be
patched (either as an object or a string to fetch the object by importing)
and keyword arguments for the patches::
with patch.multiple(settings, FIRST_PATCH='one', SECOND_PATCH='two'):
...
Use `DEFAULT` as the value if you want `patch.multiple` to create
mocks for you. In this case the created mocks are passed into a decorated
function by keyword, and a dictionary is returned when `patch.multiple` is
used as a context manager.
`patch.multiple` can be used as a decorator, class decorator or a context
manager. The arguments `spec`, `spec_set`, `create`,
`autospec` and `new_callable` have the same meaning as for `patch`. These
arguments will be applied to *all* patches done by `patch.multiple`.
When used as a class decorator `patch.multiple` honours `patch.TEST_PREFIX`
for choosing which methods to wrap.
c s t S )N)r r )r r r r s z!_patch_multiple..c s S )Nr r )r r r r s z=Must supply at least one keyword argument with patch.multipler r N) r strr& r> r r r r r )
r r r r r r r, r r r r r Z this_patcherr )r r _patch_multiple s&
r# c
K s$ t | \}} t|| ||||||| S )a
`patch` acts as a function decorator, class decorator or a context
manager. Inside the body of the function or with statement, the `target`
is patched with a `new` object. When the function/with statement exits
the patch is undone.
If `new` is omitted, then the target is replaced with a
`MagicMock`. If `patch` is used as a decorator and `new` is
omitted, the created mock is passed in as an extra argument to the
decorated function. If `patch` is used as a context manager the created
mock is returned by the context manager.
`target` should be a string in the form `'package.module.ClassName'`. The
`target` is imported and the specified object replaced with the `new`
object, so the `target` must be importable from the environment you are
calling `patch` from. The target is imported when the decorated function
is executed, not at decoration time.
The `spec` and `spec_set` keyword arguments are passed to the `MagicMock`
if patch is creating one for you.
In addition you can pass `spec=True` or `spec_set=True`, which causes
patch to pass in the object being mocked as the spec/spec_set object.
`new_callable` allows you to specify a different class, or callable object,
that will be called to create the `new` object. By default `MagicMock` is
used.
A more powerful form of `spec` is `autospec`. If you set `autospec=True`
then the mock will be created with a spec from the object being replaced.
All attributes of the mock will also have the spec of the corresponding
attribute of the object being replaced. Methods and functions being
mocked will have their arguments checked and will raise a `TypeError` if
they are called with the wrong signature. For mocks replacing a class,
their return value (the 'instance') will have the same spec as the class.
Instead of `autospec=True` you can pass `autospec=some_object` to use an
arbitrary object as the spec instead of the one being replaced.
By default `patch` will fail to replace attributes that don't exist. If
you pass in `create=True`, and the attribute doesn't exist, patch will
create the attribute for you when the patched function is called, and
delete it again afterwards. This is useful for writing tests against
attributes that your production code creates at runtime. It is off by
default because it can be dangerous. With it switched on you can write
passing tests against APIs that don't actually exist!
Patch can be used as a `TestCase` class decorator. It works by
decorating each test method in the class. This reduces the boilerplate
code when your test methods share a common patchings set. `patch` finds
tests by looking for method names that start with `patch.TEST_PREFIX`.
By default this is `test`, which matches the way `unittest` finds tests.
You can specify an alternative prefix by setting `patch.TEST_PREFIX`.
Patch can be used as a context manager, with the with statement. Here the
patching applies to the indented block after the with statement. If you
use "as" then the patched object will be bound to the name after the
"as"; very useful if `patch` is creating a mock object for you.
`patch` takes arbitrary keyword arguments. These will be passed to
the `Mock` (or `new_callable`) on construction.
`patch.dict(...)`, `patch.multiple(...)` and `patch.object(...)` are
available for alternate use-cases.
)r r )
r r r r r r r r, r r r r r r s E
c @ sV e Zd ZdZf dfddZdd Zdd Zd d
Zd d Zd
d Z dd Z
eZ e
Z dS ) _patch_dicta#
Patch a dictionary, or dictionary like object, and restore the dictionary
to its original state after the test.
`in_dict` can be a dictionary or a mapping like container. If it is a
mapping then it must at least support getting, setting and deleting items
plus iterating over keys.
`in_dict` can also be a string specifying the name of the dictionary, which
will then be fetched by importing it.
`values` can be a dictionary of values to set in the dictionary. `values`
can also be an iterable of `(key, value)` pairs.
If `clear` is True then the dictionary will be cleared before the new
values are set.
`patch.dict` can also be called with arbitrary keyword arguments to set
values in the dictionary::
with patch.dict('sys.modules', mymodule=Mock(), other_module=Mock()):
...
`patch.dict` can be used as a context manager, decorator or class
decorator. When used as a class decorator `patch.dict` honours
`patch.TEST_PREFIX` for choosing which methods to wrap.
Fc K s> t |trt|}|| _t|| _| jj| || _d | _d S )N) r r" r in_dictrn r r
clear _original)rf r% r r& r, r r r r! 0 s
z_patch_dict.__init__c s. t trj S t fdd}|S )Nc
s" j z
| |S j X d S )N)r$
_unpatch_dict)r+ r )frf r r _inner> s
z$_patch_dict.__call__.._inner)r r r r )rf r) r* r )r) rf r r$ ; s
z_patch_dict.__call__c C s\ xVt |D ]J}t||}|jtjr
t|dr
t| j| j| j }||}t
||| q
W |S )Nr$ ) r r< r r r r r$ r% r r& rz )rf r r r Z decoratorZ decoratedr r r r I s
z_patch_dict.decorate_classc C s | j dS )zPatch the dict.N)r$ )rf r r r r T s z_patch_dict.__enter__c C s | j }| j}| j}y |j }W n2 tk
rP i }x|D ]}|| ||< q8W Y nX || _|rdt| y|j| W n. tk
r x|D ]}|| ||< qW Y nX d S )N)r r% r& r r" r' _clear_dictr
)rf r r% r& rI r r r r r$ Y s"
z_patch_dict._patch_dictc C sV | j }| j}t| y|j| W n. tk
rP x|D ]}|| ||< q8W Y nX d S )N)r% r' r+ r
r" )rf r% rI r r r r r( s s
z_patch_dict._unpatch_dictc G s | j dS )zUnpatch the dict.F)r( )rf r+ r r r r
s z_patch_dict.__exit__N)
r5 r8 rh r6 r! r$ r r r$ r( r
r r r r r r r$ s
r$ c C sB y | j W n0 tk
r< t| }x|D ]
}| |= q*W Y nX d S )N)r& r" r> )r% keysr r r r r+ s
r+ c C s xt tjD ] } | j q W dS )z7Stop all active patches. LIFO to unroll nested patches.N)r r r r )r r r r _patch_stopall s r- Ztestzlt le gt ge eq ne getitem setitem delitem len contains iter hash str sizeof enter exit divmod rdivmod neg pos abs invert complex int float index trunc floor ceil bool next zHadd sub mul matmul div floordiv mod lshift rshift and xor or pow truediv c c s | ]}d | V qdS )zi%sNr )r nr r r r s r c c s | ]}d | V qdS )zr%sNr )r r/ r r r r s __get____set__
__delete__ __reversed__ __missing__
__reduce__
__reduce_ex__Z__getinitargs____getnewargs__ __getstate__ __setstate__
__getformat__
__setformat__rg r __subclasses__
__format____getnewargs_ex__c s fdd}| |_ |S )z:Turns a callable object (like a mock) into a real functionc s | f||S )Nr )rf r+ r )r' r r method s z_get_method..method)r5 )r r' r? r )r' r r s r c C s h | ] }d | qS )z__%s__r )r r? r r r r s rm r r! r __prepare____instancecheck____subclasscheck____del__c C s
t j| S )N)r __hash__)rf r r r r s r c C s
t j| S )N)r __str__)rf r r r r s c C s
t j| S )N)r
__sizeof__)rf r r r r s )rD rE rF r y ?g ?) __lt____gt____le____ge____int__r __len__r
__complex__ __float____bool__ __index__c s fdd}|S )Nc s$ j j}|tk r|S | kr dS tS )NT)__eq__rr r NotImplemented)otherret_val)rf r r rQ s z_get_eq..__eq__r )rf rQ r )rf r _get_eq s rU c s fdd}|S )Nc s j jtk rtS | kr dS tS )NF)__ne__rr r rR )rS )rf r r rV s
z_get_ne..__ne__r )rf rV r )rf r _get_ne s rW c s fdd}|S )Nc s j j} | tkrtg S t| S )N)r rr r r )rT )rf r r r s z_get_iter..__iter__r )rf r r )rf r _get_iter s rX )rQ rV r c C s t j|t}|tk r ||_d S tj|}|d k rdy || }W n tk
rX t|}Y nX ||_d S tj|}|d k r|| |_d S )N)_return_valuesrC r rY _calculate_return_valuer" _side_effect_methodsr_ )r1 r? r ZfixedZreturn_calulatorrY Z
side_effectorr r r _set_return_value s
r\ c @ s e Zd Zdd Zdd ZdS )
MagicMixinc O s& | j tt| j|| | j d S )N)_mock_set_magicsr r] r! )rf r+ r r r r r! ( s zMagicMixin.__init__c C s t }t| dd d k rTt j| j}t }t | }x$|D ] }|t| jkr4t| | q4W |tt| j }t| }x |D ]}t||t || qtW d S )Nr )
_magicsr< intersectionr ro r rB r rz
MagicProxy)rf Z these_magicsZ
remove_magicsr r r r r r^ . s
zMagicMixin._mock_set_magicsN)r5 r8 rh r! r^ r r r r r] ' s r] c @ s e Zd ZdZdddZdS )r z-A version of `MagicMock` that isn't callable.Fc C s | j || | j dS )zAdd a spec to a mock. `spec` can either be an object or a
list of strings. Only attributes on the `spec` can be fetched as
attributes from the mock.
If `spec_set` is True then only attributes on the spec can be set.N)r r^ )rf r r r r r r G s z"NonCallableMagicMock.mock_add_specN)F)r5 r8 rh r6 r r r r r r E s c @ s e Zd ZdZdddZdS )r a
MagicMock is a subclass of Mock with default implementations
of most of the magic methods. You can use MagicMock without having to
configure the magic methods yourself.
If you use the `spec` or `spec_set` arguments then *only* magic
methods that exist in the spec will be created.
Attributes and the return value of a `MagicMock` will also be `MagicMocks`.
Fc C s | j || | j dS )zAdd a spec to a mock. `spec` can either be an object or a
list of strings. Only attributes on the `spec` can be fetched as
attributes from the mock.
If `spec_set` is True then only attributes on the spec can be set.N)r r^ )rf r r r r r r ] s zMagicMock.mock_add_specN)F)r5 r8 rh r6 r r r r r r R s
c @ s. e Zd Zdd Zdd Zdd Zd
dd ZdS ) ra c C s || _ || _d S )N)r r )rf r r r r r r! i s zMagicProxy.__init__c O s | j }|||S )N) create_mock)rf r+ r, mr r r r$ m s zMagicProxy.__call__c C s8 | j }| j}|j|||d}t||| t||| |S )N)r r r )r r r rz r\ )rf r r rc r r r rb q s zMagicProxy.create_mockNc C s | j S )N)rb )rf r r r r r r0 z s zMagicProxy.__get__)N)r5 r8 rh r! r$ rb r0 r r r r ra h s ra c @ s( e Zd ZdZdd Zdd Zdd ZdS ) _ANYz2A helper object that compares equal to everything.c C s dS )NTr )rf rS r r r rQ s z _ANY.__eq__c C s dS )NFr )rf rS r r r rV s z _ANY.__ne__c C s dS )Nzr )rf r r r rg s z
_ANY.__repr__N)r5 r8 rh r6 rQ rV rg r r r r rd s rd c C sd d| }d}dj dd |D }dj dd t|j D }|rD|}|r\|rT|d7 }||7 }|| S )Nz%s(%%s)r z, c S s g | ] }t |qS r )repr)r r r r r r s z*_format_call_signature..c S s g | ]\}}d ||f qS )z%s=%rr )r r rp r r r r s )r r r )r r+ r, r Zformatted_argsZ args_stringZ
kwargs_stringr r r r s r c @ sv e Zd ZdZf ddddfddZf ddddfdd Zd
d ZejZd d
Z dd Z
dd Z dd Z dd Z
dd ZdS )r a
A tuple for holding the results of a call to a mock, either in the form
`(args, kwargs)` or `(name, args, kwargs)`.
If args or kwargs are empty then a call tuple will compare equal to
a tuple without those values. This makes comparisons less verbose::
_Call(('name', (), {})) == ('name',)
_Call(('name', (1,), {})) == ('name', (1,))
_Call(((), {'a': 'b'})) == ({'a': 'b'},)
The `_Call` object provides a useful shortcut for comparing with call::
_Call(((1, 2), {'a': 3})) == call(1, 2, a=3)
_Call(('foo', (1, 2), {'a': 3})) == call.foo(1, 2, a=3)
If the _Call has no name then it will match any name.
r NFTc C s f }i }t |}|dkr$|\}}}nr|dkrd|\} }
t| trX| }t|
trR|
}qb|
}q| |
}}n2|dkr|\}t|tr|}nt|tr|}n|}|rtj| ||fS tj| |||fS )Nr rb r )r r r" r? r ) r rp r r r from_kallr+ r, _lenfirstsecondr r r r s.
z
_Call.__new__c C s || _ || _|| _d S )N)rv rt _mock_from_kall)rf rp r r r rf r r r r! s z_Call.__init__c C sf |t kr dS y t|}W n tk
r, dS X d}t| dkrH| \}}n
| \}}}t| dd rzt|dd rz| j|jkrzdS d}|dkrf i }}n|dkr|\}}}n|dkr|\} t| tr| }i }n"t| tr| }f i }}nf }| }nV|dkr>|\}
} t|
tr2|
}t| tr&| i }}n
f | }}n
|
| }}ndS |rV||krVdS ||f||fkS ) NTFr rb rt r r r )r r r r< rt r r? r" ) rf rS Z len_otherZ self_nameZ self_argsZ self_kwargsZ
other_nameZ
other_argsZ other_kwargsrp rh ri r r r rQ sP
z _Call.__eq__c O s< | j d kr td||fddS | j d }t| j ||f|| dS )Nr z())r )r r )rv r )rf r+ r, r r r r r$ s
z_Call.__call__c C s2 | j d krt|ddS d| j |f }t|| ddS )NF)r rf z%s.%s)r r rf )rv r )rf r r r r r rm s
z_Call.__getattr__c O s | j d||S )Nr )rm )rf r+ r, r r r r $ s z _Call.countc O s | j d||S )Nindex)rm )rf r+ r, r r r rk ' s z _Call.indexc C s| | j s&| jpd}|jdr"d| }|S t| dkr@d}| \}}n0| \}}}|sTd}n |jdshd| }nd| }t|||S )Nr z()zcall%srb zcall.%s)rj rv r r r )rf r r+ r, r r r rg * s
z_Call.__repr__c C s8 g }| }x"|dk r*|j r"|j| |j}q
W tt|S )zFor a call object that represents multiple calls, `call_list`
returns a list of all the intermediate calls as well as the
final call.N)rj r rt rU r )rf valsr r r r call_list? s
z_Call.call_list)r5 r8 rh r6 r r! rQ r rV r$ rm r rk rg rm r r r r r s 7r )rf c K s, t | rt| } t| t}d| i}|r0d| i}n | dkr.r rx rx rx rx )r bytesr ) read_dataZ data_as_listliner )rw r _iterate_read_data s
r| r c s fdd}fdd}fdd t dkr`ddl}ttt|jjtt|ja | dkrttd t d
} tt d j
_ t dgdj
_ dj_ dj_ dj_ |j_ d < d j_|j_ fd
d}|| _| _ | S )a
A helper function to create a mock to replace the use of `open`. It works
for `open` called directly or used as a context manager.
The `mock` argument is the mock object to configure. If `None` (the
default) then a `MagicMock` will be created for you, with the API limited
to methods or attributes available on standard file handles.
`read_data` is a string for the `read` methoddline`, and `readlines` of the
file handle to return. This is an empty string by default.
c s j jd k rj jS t d S )Nr ) readlinesrY r> )r+ r, )_statehandler r _readlines_side_effect( s z)mock_open.._readlines_side_effectc s( j jd k rj jS t j d S )Nr )readrY r r )r+ r, )r~ r rz r r _read_side_effect- s z$mock_open.._read_side_effectc 3 sJ j jd k r xj jV qW x d D ]
} | V q&W xt V q6W d S )Nr )readlinerY r )r{ )r~ r rz r r _readline_side_effect2 s
z(mock_open.._readline_side_effectNr open)r r )r r c s6 t d< jjd kr2 d< d j_tS )Nr r )r| r r_ r )r+ r, )r r~ r rz r r
reset_dataS s
z mock_open..reset_data) file_spec_ior> ro r
TextIOWrapperunionBytesIOr r r rY r| writer r r} r_ )r1 rz r r r r r )r r~ r rz r r s. "
c @ s( e Zd ZdZdd Zdd Zdd ZdS ) r
aW
A mock intended to be used as a property, or other descriptor, on a class.
`PropertyMock` provides `__get__` and `__set__` methods so you can specify
a return value when it is fetched.
Fetching a `PropertyMock` instance from an object calls the mock, with
no args. Setting it calls the mock with the value being set.
c K s
t f |S )N)r )rf r, r r r r i s z PropertyMock._get_child_mockc C s | S )Nr )rf r Zobj_typer r r r0 l s zPropertyMock.__get__c C s | | d S )Nr )rf r r r r r r1 n s zPropertyMock.__set__N)r5 r8 rh r6 r r0 r1 r r r r r
` s )
r r r r r r r r r r
r r r
)F)F)NFNNN)FFNN)Nr )o__all__ __version__r% r r builtinstypesr functoolsr r r r
BaseExceptionr platformr ZlangZ Throwabler superr r r r( r4 r/ r= r@ rE rM rH rd r re ri r r ZMISSINGr ZDELETEDr rq r| r r> rU r r r r
r r r r r r r r r! r# r r$ r+ r- rn ZmultipleZstopallr Z
magic_methodsZnumericsr r ZinplacerightZ
_non_defaultsr r_ r r rZ rR rY rU rW rX r[ r\ r] r r ra rd r r r? r r r rn r r r rQ r# r0 rr r r| r r
r r r r s
4 i5 ,
1Jv
,
w
DPK ! +eQd d mock.cpython-36.opt-2.pycnu [ 3
\7 @ sF dZ d
ZddlZddlZddlZddlZddlmZ ddlm Z m
Z
dd e eD Z e
fZdejkrzddlZe
ejjfZdZeZdd Zdd Zdd Zdd d Zd d Zd!d" Zd#d$ Z d%d& Z dd'd(Z d)d* Z d+d, Z G d-d. d.e!Z"G d/d0 d0e!Z#e# Z$e$j%Z%e$j&Z'e$j(Z)d1d2 Z*d3d4d5d6d7d8d9d:hZ+d;d< Z,G d=d> d>e-Z.d?d@ Z/G dAdB dBe!Z0G dCdD dDe!Z1G dEd d e1Z2dFdG Z3G dHdI dIe1Z4G dJd d e4e2Z5dKdL Z6dMdN Z7dOdP Z8G dQdR dRe!Z9dSdT Z:e%dd dddfdUdVZ;ddWdXZd\d] Z?d^d_ Z@e;e=_!e>e=_Ae# s r javaTc C s t t| tS )N)
issubclasstyper
)objr r r _is_instance_mock2 s r c C s t | tp t | to t| tS )N)
isinstanceBaseExceptionsr r )r r r r
_is_exception8 s
r c C s t | tr6| r6y
| j} W n tk
r. d S X d}n*t | ts`y
| j} W n tk
r^ d S X |rpt| d }n| }y| tj|fS t k
r d S X d S )NT)
r r __init__AttributeError
FunctionTypes__call__r inspectZ signature
ValueError)funcZ as_instanceZeat_selfZsig_funcr r r _get_signature_object? s$
r( Fc sD t | || d krd S \} fdd}t| | |t|_d S )Nc s j || d S )N)bind)
_mock_selfargskwargs)sigr r checksigd s z"_check_signature..checksig)r( _copy_func_detailsr _mock_check_sig)r' mock skipfirstinstancer. r )r- r _check_signature_ s
r4 c # C s | j |_ | j|_y | j|_W n tk
r0 Y nX y | j|_W n tk
rR Y nX y | j|_W n tk
rt Y nX y | j|_W n tk
r Y nX d S )N)__name____doc____text_signature__r"
__module__ __defaults____kwdefaults__)r' funcopyr r r r/ j s$ r/ c C s& t | trdS t| dd d k r"dS dS )NTr$ F)r r getattr)r r r r _callable s
r= c C s t | ttfkS )N)r listtuple)r r r r _is_list s r@ c C sH t | tst| dd d k S x(| f| j D ]}|jjdd k r(dS q(W dS )Nr$ TF)r r r< __mro____dict__get)r baser r r _instance_callable s
rE c s t |s d S t|t}t|||}|d kr.| S |\} fdd}t|| |j}|j s^d}|| d}d| } t| | || }
t|
| |
S )Nc s j | | d S )N)r) )r+ r, )r- r r r. s z _set_signature..checksigr; )Z
_checksig_r1 zYdef %s(*args, **kwargs):
_checksig_(*args, **kwargs)
return mock(*args, **kwargs)) r= r r r( r/ r5 isidentifierexec _setup_func) r1 originalr3 r2 resultr' r. r contextsrcr; r )r- r _set_signature s$
rM c
s _ tsd S fdd}fdd}fdd}fdd}fd d
}fd d }fd
d} fdd} d _d _d _t _t _t _j _ j
_
j _ | _ | _
| _| _| _| _| _| _ _d S )Nc s j | |S )N)assert_called_with)r+ r, )r1 r r rN s z'_setup_func..assert_called_withc s j | |S )N)
assert_called)r+ r, )r1 r r rO s z"_setup_func..assert_calledc s j | |S )N)assert_not_called)r+ r, )r1 r r rP s z&_setup_func..assert_not_calledc s j | |S )N)assert_called_once)r+ r, )r1 r r rQ s z'_setup_func..assert_called_oncec s j | |S )N)assert_called_once_with)r+ r, )r1 r r rR s z,_setup_func..assert_called_once_withc s j | |S )N)assert_has_calls)r+ r, )r1 r r rS s z%_setup_func..assert_has_callsc s j | |S )N)assert_any_call)r+ r, )r1 r r rT s z$_setup_func..assert_any_callc s: t _t _j j} t| r6| k r6| j d S )N) _CallList method_calls
mock_calls
reset_mock return_valuer )ret)r; r1 r r rX s z _setup_func..reset_mockFr )r1 r called
call_count call_argsrU call_args_listrV rW rY side_effect_mock_childrenrN rR rS rT rX rO rP rQ _mock_delegate)
r; r1 rN rO rP rQ rR rS rT rX r )r; r1 r rH s: rH c C s d| dd | kS )Nz__%s__ r )r r r r _is_magic s rd c @ s e Zd Zdd Zdd ZdS )_SentinelObjectc C s
|| _ d S )N)r )selfr r r r r! s z_SentinelObject.__init__c C s
d| j S )Nz sentinel.%s)r )rf r r r __repr__ s z_SentinelObject.__repr__N)r5 r8 __qualname__r! rg r r r r re s re c @ s e Zd Zdd Zdd ZdS ) _Sentinelc C s
i | _ d S )N)
_sentinels)rf r r r r! s z_Sentinel.__init__c C s |dkr t | jj|t|S )N __bases__)r" rj
setdefaultre )rf r r r r __getattr__ s z_Sentinel.__getattr__N)r5 r8 rh r! rm r r r r ri s ri c C s$ t | ttttfkr t | | S | S )N)r dictr> r? set)valuer r r _copy s rq rY _mock_return_valuer_ _mock_side_effect _mock_parent_mock_new_parent
_mock_name_mock_new_namec C s8 t j| d| }| |fdd}| |fdd}t||S )NZ_mock_c S s" | j }|d krt| |S t||S )N)ra r< )rf r _the_namer- r r r _get s
z"_delegating_property.._getc S s* | j }|d kr|| j|< n t||| d S )N)ra rB setattr)rf rp r rx r- r r r _set$ s z"_delegating_property.._set)_allowed_namesaddproperty)r rx ry r{ r r r _delegating_property s
r c @ s e Zd Zdd Zdd ZdS )rU c C sn t |tstj| |S t|}t| }||kr2dS x6td|| d D ] }| ||| }||krFdS qFW dS )NFr T)r r> __contains__lenrange)rf rp Z len_valueZlen_selfiZsub_listr r r r 1 s
z_CallList.__contains__c C s t jt| S )N)pprintZpformatr> )rf r r r rg ? s z_CallList.__repr__N)r5 r8 rh r rg r r r r rU / s rU c C sx t |s dS |js,|js,|jd k s,|jd k r0dS | }x |d k rR||krJdS |j}q6W |rd| |_||_|rt| |_||_dS )NFT)r rv rw rt ru )parentrp r new_name_parentr r r _check_and_set_parentC s$
r c @ s$ e Zd Zdd Zdd Zdd ZdS ) _MockIterc C s t || _d S )N)iterr )rf r r r r r! ] s z_MockIter.__init__c C s | S )Nr )rf r r r __iter___ s z_MockIter.__iter__c C s
t | jS )N)nextr )rf r r r __next__a s z_MockIter.__next__N)r5 r8 rh r! r r r r r r r \ s r c @ s e Zd ZeZdZdd ZdS )BaseNc O s d S )Nr )rf r+ r, r r r r! g s z
Base.__init__)r5 r8 rh r rr rs r! r r r r r d s r c @ sD e Zd Zdd ZdCddZdd ZdDd
d ZdEd d
Zdd Zdd Z dZ
e ee e
Z e dd Z
edZedZedZedZedZdd Zd d Ze eeZdFddd d d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Z d+d, Z d-d. Z d/d0 Z d1d2 Z d3d4 Z!d5d6 Z"d7d8 Z#d9d: Z$d;d< Z%dGd=d>Z&d?d@ Z'dAdB Z(dS )Hr
c O s$ t | j| fd| ji}tj|}|S )Nr6 )r r5 r6 object__new__)clsr+ kwnewr3 r r r r o s
zNonCallableMock.__new__N Fc K s |d kr |}| j }
||
d< ||
d< ||
d< ||
d< |d k rB|}d}|
d krR|d k }
| j||| |
i |
d< ||
d< d |
d< d |
d
< d |
d < d |
d
< t |
d< t |
d< t |
d< | |
d< | r| jf | tt| j|||||| d S )Nrt rv rw ru Tr` _mock_wrapsra FZ _mock_calledZ_mock_call_argsr Z_mock_call_countZ_mock_call_args_listZ_mock_mock_callsrV _mock_unsafe)rB _mock_add_specrU configure_mock _safe_superr
r! )rf specr r spec_setr _spec_state _new_name _new_parent_spec_as_instance _eat_selfZunsafer, rB r r r r! x s8
zNonCallableMock.__init__c C s( d |_ d |_d|_d |_t| || d S )Nr )rt ru rv rw rz )rf r1 attributer r r attach_mock s
zNonCallableMock.attach_mockc C s | j || d S )N)r )rf r r r r r
mock_add_spec s z NonCallableMock.mock_add_specc C s| d }d }|d k rRt | rRt|tr*|}nt|}t|||}|oH|d }t|}| j}||d< ||d< ||d< ||d< d S )Nr _spec_class _spec_set_spec_signature
_mock_methods)r@ r r
_get_classr( dirrB ) rf r r r r r r resrB r r r r s
z NonCallableMock._mock_add_specc C s8 | j }| jd k r| jj}|tkr4| j| dd}|| _|S )Nz())r r )rr ra rY r _get_child_mock)rf rZ r r r Z__get_return_value s
z"NonCallableMock.__get_return_valuec C s, | j d k r|| j _n|| _t| |d d d S )Nz())ra rY rr r )rf rp r r r Z__set_return_value s
z"NonCallableMock.__set_return_valuez1The value to be returned when the mock is called.c C s | j d krt| S | j S )N)r r )rf r r r __class__ s
zNonCallableMock.__class__r[ r\ r] r^ rW c C sT | j }|d kr| jS |j}|d k rPt| rPt|t rPt| rPt|}||_|S )N)ra rs r_ callabler r r )rf delegatedZsfr r r Z__get_side_effect s z!NonCallableMock.__get_side_effectc C s( t |}| j}|d kr || _n||_d S )N) _try_iterra rs r_ )rf rp r r r r Z__set_side_effect s
z!NonCallableMock.__set_side_effect)rY r_ c C s |d kr g }t | |kr d S |jt | d| _d | _d| _t | _t | _t | _|r^t | _
|rhd | _ x0| j j
D ]"}t|tst|tkrqt|j| qtW | j
}t|r|| k r|j| d S )NFr )idappendr[ r] r\ rU rW r^ rV r rr rs r` valuesr
_SpecState_deletedrX r )rf ZvisitedrY r_ ZchildrZ r r r rX
s, zNonCallableMock.reset_mockc K s` xZt |j dd dD ]B\}}|jd}|j }| }x|D ]}t||}q:W t||| qW d S )Nc S s | d j dS )Nr .)count)entryr r r 8 s z0NonCallableMock.configure_mock..)keyr )sorteditemssplitpopr< rz )rf r, argvalr+ finalr r r r r r + s
z NonCallableMock.configure_mockc C s |dkrt |n:| jd k r<|| jks.|tkrLt d| nt|rLt || jsd|jdrdt || jj|}|tkrt |np|d krd }| j d k rt
| j |}| j | |||| d}|| j|< n.t |t
rt|j|j|j|j|j}|| j|< |S ) Nr r z Mock object has no attribute %rassertassret)r r r r r > r r )r r )r" r _all_magicsrd r r r` rC r r r< r r r r r r r3 r r )rf r rJ r r r r rm A s6
zNonCallableMock.__getattr__c
C s" | j g}| j}| }d}|dgkr$d}t }xT|d k r~|}|j|j | d}|j dkrZd}|j}t||krnP |jt| q,W tt|}|jpd}t |dkr|d d
kr|d7 }||d< dj
|}d}|dkrd | }d} | j d k rd
} | j rd } | | j j
} d t| j
|| t| f S )Nr ()r r1 r ().r mock.z name=%rz spec=%rz spec_set=%rz<%s%s%s id='%s'>)r r )r1 r )rw ru ro r r r} r> reversedrv r joinr r r5 r )
rf Z
_name_listr ZlastdotseenZ_firstr Z name_stringZ spec_stringr r r rg g sL
zNonCallableMock.__repr__c C sh t stj| S | jpg }tt| }t| j}dd |D }dd |D }tt || | t| j
S )Nc S s g | ]}|j d s|qS )r )r )r er r r
s z+NonCallableMock.__dir__..c S s$ g | ] }|j d s t|r|qS )r )r rd )r r r r r r s ) r r __dir__r r r r> rB r ro r` )rf ZextrasZ from_typeZ from_dictr r r r s
zNonCallableMock.__dir__c s" |t krtj||S jrHjd k rH|jkrH|jkrHtd| n|tkrbd| }t|n|tkrjd k r|jkrtd| t |st
t |t || | fdd}n(t
|d | t
t || |j|< n.|dk r|_d S t
|||r|j|< tj||S )Nz!Mock object has no attribute '%s'z.Attempting to set unsupported magic method %r.c s f| |S )Nr )r+ r )rI rf r r r s z-NonCallableMock.__setattr__..r )r| r __setattr__r r rB r" _unsupported_magicsr r rz r _get_methodr r` r )rf r rp msgr )rI rf r r s2
zNonCallableMock.__setattr__c C s |t kr2|t| jkr2tt| | || jkr2d S || jkrHtj| | | jj|t}|t krft
||tk rv| j|= t | j|< d S )N) r r rB delattrr __delattr__r` rC _missingr r" )rf r r r r r r s
zNonCallableMock.__delattr__c C s | j pd}t|||S )Nr1 )rv _format_call_signature)rf r+ r, r r r r _format_mock_call_signature s
z+NonCallableMock._format_mock_call_signaturec C sD d}| j ||}| j}t|dkr.|dd }| j | }|||f S )Nz!Expected call: %s
Actual call: %s r )r r] r )rf r+ r, messageexpected_stringr] Z
actual_stringr r r _format_mock_failure_message s
z,NonCallableMock._format_mock_failure_messagec C st | j }|d k rlt|dkr(d}|\}}n
|\}}}y||j||fS tk
rh } z
|jd S d }~X qpX n|S d S )Nrb r )r r r) TypeErrorwith_traceback)rf _callr- r r+ r, r r r r
_call_matcher s
z NonCallableMock._call_matcherc C s. | }|j dkr*d|jpd|j f }t|d S )Nr z7Expected '%s' to not have been called. Called %s times.r1 )r\ rv AssertionError)r* rf r r r r rP s
z!NonCallableMock.assert_not_calledc C s( | }|j dkr$d|j pd}t|d S )Nr z"Expected '%s' to have been called.r1 )r\ rv r )r* rf r r r r rO s
z NonCallableMock.assert_calledc C s. | }|j dks*d|jpd|j f }t|d S )Nr z8Expected '%s' to have been called once. Called %s times.r1 )r\ rv r )r* rf r r r r rQ s
z"NonCallableMock.assert_called_oncec s| | j d kr(j }td|f fdd}j f}jj }||krxt|trh|nd }t| |d S )Nz Expected call: %s
Not calledc s j } | S )N)r )r )r+ r, rf r r _error_message' s z:NonCallableMock.assert_called_with.._error_message)r] r r r r Exception)r* r+ r, expectedr actualcauser )r+ r, rf r rN s
z"NonCallableMock.assert_called_withc O s6 | }|j dks*d|jpd|j f }t||j||S )Nr z1Expected '%s' to be called once. Called %s times.r1 )r\ rv r rN )r* r+ r, rf r r r r rR 1 s
z'NonCallableMock.assert_called_once_withc s fdd|D }t |tr |nd }t fdd jD }|sd||kr`tdt| jf |d S t|}g }x:|D ]2}y|j| W qv tk
r |j| Y qvX qvW |rtdt |f |d S )Nc s g | ]} j |qS r )r )r c)rf r r r F s z4NonCallableMock.assert_has_calls..c 3 s | ]} j |V qd S )N)r )r r )rf r r H s z3NonCallableMock.assert_has_calls..z(Calls not found.
Expected: %r
Actual: %rz %r not all found in call list)
r r rU rW r r> remover& r r? )rf ZcallsZ any_orderr r Z all_callsZ not_foundZkallr )rf r rS < s*
z NonCallableMock.assert_has_callsc sZ j ||f} fdd jD }||krVt|tr8|nd } j||}td| |d S )Nc s g | ]} j |qS r )r )r r )rf r r r f s z3NonCallableMock.assert_any_call..z%s call not found)r r^ r r r r )rf r+ r, r r r r r )rf r rT _ s z NonCallableMock.assert_any_callc K sF t | }t|ts2t|tr"t}q.) r r r r r r r r r r, attribute_namer )rf r r r r r i s z _patch.copyc C s t |tr| j|S | j|S )N)r r decorate_classdecorate_callable)rf r' r r r r$ v s
z_patch.__call__c C sR xLt |D ]@}|jtjs q
t||}t|ds2q
| j }t|||| q
W |S )Nr$ )r r r TEST_PREFIXr< r r rz )rf r attr
attr_valuer r r r r | s
z_patch.decorate_classc s< t dr jj| S t fdd| g_S )N patchingsc s g }g }t }zybxJjD ]@}|j }|j| |jd k rF|j| q|jtkr|j| qW | t |7 } | |S ||krt|r|j| t j
} Y nX W d xt |D ]}|j | qW X d S )N)
r? r __enter__r r updater r r sysexc_infor __exit__)r+ Zkeywargs
extra_argsZentered_patchersr patchingr )r' patchedr r r s*
z)_patch.decorate_callable..patched)r r r r )rf r' r )r' r r r s
z_patch.decorate_callablec C s | j }| j}t}d}y|j| }W n$ ttfk
rH t||t}Y nX d}|tkrft|t rfd| _
| j
r|tkrtd||f ||fS )NFTz!%s does not have the attribute %r) r r r rB r" KeyErrorr< _builtinsr r r )rf r r rI localr r r get_original s z_patch.get_originalc C s> | j | j| j }}}| j| j }}| j}| j | _|dkr@d }|dkrLd }|dkrXd }|d k rp|d k rptd|d k s|d k r|dkrtd| j \}}|t
ko|d krTd} |dkr|}|dkr|}d }n&|d k r|dkr|}d }n |dkr|}|d k s
|d k r,|t
kr tdt |t r,d} t
}
i } |d k rD|}
nN|d k sX|d k r|} |d k rj|} t| r~d| k}
n
t| }
|
rt}
|d k r|| d< |d k r|| d< t |
t rt|
tr| jr| j| d < | j| |
f | }| rt|r|} |d k r|} t| r2t| r2t}
| jd |
f |d
d | |_nl|d k r|t
k rptd |t
krtd
t|}|dkr|}t|f|| jd|}n|rtd|}|| _|| _ t | j| j| | j d k r:i }| j t
kr||| j < x.| j D ]$}|j }|j t
kr|j| qW |S |S )NFz Can't specify spec and autospecTz6Can't provide explicit spec_set *and* spec or autospecz!Can't use 'spec' with create=Truer$ r r r z())r r zBautospec creates the mock for you. Can't specify autospec and new.z%Can't use 'autospec' with create=True)r _namez.Can't pass kwargs to a mock we aren't creating)TN)!r r r r r, r r r r r r r r r r@ r r r r
r r
r rE r rY boolr
temp_originalr rz r r r )rf r r r r r, r rI r ZinheritKlass_kwargsZ this_specZ not_callableZnew_attrr r r r r r r s
z_patch.__enter__c G s t | std| jr4| jtk r4t| j| j| j nBt| j| j | j rvt
| j| j sd| jdkrvt| j| j| j | `| `| `x$t | j D ]}t |r|j
| qW d S )Nz stop called on unstarted patcherr6 r8 r9 __annotations__r: )r6 r8 r9 r r: )r RuntimeErrorr r r rz r r r r r r r r
)rf r r r r r r
H s z_patch.__exit__c C s | j }| jj| |S )N)r _active_patchesr )rf rJ r r r start` s z _patch.startc C s. y| j j| W n tk
r$ Y nX | j S )N)r r r& r
)rf r r r stopg s
z _patch.stop)r5 r8 rh r r r! r r$ r r r r r
r r r r r r r I s
(~r c sP y j dd\ }W n& ttfk
r: td f Y nX fdd}||fS )Nr r z.Need a valid target to patch. You supplied: %rc s t S )N)r r )r r r r y s z _get_target..)rsplitr r& )r r r r )r r _get_targets s r c
s$ fdd} t | |||||||| S )Nc s S )Nr r )r r r r s z _patch_object..)r )
r r r r r r r r r, r r )r r
_patch_object} s
r! c
s t tkr fdd}n fdd}|s2tdt|j }|d \} }
t|| |
|||||i } | | _xB|dd D ]2\} }
t|| |
|||||i } | | _| jj| qvW | S )Nc s t S )N)r r )r r r r s z!_patch_multiple..c s S )Nr r )r r r r s z=Must supply at least one keyword argument with patch.multipler r ) r strr& r> r r r r r )
r r r r r r r, r r r r r Z this_patcherr )r r _patch_multiple s&
r# c
K s$ t | \}} t|| ||||||| S )N)r r )
r r r r r r r r, r r r r r r s E
c @ sR e Zd Zf dfddZdd Zdd Zdd Zd
d Zd d
Zdd Z eZ
e Z dS ) _patch_dictFc K s> t |trt|}|| _t|| _| jj| || _d | _d S )N) r r" r in_dictrn r r
clear _original)rf r% r r& r, r r r r! 0 s
z_patch_dict.__init__c s. t trj S t fdd}|S )Nc
s" j z
| |S j X d S )N)r$
_unpatch_dict)r+ r )frf r r _inner> s
z$_patch_dict.__call__.._inner)r r r r )rf r) r* r )r) rf r r$ ; s
z_patch_dict.__call__c C s\ xVt |D ]J}t||}|jtjr
t|dr
t| j| j| j }||}t
||| q
W |S )Nr$ ) r r< r r r r r$ r% r r& rz )rf r r r Z decoratorZ decoratedr r r r I s
z_patch_dict.decorate_classc C s | j d S )N)r$ )rf r r r r T s z_patch_dict.__enter__c C s | j }| j}| j}y |j }W n2 tk
rP i }x|D ]}|| ||< q8W Y nX || _|rdt| y|j| W n. tk
r x|D ]}|| ||< qW Y nX d S )N)r r% r& r r" r' _clear_dictr
)rf r r% r& rI r r r r r$ Y s"
z_patch_dict._patch_dictc C sV | j }| j}t| y|j| W n. tk
rP x|D ]}|| ||< q8W Y nX d S )N)r% r' r+ r
r" )rf r% rI r r r r r( s s
z_patch_dict._unpatch_dictc G s | j dS )NF)r( )rf r+ r r r r
s z_patch_dict.__exit__N) r5 r8 rh r! r$ r r r$ r( r
r r r r r r r$ s
r$ c C sB y | j W n0 tk
r< t| }x|D ]
}| |= q*W Y nX d S )N)r& r" r> )r% keysr r r r r+ s
r+ c C s xt tjD ] } | j q W d S )N)r r r r )r r r r _patch_stopall s r- Ztestzlt le gt ge eq ne getitem setitem delitem len contains iter hash str sizeof enter exit divmod rdivmod neg pos abs invert complex int float index trunc floor ceil bool next zHadd sub mul matmul div floordiv mod lshift rshift and xor or pow truediv c c s | ]}d | V qdS )zi%sNr )r nr r r r s r c c s | ]}d | V qdS )zr%sNr )r r/ r r r r s __get____set__
__delete__ __reversed__ __missing__
__reduce__
__reduce_ex__Z__getinitargs____getnewargs__ __getstate__ __setstate__
__getformat__
__setformat__rg r __subclasses__
__format____getnewargs_ex__c s fdd}| |_ |S )Nc s | f||S )Nr )rf r+ r )r' r r method s z_get_method..method)r5 )r r' r? r )r' r r s r c C s h | ] }d | qS )z__%s__r )r r? r r r r s rm r r! r __prepare____instancecheck____subclasscheck____del__c C s
t j| S )N)r __hash__)rf r r r r s r c C s
t j| S )N)r __str__)rf r r r r s c C s
t j| S )N)r
__sizeof__)rf r r r r s )rD rE rF r y ?g ?) __lt____gt____le____ge____int__r __len__r
__complex__ __float____bool__ __index__c s fdd}|S )Nc s$ j j}|tk r|S | kr dS tS )NT)__eq__rr r NotImplemented)otherret_val)rf r r rQ s z_get_eq..__eq__r )rf rQ r )rf r _get_eq s rU c s fdd}|S )Nc s j jtk rtS | kr dS tS )NF)__ne__rr r rR )rS )rf r r rV s
z_get_ne..__ne__r )rf rV r )rf r _get_ne s rW c s fdd}|S )Nc s j j} | tkrtg S t| S )N)r rr r r )rT )rf r r r s z_get_iter..__iter__r )rf r r )rf r _get_iter s rX )rQ rV r c C s t j|t}|tk r ||_d S tj|}|d k rdy || }W n tk
rX t|}Y nX ||_d S tj|}|d k r|| |_d S )N)_return_valuesrC r rY _calculate_return_valuer" _side_effect_methodsr_ )r1 r? r ZfixedZreturn_calulatorrY Z
side_effectorr r r _set_return_value s
r\ c @ s e Zd Zdd Zdd ZdS )
MagicMixinc O s& | j tt| j|| | j d S )N)_mock_set_magicsr r] r! )rf r+ r r r r r! ( s zMagicMixin.__init__c C s t }t| dd d k rTt j| j}t }t | }x$|D ] }|t| jkr4t| | q4W |tt| j }t| }x |D ]}t||t || qtW d S )Nr )
_magicsr< intersectionr ro r rB r rz
MagicProxy)rf Z these_magicsZ
remove_magicsr r r r r r^ . s
zMagicMixin._mock_set_magicsN)r5 r8 rh r! r^ r r r r r] ' s r] c @ s e Zd ZdddZdS )r Fc C s | j || | j d S )N)r r^ )rf r r r r r r G s z"NonCallableMagicMock.mock_add_specN)F)r5 r8 rh r r r r r r E s c @ s e Zd ZdddZdS )r Fc C s | j || | j d S )N)r r^ )rf r r r r r r ] s zMagicMock.mock_add_specN)F)r5 r8 rh r r r r r r R s c @ s. e Zd Zdd Zdd Zdd Zd
dd ZdS ) ra c C s || _ || _d S )N)r r )rf r r r r r r! i s zMagicProxy.__init__c O s | j }|||S )N) create_mock)rf r+ r, mr r r r$ m s zMagicProxy.__call__c C s8 | j }| j}|j|||d}t||| t||| |S )N)r r r )r r r rz r\ )rf r r rc r r r rb q s zMagicProxy.create_mockNc C s | j S )N)rb )rf r r r r r r0 z s zMagicProxy.__get__)N)r5 r8 rh r! r$ rb r0 r r r r ra h s ra c @ s$ e Zd Zdd Zdd Zdd ZdS )_ANYc C s dS )NTr )rf rS r r r rQ s z _ANY.__eq__c C s dS )NFr )rf rS r r r rV s z _ANY.__ne__c C s dS )Nzr )rf r r r rg s z
_ANY.__repr__N)r5 r8 rh rQ rV rg r r r r rd s rd c C sd d| }d}dj dd |D }dj dd t|j D }|rD|}|r\|rT|d7 }||7 }|| S )Nz%s(%%s)r z, c S s g | ] }t |qS r )repr)r r r r r r s z*_format_call_signature..c S s g | ]\}}d ||f qS )z%s=%rr )r r rp r r r r s )r r r )r r+ r, r Zformatted_argsZ args_stringZ
kwargs_stringr r r r s r c @ sr e Zd Zf ddddfddZf ddddfddZd d
ZejZd d Zd
d Z dd Z
dd Z dd Z dd Z
dS )r r NFTc C s f }i }t |}|dkr$|\}}}nr|dkrd|\} }
t| trX| }t|
trR|
}qb|
}q| |
}}n2|dkr|\}t|tr|}nt|tr|}n|}|rtj| ||fS tj| |||fS )Nr rb r )r r r" r? r ) r rp r r r from_kallr+ r, _lenfirstsecondr r r r s.
z
_Call.__new__c C s || _ || _|| _d S )N)rv rt _mock_from_kall)rf rp r r r rf r r r r! s z_Call.__init__c C sf |t kr dS y t|}W n tk
r, dS X d}t| dkrH| \}}n
| \}}}t| dd rzt|dd rz| j|jkrzdS d}|dkrf i }}n|dkr|\}}}n|dkr|\} t| tr| }i }n"t| tr| }f i }}nf }| }nV|dkr>|\}
} t|
tr2|
}t| tr&| i }}n
f | }}n
|
| }}ndS |rV||krVdS ||f||fkS ) NTFr rb rt r r r )r r r r< rt r r? r" ) rf rS Z len_otherZ self_nameZ self_argsZ self_kwargsZ
other_nameZ
other_argsZ other_kwargsrp rh ri r r r rQ sP
z _Call.__eq__c O s< | j d kr td||fddS | j d }t| j ||f|| dS )Nr z())r )r r )rv r )rf r+ r, r r r r r$ s
z_Call.__call__c C s2 | j d krt|ddS d| j |f }t|| ddS )NF)r rf z%s.%s)r r rf )rv r )rf r r r r r rm s
z_Call.__getattr__c O s | j d||S )Nr )rm )rf r+ r, r r r r $ s z _Call.countc O s | j d||S )Nindex)rm )rf r+ r, r r r rk ' s z _Call.indexc C s| | j s&| jpd}|jdr"d| }|S t| dkr@d}| \}}n0| \}}}|sTd}n |jdshd| }nd| }t|||S )Nr z()zcall%srb zcall.%s)rj rv r r r )rf r r+ r, r r r rg * s
z_Call.__repr__c C s8 g }| }x"|d k r*|j r"|j| |j}q
W tt|S )N)rj r rt rU r )rf valsr r r r call_list? s
z_Call.call_list)r5 r8 rh r r! rQ r rV r$ rm r rk rg rm r r r r r s 7r )rf c K s, t | rt| } t| t}d| i}|r0d| i}n | d kr.r rx rx rx rx )r bytesr ) read_dataZ data_as_listliner )rw r _iterate_read_data s
r| r c s fdd}fdd}fdd t d kr`dd l}ttt|jjtt|ja | d krttdt d } tt d
j
_ t d gd j
_ d j_ d j_ d j_ |j_ d < d j_|j_ fd d
}|| _| _ | S )Nc s j jd k rj jS t d S )Nr ) readlinesrY r> )r+ r, )_statehandler r _readlines_side_effect( s z)mock_open.._readlines_side_effectc s( j jd k rj jS t j d S )Nr )readrY r r )r+ r, )r~ r rz r r _read_side_effect- s z$mock_open.._read_side_effectc 3 sJ j jd k r xj jV qW x d D ]
} | V q&W xt V q6W d S )Nr )readlinerY r )r{ )r~ r rz r r _readline_side_effect2 s
z(mock_open.._readline_side_effectr open)r r )r r c s6 t d< jjd kr2 d< d j_tS )Nr r )r| r r_ r )r+ r, )r r~ r rz r r
reset_dataS s
z mock_open..reset_data) file_spec_ior> ro r
TextIOWrapperunionBytesIOr r r rY r| writer r r} r_ )r1 rz r r r r r )r r~ r rz r r s. "
c @ s$ e Zd Zdd Zdd Zdd ZdS )r
c K s
t f |S )N)r )rf r, r r r r i s z PropertyMock._get_child_mockc C s | S )Nr )rf r Zobj_typer r r r0 l s zPropertyMock.__get__c C s | | d S )Nr )rf r r r r r r1 n s zPropertyMock.__set__N)r5 r8 rh r r0 r1 r r r r r
` s )
r r r r r r r r r r
r r r
)F)F)NFNNN)FFNN)Nr )o__all__ __version__r% r r builtinstypesr functoolsr r r r
BaseExceptionr platformr ZlangZ Throwabler superr r r r( r4 r/ r= r@ rE rM rH rd r re ri r r ZMISSINGr ZDELETEDr rq r| r r> rU r r r r
r r r r r r r r r! r# r r$ r+ r- rn ZmultipleZstopallr Z
magic_methodsZnumericsr r ZinplacerightZ
_non_defaultsr r_ r r rZ rR rY rU rW rX r[ r\ r] r r ra rd r r r? r r r rn r r r rQ r# r0 rr r r| r r
r r r r s
4 i5 ,
1Jv
,
w
DPK ! g"\ \ mock.cpython-36.pycnu [ 3
\7 @ sF dZ d
ZddlZddlZddlZddlZddlmZ ddlm Z m
Z
dd e eD Z e
fZdejkrzddlZe
ejjfZdZeZdd Zdd Zdd Zdd d Zd d Zd!d" Zd#d$ Z d%d& Z dd'd(Z d)d* Z d+d, Z G d-d. d.e!Z"G d/d0 d0e!Z#e# Z$e$j%Z%e$j&Z'e$j(Z)d1d2 Z*d3d4d5d6d7d8d9d:hZ+d;d< Z,G d=d> d>e-Z.d?d@ Z/G dAdB dBe!Z0G dCdD dDe!Z1G dEd d e1Z2dFdG Z3G dHdI dIe1Z4G dJd d e4e2Z5dKdL Z6dMdN Z7dOdP Z8G dQdR dRe!Z9dSdT Z:e%dd dddfdUdVZ;ddWdXZd\d] Z?d^d_ Z@e;e=_!e>e=_Ae# s r javaTc C s t t| tS )N)
issubclasstyper
)objr r r _is_instance_mock2 s r c C s t | tp t | to t| tS )N)
isinstanceBaseExceptionsr r )r r r r
_is_exception8 s
r c C s t | tr6| r6y
| j} W n tk
r. dS X d}n*t | ts`y
| j} W n tk
r^ dS X |rpt| d}n| }y| tj|fS t k
r dS X dS )z
Given an arbitrary, possibly callable object, try to create a suitable
signature object.
Return a (reduced func, signature) tuple, or None.
NT)
r r __init__AttributeError
FunctionTypes__call__r inspectZ signature
ValueError)funcZ as_instanceZeat_selfZsig_funcr r r _get_signature_object? s$
r( Fc sD t | || d krd S \} fdd}t| | |t|_d S )Nc s j || d S )N)bind)
_mock_selfargskwargs)sigr r checksigd s z"_check_signature..checksig)r( _copy_func_detailsr _mock_check_sig)r' mock skipfirstinstancer. r )r- r _check_signature_ s
r4 c # C s | j |_ | j|_y | j|_W n tk
r0 Y nX y | j|_W n tk
rR Y nX y | j|_W n tk
rt Y nX y | j|_W n tk
r Y nX d S )N)__name____doc____text_signature__r"
__module__ __defaults____kwdefaults__)r' funcopyr r r r/ j s$ r/ c C s& t | trdS t| dd d k r"dS dS )NTr$ F)r r getattr)r r r r _callable s
r= c C s t | ttfkS )N)r listtuple)r r r r _is_list s r@ c C sH t | tst| dddk S x(| f| j D ]}|jjddk r(dS q(W dS )ztGiven an object, return True if the object is callable.
For classes, return True if instances would be callable.r$ NTF)r r r< __mro____dict__get)r baser r r _instance_callable s
rE c s t |s d S t|t}t|||}|d kr.| S |\} fdd}t|| |j}|j s^d}|| d}d| } t| | || }
t|
| |
S )Nc s j | | d S )N)r) )r+ r, )r- r r r. s z _set_signature..checksigr; )Z
_checksig_r1 zYdef %s(*args, **kwargs):
_checksig_(*args, **kwargs)
return mock(*args, **kwargs)) r= r r r( r/ r5 isidentifierexec _setup_func) r1 originalr3 r2 resultr' r. r contextsrcr; r )r- r _set_signature s$
rM c
s _ tsd S fdd}fdd}fdd}fdd}fd d
}fd d }fd
d} fdd} d _d _d _t _t _t _j _ j
_
j _ | _ | _
| _| _| _| _| _| _ _d S )Nc s j | |S )N)assert_called_with)r+ r, )r1 r r rN s z'_setup_func..assert_called_withc s j | |S )N)
assert_called)r+ r, )r1 r r rO s z"_setup_func..assert_calledc s j | |S )N)assert_not_called)r+ r, )r1 r r rP s z&_setup_func..assert_not_calledc s j | |S )N)assert_called_once)r+ r, )r1 r r rQ s z'_setup_func..assert_called_oncec s j | |S )N)assert_called_once_with)r+ r, )r1 r r rR s z,_setup_func..assert_called_once_withc s j | |S )N)assert_has_calls)r+ r, )r1 r r rS s z%_setup_func..assert_has_callsc s j | |S )N)assert_any_call)r+ r, )r1 r r rT s z$_setup_func..assert_any_callc s: t _t _j j} t| r6| k r6| j d S )N) _CallList method_calls
mock_calls
reset_mock return_valuer )ret)r; r1 r r rX s z _setup_func..reset_mockFr )r1 r called
call_count call_argsrU call_args_listrV rW rY side_effect_mock_childrenrN rR rS rT rX rO rP rQ _mock_delegate)
r; r1 rN rO rP rQ rR rS rT rX r )r; r1 r rH s: rH c C s d| dd | kS )Nz__%s__ r )r r r r _is_magic s rd c @ s e Zd ZdZdd Zdd ZdS )_SentinelObjectz!A unique, named, sentinel object.c C s
|| _ d S )N)r )selfr r r r r! s z_SentinelObject.__init__c C s
d| j S )Nz sentinel.%s)r )rf r r r __repr__ s z_SentinelObject.__repr__N)r5 r8 __qualname__r6 r! rg r r r r re s re c @ s e Zd ZdZdd Zdd ZdS ) _SentinelzAAccess attributes to return a named object, usable as a sentinel.c C s
i | _ d S )N)
_sentinels)rf r r r r! s z_Sentinel.__init__c C s |dkr t | jj|t|S )N __bases__)r" rj
setdefaultre )rf r r r r __getattr__ s z_Sentinel.__getattr__N)r5 r8 rh r6 r! rm r r r r ri s ri c C s$ t | ttttfkr t | | S | S )N)r dictr> r? set)valuer r r _copy s rq rY _mock_return_valuer_ _mock_side_effect _mock_parent_mock_new_parent
_mock_name_mock_new_namec C s8 t j| d| }| |fdd}| |fdd}t||S )NZ_mock_c S s" | j }|d krt| |S t||S )N)ra r< )rf r _the_namer- r r r _get s
z"_delegating_property.._getc S s* | j }|d kr|| j|< n t||| d S )N)ra rB setattr)rf rp r rx r- r r r _set$ s z"_delegating_property.._set)_allowed_namesaddproperty)r rx ry r{ r r r _delegating_property s
r c @ s e Zd Zdd Zdd ZdS )rU c C sn t |tstj| |S t|}t| }||kr2dS x6td|| d D ] }| ||| }||krFdS qFW dS )NFr T)r r> __contains__lenrange)rf rp Z len_valueZlen_selfiZsub_listr r r r 1 s
z_CallList.__contains__c C s t jt| S )N)pprintZpformatr> )rf r r r rg ? s z_CallList.__repr__N)r5 r8 rh r rg r r r r rU / s rU c C sx t |s dS |js,|js,|jd k s,|jd k r0dS | }x |d k rR||krJdS |j}q6W |rd| |_||_|rt| |_||_dS )NFT)r rv rw rt ru )parentrp r new_name_parentr r r _check_and_set_parentC s$
r c @ s$ e Zd Zdd Zdd Zdd ZdS ) _MockIterc C s t || _d S )N)iterr )rf r r r r r! ] s z_MockIter.__init__c C s | S )Nr )rf r r r __iter___ s z_MockIter.__iter__c C s
t | jS )N)nextr )rf r r r __next__a s z_MockIter.__next__N)r5 r8 rh r! r r r r r r r \ s r c @ s e Zd ZeZdZdd ZdS )BaseNc O s d S )Nr )rf r+ r, r r r r! g s z
Base.__init__)r5 r8 rh r rr rs r! r r r r r d s r c @ sH e Zd ZdZdd ZdDddZd d
ZdEd d ZdFd
dZdd Z dd Z
dZ e e e
e Z
e dd ZedZedZedZedZedZdd Zd d Ze eeZdGddd d d!Zd"d# Zd$d% Zd&d' Zd(d) Z d*d+ Z d,d- Z d.d/ Z d0d1 Z d2d3 Z!d4d5 Z"d6d7 Z#d8d9 Z$d:d; Z%dd?Z'd@dA Z(dBdC Z)dS )Ir
z A non-callable version of `Mock`c O s$ t | j| fd| ji}tj|}|S )Nr6 )r r5 r6 object__new__)clsr+ kwnewr3 r r r r o s
zNonCallableMock.__new__N Fc K s |d kr |}| j }
||
d< ||
d< ||
d< ||
d< |d k rB|}d}|
d krR|d k }
| j||| |
i |
d< ||
d< d |
d< d |
d
< d |
d < d |
d
< t |
d< t |
d< t |
d< | |
d< | r| jf | tt| j|||||| d S )Nrt rv rw ru Tr` _mock_wrapsra FZ _mock_calledZ_mock_call_argsr Z_mock_call_countZ_mock_call_args_listZ_mock_mock_callsrV _mock_unsafe)rB _mock_add_specrU configure_mock _safe_superr
r! )rf specr r spec_setr _spec_state _new_name _new_parent_spec_as_instance _eat_selfZunsafer, rB r r r r! x s8
zNonCallableMock.__init__c C s( d|_ d|_d|_d|_t| || dS )z
Attach a mock as an attribute of this one, replacing its name and
parent. Calls to the attached mock will be recorded in the
`method_calls` and `mock_calls` attributes of this one.Nr )rt ru rv rw rz )rf r1 attributer r r attach_mock s
zNonCallableMock.attach_mockc C s | j || dS )zAdd a spec to a mock. `spec` can either be an object or a
list of strings. Only attributes on the `spec` can be fetched as
attributes from the mock.
If `spec_set` is True then only attributes on the spec can be set.N)r )rf r r r r r
mock_add_spec s z NonCallableMock.mock_add_specc C s| d }d }|d k rRt | rRt|tr*|}nt|}t|||}|oH|d }t|}| j}||d< ||d< ||d< ||d< d S )Nr _spec_class _spec_set_spec_signature
_mock_methods)r@ r r
_get_classr( dirrB ) rf r r r r r r resrB r r r r s
z NonCallableMock._mock_add_specc C s8 | j }| jd k r| jj}|tkr4| j| dd}|| _|S )Nz())r r )rr ra rY r _get_child_mock)rf rZ r r r Z__get_return_value s
z"NonCallableMock.__get_return_valuec C s, | j d k r|| j _n|| _t| |d d d S )Nz())ra rY rr r )rf rp r r r Z__set_return_value s
z"NonCallableMock.__set_return_valuez1The value to be returned when the mock is called.c C s | j d krt| S | j S )N)r r )rf r r r __class__ s
zNonCallableMock.__class__r[ r\ r] r^ rW c C sT | j }|d kr| jS |j}|d k rPt| rPt|t rPt| rPt|}||_|S )N)ra rs r_ callabler r r )rf delegatedZsfr r r Z__get_side_effect s z!NonCallableMock.__get_side_effectc C s( t |}| j}|d kr || _n||_d S )N) _try_iterra rs r_ )rf rp r r r r Z__set_side_effect s
z!NonCallableMock.__set_side_effect)rY r_ c C s |dkr g }t | |kr dS |jt | d| _d| _d| _t | _t | _t | _|r^t | _
|rhd| _ x0| j j
D ]"}t|tst|tkrqt|j| qtW | j
}t|r|| k r|j| dS )z-Restore the mock object to its initial state.NFr )idappendr[ r] r\ rU rW r^ rV r rr rs r` valuesr
_SpecState_deletedrX r )rf ZvisitedrY r_ ZchildrZ r r r rX
s, zNonCallableMock.reset_mockc K s` xZt |j dd dD ]B\}}|jd}|j }| }x|D ]}t||}q:W t||| qW dS )aZ Set attributes on the mock through keyword arguments.
Attributes plus return values and side effects can be set on child
mocks using standard dot notation and unpacking a dictionary in the
method call:
>>> attrs = {'method.return_value': 3, 'other.side_effect': KeyError}
>>> mock.configure_mock(**attrs)c S s | d j dS )Nr .)count)entryr r r 8 s z0NonCallableMock.configure_mock..)keyr N)sorteditemssplitpopr< rz )rf r, argvalr+ finalr r r r r r + s
z NonCallableMock.configure_mockc C s |dkrt |n:| jd k r<|| jks.|tkrLt d| nt|rLt || jsd|jdrdt || jj|}|tkrt |np|d krd }| j d k rt
| j |}| j | |||| d}|| j|< n.t |t
rt|j|j|j|j|j}|| j|< |S ) Nr r z Mock object has no attribute %rassertassret)r r r r r > r r )r r )r" r _all_magicsrd r r r` rC r r r< r r r r r r r3 r r )rf r rJ r r r r rm A s6
zNonCallableMock.__getattr__c
C s" | j g}| j}| }d}|dgkr$d}t }xT|d k r~|}|j|j | d}|j dkrZd}|j}t||krnP |jt| q,W tt|}|jpd}t |dkr|d d
kr|d7 }||d< dj
|}d}|dkrd | }d} | j d k rd
} | j rd } | | j j
} d t| j
|| t| f S )Nr ()r r1 r ().r mock.z name=%rz spec=%rz spec_set=%rz<%s%s%s id='%s'>)r r )r1 r )rw ru ro r r r} r> reversedrv r joinr r r5 r )
rf Z
_name_listr ZlastdotseenZ_firstr Z name_stringZ spec_stringr r r rg g sL
zNonCallableMock.__repr__c C sh t stj| S | jpg }tt| }t| j}dd |D }dd |D }tt || | t| j
S )z8Filter the output of `dir(mock)` to only useful members.c S s g | ]}|j d s|qS )r )r )r er r r
s z+NonCallableMock.__dir__..c S s$ g | ] }|j d s t|r|qS )r )r rd )r r r r r r s ) r r __dir__r r r r> rB r ro r` )rf ZextrasZ from_typeZ from_dictr r r r s
zNonCallableMock.__dir__c s" |t krtj||S jrHjd k rH|jkrH|jkrHtd| n|tkrbd| }t|n|tkrjd k r|jkrtd| t |st
t |t || | fdd}n(t
|d | t
t || |j|< n.|dk r|_d S t
|||r|j|< tj||S )Nz!Mock object has no attribute '%s'z.Attempting to set unsupported magic method %r.c s f| |S )Nr )r+ r )rI rf r r r s z-NonCallableMock.__setattr__..r )r| r __setattr__r r rB r" _unsupported_magicsr r rz r _get_methodr r` r )rf r rp msgr )rI rf r r s2
zNonCallableMock.__setattr__c C s |t kr2|t| jkr2tt| | || jkr2d S || jkrHtj| | | jj|t}|t krft
||tk rv| j|= t | j|< d S )N) r r rB delattrr __delattr__r` rC _missingr r" )rf r r r r r r s
zNonCallableMock.__delattr__c C s | j pd}t|||S )Nr1 )rv _format_call_signature)rf r+ r, r r r r _format_mock_call_signature s
z+NonCallableMock._format_mock_call_signaturec C sD d}| j ||}| j}t|dkr.|dd }| j | }|||f S )Nz!Expected call: %s
Actual call: %s r )r r] r )rf r+ r, messageexpected_stringr] Z
actual_stringr r r _format_mock_failure_message s
z,NonCallableMock._format_mock_failure_messagec C st | j }|dk rlt|dkr(d}|\}}n
|\}}}y||j||fS tk
rh } z
|jdS d}~X qpX n|S dS )a
Given a call (or simply an (args, kwargs) tuple), return a
comparison key suitable for matching with other calls.
This is a best effort method which relies on the spec's signature,
if available, or falls back on the arguments themselves.
Nrb r )r r r) TypeErrorwith_traceback)rf _callr- r r+ r, r r r r
_call_matcher s
z NonCallableMock._call_matcherc C s. | }|j dkr*d|jpd|j f }t|dS )z/assert that the mock was never called.
r z7Expected '%s' to not have been called. Called %s times.r1 N)r\ rv AssertionError)r* rf r r r r rP s
z!NonCallableMock.assert_not_calledc C s( | }|j dkr$d|j pd}t|dS )z6assert that the mock was called at least once
r z"Expected '%s' to have been called.r1 N)r\ rv r )r* rf r r r r rO s
z NonCallableMock.assert_calledc C s. | }|j dks*d|jpd|j f }t|dS )z3assert that the mock was called only once.
r z8Expected '%s' to have been called once. Called %s times.r1 N)r\ rv r )r* rf r r r r rQ s
z"NonCallableMock.assert_called_oncec s| | j dkr(j }td|f fdd}j f}jj }||krxt|trh|nd}t| |dS )zassert that the mock was called with the specified arguments.
Raises an AssertionError if the args and keyword args passed in are
different to the last call to the mock.Nz Expected call: %s
Not calledc s j } | S )N)r )r )r+ r, rf r r _error_message' s z:NonCallableMock.assert_called_with.._error_message)r] r r r r Exception)r* r+ r, expectedr actualcauser )r+ r, rf r rN s
z"NonCallableMock.assert_called_withc O s6 | }|j dks*d|jpd|j f }t||j||S )ziassert that the mock was called exactly once and that that call was
with the specified arguments.r z1Expected '%s' to be called once. Called %s times.r1 )r\ rv r rN )r* r+ r, rf r r r r rR 1 s
z'NonCallableMock.assert_called_once_withc s fdd|D }t |tr |nd}t fdd jD }|sd||kr`tdt| jf |dS t|}g }x:|D ]2}y|j| W qv tk
r |j| Y qvX qvW |rtdt |f |dS )a assert the mock has been called with the specified calls.
The `mock_calls` list is checked for the calls.
If `any_order` is False (the default) then the calls must be
sequential. There can be extra calls before or after the
specified calls.
If `any_order` is True then the calls can be in any order, but
they must all appear in `mock_calls`.c s g | ]} j |qS r )r )r c)rf r r r F s z4NonCallableMock.assert_has_calls..Nc 3 s | ]} j |V qd S )N)r )r r )rf r r H s z3NonCallableMock.assert_has_calls..z(Calls not found.
Expected: %r
Actual: %rz %r not all found in call list)
r r rU rW r r> remover& r r? )rf ZcallsZ any_orderr r Z all_callsZ not_foundZkallr )rf r rS < s*
z NonCallableMock.assert_has_callsc sZ j ||f} fdd jD }||krVt|tr8|nd} j||}td| |dS )zassert the mock has been called with the specified arguments.
The assert passes if the mock has *ever* been called, unlike
`assert_called_with` and `assert_called_once_with` that only pass if
the call is the most recent one.c s g | ]} j |qS r )r )r r )rf r r r f s z3NonCallableMock.assert_any_call..Nz%s call not found)r r^ r r r r )rf r+ r, r r r r r )rf r rT _ s z NonCallableMock.assert_any_callc K sF t | }t|ts2t|tr"t}q.) r r r r r r r r r r, attribute_namer )rf r r r r r i s z _patch.copyc C s t |tr| j|S | j|S )N)r r decorate_classdecorate_callable)rf r' r r r r$ v s
z_patch.__call__c C sR xLt |D ]@}|jtjs q
t||}t|ds2q
| j }t|||| q
W |S )Nr$ )r r r TEST_PREFIXr< r r rz )rf r attr
attr_valuer r r r r | s
z_patch.decorate_classc s< t dr jj| S t fdd| g_S )N patchingsc s g }g }t }zybxJjD ]@}|j }|j| |jd k rF|j| q|jtkr|j| qW | t |7 } | |S ||krt|r|j| t j
} Y nX W d xt |D ]}|j | qW X d S )N)
r? r __enter__r r updater r r sysexc_infor __exit__)r+ Zkeywargs
extra_argsZentered_patchersr patchingr )r' patchedr r r s*
z)_patch.decorate_callable..patched)r r r r )rf r' r )r' r r r s
z_patch.decorate_callablec C s | j }| j}t}d}y|j| }W n$ ttfk
rH t||t}Y nX d}|tkrft|t rfd| _
| j
r|tkrtd||f ||fS )NFTz!%s does not have the attribute %r) r r r rB r" KeyErrorr< _builtinsr r r )rf r r rI localr r r get_original s z_patch.get_originalc C s> | j | j| j }}}| j| j }}| j}| j | _|dkr@d}|dkrLd}|dkrXd}|dk rp|dk rptd|dk s|dk r|dkrtd| j \}}|t
ko|dkrTd} |dkr|}|dkr|}d}n&|dk r|dkr|}d}n |dkr|}|dk s
|dk r,|t
kr tdt |t r,d} t
}
i } |dk rD|}
nN|dk sX|dk r|} |dk rj|} t| r~d| k}
n
t| }
|
rt}
|dk r|| d< |dk r|| d < t |
t rt|
tr| jr| j| d
< | j| |
f | }| rt|r|} |dk r|} t| r2t| r2t}
| jd
|
f |d d | |_nl|dk r|t
k rptd
|t
krtdt|}|dkr|}t|f|| jd|}n|rtd|}|| _|| _ t | j| j| | j dk r:i }| j t
kr||| j < x.| j D ]$}|j }|j t
kr|j| qW |S |S )zPerform the patch.FNz Can't specify spec and autospecTz6Can't provide explicit spec_set *and* spec or autospecz!Can't use 'spec' with create=Truer$ r r r z())r r zBautospec creates the mock for you. Can't specify autospec and new.z%Can't use 'autospec' with create=True)r _namez.Can't pass kwargs to a mock we aren't creating)TN)!r r r r r, r r r r r r r r r r@ r r r r
r r
r rE r rY boolr
temp_originalr rz r r r )rf r r r r r, r rI r ZinheritKlass_kwargsZ this_specZ not_callableZnew_attrr r r r r r r s
z_patch.__enter__c G s t | std| jr4| jtk r4t| j| j| j nBt| j| j | j rvt
| j| j sd| jdkrvt| j| j| j | `| `| `x$t | j D ]}t |r|j
| qW dS ) zUndo the patch.z stop called on unstarted patcherr6 r8 r9 __annotations__r: N)r6 r8 r9 r r: )r RuntimeErrorr r r rz r r r r r r r r
)rf r r r r r r
H s z_patch.__exit__c C s | j }| jj| |S )z-Activate a patch, returning any created mock.)r _active_patchesr )rf rJ r r r start` s z _patch.startc C s. y| j j| W n tk
r$ Y nX | j S )zStop an active patch.)r r r& r
)rf r r r stopg s
z _patch.stop)r5 r8 rh r r r! r r$ r r r r r
r r r r r r r I s
(~r c sP y j dd\ }W n& ttfk
r: td f Y nX fdd}||fS )Nr r z.Need a valid target to patch. You supplied: %rc s t S )N)r r )r r r r y s z _get_target..)rsplitr r& )r r r r )r r _get_targets s r c
s$ fdd} t | |||||||| S )a
patch the named member (`attribute`) on an object (`target`) with a mock
object.
`patch.object` can be used as a decorator, class decorator or a context
manager. Arguments `new`, `spec`, `create`, `spec_set`,
`autospec` and `new_callable` have the same meaning as for `patch`. Like
`patch`, `patch.object` takes arbitrary keyword arguments for configuring
the mock object it creates.
When used as a class decorator `patch.object` honours `patch.TEST_PREFIX`
for choosing which methods to wrap.
c s S )Nr r )r r r r s z _patch_object..)r )
r r r r r r r r r, r r )r r
_patch_object} s
r! c
s t tkr fdd}n fdd}|s2tdt|j }|d \} }
t|| |
|||||i } | | _xB|dd D ]2\} }
t|| |
|||||i } | | _| jj| qvW | S )a Perform multiple patches in a single call. It takes the object to be
patched (either as an object or a string to fetch the object by importing)
and keyword arguments for the patches::
with patch.multiple(settings, FIRST_PATCH='one', SECOND_PATCH='two'):
...
Use `DEFAULT` as the value if you want `patch.multiple` to create
mocks for you. In this case the created mocks are passed into a decorated
function by keyword, and a dictionary is returned when `patch.multiple` is
used as a context manager.
`patch.multiple` can be used as a decorator, class decorator or a context
manager. The arguments `spec`, `spec_set`, `create`,
`autospec` and `new_callable` have the same meaning as for `patch`. These
arguments will be applied to *all* patches done by `patch.multiple`.
When used as a class decorator `patch.multiple` honours `patch.TEST_PREFIX`
for choosing which methods to wrap.
c s t S )N)r r )r r r r s z!_patch_multiple..c s S )Nr r )r r r r s z=Must supply at least one keyword argument with patch.multipler r N) r strr& r> r r r r r )
r r r r r r r, r r r r r Z this_patcherr )r r _patch_multiple s&
r# c
K s$ t | \}} t|| ||||||| S )a
`patch` acts as a function decorator, class decorator or a context
manager. Inside the body of the function or with statement, the `target`
is patched with a `new` object. When the function/with statement exits
the patch is undone.
If `new` is omitted, then the target is replaced with a
`MagicMock`. If `patch` is used as a decorator and `new` is
omitted, the created mock is passed in as an extra argument to the
decorated function. If `patch` is used as a context manager the created
mock is returned by the context manager.
`target` should be a string in the form `'package.module.ClassName'`. The
`target` is imported and the specified object replaced with the `new`
object, so the `target` must be importable from the environment you are
calling `patch` from. The target is imported when the decorated function
is executed, not at decoration time.
The `spec` and `spec_set` keyword arguments are passed to the `MagicMock`
if patch is creating one for you.
In addition you can pass `spec=True` or `spec_set=True`, which causes
patch to pass in the object being mocked as the spec/spec_set object.
`new_callable` allows you to specify a different class, or callable object,
that will be called to create the `new` object. By default `MagicMock` is
used.
A more powerful form of `spec` is `autospec`. If you set `autospec=True`
then the mock will be created with a spec from the object being replaced.
All attributes of the mock will also have the spec of the corresponding
attribute of the object being replaced. Methods and functions being
mocked will have their arguments checked and will raise a `TypeError` if
they are called with the wrong signature. For mocks replacing a class,
their return value (the 'instance') will have the same spec as the class.
Instead of `autospec=True` you can pass `autospec=some_object` to use an
arbitrary object as the spec instead of the one being replaced.
By default `patch` will fail to replace attributes that don't exist. If
you pass in `create=True`, and the attribute doesn't exist, patch will
create the attribute for you when the patched function is called, and
delete it again afterwards. This is useful for writing tests against
attributes that your production code creates at runtime. It is off by
default because it can be dangerous. With it switched on you can write
passing tests against APIs that don't actually exist!
Patch can be used as a `TestCase` class decorator. It works by
decorating each test method in the class. This reduces the boilerplate
code when your test methods share a common patchings set. `patch` finds
tests by looking for method names that start with `patch.TEST_PREFIX`.
By default this is `test`, which matches the way `unittest` finds tests.
You can specify an alternative prefix by setting `patch.TEST_PREFIX`.
Patch can be used as a context manager, with the with statement. Here the
patching applies to the indented block after the with statement. If you
use "as" then the patched object will be bound to the name after the
"as"; very useful if `patch` is creating a mock object for you.
`patch` takes arbitrary keyword arguments. These will be passed to
the `Mock` (or `new_callable`) on construction.
`patch.dict(...)`, `patch.multiple(...)` and `patch.object(...)` are
available for alternate use-cases.
)r r )
r r r r r r r r, r r r r r r s E
c @ sV e Zd ZdZf dfddZdd Zdd Zd d
Zd d Zd
d Z dd Z
eZ e
Z dS ) _patch_dicta#
Patch a dictionary, or dictionary like object, and restore the dictionary
to its original state after the test.
`in_dict` can be a dictionary or a mapping like container. If it is a
mapping then it must at least support getting, setting and deleting items
plus iterating over keys.
`in_dict` can also be a string specifying the name of the dictionary, which
will then be fetched by importing it.
`values` can be a dictionary of values to set in the dictionary. `values`
can also be an iterable of `(key, value)` pairs.
If `clear` is True then the dictionary will be cleared before the new
values are set.
`patch.dict` can also be called with arbitrary keyword arguments to set
values in the dictionary::
with patch.dict('sys.modules', mymodule=Mock(), other_module=Mock()):
...
`patch.dict` can be used as a context manager, decorator or class
decorator. When used as a class decorator `patch.dict` honours
`patch.TEST_PREFIX` for choosing which methods to wrap.
Fc K s> t |trt|}|| _t|| _| jj| || _d | _d S )N) r r" r in_dictrn r r
clear _original)rf r% r r& r, r r r r! 0 s
z_patch_dict.__init__c s. t trj S t fdd}|S )Nc
s" j z
| |S j X d S )N)r$
_unpatch_dict)r+ r )frf r r _inner> s
z$_patch_dict.__call__.._inner)r r r r )rf r) r* r )r) rf r r$ ; s
z_patch_dict.__call__c C s\ xVt |D ]J}t||}|jtjr
t|dr
t| j| j| j }||}t
||| q
W |S )Nr$ ) r r< r r r r r$ r% r r& rz )rf r r r Z decoratorZ decoratedr r r r I s
z_patch_dict.decorate_classc C s | j dS )zPatch the dict.N)r$ )rf r r r r T s z_patch_dict.__enter__c C s | j }| j}| j}y |j }W n2 tk
rP i }x|D ]}|| ||< q8W Y nX || _|rdt| y|j| W n. tk
r x|D ]}|| ||< qW Y nX d S )N)r r% r& r r" r' _clear_dictr
)rf r r% r& rI r r r r r$ Y s"
z_patch_dict._patch_dictc C sV | j }| j}t| y|j| W n. tk
rP x|D ]}|| ||< q8W Y nX d S )N)r% r' r+ r
r" )rf r% rI r r r r r( s s
z_patch_dict._unpatch_dictc G s | j dS )zUnpatch the dict.F)r( )rf r+ r r r r
s z_patch_dict.__exit__N)
r5 r8 rh r6 r! r$ r r r$ r( r
r r r r r r r$ s
r$ c C sB y | j W n0 tk
r< t| }x|D ]
}| |= q*W Y nX d S )N)r& r" r> )r% keysr r r r r+ s
r+ c C s xt tjD ] } | j q W dS )z7Stop all active patches. LIFO to unroll nested patches.N)r r r r )r r r r _patch_stopall s r- Ztestzlt le gt ge eq ne getitem setitem delitem len contains iter hash str sizeof enter exit divmod rdivmod neg pos abs invert complex int float index trunc floor ceil bool next zHadd sub mul matmul div floordiv mod lshift rshift and xor or pow truediv c c s | ]}d | V qdS )zi%sNr )r nr r r r s r c c s | ]}d | V qdS )zr%sNr )r r/ r r r r s __get____set__
__delete__ __reversed__ __missing__
__reduce__
__reduce_ex__Z__getinitargs____getnewargs__ __getstate__ __setstate__
__getformat__
__setformat__rg r __subclasses__
__format____getnewargs_ex__c s fdd}| |_ |S )z:Turns a callable object (like a mock) into a real functionc s | f||S )Nr )rf r+ r )r' r r method s z_get_method..method)r5 )r r' r? r )r' r r s r c C s h | ] }d | qS )z__%s__r )r r? r r r r s rm r r! r __prepare____instancecheck____subclasscheck____del__c C s
t j| S )N)r __hash__)rf r r r r s r c C s
t j| S )N)r __str__)rf r r r r s c C s
t j| S )N)r
__sizeof__)rf r r r r s )rD rE rF r y ?g ?) __lt____gt____le____ge____int__r __len__r
__complex__ __float____bool__ __index__c s fdd}|S )Nc s$ j j}|tk r|S | kr dS tS )NT)__eq__rr r NotImplemented)otherret_val)rf r r rQ s z_get_eq.