File: //lib/python3/dist-packages/twisted/conch/test/__pycache__/test_manhole_tap.cpython-38.pyc
U
W[� � @ s� d Z z ddlZW n ek
r( dZY nX z ddlZW n ek
rN dZY nX erherhddlmZmZ ddlmZ ddl m
Z
ddl m Z ddl
mZ ddlmZ dd lmZ dd
lmZ G d d � d e�ZdS )
z)
Tests for L{twisted.conch.manhole_tap}.
� N)� manhole_tap� manhole_ssh)�StreamServerEndpointService)� MultiService)�error)�UsernamePassword)�telnet)�usage)�TestCasec @ sL e Zd ZdZesdZes dZdZdd� Zdd� Z d d
� Z
d d � Z d
d� Z dS )�MakeServiceTestsz/
Tests for L{manhole_tap.makeService}.
z can't run without cryptographyzCannot run without PyASN1)s iamusers thisispasswordc C sB | � � | _t| jd��}|�d�| j�� W 5 Q R X t�� | _dS )z8
Create a passwd-like file with a user.
�wb� :N) �mktemp�filename�open�write�join�usernamePasswordr �Options�options)�self�f� r �E/usr/lib/python3/dist-packages/twisted/conch/test/test_manhole_tap.py�setUp0 s
zMakeServiceTests.setUpc C s> | � tj��}t�� �g � W 5 Q R X | �|jjd d� dS )zk
L{manhole_tap.makeService} requires either 'telnetPort' or 'sshPort' to
be given.
r z<At least one of --telnetPort and --sshPort must be specifiedN) Z assertRaisesr Z
UsageErrorr r � parseOptions� assertEqualZ exception�args)r �er r r �test_requiresPort: s z"MakeServiceTests.test_requiresPortc C s~ | j �ddg� t�| j �}| �|t� | �t|j�d� | �|jd t � | �|jd j
j tj � | �|jd j
jd� dS )z�
L{manhole_tap.makeService} will make a telnet service on the port
defined by C{--telnetPort}. It will not make a SSH service.
� --telnetPortztcp:222� r �� N)r r r � makeService�assertIsInstancer r �len�servicesr �factory�protocolZmakeTelnetProtocol�endpoint�_port�r �servicer r r �test_telnetPortF s �z MakeServiceTests.test_telnetPortc C s� | j �d| �� ddddg� t�| j �}| �|t� | �t|j �d� | �|j d t
� | �|j d j t j
� | �|j d jjd� d S )
z�
L{manhole_tap.makeService} will make a SSH service on the port
defined by C{--sshPort}. It will not make a telnet service.
z --sshKeyDirz --sshKeySizeZ512z --sshPortztcp:223r! r �� N)r r r r r# r$ r r r% r& r r' r Z ConchFactoryr) r* r+ r r r � test_sshPortU s � �z MakeServiceTests.test_sshPortc C s� | j �ddd| jg� t�| j �}|jd jjj}| � t
|j � � �d� | �
| �|�t| j� dtj��� | �| �|�tdd�dtj��jtj� dS ) zS
The C{--passwd} command-line option will load a passwd-like file.
r ztcp:22z--passwdr � Ns wrongs user)r r r r r# r& r' r( �portalr r% Zcheckers�keysZ
assertTrueZsuccessResultOfZloginr r r ZITelnetProtocolr$ ZfailureResultOf�valuer ZUnauthorizedLogin)r r, r1 r r r � test_passwdi s*
� �� ���z MakeServiceTests.test_passwdN)
�__name__�
__module__� __qualname__�__doc__� cryptography�skip�pyasn1r r r r- r/ r4 r r r r r # s
r )r8 r9 � ImportErrorr; Z
twisted.conchr r Z twisted.application.internetr Ztwisted.application.servicer Z twisted.credr Ztwisted.cred.credentialsr r Ztwisted.pythonr Ztwisted.trial.unittestr
r r r r r �<module> s$