← Back
Editing: 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*
Save File
Cancel