File: //usr/lib/python3/dist-packages/twisted/conch/test/__pycache__/test_manhole_tap.cpython-310.pyc
o
�b� � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZ ddl m
Z
ddl m Z ddl
mZ dd lmZ ed
�Zed �ZerFerFdd lmZmZ G d
d� de�ZdS )z)
Tests for L{twisted.conch.manhole_tap}.
� )�StreamServerEndpointService)� MultiService)�telnet)�error)�UsernamePassword)�usage)�
requireModule)�TestCase� cryptography�pyasn1)� manhole_ssh� manhole_tapc @ sL e Zd ZdZes
dZesdZdZdd� 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 sV | � � | _t| jd��}|�d�| j�� W d � n1 s w Y 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�setUp% s
�zMakeServiceTests.setUpc C sR | � tj��}t�� �g � W d � n1 sw Y | �|jjd d� dS )zk
L{manhole_tap.makeService} requires either 'telnetPort' or 'sshPort' to
be given.
Nr z<At least one of --telnetPort and --sshPort must be specified) � assertRaisesr �
UsageErrorr
r � parseOptions� assertEqual� 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�protocol�makeTelnetProtocol�endpoint�_port�r �servicer r r �test_telnetPort; 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 --sshKeySize�512z --sshPortztcp:223r' r �� N)r r r r
r) r* r r! r+ r, r r- r � ConchFactoryr0 r1 r2 r r r � test_sshPortJ 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+ �checkers�keys�
assertTrue�successResultOf�loginr r r �ITelnetProtocolr* �failureResultOf�valuer �UnauthorizedLogin)r r3 r: r r r � test_passwdc s, ��� ���z MakeServiceTests.test_passwdN)
�__name__�
__module__� __qualname__�__doc__r
�skipr r r r% r4 r8 rD r r r r r s
r N)rH � twisted.application.internetr �twisted.application.servicer �
twisted.conchr � twisted.credr �twisted.cred.credentialsr �twisted.pythonr �twisted.python.reflectr �twisted.trial.unittestr r
r r r
r r r r r �<module> s