Your IP : 216.73.216.213
�
��\c @ sr d Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z m
Z
d d l m Z y d d l
m Z Wn e k
r� e Z n Xy d d l m Z Wn9 e k
r� y d d l m Z Wn e k
r� n Xn Xd d l m Z d d l m Z d d l m Z d d
l m Z m Z m Z e rcd d l m Z n e r|d d l m Z n e j d
f k r�d d l Z n d d l Z e j! d k r�d n d Z" e j! d k r�d n d Z# d Z$ e% e& e% e% d � Z' d e f d � � YZ( d e j) e( f d � � YZ) d e f d � � YZ* d e f d � � YZ+ d e f d � � YZ, d S( sG This module provides a base class derived from `unittest.TestClass`
for unit tests using the :py:class:`pyfakefs` module.
`fake_filesystem_unittest.TestCase` searches `sys.modules` for modules
that import the `os`, `io`, `path` `shutil`, and `pathlib` modules.
The `setUpPyfakefs()` method binds these modules to the corresponding fake
modules from `pyfakefs`. Further, the `open()` built-in is bound to a fake
`open()`. In Python 2, built-in `file()` is similarly bound to the fake
`open()`.
It is expected that `setUpPyfakefs()` be invoked at the beginning of the
derived class' `setUp()` method. There is no need to add anything to the
derived class' `tearDown()` method.
During the test, everything uses the fake file system and modules. This means
that even in your test fixture, familiar functions like `open()` and
`os.makedirs()` manipulate the fake file system.
Existing unit tests that use the real file system can be retrofitted to use
pyfakefs by simply changing their base class from `:py:class`unittest.TestCase`
to `:py:class`pyfakefs.fake_filesystem_unittest.TestCase`.
i����N( t IS_PY2t IS_PYPY( t
Deprecator( t
ModuleSpec( t reload( t fake_filesystem( t fake_filesystem_shutil( t mox3_stubout( t pathlibt pathlib2t use_scandir( t fake_pathlib( t fake_scandiri t win32t ntt posixt ntpatht posixpatht __builtin__c
C sh t d | d | d | d | � } | j t | � � } | j t j | d | d | j d | j �� | S( sY Load the doctest tests for the specified module into unittest.
Args:
loader, tests, ignore : arguments passed in from `load_tests()`
module: module under test
remaining args: see :py:class:`TestCase` for an explanation
File `example_test.py` in the pyfakefs release provides a usage example.
t additional_skip_namest use_dynamic_patcht modules_to_reloadt modules_to_patcht globst setUpt tearDown( t Patchert
replace_globst varst addTestst doctestt DocTestSuiteR R (
t loadert testst ignoret moduleR R R R t _patcherR ( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyt
load_doctestsX s
t
TestCaseMixinc B s\ e Z d Z d Z d Z e Z d Z e d � � Z
d d d d d � Z d � Z d � Z
RS( sa Test case mixin that automatically replaces file-system related
modules by fake implementations.
Attributes:
additional_skip_names: names of modules inside of which no module
replacement shall be performed, in addition to the names in
:py:attr:`fake_filesystem_unittest.Patcher.SKIPNAMES`.
modules_to_reload: A list of modules that need to be reloaded
to be patched dynamically; may be needed if the module
imports file system modules under an alias
.. note:: This is done independently of `use_dynamic_patch`
.. caution:: Reloading modules may have unwanted side effects.
use_dynamic_patch: If `True`, dynamic patching after setup is used
(for example for modules loaded locally inside of functions).
Can be switched off if it causes unwanted side effects.
modules_to_patch: A dictionary of fake modules mapped to the
fully qualified patched module names. Can be used to add patching
of modules not provided by `pyfakefs`.
If you specify some of these attributes here and you have DocTests,
consider also specifying the same arguments to :py:func:`load_doctests`.
Example usage in derived test classes::
from unittest import TestCase
from fake_filesystem_unittest import TestCaseMixin
class MyTestCase(TestCase, TestCaseMixin):
def __init__(self, methodName='runTest'):
super(MyTestCase, self).__init__(
methodName=methodName,
additional_skip_names=['posixpath'])
import sut
class AnotherTestCase(TestCase, TestCaseMixin):
def __init__(self, methodName='runTest'):
super(MyTestCase, self).__init__(
methodName=methodName, modules_to_reload=[sut])
c C s
| j j S( N( t _stubbert fs( t self( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyR( � s c C s� | d k r | j } n | d k r0 | j } n | d k rH | j } n | d k r` | j } n t d | d | d | d | � | _ | j j � | j | j j � d S( s\ Bind the file-related modules to the :py:class:`pyfakefs` fake file
system instead of the real file system. Also bind the fake `open()`
function, and on Python 2, the `file()` function.
Invoke this at the beginning of the `setUp()` method in your unit test
class.
For the arguments, see the `TestCaseMixin` attribute description.
If any of the arguments is not None, it overwrites the settings for
the current test case. Settings the arguments here may be a more
convenient way to adapt the setting than overwriting `__init__()`.
R R R R N(
t NoneR R R R R R' R t
addCleanupR ( R) R R R R ( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyt
setUpPyfakefs� s
c C s | j j � d S( s� Pause the patching of the file system modules until `resume` is
called. After that call, all file system calls are executed in the
real file system.
Calling pause() twice is silently ignored.
N( R' t pause( R) ( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyR- � s c C s | j j � d S( s� Resume the patching of the file system modules if `pause` has
been called before. After that call, all file system calls are
executed in the fake file system.
Does nothing if patching is not paused.
N( R' t resume( R) ( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyR. � s N( t __name__t
__module__t __doc__R* R R t TrueR R t propertyR( R, R- R. ( ( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyR& q s * t TestCasec B sP e Z d Z d d d e d d � Z e d � d e d � � Z e d � � Z RS( s� Test case class that automatically replaces file-system related
modules by fake implementations. Inherits :py:class:`TestCaseMixin`.
The arguments are explained in :py:class:`TestCaseMixin`.
t runTestc C s> t t | � j | � | | _ | | _ | | _ | | _ d S( s� Creates the test class instance and the patcher used to stub out
file system related modules.
Args:
methodName: The name of the test method (same as in
unittest.TestCase)
N( t superR4 t __init__R R R R ( R) t
methodNameR R R R ( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyR7 � s
t
add_real_filec C sU | d k r' | | k r' t d � � n | s< t d � � n | j j j | d t �S( sh Add the file `real_file_path` in the real file system to the same
path in the fake file system.
**This method is deprecated** in favor of
:py:meth:`FakeFilesystem..add_real_file`.
`copyRealFile()` is retained with limited functionality for backward
compatibility only.
Args:
real_file_path: Path to the file in both the real and fake
file systems
fake_file_path: Deprecated. Use the default, which is
`real_file_path`.
If a value other than `real_file_path` is specified, a `ValueError`
exception will be raised.
create_missing_dirs: Deprecated. Use the default, which creates
missing directories in the fake file system. If `False` is
specified, a `ValueError` exception is raised.
Returns:
The newly created FakeFile object.
Raises:
IOError: If the file already exists in the fake file system.
ValueError: If deprecated argument values are specified.
See:
:py:meth:`FakeFileSystem.add_real_file`
sV CopyRealFile() is deprecated and no longer supports different real and fake file pathssT CopyRealFile() is deprecated and no longer supports NOT creating missing directoriest read_onlyN( R* t
ValueErrorR' R( R9 t False( R) t real_file_patht fake_file_patht create_missing_dirs( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyt copyRealFile� s
c C s d S( sg This method is deprecated and exists only for backward
compatibility. It does nothing.
N( ( R) ( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyt tearDownPyfakefs s N(
R/ R0 R1 R* R2 R7 R R@ t DeprecationWarningRA ( ( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyR4 � s &R c B s1 e Z d Z d e e e h Z d e k s6 t d � � e j d k Z
d d d d e e h Z
e rs e
j d � n e r� e
j d � n e r� e j j d
� r� e
j d � n d d e d d � Z d
� Z d � Z d � Z d � Z d d � Z d � Z d � Z d d � Z d � Z d � Z d � Z! RS( st
Instantiate a stub creator to bind and un-bind the file-related modules to
the :py:mod:`pyfakefs` fake modules.
The arguments are explained in :py:class:`TestCaseMixin`.
:py:class:`Patcher` is used in :py:class:`TestCaseMixin`.
:py:class:`Patcher` also works as a context manager for other tests::
with Patcher():
doStuff()
s3 sys.modules contains 'None' values; must skip them.R
t cygwint ost patht iot genericpathR R t PYCHARM_HOSTEDt pydevdc
C s | j j � | _ t | _ d | _ | d k rC | j j | � n t g | _ | d k rn | j j
| � n | | _ i t j
d 6t j d 6t j d 6| _ t s� t r� t j | j d <n i | _ t rFg | j d <t rt j | j d <| j d j d � n t r3t j | j d <| j d j d � n t j | j d <n t r_t j | j d <n | d k r�x* | j � D] \ } } | | j | <qxWn i | _ x� | j j � D]� \ } } t | d � r�t! j" | j# � r�xm | j# � D]\ } t$ | | � | f } | | j j% | i � | <| d k r�| | j j% | i � t& <q�q�Wq�q�Wt j' } x[ | j# � D]M } t$ | | � t( f } | | j j% | i � d
<| | j j% | i � t( <qhWi | _) i | _* i | _+ d | _, d | _- i | _. d | _/ t0 | _1 t2 | _3 d S( s9 For a description of the arguments, see TestCase.__init__RD t shutilRF t _iot PathR R t scandirt dirRG N(4 t SKIPNAMESt copyt
_skipNamest opent
original_openR* t fake_opent updatet tempfileR t extendt _use_dynamic_patchR t FakeOsModuleR t FakeShutilModulet FakeIoModulet _fake_module_classesR R t _class_modulesR R t FakePathlibModulet appendR t FakePathlibPathModuleR
R t FakeScanDirModulet itemst _fake_module_functionst hasattrt inspectt
isfunctionRN t getattrt
setdefaultt OS_MODULEt FakePathModulet PATH_MODULEt _modulest _fct_modulest _open_functionst _stubsR( t fake_modulest _dyn_patcherR2 t _isStaleR< t _patching(
R) R R R R t namet fake_modulet mod_namet fct_namet module_attr( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyR7 D sj
' c C s | j � | S( s� Context manager for usage outside of
fake_filesystem_unittest.TestCase.
Ensure that all patched modules are removed in case of an
unhandled exception.
( R ( R) ( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyt __enter__� s
c C s | j � d S( N( R ( R) t exc_typet exc_valt exc_tb( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyt __exit__� s c s� � � f d � � d � � � f d � � t � j j � � t g � x�t t j j � � D]�\ } } yH | � j k s� t j
| � s� | j j d � d � j
k r� w\ n Wn t k
r� q\ n X� f d � | j j � D� } xB | j � D]4 \ } } � j j | t � � j | | j f � q� W� f d � | j j � D� } xH | j � D]: \ } } � j j | | j | j f t � � j | � qWWt r\ � f d � | j j � D� } x0 | D]% } � j j | t � � j | � q�Wq\ q\ Wd S(
s� Find and cache all modules that import file system modules.
Later, `setUp()` will stub these with the fake file system
modules.
c sf yM t j | � r! | j � k pK t j | � oK | j � j j | g � k SWn t k
ra t SXd S( N( Re t ismoduleR/ t isclassR0 R] t gett AttributeErrorR< ( t modRt ( t module_namesR) ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyt is_fs_module� s
c S sW y> t j | � s! t j | � o< | j d k o< | j t k SWn t k
rR t SXd S( NRR ( Re Rf t isbuiltinR/ R0 t BUILTIN_MODULER� R< ( t fct( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyt is_open_function� s
c sd yK t j | � s! t j | � oI | j � j k oI | j � j | j k SWn t k
r_ t SXd S( N( Re Rf R� R/ Rc R0 R� R< ( R� ( R) ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyt is_fs_function� s
t .i c s. i | ]$ \ } } � | | � r | | � q S( ( ( t .0Rt R� ( R� ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pys
<dictcomp>� s c s+ i | ]! \ } } � | � r | | � q S( ( ( R� Rt R� ( R� ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pys
<dictcomp>� s c s( h | ] \ } } � | � r | � q S( ( ( R� Rt R� ( R� ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pys <setcomp>� s N( t listR\ t keysRk t sett syst modulesRb t SKIPMODULESRe R~ R/ t splitRQ R� t __dict__Rl Rh t addRm R0 R Rn ( R) Rt R# R� R� t functionsR� t open_fcts( ( R� R� R� R� R) sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyt
_find_modules� s0 "
)
c C s� | j d k r | j j � n t j � | _ t j d | � | _ x. | j D]# } | j | | j � | j | <qM W| j d j
| j t <t j | j � | _
t | _ d S( s@ Renew the fake file system and set the _isStale flag to `False`.t patcherRD N( Ro R* t smart_unset_allR t StubOutForTestingR t FakeFilesystemR( R\ Rp RE Rk t FakeFileOpenRT R< Rr ( R) Rt ( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyt _refresh� s !c C sb t j � } | j � | j � | d k rD | j | j � | _ n | j � | j j | � d S( s� Bind the file-related modules to the :py:mod:`pyfakefs` fake
modules real ones. Also bind the fake `file()` and `open()` functions.
N(
RV t
gettempdirR� R� R* R R t start_patchingR( t
create_dir( R) t doctestert temp_dir( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyR s
c
C s | j st | _ t rM | j j t d | j � | j j t d | j � n xQ | j j � D]@ \ } } x1 | D]) \ } } | j j | | | j | � qp Wq] Wx� | j
j � D]u \ \ } } } } | j | | \ } } | j | } | j | | j
� } x$ | D] } | j j | | | � qWq� Wt r}xJ | j j � D]6 \ } } x' | D] } | j j | | | j � qSWq@Wn t | � | _ t j j d | j � x0 | j D]% } | j t j k r�t | � q�q�W| j s| j j � t j j d � qn d S( NRR t filei ( Rs R2 R Ro t smart_sett builtinsRT Rl Rb Rp Rm Rc t __get__t __class__Rn t DynamicPatcherRq R� t meta_patht insertR R/ R� R RX t cleanupt pop(
R) Rt R� R# t attrt ft_namet ft_modt methodRv Ru ( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyR� s6 %
$
c C sc | j � } | j r" | j � n x: | j D]/ } | | k r, | j | | j � | | <q, q, W| S( N( RP Rr R� R\ R( ( R) t globs_R Rt ( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyR 2 s
!c C s | j � d S( s8 Clear the fake filesystem bindings created by `setUp()`.N( t
stop_patching( R) R� ( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyR ; s c C sX | j rT t | _ t | _ | j j � | j rT | j j � t j
j d � qT n d S( Ni ( Rs R2 Rr R< Ro R� RX Rq R� R� R� R� ( R) ( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyR� ? s
c C s | j � d S( s� Pause the patching of the file system modules until `resume` is
called. After that call, all file system calls are executed in the
real file system.
Calling pause() twice is silently ignored.
N( R� ( R) ( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyR- H s c C s | j � d S( s� Resume the patching of the file system modules if `pause` has
been called before. After that call, all file system calls are
executed in the fake file system.
Does nothing if patching is not paused.
N( R� ( R) ( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyR. Q s N( R
RC (" R/ R0 R1 R* R R R� R� t AssertionErrort platformt
IS_WINDOWSRi Rk RO R R� R R RD t environR� R2 R7 Ry R} R� R� R R� R R R� R- R. ( ( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyR # s2 \ E t Pausec B s) e Z d Z d � Z d � Z d � Z RS( s� Simple context manager that allows to pause/resume patching the
filesystem. Patching is paused in the context manager, and resumed after
going out of it's scope.
c C sR t | t t f � r$ | j | _ n* t | t j � rB | | _ n t d � � d S( s� Initializes the context manager with the fake filesystem.
Args:
caller: either the FakeFilesystem instance, the Patcher instance
or the pyfakefs test case.
s� Invalid argument - should be of type "fake_filesystem_unittest.Patcher", "fake_filesystem_unittest.TestCase" or "fake_filesystem.FakeFilesystem"N( t
isinstanceR R& R( t _fsR R� R; ( R) t caller( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyR7 _ s
c C s | j j � | j S( N( R� R- ( R) ( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyRy p s
c G s
| j j � S( N( R� R. ( R) t args( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyR} t s ( R/ R0 R1 R7 Ry R} ( ( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyR� Z s R� c B sJ e Z d Z d � Z d � Z d � Z d d � Z d d � Z d � Z RS( s� A file loader that replaces file system related modules by their
fake implementation if they are loaded after calling `setupPyFakefs()`.
Implements the protocol needed for import hooks.
c C s� | | _ i | _ | j j | _ xP | j D]E } | j | � r+ | t j k r+ t j | | j | <t j | =q+ q+ Wx* | j j � D] \ } } | t j | <q� Wd S( N( R$ t
sysmodulesRp R� t needs_patchR� Rb ( R) R� Rt R# ( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyR7 ~ s c C sb x% | j D] } | j | t j | <q
Wx3 | j j D]% } | j t j k r5 t | � q5 q5 Wd S( N( R� R� R� R$ R R/ R ( R) R# ( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyR� � s
c C sJ | | j k r t S| t j k rF t t j | � | j | k rF t St S( s: Check if the module with the given name shall be replaced.( R� R< R� t typeR2 ( R) Rt ( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyR� � s c C s | j | � r t | | � Sd S( s Module finder for Python 3.N( R� R ( R) t fullnameRE t target( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyt find_spec� s c C s | j | � r | Sd S( s Module finder for Python 2.N( R� ( R) R� RE ( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyt find_module� s c C s | j | t j | <| j | S( s/ Replaces the module by its fake implementation.( R� R� ( R) R� ( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyt load_module� s N(
R/ R0 R1 R7 R� R� R* R� R� R� ( ( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyR� x s (- R1 R Re RD R� RV t unittestt zipfilet pyfakefs.helpersR R t pyfakefs.deprecatorR t importlib.machineryR t ImportErrort objectt importlibR t impt pyfakefsR R R t pyfakefs.extra_packagesR R R
R R t version_infoR R� R� Ri Rk R� R* R2 R% R& R4 R R� R� ( ( ( sR /opt/alt/python27/lib/python2.7/site-packages/pyfakefs/fake_filesystem_unittest.pyt <module>& sX
h J� 8