Your IP : 216.73.216.213
B
��\); � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl m
Z
ddl m Z G dd� de
�ZG dd� dej�ZG d d
� d
e
�ZG d d � d ee�ZdS )
z;Common helper classes used in tests, or as test class base.� N)�fake_filesystem)�is_byte_stringc @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � DummyTimezBMock replacement for time.time. Increases returned time on access.c C s || _ || _d| _d S )NF)� curr_time� increment�started)�selfr r � r �J/opt/alt/python37/lib/python3.7/site-packages/pyfakefs/tests/test_utils.py�__init__" s zDummyTime.__init__c C s
d| _ d S )NT)r )r r r r
�start' s zDummyTime.startc O s | j r| j| j7 _| jS )N)r r r )r �args�kwargsr r r
�__call__* s zDummyTime.__call__N)�__name__�
__module__� __qualname__�__doc__r r r r r r r
r s r c @ sX e Zd ZdZejdkZejdkZejdkZej d dk Z
dZ dd � Z d
d � Z
d d
� ZdS )�TestCasez<Test base class with some convenience methods and attributes�win32�cygwin�darwinr � Nc C s | � t�|�t�|��S )N)� assertEqual�stat�S_IMODE)r Zexpected�actualr r r
�assert_mode_equal8 s zTestCase.assert_mode_equalc
O sN y|||� | � d� W n0 tk
rH } z| �||j� W dd}~X Y nX dS )z5Asserts that a specific subtype of IOError is raised.z)No exception was raised, IOError expectedN)�fail�IOErrorr �errno)r �subtype�
expressionr
r �excr r r
�assert_raises_io_error; s
z TestCase.assert_raises_io_errorc
O sN y|||� | � d� W n0 tk
rH } z| �||j� W dd}~X Y nX dS )z5Asserts that a specific subtype of OSError is raised.z)No exception was raised, OSError expectedN)r �OSErrorr r )r r! r" r
r r# r r r
�assert_raises_os_errorC s
z TestCase.assert_raises_os_error)r r r r �sys�platform�
is_windowsZ is_cygwin�is_macos� version_info�
is_python2�symlinks_can_be_testedr r$ r& r r r r
r 0 s
r c @ s� e Zd ZdZdd� Zedd� �Zdd� Zedd � �Zed
d � �Z d d
� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd6d!d"�Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd7d,d-�Zd.d/� Zd0d1� Zd2d3� Z d4d5� Z d+S )8�RealFsTestMixina� Test mixin to allow tests to run both in the fake filesystem and in the
real filesystem.
To run tests in the real filesystem, a new test class can be derived from
the test class testing the fake filesystem which overwrites
`use_real_fs()` to return `True`.
All tests in the real file system operate inside the local temp path.
In order to make a test able to run in the real FS, it must not use the
fake filesystem functions directly. For access to `os` and `open`,
the respective attributes must be used, which point either to the native
or to the fake modules. A few convenience methods allow to compose
paths, create files and directories.
c C s> d | _ t| _t| _tjd dk| _d | _| �� r:t� � | _d S )Nr � )
�
filesystem�open�osr' r+ r, � base_path� use_real_fs�tempfile�mkdtemp)r r r r
r [ s zRealFsTestMixin.__init__c C s t jS )N)r r) )r r r r
�
is_windows_fsd s z RealFsTestMixin.is_windows_fsc C s* | j d k r&|| j _|r d| j _| �� d S )NF)r0 r7 r* �create_basepath)r �valuer r r
�set_windows_fsh s
z RealFsTestMixin.set_windows_fsc C s t jS )N)r r* )r r r r
r* o s zRealFsTestMixin.is_macosc C s t �� dkS )N�PyPy)r( �python_implementation)r r r r
�is_pypys s zRealFsTestMixin.is_pypyc C s dS )z4Return True if the real file system shall be tested.Fr )r r r r
r4 w s zRealFsTestMixin.use_real_fsc C s | � � rtjjS dS )zNCan be overwritten to use a specific separator in the
fake filesystem.�/)r4 r2 �path�sep)r r r r
�path_separator{ s z RealFsTestMixin.path_separatorc C s( | � � rtjs$t�d��n
| �d� dS )z�If called at test start, the real FS test is executed only under
Windows, and the fake filesystem test emulates a Windows system.
z&Testing Windows specific functionalityTN)r4 r r) �unittest�SkipTestr: )r r r r
�check_windows_only� s
z"RealFsTestMixin.check_windows_onlyc C s6 | � � r tjstjr2t�d��n| �d� d| j_dS )z�If called at test start, the real FS test is executed only under
Linux, and the fake filesystem test emulates a Linux system.
z$Testing Linux specific functionalityFN)r4 r r* r) rB rC r: r0 )r r r r
�check_linux_only� s
z RealFsTestMixin.check_linux_onlyc C s0 | � � rtjs,t�d��n| �d� d| j_dS )z�If called at test start, the real FS test is executed only under
MacOS, and the fake filesystem test emulates a MacOS system.
z$Testing MacOS specific functionalityFTN)r4 r r* rB rC r: r0 )r r r r
�check_macos_only� s
z RealFsTestMixin.check_macos_onlyc C s& | � � rtjr"t�d��nd| j_dS )z�If called at test start, the real FS test is executed only under
Linux and Windows, and the fake filesystem test emulates a Linux
system under MacOS.
z Testing non-MacOs functionalityFN)r4 r r* rB rC r0 )r r r r
�check_linux_and_windows� s
z'RealFsTestMixin.check_linux_and_windowsc C s, | � � r tjs(tjs(t�d��nd| j_dS )z�If called at test start, the real FS test is executed only in a
case-insensitive FS (e.g. Windows or MacOS), and the fake filesystem
test emulates a case-insensitive FS under the running OS.
z/Testing case insensitive specific functionalityFN)r4 r r* r) rB rC r0 �is_case_sensitive)r r r r
�check_case_insensitive_fs� s
z)RealFsTestMixin.check_case_insensitive_fsc C s, | � � r tjstjr(t�d��nd| j_dS )z�If called at test start, the real FS test is executed only in a
case-sensitive FS (e.g. under Linux), and the fake file system test
emulates a case-sensitive FS under the running OS.
z-Testing case sensitive specific functionalityTN)r4 r r* r) rB rC r0 rH )r r r r
�check_case_sensitive_fs� s
z'RealFsTestMixin.check_case_sensitive_fsc C s( | � � rtjr$t�d��n
| �d� dS )z�If called at test start, the real FS test is executed only under
Linux and MacOS, and the fake filesystem test emulates a Linux
system under Windows.
z$Testing Posix specific functionalityFN)r4 r r) rB rC r: )r r r r
�check_posix_only� s
z RealFsTestMixin.check_posix_onlyc C s | � � rt�d��dS )z5If called at test start, no real FS test is executed.zOnly tests fake FSN)r4 rB rC )r r r r
� skip_real_fs� s z RealFsTestMixin.skip_real_fsTc C sf | � � rbtjr|sDtjs |r |sDtjr*|sDtjs:tjs:|sDtjsb|rbtjrN|sXtjsb|rbt�d��dS )z�If called at test start, no real FS test is executed for the given
conditions. This is used to mark tests that do not pass correctly under
certain systems and shall eventually be fixed.
z.Skipping because FakeFS does not match real FSN)r4 r r) r* r, rB rC )r Z skip_windowsZ
skip_posixZ
skip_macosZ
skip_linuxZ skip_python2Z skip_python3r r r
�skip_real_fs_failure� s
z$RealFsTestMixin.skip_real_fs_failurec C sr t jr| �� sdS t jdkrl| �d�}y&| j�| j|� dt _| j�|� W n t t
fk
rj dt _Y nX t jS )z�Used to check if symlinks and hard links can be tested under
Windows. All tests are skipped under Windows for Python versions
not supporting links, and real tests are skipped if running without
administrator rights.
TN�linkF) r r) r4 r- � make_pathr2 �symlinkr3 �remover% �NotImplementedError)r � link_pathr r r
�symlink_can_be_tested� s
z%RealFsTestMixin.symlink_can_be_testedc C sH | � � rtjs | � � s2| jjr2tjdk r2t�d��| � � sDt�d��dS )zQIf called at test start, tests are skipped if symlinks are not
supported.)r r z:Symlinks are not supported under Windows before Python 3.3z,Symlinks under Windows need admin privilegesN)
r4 r r) r0 r7 r' r+ rB rC rT )r r r r
� skip_if_symlink_not_supported s
z-RealFsTestMixin.skip_if_symlink_not_supportedc G sv | j s t|d t�r | j�� }n| j}t|d ttf�rb|}x |d D ]}| jj� ||�}qFW |S | jjj |f|�� S )a$ Create a path with the given component(s). A base path is prepended
to the path which represents a temporary directory in the real FS,
and a fixed path in the fake filesystem.
Always use to compose absolute paths for tests also running in the
real FS.
r )
r, �
isinstance�bytesr3 �encode�list�tupler2 r? �join)r r
r3 r? �argr r r
rO s zRealFsTestMixin.make_pathc C sz |}g }x4|r<| j j�|�s<| j j�|�\}}|�d|� q
W x6|D ].}| j j�||�}| j �|� | j �|d� qDW dS )z�Create the directory at `dir_path`, including subdirectories.
`dir_path` shall be composed using `make_path()`.
r i� N)r2 r? �exists�split�insertr[ �mkdir�chmod)r �dir_pathZ
existing_path�
components� componentr r r
�
create_dir! s
zRealFsTestMixin.create_dirNc C s� | � | jj�|�� |dk s$t|�r(dnd}|dk rF|dk rF|�|�}| �||��}|dk rf|�|� W dQ R X | j�|d� dS )z�Create the given file at `file_path` with optional contents,
including subdirectories. `file_path` shall be composed using
`make_path()`.
N�wb�wi� ) re r2 r? �dirnamer rX r1 �writera )r � file_path�contents�encoding�mode�fr r r
� create_file/ s
zRealFsTestMixin.create_filec C s& | � | jj�|�� | j�||� dS )z�Create the path at `link_path`, and a symlink to this path at
`target_path`. `link_path` shall be composed using `make_path()`.
N)re r2 r? rh rP )r rS � target_pathr r r
�create_symlink? s z RealFsTestMixin.create_symlinkc C s< t |�r dnd}| �||��}| �||�� � W dQ R X dS )zcCompare `contents` with the contents of the file at `file_path`.
Asserts equality.
�rb�rN)r r1 r �read)r rj rk rm rn r r r
�check_contentsF s z RealFsTestMixin.check_contentsc C s t j}| jr| jrt j}|S )N)r �ENOTDIRr7 r, �EINVAL)r �errorr r r
�
not_dir_errorN s z RealFsTestMixin.not_dir_errorc C s| | j dk rx| j}| j jd | _| jr0d| j | _|| jkrx|dk rL| j �� | j �| j�sh| j �| j� |dk rx| �� dS )z1Create the path used as base path in `make_path`.NZbasepathzC:)r0 r3 rA r7 �resetr] re �setUpFileSystem)r Z
old_base_pathr r r
r8 T s
z RealFsTestMixin.create_basepath)TTTTTT)NN) r r r r r �propertyr7 r: r* r= r4 rA rD rE rF rG rI rJ rK rL rM rT rU rO re ro rq ru ry r8 r r r r
r. L s8
r. c @ sJ e Zd ZdZddd�Zdd� Zdd� Zed d
� �Zed d � �Z d
d� Z
dS )�RealFsTestCasezMCan be used as base class for tests also running in the real
file system.�runTestc C s t �| |� t�| � d S )N)r r r. )r Z
methodNamer r r
r h s zRealFsTestCase.__init__c C sj t �� | _| �� sJtj| �� d�| _t�| j�| _ t�
| j�| _ | � � nt j �
d�s^| �� | �� d S )N)rA Z TEST_REAL_FS)r2 �getcwd�cwdr4 r �FakeFilesystemrA r0 � FakeFileOpenr1 � FakeOsModuler8 �environ�getrL r{ )r r r r
�setUpl s
zRealFsTestCase.setUpc C s d S )Nr )r r r r
r{ y s z RealFsTestCase.setUpFileSystemc C s | � � r| jS | jjS )N)r4 r) r0 r7 )r r r r
r7 | s z RealFsTestCase.is_windows_fsc C s | � � rtjS | jjS )N)r4 r r* r0 )r r r r
r* � s zRealFsTestCase.is_macosc C s@ | � � r<| j�tj�| j�� tj| jdd� | j�| j� d S )NT)�
ignore_errors) r4 r2 �chdirr? rh r3 �shutil�rmtreer� )r r r r
�tearDown� s zRealFsTestCase.tearDownN)r~ ) r r r r r r� r{ r| r7 r* r� r r r r
r} d s
r} )r r r2 r( r� r r' r5 rB �pyfakefsr �pyfakefs.helpersr �objectr r r. r} r r r r
�<module> s