File: //lib/python3/dist-packages/twisted/conch/test/__pycache__/test_tap.cpython-38.pyc
U
W[W � @ s� d Z z ddlZW n ek
r( dZY nX z ddlZW n ek
rN dZY nX zddlmZ W n ek
rx dZY nX er�er�er�ddlmZ ddlmZ ddl m
Z
ddl m Z ddl
mZ dd l
mZmZ dd
lmZ G d d � d e�ZdS )
z!
Tests for L{twisted.conch.tap}.
� N)�unix)�tap)�OpenSSHFactory)�StreamServerEndpointService)�error)�ISSHPrivateKey)�IUsernamePassword�UsernamePassword)�TestCasec @ sl e Zd ZdZesdZes dZes$dZdZdd� Z dd � Z
d
d � Z d d
� Z dd� Z
dd� Zdd� Zdd� ZdS )�MakeServiceTestsz'
Tests for L{tap.makeService}.
z can't run without cryptographyzCannot run without PyASN1z can't run on non-posix computers)s iamusers thisispasswordc C sB | � � | _t| jd��}|�d�| j�� W 5 Q R X t�� | _dS )z/
Create a file with two users.
zwb+� :N) Zmktemp�filename�open�write�join�usernamePasswordr �Options�options)�self�f� r �=/usr/lib/python3/dist-packages/twisted/conch/test/test_tap.py�setUp4 s
zMakeServiceTests.setUpc C s@ t �� }t �|�}| �|t� | �|jjd� | �|jt � dS )z�
L{tap.makeService} returns a L{StreamServerEndpointService} instance
running on TCP port 22, and the linked protocol factory is an instance
of L{OpenSSHFactory}.
� N)
r r � makeServiceZassertIsInstancer � assertEqualZendpointZ_port�factoryr )r �config�servicer r r �
test_basic> s
zMakeServiceTests.test_basicc C sN d}| � t| jd d� | � t| jd d� | �|t| jd �d|f � dS )z�
Make sure that if the C{--auth} command-line option is not passed,
the default checkers are (for backwards compatibility): SSH and UNIX
� ZcredInterfacesz)SSH should be one of the default checkersz*UNIX should be one of the default checkers� credCheckersz&There should be %d checkers by defaultN)ZassertInr r r r �len)r Z numCheckersr r r �test_defaultAuthsK s ���z"MakeServiceTests.test_defaultAuthsc C s0 | j �dd| j g� | �t| j d �d� dS )z�
The C{--auth} command-line option will add a checker to the list of
checkers, and it should be the only auth checker
�--auth�file:r! � N�r � parseOptionsr
r r" �r r r r �test_authAddedZ s z MakeServiceTests.test_authAddedc C s4 | j �dd| j ddg� | �t| j d �d� dS )z�
Multiple C{--auth} command-line options will add all checkers specified
to the list ofcheckers, and there should only be the specified auth
checkers (no default checkers).
r$ r% z memory:testuser:testpasswordr! r Nr' r) r r r �test_multipleAuthAddedc s
�z'MakeServiceTests.test_multipleAuthAddedc C sH | j �dd| j g� | j d d }t| jd d�}| �|�|�tj�S )z�
The checker created by the C{--auth} command-line option returns a
L{Deferred} that fails with L{UnauthorizedLogin} when
presented with credentials that are unknown to that checker.
r$ r% r! ���r Zfake) r r( r
r r Z
assertFailure�requestAvatarIdr ZUnauthorizedLogin)r �checkerZinvalidr r r �test_authFailuren s �z!MakeServiceTests.test_authFailurec sP �j �dd�j g� �j d d }t�j� � |�� �}� �fdd�}|�|�S )z�
The checker created by the C{--auth} command-line option returns a
L{Deferred} that returns the avatar id when presented with credentials
that are known to that checker.
r$ r% r! r, c s �� | � j� d S )N)r �username)r0 �Zcorrectr r r � checkSuccess� s z7MakeServiceTests.test_authSuccess.<locals>.checkSuccess)r r( r
r r r- Z addCallback)r r. �dr2 r r1 r �test_authSuccess| s
z!MakeServiceTests.test_authSuccessc C s< t �� }t �|�}|jj}| �t|j�� �tt t
g�� dS )z�
The L{OpenSSHFactory} built by L{tap.makeService} has a portal with
L{ISSHPrivateKey} and L{IUsernamePassword} interfaces registered as
checkers.
N) r r r r �portalr �setZcheckers�keysr r )r r r r5 r r r �
test_checkers� s
�z MakeServiceTests.test_checkersN)�__name__�
__module__� __qualname__�__doc__� cryptography�skip�pyasn1r r r r r# r* r+ r/ r4 r8 r r r r r $ s
r )r< r= � ImportErrorr? Z
twisted.conchr r Z$twisted.conch.openssh_compat.factoryr Z twisted.application.internetr Z twisted.credr Ztwisted.cred.credentialsr r r Ztwisted.trial.unittestr
r r r r r �<module> s*