Your IP : 216.73.216.213
B
R�X� � @ sf d dl mZ d dlmZ d dlmZ d dlmZ d dlm Z G dd� de
�Z dZ e e e j
e jg� d S )
� )�Mock)�Popen)�
TemporaryFile)�
basestring)�extend_docstringc
@ st e Zd ZdZdZdd� Zd d d
�Zd d d �Zd
d� Zd dd�Z dd� Z
d!dd�Z dd� Z dd� Z
dd� Zd d � ZdS )"� MockPopenz�
A specialised mock for testing use of :class:`subprocess.Popen`.
An instance of this class can be used in place of the
:class:`subprocess.Popen` and is often inserted where it's needed using
:func:`mock.patch` or a :class:`Replacer`.
Nc C sx i | _ t� | _}| j| jj_ttd�|_|j }|j_| j|j_| j|j_| j |j _| j
|j
_| j |j _| j |j _d S )N)�spec)
�commandsr �mockr Z side_effect�Popen_instanceZ return_value� communicate�wait� send_signal� terminate�kill�poll)�selfr
�inst� r �C/opt/alt/python37/lib/python3.7/site-packages/testfixtures/popen.py�__init__ s
zMockPopen.__init__� r �� � c C s |||||f| j |< dS )z�
Set the behaviour of this mock when it is used to simulate the
specified command.
:param command: A string representing the command to be simulated.
N)r )r �command�stdout�stderr�
returncode�pid�
poll_countr r r � set_command! s zMockPopen.set_commandc C s |||||f| _ dS )z�
Set the behaviour of this mock when it is used to simulate commands
that have no explicit behavior specified using
:meth:`~MockPopen.set_command`.
N)�default_command)r r r r r r r r r � set_default+ s zMockPopen.set_defaultc O s | j j||�S )N)r
r )r �args�kwr r r �__call__4 s zMockPopen.__call__Fc C s� t |t�r|}n
d�|�}| j�|| j�}|d kr>td| ��|\| _| _| _ }}|| _
xDdD ]<}t � }|� t
| |�� |�� |�d� t| jj||� q^W || jj_d | jj_ | jjS )N� z Nothing specified for command %r)r r r )�
isinstancer �joinr �getr! �KeyErrorr r r r r �write�getattr�flush�seek�setattrr
r r )r r# �bufsize�
executable�stdinr r �
preexec_fn� close_fds�shell�cwd�env�universal_newlines� startupinfo�
creationflags�cmdZ behaviourr r �name�fr r r r 7 s"
zMockPopen.Popenc C s | j | jj_ | j S )z/Simulate calls to :meth:`subprocess.Popen.wait`)r r
r )r r r r r
T s zMockPopen.waitc C s | � � | j| jfS )z6Simulate calls to :meth:`subprocess.Popen.communicate`)r
r r )r �inputr r r r Y s zMockPopen.communicatec C s2 x(| j r(| jjjdkr(| j d8 _ dS W | �� S )z/Simulate calls to :meth:`subprocess.Popen.poll`N� )r r
r r r
)r r r r r ^ s zMockPopen.pollc C s dS )z6Simulate calls to :meth:`subprocess.Popen.send_signal`Nr )r �signalr r r r j s zMockPopen.send_signalc C s dS )z4Simulate calls to :meth:`subprocess.Popen.terminate`Nr )r r r r r n s zMockPopen.terminatec C s dS )z/Simulate calls to :meth:`subprocess.Popen.kill`Nr )r r r r r r s zMockPopen.kill)r r r r r )r r r r r )
r NNNNNFFNNFNr )N)�__name__�
__module__� __qualname__�__doc__r! r r r" r% r r
r r r r r r r r r r
s&
r a�
:param stdout:
A string representing the simulated content written by the process
to the stdout pipe.
:param stderr:
A string representing the simulated content written by the process
to the stderr pipe.
:param returncode:
An integer representing the return code of the simulated process.
:param pid:
An integer representing the process identifier of the simulated
process. This is useful if you have code the prints out the pids
of running processes.
:param poll_count:
Specifies the number of times :meth:`MockPopen.poll` can be
called before :attr:`MockPopen.returncode` is set and returned
by :meth:`MockPopen.poll`.
N)r
r �
subprocessr �tempfiler Ztestfixtures.compatr Ztestfixtures.utilsr �objectr Zset_command_paramsr r" r r r r �<module> s ~