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