Файловый менеджер - Редактировать - /opt/imh-python/lib/python3.9/site-packages/libcloud/test/compute/__pycache__/test_ssh_client.cpython-39.pyc
Ðазад
a q�h�^ � @ s d dl mZ d dl mZ d dlZd dlZd dlZd dlmZ d dlm Z d dlm Z d dl m Z d dl m Z d d l mZ d d lmZ d d lmZ d d lmZ d d lmZmZmZ es�dZ dZnd dlZejZe �e d�G dd� de ��ZG dd� de �Z e dk�re� e � � � dS )� )�absolute_import)�with_statementN)� _init_once)�LibcloudTestCase)�unittest)�ParamikoSSHClient)�ShellOutSSHClient)� have_paramiko)�StringIO)�u)�assertRaisesRegex)�patch�Mock� MagicMockz0.0.0z*Skipping because paramiko is not availablec @ s� e Zd Zede�dd� �Zdd� Zede�dd� �Zede�dd � �Zd d � Z ede�d d � �Z ede�dd� �Z ede�e � edkd�dd� ��Zede�dd� �Zede�dd� �Zede�dd� �Zede�dd� �Zdd � Zd d � Zede�d d � �Zede�d!d"� �Zede�d#d$� �Ze�ed%eed&�d'��e�ed(eed&�d'��d)d*� ��Z d+d,� Z d-d.� Z d/d0� Z d1d2� Z d3d4� Z!d5d6� Z"d7d8� Z#d9d:� Z$d;S )<�ParamikoSSHClientTestszparamiko.SSHClientc C sD dddddd�}t �� \}| _| jtjd< t� tf i |��| _dS ) zD Creates the object patching the actual connection. �dummy.host.org�v" �ubuntu�~/.ssh/ubuntu_ssh�600)�hostname�port�username�key�timeout�LIBCLOUD_DEBUGN)�tempfile�mkstemp�tmp_file�os�environr r �ssh_cli)�self� conn_params�_� r% ��/root/rpmbuild/BUILDROOT/imh-python39-modules-3.9.7-92.el8.x86_64/opt/imh-python/lib/python3.9/site-packages/libcloud/test/compute/test_ssh_client.py�setUp0 s � z ParamikoSSHClientTests.setUpc C s dt jv rt jd= d S )Nr )r r )r" r% r% r& �tearDown? s z ParamikoSSHClientTests.tearDownc C sV dddd�}t f i |��}|�� ddddddd�}|jjjf i |�� | �d� d S )Nr r �r r �passwordF� )r r* � allow_agentr � look_for_keysr �Connecting to server�r �connect�client�assert_called_once_with� assertLogMsg�r" r# �mock� expected_connr% r% r& �test_create_with_passwordC s ��z0ParamikoSSHClientTests.test_create_with_passwordc C sV dddd�}t f i |��}|�� ddddddd�}|jjjf i |�� | �d� d S ) Nr r �id_rsa�r r r Fr+ �r r, r r- � key_filenamer r. r/ r4 r% r% r& � test_deprecated_key_argumentT s ��z3ParamikoSSHClientTests.test_deprecated_key_argumentc C s, ddddd�}d}t | t|tfi |�� d S )Nr r r8 r )r r � key_files� key_materialz;key_files and key_material arguments are mutually exclusive)r � ValueErrorr )r" r# � expected_msgr% r% r& �>test_key_files_and_key_material_arguments_are_mutual_exclusivee s ���zUParamikoSSHClientTests.test_key_files_and_key_material_arguments_are_mutual_exclusivec C s� t j�t j�t�ddd�}t|d��}|�� }W d � n1 sB0 Y dd|d�}tf i |��}|�� t j � t |��}dddd|d d �}|j jjf i |�� | �d � d S ) N�fixtures�misc� test_rsa.key�rr r �r r r> Fr+ )r r, r r- �pkeyr r. )r �path�join�dirname�__file__�open�readr r0 �paramiko�RSAKeyZfrom_private_keyr r1 r2 r3 )r" rH �fp� private_keyr# r5 rG r6 r% r% r& �test_key_material_argumentp s* � &��z1ParamikoSSHClientTests.test_key_material_argumentc C s6 dddd�}t f i |��}d}t| tjj||j� d S )Nr r r8 rF � Invalid or unsupported key type)r r rN � ssh_exception� SSHExceptionr0 )r" r# r5 r@ r% r% r& �&test_key_material_argument_invalid_key� s � �z=ParamikoSSHClientTests.test_key_material_argument_invalid_keyz2.7.0z3New versions of paramiko support OPENSSH key formatc C s� t j�t j�t�ddd�}t|d��}|�� }W d � n1 sB0 Y dd|d�}tf i |��}d}t| t j j ||j � d S ) NrB rC ztest_rsa_non_pem_format.keyrE r r rF rS � r rH rI rJ rK rL rM r r rN rT rU r0 �r" rH rP rQ r# r5 r@ r% r% r& �"test_key_file_non_pem_format_error� s � &� �z9ParamikoSSHClientTests.test_key_file_non_pem_format_errorc C s� t j�t j�t�ddd�}t|d��}|�� }W d � n1 sB0 Y dd|d�}tf i |��}d}t| t j j ||j � dd|d �}tf i |��}d}t| t j j ||j � d S ) NrB rC � test_rsa_2048b_pass_foobar.keyrE r r rF z private key file is encrypted�r r r= ) r rH rI rJ rK rL rM r r rN rT ZPasswordRequiredExceptionr0 rX r% r% r& �0test_password_protected_key_no_password_provided� s0 � &� �� �zGParamikoSSHClientTests.test_password_protected_key_no_password_providedc C s� t j�t j�t�ddd�}t|d��}|�� }W d � n1 sB0 Y dd|dd�}tf i |��}d }t| t j j ||j � d S ) NrB rC rZ rE r r �invalid�r r r> r* z/OpenSSH private key file checkints do not matchrW rX r% r% r& r\ � s � &� �c C s� t j�t j�t�ddd�}t|d��}|�� }W d � n1 sB0 Y dd|dd�}tf i |��}| �|� � � dd|dd �}tf i |��}| �|� � � d S ) NrB rC rZ rE r r Zfoobarr^ )r r r= r* � r rH rI rJ rK rL rM r � assertTruer0 �r" rH rP rQ r# r5 r% r% r& �3test_password_protected_key_valid_password_provided� s( � &��zJParamikoSSHClientTests.test_password_protected_key_valid_password_providedc C sx t j�t j�t�ddd�}t|d��}|�� }W d � n1 sB0 Y dd|d�}tf i |��}| �|� � � d S )NrB rC ztest_ed25519.keyrE r r rF r_ ra r% r% r& �test_ed25519_key_type� s � &�z,ParamikoSSHClientTests.test_ed25519_key_typec C sz ddd�}t f i |��}tj�tj�t�ddd�}t|d��}|�� }W d � n1 sZ0 Y |j|d�}| � |� | � t |t j �� tj�tj�t�ddd �}t|d��}|�� }W d � n1 s�0 Y |j|d�}| � |� | � t |t j �� tj�tj�t�ddd �}t|d��}|�� }W d � n1 �sD0 Y |j|d�}| � |� | � t |t j�� d S ) Nr r �r r rB rC z+test_rsa_non_paramiko_recognized_header.keyrE �r z+test_dsa_non_paramiko_recognized_header.keyz-test_ecdsa_non_paramiko_recognized_header.key�r r rH rI rJ rK rL rM Z_get_pkey_objectr` � isinstancerN rO ZDSSKeyZECDSAKey�r" r# r1 rH rP rQ rG r% r% r& �?test_key_material_valid_pem_keys_invalid_header_auto_conversion s> �� & � & � ( zVParamikoSSHClientTests.test_key_material_valid_pem_keys_invalid_header_auto_conversionc C sz ddd�}t f i |��}tj�tj�t�ddd�}t|d��}|�� }W d � n1 sZ0 Y |j|d�}| � |� | � t |t j �� tj�tj�t�ddd �}t|d��}|�� }W d � n1 s�0 Y |j|d�}| � |� | � t |t j �� tj�tj�t�ddd �}t|d��}|�� }W d � n1 �sD0 Y |j|d�}| � |� | � t |t j�� d S ) Nr r rd rB rC rD rE re z test_dsa.keyztest_ecdsa.keyrf rh r% r% r& � test_key_material_valid_pem_keys4 s> �� & � & � ( z7ParamikoSSHClientTests.test_key_material_valid_pem_keysc C sV dddd�}t f i |��}|�� ddddddd�}|jjjf i |�� | �d� d S ) Nr r r8 r[ Fr+ r: r. r/ r4 r% r% r& �test_create_with_key] s ��z+ParamikoSSHClientTests.test_create_with_keyc C sZ ddddd�}t f i |��}|�� dddddddd�}|jjjf i |�� | �d� d S ) Nr r r8 )r r r* r Fr+ )r r* r, r r- r; r r. r/ r4 r% r% r& �!test_create_with_password_and_keyn s" ��z8ParamikoSSHClientTests.test_create_with_password_and_keyc C sH ddd�}t f i |��}|�� dddddd�}|jjjf i |�� dS )z� Initialize object with no credentials. Just to have better coverage, initialize the object without 'password' neither 'key'. r r rd Tr+ )r r r, r- r N)r r0 r1 r2 r4 r% r% r& � test_create_without_credentials� s ��z6ParamikoSSHClientTests.test_create_without_credentials�_consume_stdout� )� return_value�_consume_stderrc C s� | j }d}|�� |j}dddddddd�}|jjf i |�� |�|� |�� j�d � |�� jjd d d � |� |� |� � � � j �|� | � d � | � d� |�� dS )z" Basic execution. �/root/random_script.shr r Fr r r )r r; r, r r- r r �rootzrandom_script.sh�w)�modez.Executing command (cmd=/root/random_script.sh)zCommand finishedN)r! r0 r1 r2 �put� open_sftp�chdir�assert_called_with�file�runZ get_transportZ open_sessionZ exec_commandr3 �close)r" r5 �sdZmock_clir6 r% r% r& � test_basic_usage_absolute_path� s2 � � � z5ParamikoSSHClientTests.test_basic_usage_absolute_pathc C sN | j }d}|�� |�|� |j�� j�|� | �d� |�� | �d� dS )z< Provide a basic test with 'delete' action. rr z Deleting filezClosing server connectionN) r! r0 �deleter1 rw �unlinkry r3 r| )r" r5 r} r% r% r& �test_delete_script� s z)ParamikoSSHClientTests.test_delete_scriptc C sL t | jd��}|�� }W d � n1 s*0 Y | �|�|�dk� d S )NrE ���)rL r rM r` �find)r" r@ rP �contentr% r% r& r3 � s &z#ParamikoSSHClientTests.assertLogMsgc C s� ddd�}t f i |��}d|_t� }g d�|j_ddg|j_|�|��� }| �t d�|� | �t |�d � ddd�}t f i |��}d|_t� }g d�|j_d d g|j_|�|��� }| �t d �|� | �t |�d � d S �Nr r rd i )TTFZ123Z456Z123456� Z987Z6543210Z 9876543210� ) r � CHUNK_SIZEr � recv_ready� side_effect�recvrn �getvalue� assertEqualr �len�r" r# r1 �chan�stdoutr% r% r& �test_consume_stdout� s, � � z*ParamikoSSHClientTests.test_consume_stdoutc C s� ddd�}t f i |��}d|_t� }g d�|j_ddg|j_|�|��� }| �t d�|� | �t |�d � ddd�}t f i |��}d|_t� }g d�|j_d d g|j_|�|��� }| �t d �|� | �t |�d � d S r� ) r r� r �recv_stderr_readyr� � recv_stderrrq r� r� r r� �r" r# r1 r� �stderrr% r% r& �test_consume_stderr� s, � � z*ParamikoSSHClientTests.test_consume_stderrc C sj ddd�}t f i |��}d|_t� }g d�|j_g d�|j_|�|��� }| �d|� | �t |�d� d S � Nr r rd � )TTTTF)� ð� � � u ð� ) r r� r r� r� r� rn r� r� r� r� r% r% r& �Dtest_consume_stdout_chunk_contains_part_of_multi_byte_utf8_character s � z[ParamikoSSHClientTests.test_consume_stdout_chunk_contains_part_of_multi_byte_utf8_characterc C sj ddd�}t f i |��}d|_t� }g d�|j_g d�|j_|�|��� }| �d|� | �t |�d� d S r� ) r r� r r� r� r� rq r� r� r� r� r% r% r& �Dtest_consume_stderr_chunk_contains_part_of_multi_byte_utf8_character s � z[ParamikoSSHClientTests.test_consume_stderr_chunk_contains_part_of_multi_byte_utf8_characterc C sr ddd�}t f i |��}d|_t� }g d�|j_d�d�dd g|j_|�|��� }| � d |� | � t |�d � d S � Nr r rd r� )TTTFu 🤦zutf-32�a�bz & ab� ) r r� r r� r� �encoder� rn r� r� r� r� r% r% r& �5test_consume_stdout_chunk_contains_non_utf8_character' s � zLParamikoSSHClientTests.test_consume_stdout_chunk_contains_non_utf8_characterc C sr ddd�}t f i |��}d|_t� }g d�|j_d�d�dd g|j_|�|��� }| � d |� | � t |�d � d S r� ) r r� r r� r� r� r� rq r� r� r� r� r% r% r& �5test_consume_stderr_chunk_contains_non_utf8_character5 s � zLParamikoSSHClientTests.test_consume_stderr_chunk_contains_non_utf8_characterN)%�__name__� __module__� __qualname__r r r' r( r7 r< rA rR rV r �skipIf�paramiko_versionrY r\ rb rc ri rj rk rl rm �objectr r r r~ r� r3 r� r� r� r� r� r� r% r% r% r& r - s` � ,) � � $r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �ShellOutSSHClientTestsc C s\ zt dddd� W n: tyL } z"t|�}| �d|v � W Y d }~nd }~0 0 | �d� d S )N� localhost�foo�barr) z(ShellOutSSHClient only supports key auth�Exception was not thrown)r r? �strr` �fail)r" �e�msgr% r% r& � test_password_auth_not_supportedF s � $z7ShellOutSSHClientTests.test_password_auth_not_supportedc s� G dd� dt �� � fdd�}td|��f ztddd� W n: tyr } z"t|�}| �d |v � W Y d }~nd }~0 0 | �d � W d � n1 s�0 Y d S ) Nc @ s e Zd ZdZdd� ZdS )zKShellOutSSHClientTests.test_ssh_executable_not_available.<locals>.MockChild� c _ s d S �Nr% ��args�kwargsr% r% r& � communicateT s zWShellOutSSHClientTests.test_ssh_executable_not_available.<locals>.MockChild.communicateN)r� r� r� � returncoder� r% r% r% r& � MockChildQ s r� c s � � S r� r% r� �r� r% r& � mock_popenW s zLShellOutSSHClientTests.test_ssh_executable_not_available.<locals>.mock_popenzsubprocess.Popenr� r� rd zssh client is not availabler� )r� r r r? r� r` r� )r" r� r� r� r% r� r& �!test_ssh_executable_not_availableP s $z8ShellOutSSHClientTests.test_ssh_executable_not_availablec C s t ddd�}| �|�� � d S �Nr� rs rd )r r` r0 �r" r1 r% r% r& �test_connect_successc s z+ShellOutSSHClientTests.test_connect_successc C s t ddd�}| �|�� � d S r� )r r` r| r� r% r% r& �test_close_successg s z)ShellOutSSHClientTests.test_close_successc C sv t ddd�}t dddd�}t ddddd�}|�� }|�� }|�� }| �|dd g� | �|g d �� | �|g d �� d S ) Nr� rs rd � /home/my.keyr9 � )r r r r �ssh�root@localhost)r� �-ir� r� )r� r� r� z-oConnectTimeout=5r� )r Z_get_base_ssh_commandr� )r" Zclient1Zclient2Zclient3Zcmd1Zcmd2Zcmd3r% r% r& �test_get_base_ssh_commandk s ��z0ShellOutSSHClientTests.test_get_base_ssh_commandN)r� r� r� r� r� r� r� r� r% r% r% r& r� D s r� �__main__) � __future__r r r �sysr Zlibcloudr Z libcloud.testr r Zlibcloud.compute.sshr r r Zlibcloud.utils.py3r r r r5 r r r r� rN � __version__r� r r� r� �exit�mainr% r% r% r& �<module> s8 9
| ver. 1.1 | |
.
| PHP 8.3.30 | Ð“ÐµÐ½ÐµÑ€Ð°Ñ†Ð¸Ñ Ñтраницы: 0 |
proxy
|
phpinfo
|
ÐаÑтройка