File: //lib/python3/dist-packages/twisted/conch/test/__pycache__/test_default.cpython-38.pyc
U
W[@- � @ s* d Z ddlmZmZ ddlZddlmZ ed�rzed�rzddlmZ ddl m
Z
dd l m Z dd
l
mZ dd lmZ dZnd Zdd
lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddl m Z ddl m Z e � � r�dZ!neZ!dZ"ej#�$� s�dZ"ej%�$� �sdZ"G dd� de�Z&G dd� de�Z'dS )z,
Tests for L{twisted.conch.client.default}.
� )�absolute_import�divisionN)�
requireModuleZ cryptographyZpyasn1)�SSHAgentClient)�SSHUserAuthClient)� ConchOptions)�default)�KeyzBcryptography and PyASN1 required for twisted.conch.client.default.)�TestCase)�FilePath)�
ConchError)�keydata)�StringTransport)� nativeString)�platformzlgenericAnswers and getPassword does not work on Windows. Should be fixed as part of fixing bug 6409 and 6410z#sys.stdin is not an interactive ttyz$sys.stdout is not an interactive ttyc @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z e
pZee _dd� Ze
plee_dd� Ze
p~ee_dd� Ze
p�ee_dS )�SSHUserAuthClientTestszm
Tests for L{SSHUserAuthClient}.
@type rsaPublic: L{Key}
@ivar rsaPublic: A public RSA key.
c C sZ t �tj�| _t| �� �| _| j�� | j� d�| _
| j
� tj � | j� d�� tj� d S )NZid_rsa�
id_rsa.pub)
r �
fromStringr
ZpublicRSA_openssh� rsaPublicr Zmktemp�tmpdir�makedirs�child�rsaFile�
setContent�privateRSA_openssh��self� r �A/usr/lib/python3/dist-packages/twisted/conch/test/test_default.py�setUp7 s
z SSHUserAuthClientTests.setUpc C sf t dt� d�}t� }t� }|�|� ||_d}|�| j|� | �|� � d| j�
� d | d � dS )z�
When connected to an agent, L{SSHUserAuthClient} can use it to
request signatures of particular data with a particular L{Key}.
� userNs Sign heres -
s s ) r r r r ZmakeConnection�keyAgentZsignDatar � assertEqual�value�blob)r �client�agent� transportZ cleartextr r r �test_signDataWithAgent@ s"
����z-SSHUserAuthClientTests.test_signDataWithAgentc C sJ t � }| j�� g|_|�� }| �|�� � | �|| j� | �|�� � dS )a
L{SSHUserAuthClient} looks up public keys from the agent using the
L{SSHAgentClient} class. That L{SSHAgentClient.getPublicKey} returns a
L{Key} object with one of the public keys in the agent. If no more
keys are present, it returns L{None}.
N) r r r$ Zblobs� getPublicKey�
assertTrue�isPublicr" Z assertIsNone)r r& �keyr r r �test_agentGetPublicKeyS s z-SSHUserAuthClientTests.test_agentGetPublicKeyc C sF t � }| jjg|_td|d�}|�� }| �|�� � | �|| j � dS )z�
L{SSHUserAuthClient.getPublicKey()} is able to get a public key from
the first file described by its options' C{identitys} list, and return
the corresponding public L{Key} object.
r N)
r r �path� identitysr r) r* r+ r" r )r �optionsr% r, r r r �test_getPublicKeyFromFileb s z0SSHUserAuthClientTests.test_getPublicKeyFromFilec C sR t � }| jjg|_t� }td|d�}||_|�� }| �|� � � | �
|| j � dS )z�
If an agent is present, but doesn't return a key,
L{SSHUserAuthClient.getPublicKey} continue with the normal key lookup.
r N) r r r. r/ r r r! r) r* r+ r" r )r r0 r& r% r, r r r � test_getPublicKeyAgentFallbackp s z5SSHUserAuthClientTests.test_getPublicKeyAgentFallbackc C s� t � }| j�d��tj� | j�d�}|�tj� | jj|jg|_ | j�d��d� t
d|d�}|� � }| � |�
� � | �|t�tj�� | �|j| jj|jg� dS )z�
If L{keys.Key.fromFile} raises a L{keys.BadKeyError}, the
L{SSHUserAuthClient.getPublicKey} tries again to get a public key by
calling itself recursively.
z
id_dsa.pubZid_dsar s
not a key!r N)r r r r r
ZpublicDSA_opensshZprivateDSA_opensshr r. r/ r r) r* r+ r" r r Z usedFiles)r r0 ZdsaFiler% r, r r r � test_getPublicKeyBadKeyError s z3SSHUserAuthClientTests.test_getPublicKeyBadKeyErrorc sN t �tj�� t� }�jjg|_td|d�}|� � � �fdd�}|�
� � |�S )z�
L{SSHUserAuthClient.getPrivateKey} will load a private key from the
last used file populated by L{SSHUserAuthClient.getPublicKey}, and
return a L{Deferred} which fires with the corresponding private L{Key}.
r Nc s �� | �� � ��| � � d S �N�Z assertFalser+ r" �r, ��
rsaPrivater r r �_cbGetPrivateKey� s zCSSHUserAuthClientTests.test_getPrivateKey.<locals>._cbGetPrivateKey) r r r
r r r r. r/ r r) �
getPrivateKey� addCallback)r r0 r% r9 r r7 r �test_getPrivateKey� s z)SSHUserAuthClientTests.test_getPrivateKeyc s� t �tj��d� �j���d� �� t� }�jjg|_ t
d|d�}|� � � �fdd�}��fdd�}�� |d |� |�
� �|�S )
z�
L{SSHUserAuthClient} can get a private key from a file, and return a
Deferred called back with a private L{Key} object, even if the key is
encrypted.
s this is the passphraseZopensshr Nc s �� | d�jjf � t� �S )Nz Enter passphrase for key '%s': )r" r r. r ��prompt)�
passphraser r r � _getPassword� s
�zISSHUserAuthClientTests.test_getPrivateKeyPassphrase.<locals>._getPasswordc s �� | �� � ��| � � d S r4 r5 r6 r7 r r r9 � s zMSSHUserAuthClientTests.test_getPrivateKeyPassphrase.<locals>._cbGetPrivateKeyr@ )r r r
r r r ZtoStringr r. r/ r r) �patchr: r; )r r0 r% r@ r9 r )r? r8 r r � test_getPrivateKeyPassphrase� s z3SSHUserAuthClientTests.test_getPrivateKeyPassphrasec s` G dd� d�}t � }td|d�}|d�|_� fdd�}� �tjd|� |�� }|�� jd � |S )
zn
Get the password using
L{twisted.conch.client.default.SSHUserAuthClient.getPassword}
c @ s e Zd Zdd� Zdd� ZdS )z>SSHUserAuthClientTests.test_getPassword.<locals>.FakeTransportc S s | | _ || _d S r4 )r' �host)r rC r r r �__init__� s zGSSHUserAuthClientTests.test_getPassword.<locals>.FakeTransport.__init__c S s | S r4 r r r r r �getPeer� s zFSSHUserAuthClientTests.test_getPassword.<locals>.FakeTransport.getPeerN)�__name__�
__module__� __qualname__rD rE r r r r �
FakeTransport� s rI r Nz 127.0.0.1c s � � | d� dS )Nzuser@127.0.0.1's password: � bad password�r" r= r r r �getpass� s z8SSHUserAuthClientTests.test_getPassword.<locals>.getpassrL � bad password) r r r' rA r rL � getPasswordr; r" )r rI r0 r% rL �dr r r �test_getPassword� s
z'SSHUserAuthClientTests.test_getPasswordc sP t � }td|d�}d� � �fdd�}��tjd|� |�� �}|��jd� |S )z�
Get the password using
L{twisted.conch.client.default.SSHUserAuthClient.getPassword}
using a different prompt.
r Ns Give up your passwordc s �� | t� �� dS )NrJ )r" r )�p�r> r r r rL � s z>SSHUserAuthClientTests.test_getPasswordPrompt.<locals>.getpassrL rM )r r rA r rL rN r; r" )r r0 r% rL rO r rR r �test_getPasswordPrompt� s
z-SSHUserAuthClientTests.test_getPasswordPromptc sh t � }td|d�}dd� }� �tjd|� tjtj ��|�d�}|j � ��fdd��}� �
|t � dS ) z�
Get the password using
L{twisted.conch.client.default.SSHUserAuthClient.getPassword}
and trigger a {twisted.conch.error import ConchError}.
r Nc S s t d��d S )NzUser pressed CTRL-C)�KeyboardInterruptr= r r r rL � s zBSSHUserAuthClientTests.test_getPasswordConchError.<locals>.getpassrL � ?c s � � ��gtjtjg� | S r4 )r" �sys�stdout�stdin)Zfail�r rX rW r r � check_sys s
�zDSSHUserAuthClientTests.test_getPasswordConchError.<locals>.check_sys) r r rA r rL rV rW rX rN Z
addErrbackZ
assertFailurer )r r0 r% rL rO rZ r rY r �test_getPasswordConchError� s
z1SSHUserAuthClientTests.test_getPasswordConchErrorc sp t � }td|d�}� fdd�}� �tjd|� � fdd�}� �td|� |�d d
d d g�}|�� jddg� |S )
zU
L{twisted.conch.client.default.SSHUserAuthClient.getGenericAnswers}
r Nc s � � | d� dS )Nz pass promptrL rK r= r r r rL s z>SSHUserAuthClientTests.test_getGenericAnswers.<locals>.getpassrL c s � � | d� dS )Nzraw_input prompt� raw_inputrK r= r r r r\ s z@SSHUserAuthClientTests.test_getGenericAnswers.<locals>.raw_inputr\ s Names Instruction)s pass promptF)s raw_input promptT)r r rA r rL ZgetGenericAnswersr; ZassertListEqual)r r0 r% rL r\ rO r r r �test_getGenericAnswers s$ �� �z-SSHUserAuthClientTests.test_getGenericAnswersN)rF rG rH �__doc__r r( r- r1 r2 r3 r< rB rP � windowsSkip�ttySkip�skiprS r[ r] r r r r r / s"
r c @ s e Zd ZdZdd� Zdd� ZdS )�ConchOptionsParsingz
Options parsing.
c C s� t � }| �t|jd�}| �d|j� t � }|�d� | �|d dg� |�d� | �|d dg� |�d� | �|d ddd g� d
S ) z)
Specify MAC algorithms.
z invalid-maczUnknown mac typez
hmac-sha2-512Zmacss
hmac-sha2-512z hmac-sha2-256,hmac-sha1,hmac-md5s
hmac-sha2-256s hmac-sha1s hmac-md5N)r � assertRaises�
SystemExitZopt_macs�assertIn�coder" �r Zopts�er r r � test_macs0 s
z ConchOptionsParsing.test_macsc C s� t � }| �t|jd�}| �d|j� t � }|�d� | �|d dg� |�d� | �|d dg� |�d� | �|d ddg� dS ) z.
Specify host key algorithms.
z invalid-keyzUnknown host key typezssh-rsazhost-key-algorithmss ssh-rsas ssh-dsszssh-rsa,ssh-dssN)r rc rd Zopt_host_key_algorithmsre rf r" rg r r r �test_host_key_algorithms@ s
z,ConchOptionsParsing.test_host_key_algorithmsN)rF rG rH r^ ri rj r r r r rb , s rb )(r^ Z
__future__r r rV Ztwisted.python.reflectr Ztwisted.conch.client.agentr Z twisted.conch.client.defaultr Z twisted.conch.client.optionsr Ztwisted.conch.clientr Ztwisted.conch.ssh.keysr ra Ztwisted.trial.unittestr
Ztwisted.python.filepathr Ztwisted.conch.errorr Ztwisted.conch.testr
Ztwisted.test.proto_helpersr Ztwisted.python.compatr Ztwisted.python.runtimer Z isWindowsr_ r` rX �isattyrW r rb r r r r �<module> s< �
~