File: //lib/python3/dist-packages/twisted/conch/test/__pycache__/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