File: //usr/lib/python3/dist-packages/twisted/conch/test/__pycache__/test_unix.cpython-38.pyc
U
W[� � @ s� d dl mZ d dlmZ d dlmZ d dlmZ d dlm Z ed�Z
ed�Z ee�G dd � d e ��Z
G d
d � d e �ZG d d
� d
e j�ZdS )� )�absolute_import)� implementer)�IReactorProcess)�
requireModule)�unittest� cryptographyztwisted.conch.unixc @ s0 e Zd ZdZdd� Zdi dddddfdd�ZdS ) �MockProcessSpawnerzC
An L{IReactorProcess} that logs calls to C{spawnProcess}.
c C s
g | _ d S �N)�_spawnProcessCalls��self� r
�>/usr/lib/python3/dist-packages/twisted/conch/test/test_unix.py�__init__ s zMockProcessSpawner.__init__r
Nr c
C s$ | j �||||||||| d� � dS )zQ
Log a call to C{spawnProcess}. Do not actually spawn a process.
) �processProtocol�
executable�args�env�path�uid�gid�usePTY�childFDsN)r
�append)
r r r r r r r r r r r
r
r � spawnProcess s ��z MockProcessSpawner.spawnProcess)�__name__�
__module__� __qualname__�__doc__r r r
r
r
r r s �r c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �StubUnixConchUserz`
Enough of UnixConchUser to exercise SSHSessionForUnixConchUser in the
tests below.
c C s( ddl m}m} || _||� d�| _d S )N� )�StubConnection�
StubClient)Z transport)Z test_sessionr! r" �_homeDirectoryZconn)r �
homeDirectoryr! r" r
r
r r 4 s zStubUnixConchUser.__init__c C s dS )N)NNr
r r
r
r �getUserGroupId; s z StubUnixConchUser.getUserGroupIdc C s | j S r )r# r r
r
r �
getHomeDir? s z StubUnixConchUser.getHomeDirc C s d S r r
r r
r
r �getShellC s zStubUnixConchUser.getShellN)r r r r r r% r& r' r
r
r
r r . s
r c @ s. e Zd ZedkrdZn edkr"dZdd� ZdS )� TestSSHSessionForUnixConchUserNz Cannot run without cryptographyzUnix system requiredc C sV t � }d}t|�}tj||d�}d}dg}|�||� |j\}| �||d d � dS )zn
C{execCommand} sets the C{HOME} environment variable to the avatar's home
directory.
z/made/up/path/)ZreactorNznot-actually-executedr �HOME)r r �unixZSSHSessionForUnixConchUserZ execCommandr
Z assertEqual)r Z mockReactorr$ ZavatarZsessionZprotocolZcommand�callr
r
r �testExecCommandEnvironmentP s z9TestSSHSessionForUnixConchUser.testExecCommandEnvironment)r r r r �skipr* r, r
r
r
r r( H s
r( N)Z
__future__r Zzope.interfacer Ztwisted.internet.interfacesr Ztwisted.python.reflectr Z
twisted.trialr r r* �objectr r ZTestCaser( r
r
r
r �<module> s