← Back
Editing: test_openssh_compat.cpython-38.pyc
U W[ � @ s� d Z ddlZddlmZ ddlmZ ddlmZ ed�rNed�rNddlm Z ndZ dd l m Z dd l m Z dd lmZ G d d � d e�ZdS )z, Tests for L{twisted.conch.openssh_compat}. � N)�TestCase)�FilePath)� requireModuleZ cryptographyZpyasn1)�OpenSSHFactory)�getDHGeneratorAndPrime)�keydata)�MockOSc @ sZ e Zd ZdZeedd�dkr"dZn edkr.dZdd� Zdd � Z d d � Z d d � Z dd� Z dS )�OpenSSHFactoryTestsz& Tests for L{OpenSSHFactory}. �geteuidNz geteuid/seteuid not availablez)Cannot run without cryptography or PyASN1c C s t � | _t| �� �| _| j�� | jj| j_t| �� �| _| j�� | jj| j_ | j� d�� d� | j� d�� d� | j� d�� t j � | j� d�� t j� | j� d�� d� | j� d�� t j� | j� d �� d � t� | _| �td | jj� | �td | jj� d S ) NZ ssh_host_foos fooZbar_keyZssh_host_one_key�ssh_host_two_keyZssh_host_three_keys not a key contentzssh_host_one_key.pubZmodulis� # $OpenBSD: moduli,v 1.xx 2016/07/26 12:34:56 jhacker Exp $ # Time Type Tests Tries Size Generator Modulus 20030501000000 2 6 100 2047 2 FFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD129024E088A67CC74020BBEA63B139B22514A08798E3404DDEF9519B3CD3A431B302B0A6DF25F14374FE1356D6D51C245E485B576625E7EC6F44C42E9A637ED6B0BFF5CB6F406B7EDEE386BFB5A899FA5AE9F24117C4B1FE649286651ECE45B3DC2007CB8A163BF0598DA48361C55D39A69163FA8FD24CF5F83655D23DCA3AD961C62F356208552BB9ED529077096966D670C354E4ABC9804F1746C08CA18217C32905E462E36CE3BE39E772C180E86039B2783A2EC07A28FB5C55DF06F4C52C9DE2BCBF6955817183995497CEA956AE515D2261898FA051015728E5A8AACAA68FFFFFFFFFFFFFFFF �seteuid�setegid)r �factoryr Zmktemp�keysDir�makedirs�pathZdataRootZ moduliDirZ moduliRoot�childZ setContentr ZprivateRSA_opensshZprivateDSA_opensshZpublicRSA_opensshr �mockos�patch�osr r )�self� r �H/usr/lib/python3/dist-packages/twisted/conch/test/test_openssh_compat.py�setUp" s2 � � � �zOpenSSHFactoryTests.setUpc C s8 | j �� }| �t|�d� |�� }| �t|�dg� dS )zw L{OpenSSHFactory.getPublicKeys} should return the available public keys in the data directory � � ssh-rsaN)r Z getPublicKeys� assertEqual�len�keys�list�r r �keyTypesr r r �test_getPublicKeysC s z&OpenSSHFactoryTests.test_getPublicKeysc C s^ | j �� }| �t|�d� |�� }| �t|�tddg�� | �| jjg � | �| jjg � dS )z� Will return the available private keys in the data directory, ignoring key files which failed to be loaded. � r � ssh-dssN) r �getPrivateKeysr r r �setr � seteuidCalls� setegidCallsr r r r �test_getPrivateKeysN s z'OpenSSHFactoryTests.test_getPrivateKeysc s� | j �d�� � �d� | �� jd� tj�� �fdd�}| �td|� | j�� }| � t |�d� |� � }| � t |�t dd g�� | � | j jdt�� g� | � | j jdt�� g� d S ) z� L{OpenSSHFactory.getPrivateKeys} should switch to root if the keys aren't readable by the current user. r r � c s � � d� �| �S )Nr* )�chmod)Zeuid�ZkeyFileZ savedSeteuidr r r f s z>OpenSSHFactoryTests.test_getPrivateKeysAsRoot.<locals>.seteuidr r# r r$ N)r r r+ Z addCleanupr r r r r% r r r r& r r' r r( �getegid)r r r r! r r, r �test_getPrivateKeysAsRoot[ s z-OpenSSHFactoryTests.test_getPrivateKeysAsRootc C s$ | j �� }| �|dtd�gi� dS )zq L{OpenSSHFactory.getPrimes} should return the available primes in the moduli directory. i s diffie-hellman-group14-sha1N)r Z getPrimesr r )r Zprimesr r r �test_getPrimesr s �z"OpenSSHFactoryTests.test_getPrimes) �__name__� __module__� __qualname__�__doc__�getattrr �skipr r r" r) r. r/ r r r r r s ! r )r3 r Ztwisted.trial.unittestr Ztwisted.python.filepathr Ztwisted.python.reflectr Z$twisted.conch.openssh_compat.factoryr Ztwisted.conch.ssh._kexr Ztwisted.conch.testr Ztwisted.test.test_processr r r r r r �<module> s
Save File
Cancel