|
Server : Apache System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64 User : dh_edsupp ( 6597262) PHP Version : 8.2.26 Disable Function : NONE Directory : /lib/python3/dist-packages/awscli/__pycache__/ |
Upload File : |
o
2&�a�� � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
mZ ddlmZmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ ddl Z dd l m Z ddl Z!dd l"m#Z# dd
l m$Z$ ddl!m%Z% ddlZej&r�ej'j(ej'_)e j*�+� Z,e�-d�Z.da/dd� Z0dd� Z1dd� Z2dd� Z3dd� Z4e
j5dd� �Z6dSd d �Z7d d � Z8dTd"d#�Z9G d$d%� d%ej'�Z:G d&d'� d'e:�Z;G d(d)� d)e<�Z=G d*d+� d+e<�Z>e
j5d,d-� �Z?e
j5dUd/d0��Z@G d1d2� d2ej'�ZAG d3d4� d4eA�ZBG d5d6� d6ej'�ZCG d7d8� d8e<�ZDG d9d:� d:eE�ZFG d;d<� d<e<�ZGd=d>� ZH dVdAdB�ZIdCdD� ZJdEdF� ZKdGdH� ZLG dIdJ� dJej'�ZMG dKdL� dLe�ZNG dMdN� dNe<�ZOG dOdP� dPeE�ZPG dQdR� dRe<�ZQdS )Wa^ Test utilities for the AWS CLI.
This module includes various classes/functions that help in writing
CLI unit/integration tests. This module should not be imported by
any module **except** for test code. This is included in the CLI
package so that code that is not part of the CLI can still take
advantage of all the testing utilities we provide.
� N)�pformat)�Popen�PIPE)�mock)�StringIO)�six)�Session)� ClientError)� WaiterError)� AWSResponse)� load_plugins)� CLIDriver)�EnvironmentVariableszawscli.tests.integrationc s � fdd�}|S )z�Decorator to skip tests that should not be run on windows.
Example usage:
@skip_if_windows("Not valid")
def test_some_non_windows_stuff(self):
self.assertEqual(...)
c s t �t�� dv� �| �S )N)�Darwin�Linux)�unittest�skipIf�platform�system)�func��reason� �2/usr/lib/python3/dist-packages/awscli/testutils.py� decoratorP s
��z"skip_if_windows.<locals>.decoratorr )r r r r r �skip_if_windowsF s
r c C sF t j�� } | j}|�d��tj�}|sg }tj �
|� |� dt� | S )N� data_path� data_loader) �awscli� clidriver�create_clidriver�session�get_config_variable�split�os�pathsep�_LOADER� search_paths�extend�register_component)�driverr! r r r r r V s
r c C sb dd l } td u r/tj�tj�| j��}tj�|dd�}tj�|�s-t d�}|d u r-t
d��|atS )Nr �bin�awsz�Could not find "aws" executable. Either make sure it is on your PATH, or you can explicitly set this value using "set_aws_cmd()") r �AWS_CMDr$ �path�dirname�abspath�__file__�join�isfile�_search_path_for_cmd�
ValueError)r � repo_root�aws_cmdr r r � get_aws_cmda s r8 c C sB t j�dd��t j�D ]}t j�|| �}t j�|�r | S q d S )N�PATH� )r$ �environ�getr# r% r. r2 r3 )�cmd_namer. �
full_cmd_pathr r r r4 s s �r4 c C s | a d S �N)r- )r7 r r r � set_aws_cmd{ s r@ c c s� � t �� }dttd�� }tj�||�}t|d��� z(t|| �� }|V W d � n1 s/w Y W t �
|� dS W t �
|� dS t �
|� w )af This is a cross platform temporary file creation.
tempfile.NamedTemporary file on windows creates a secure temp file
that can't be read by other processes and can't be opened a second time.
For tests, we generally *want* them to be read multiple times.
The test fixture writes the temp file contents, the test reads the
temp file.
z
tmpfile-%s� �wN) �tempfile�mkdtemp�str� random_charsr$ r. r2 �open�close�shutil�rmtree)�mode�temporary_directory�basename�
full_filename�fr r r �temporary_file� s � ��rP c
C s� |sd}| j d|d�}|r|}nt� }d|i}|dkr!d|i|d< z |jd i |�� W |S tyK } z|jd �d �d
kr?n� W Y d }~|S d }~ww )
zG
Creates a bucket
:returns: the name of the bucket created
� us-west-2�s3�� region_name�Bucket� us-east-1�LocationConstraint�CreateBucketConfiguration�Error�Code�BucketAlreadyOwnedByYouNr )�
create_client�random_bucket_name�
create_bucketr �responser< )r! �name�region�client� bucket_name�params�er r r r^ � s( �
���r^ c C s t �t�t| d ����d�S )zZReturns random hex characters.
Useful for creating resources with random names.
� �ascii)�binascii�hexlifyr$ �urandom�int�decode)� num_charsr r r rF � s rF �awscli-s3integ-� c C s | t |� S )a Generate a random S3 bucket name.
:param prefix: A prefix to use in the bucket name. Useful
for tracking resources. This default value makes it easy
to see which buckets were created from CLI integ tests.
:param num_random: Number of random chars to include in the bucket name.
:returns: The name of a randomly generated bucket name as a string.
)rF )�prefix�
num_randomr r r r] � s r] c @ s e Zd ZdZdd� Zdd� ZdS )�BaseCLIDriverTestz�Base unittest that use clidriver.
This will load all the default plugins as well so it
will simulate the behavior the user will see.
c C sH t jd ddddd�| _t�d| j�| _| j�� t� | _| jj| _d S )N�
AWS_DATA_PATHrV �
access_key�
secret_keyr: �rs �AWS_DEFAULT_REGION�AWS_ACCESS_KEY_ID�AWS_SECRET_ACCESS_KEY�AWS_CONFIG_FILE�
os.environ) r$ r; r �patch�
environ_patch�startr r* r! ��selfr r r �setUp� s �
zBaseCLIDriverTest.setUpc C s | j �� d S r? )r} �stopr r r r �tearDown� � zBaseCLIDriverTest.tearDownN)�__name__�
__module__� __qualname__�__doc__r� r� r r r r rr � s
rr c sH e Zd Z� fdd�Z� fdd�Zdd� Zdd� Zd d
� Zd d � Z� Z S )
�BaseAWSHelpOutputTestc s< t t| ��� t�d�| _| j�� | _t� | _ | j | j_
d S )Nzawscli.help.get_renderer) �superr� r� r r| �renderer_patchr~ �
renderer_mock�CapturedRenderer�renderer� return_valuer �� __class__r r r� � s
zBaseAWSHelpOutputTest.setUpc s t t| ��� | j�� d S r? )r� r� r� r� r� r r� r r r� � s z BaseAWSHelpOutputTest.tearDownc C s* || j jvr| �d|| j jf � d S d S )NzFThe expected contents:
%s
were not in the actual rendered contents:
%s�r� �rendered_contents�fail)r� �containsr r r �assert_contains� � �
��z%BaseAWSHelpOutputTest.assert_containsc C s8 | j j�|�}||kr| �d||| j j|f � d S d S )NzkThe expected contents:
%s
, with the count:
%d
were not in the actual rendered contents:
%s
with count:
%d)r� r� �countr� )r� r� r� �r_countr r r �assert_contains_with_count� s �
��z0BaseAWSHelpOutputTest.assert_contains_with_countc C s* || j jv r| �d|| j jf � d S d S )NzKThe contents:
%s
were not suppose to be in the actual rendered contents:
%sr� �r� �contentsr r r �assert_not_contains� r� z)BaseAWSHelpOutputTest.assert_not_containsc s� |� d�}t|�}| jj� | �|� � � �|��� �fdd�|D �}|d }t|dd � d�D ])\}}|dkrB| �d|| � f � ||k rV| �d|| ||d � f � |}q/d S ) N�
starting_fromc s g | ]}� � |���qS r )�find)�.0�arg�r� � start_indexr r �
<listcomp> s z;BaseAWSHelpOutputTest.assert_text_order.<locals>.<listcomp>r � ���z/The string %r was not found in the contents: %szBThe string %r came before %r, but was suppose to come after it.
%s)�pop�listr� r� �assertInr� � enumerater� )r� �args�kwargsr� � arg_indices�previous�i�indexr r� r �assert_text_order s$
���z'BaseAWSHelpOutputTest.assert_text_order)
r� r� r� r� r� r� r� r� r� �
__classcell__r r r� r r� � s r� c @ s e Zd Zdd� Zdd� ZdS )r� c C s
d| _ d S )Nr: )r� r r r r �__init__ s
zCapturedRenderer.__init__c C s |� d�| _d S )N�utf-8)rl r� r� r r r �render s zCapturedRenderer.renderN)r� r� r� r� r� r r r r r� s r� c @ � e Zd Zdd� ZdS )�CapturedOutputc C s || _ || _d S r? )�stdout�stderr)r� r� r� r r r r� s
zCapturedOutput.__init__N)r� r� r� r� r r r r r� � r� c c s� � t �� } t �� }t�d| ��- t�d|�� t|| �V W d � n1 s'w Y W d � d S W d � d S 1 s?w Y d S )Nz
sys.stderrz
sys.stdout)r r r r| r� )r� r� r r r �capture_output" s � ��"�r� � c c s^ � t �| �}t jrt�� }||_n|}t�d|�� |V W d � d S 1 s(w Y d S )Nz sys.stdin)r �BytesIO�PY3r �Mock�bufferr| )� input_bytes�
input_data� mock_objectr r r �
capture_input+ s �
"�r� c @ sX e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Z dd
d�Z dd� Z
ddd�Z dS )�BaseAWSCommandParamsTestNc C s� i | _ d | _tjd dddddd�| _t�d| j�| _| j�� td di d �| _ i | _
t�d �| _ d
| _ g | _
d | _t� | _d S ) Nrs rV rt ru r: )rs rw rx ry rz �AWS_SHARED_CREDENTIALS_FILEr{ �� z'botocore.endpoint.Endpoint.make_requestF)� last_params� last_kwargsr$ r; r r| r} r~ r �
http_response�parsed_response�make_request_patch�make_request_is_patched�operations_called�parsed_responsesr r* r r r r r� ; s$ �
z BaseAWSCommandParamsTest.setUpc C �( | j �� | jr| j�� d| _d S d S �NF)r} r� r� r� r r r r r� U s
�z!BaseAWSCommandParamsTest.tearDownc K s | � |� d S r? )�
_store_params)r� rd r� r r r � before_call\ r� z$BaseAWSCommandParamsTest.before_callc C s || _ |d | _d S )N�body)�last_request_dictr� )r� rd r r r r� _ s z&BaseAWSCommandParamsTest._store_paramsc sR � j r � j�� d� _ � j�� }� jd ur � fdd�|_n� j� jf|_d� _ d S )NFc s � j � j�d�fS �Nr )r� r� r� )r� r� r r r �<lambda>m s z=BaseAWSCommandParamsTest.patch_make_request.<locals>.<lambda>T) r� r� r� r~ r� � side_effectr� r� r� )r� r� r r r �patch_make_requestc s
z+BaseAWSCommandParamsTest.patch_make_requestr c C s� | � ||�\}}}|d ur| �||� |d urEt�| j�} |d ur4|D ]}
z| |
= W q# ty3 Y q#w || krE| �dt|�t| �f � |||fS )NzGActual params did not match expected params.
Expected:
%s
Actual:
%s
)�run_cmdr� �copyr� �KeyErrorr� r ) r� �cmdrd � expected_rc�stderr_contains�
ignore_paramsr� r� �rcr� �keyr r r �assert_params_for_cmds s$
� ��
z.BaseAWSCommandParamsTest.assert_params_for_cmdc K s || _ | j�||�� f� d S r? )r� r� �appendr� )r� rd �modelr� r r r �before_parameter_build� s z/BaseAWSCommandParamsTest.before_parameter_buildc
C s� t �d|� | �� | jj�d�}|�d| j� |�d| j � t
|t �s)|� � }n|}t
� �&}z| j�|�}W n tyL } z |j}W Y d }~nd }~ww W d � n1 sWw Y |j�� }|j�� } | �||d|||| |f � | ||fS )NzCalling cmd: %s�
event_emitterz before-callzbefore-parameter-build.*.*�MUnexpected rc (expected: %s, actual: %s) for command: %s
stdout:
%sstderr:
%s)�logging�debugr� r* r! �
get_component�registerr� �register_firstr� �
isinstancer� r# r� �main�
SystemExit�coder� �getvaluer� � assertEqual)
r� r� r� r� �cmdlist�capturedr� re r� r� r r r r� � s: �
��� �
���
z BaseAWSCommandParamsTest.run_cmd)Nr NN�r ) r� r� r� �maxDiffr� r� r� r� r� r� r� r� r r r r r� 8 s
�r� c s( e Zd Z� fdd�Z� fdd�Z� ZS )� BaseAWSPreviewCommandParamsTestc s( t �d�| _| j�� tt| ��� d S )Nz-awscli.customizations.preview.mark_as_preview)r r| �
preview_patchr~ r� r� r� r r� r r r� � s
�
z%BaseAWSPreviewCommandParamsTest.setUpc s | j �� tt| ��� d S r? )r� r� r� r� r� r r� r r r� � �
z(BaseAWSPreviewCommandParamsTest.tearDown)r� r� r� r� r� r� r r r� r r� � s r� c @ s6 e Zd Zdd� Zdd� Zdi dfdd�Zd
d
d �Zd S )�BaseCLIWireResponseTestc C sP t jd ddddd�| _t�d| j�| _| j�� t�d�| _d | _t� | _ d S )
Nrs rV rt ru r: rv r{ z botocore.endpoint.Endpoint._sendF)
r$ r; r r| r} r~ �
send_patch�send_is_patchedr r* r r r r r� � s �
z BaseCLIWireResponseTest.setUpc C r� r� )r} r� r� r� r r r r r� � s
�z BaseCLIWireResponseTest.tearDownr� r� c C s< | j r | j�� d| _ | j�� }tj|||d�|_d| _ d S )NF)� status_code�headers�contentT)r� r� r� r~ r r� r� )r� r� r� r r� r r r �
patch_send� s
�
z"BaseCLIWireResponseTest.patch_sendr c C s� t |t�s
|�� }n|}t� �&}z| j�|�}W n ty- } z |j}W Y d }~nd }~ww W d � n1 s8w Y |j� � }|j
� � }| � ||d|||||f � |||fS )Nr� ) r� r� r# r� r* r� r� r� r� r� r� r� ) r� r� r� r� r� r� re r� r� r r r r� � s,
��� �
���
z BaseCLIWireResponseTest.run_cmdNr� )r� r� r� r� r� r r� r r r r r� � s
r� c @ s6 e Zd Zdd� Zdd� Zd
dd�Zd d
� Zd d � ZdS )� FileCreatorc C s t �� | _d S r? )rC rD �rootdirr r r r r� � r� zFileCreator.__init__c C s" t j�| j�rt�| j� d S d S r? )r$ r. �existsr rI rJ r r r r �
remove_all� s �zFileCreator.remove_allNrB c C s� t j�| j|�}t j�t j�|��st �t j�|�� t||��
}|�|� W d � n1 s0w Y t j� |�}t �
|||d f� |durQt �
|||f� |S )a� Creates a file in a tmpdir
``filename`` should be a relative path, e.g. "foo/bar/baz.txt"
It will be translated into a full path in a tmp dir.
If the ``mtime`` argument is provided, then the file's
mtime will be set to the provided value (must be an epoch time).
Otherwise the mtime is left untouched.
``mode`` is the mode the file should be opened either as ``w`` or
`wb``.
Returns the full path to the file.
Ni �) r$ r. r2 r �isdirr/ �makedirsrG �write�getmtime�utime)r� �filenamer� �mtimerK � full_pathrO � current_timer r r � create_file� s � zFileCreator.create_filec C sp t j�| j|�}t j�t j�|��st �t j�|�� t|d��}|�|� W d � |S 1 s1w Y |S )z�Append contents to a file
``filename`` should be a relative path, e.g. "foo/bar/baz.txt"
It will be translated into a full path in a tmp dir.
Returns the full path to the file.
�aN) r$ r. r2 r r r/ r rG r )r� r r� r
rO r r r � append_file s
��zFileCreator.append_filec C s t j�| j|�S )zzTranslate relative path to full path in temp dir.
f.full_path('foo/bar.txt') -> /tmp/asdfasd/foo/bar.txt
)r$ r. r2 r )r� r r r r r
s zFileCreator.full_path)NrB )r� r� r� r� r r r r
r r r r r � s
r c @ � e Zd ZdS )�ProcessTerminatedErrorN�r� r� r� r r r r r ( � r c @ s" e Zd Zddd�Zedd� �ZdS )�ResultNc C sL || _ || _|| _t�d|� t�d|� t�d|� |d u r!g }|| _d S )Nzrc: %sz
stdout: %sz
stderr: %s)r� r� r� � INTEG_LOGr� � memory_usage)r� r� r� r� r r r r r� - s
zResult.__init__c C s t �| j�S r? )�json�loadsr� r r r r r 8 s z Result.jsonr? )r� r� r� r� �propertyr r r r r r , s
r c C s | � dd�} | � dd�} | S )N�"z\"�')�replace)�commandr r r �_escape_quotes= s r FTc C s t �� dkr
t| �} dtjv rtjd }ndt� }d|| f }t� }t|tj �r1tj
s1|� |�}t �
d|� tj�� } d| vrDd| d< |durJ|} |du rPt}t|tt|d | d
�}
|s^|
S d} |sui } |rjd |i} |
jd i | ��\}
}nt|
�\}
}} t|
j|
�|�|�|�| �S )
a] Run an aws command.
This help function abstracts the differences of running the "aws"
command on different platforms.
If collect_memory is ``True`` the the Result object will have a list
of memory usage taken at 2 second intervals. The memory usage
will be in bytes.
If env_vars is None, this will set the environment variables
to be used by the aws process.
If wait_for_finish is False, then the Process object is returned
to the caller. It is then the caller's responsibility to ensure
proper cleanup. This can be useful if you want to test timeout's
or how the CLI responds to various signals.
:type input_data: string
:param input_data: This string will be communicated to the process through
the stdin of the process. It essentially allows the user to
avoid having to use a file handle to pass information to the process.
Note that this string is not passed on creation of the process, but
rather communicated to the process.
:type input_file: a file handle
:param input_file: This is a file handle that will act as the
the stdin of the process immediately on creation. Essentially
any data written to the file will be read from stdin of the
process. This is needed if you plan to stream data into stdin while
collecting memory.
�Windows�AWS_TEST_COMMANDz python %sz%s %szRunning command: %srw rV NT)r� r� �stdin�shell�env�inputr )r r r r$ r; r8 �get_stdout_encodingr� r � text_typer� �encoder r� r� r r � communicate�_wait_and_collect_memr �
returncoderl )r �collect_memory�env_vars�wait_for_finishr� �
input_file� aws_command� full_command�stdout_encodingr% �process�memoryr� r� r� r r r r, G sD !
��r, c C s t tjdd �} | d u r
d} | S )N�encodingr� )�getattr�sys�
__stdout__)r6 r r r r' � s r' c C s� t �� dkr t}nt �� dkrt}ntdt �� ��g }| �� d u r>z|| j�}W n ty2 Y n w |�|� | �� d u s"| �� \}}|||fS )Nr r z0Can't collect memory for process on platform %s.) r r �_get_memory_with_psr5 �poll�pidr r� r* )r4 �
get_memoryr5 �currentr� r� r r r r+ � s( �� �
�
r+ c C s` d� � }|�t| �� t|td�}|�� d }|jdks"tt| ���t|� � d � � d �d S )Nzps u -p)r� r r� � i )
r# r� rE r r r* r, r rk �
splitlines)r< � command_list�pr� r r r r: � s
r: c @ s� e Zd ZdZg d�Zdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z d8dd�Z d9dd�Z
d:dd�Zdd� Zd d � Zd d � Zd;d!d"�Zd#d$� Zd;d%d&�Zd;d'd(�Zd)d*� Zd+d,� Zd-d.� Z d<d/d0�Z d<d1d2�Z 3d=d4d5�Zd6d7� Z dS )>�BaseS3CLICommandz�Base class for aws s3 command.
This contains convenience functions to make writing these tests easier
and more streamlined.
)�SSECustomerAlgorithm�SSECustomerKey�SSECustomerKeyMD5� RequestPayerc C s@ t � | _tj�� | _i | _d| _| jjd| jd�| _| � � d S )NrQ rR rS )
r �files�botocorer! � get_session�regionsra r\ rb � extra_setupr r r r r� � s zBaseS3CLICommand.setUpc C � d S r? r r r r r rL � � z BaseS3CLICommand.extra_setupc C s | j �� | �� d S r? )rH r �extra_teardownr r r r r� � s
zBaseS3CLICommand.tearDownc C rM r? r r r r r rO � rN z BaseS3CLICommand.extra_teardownc K s | j �d�}|jdi |�� d S )N�response_parser_factoryr )r! r� �set_parser_defaults)r� r� �factoryr r r �override_parser� s z BaseS3CLICommand.override_parserc C s$ | j �|| j�}| jjd|d�}|S )NrR rS )rK r< ra r! r\ )r� rc ra rb r r r �create_client_for_bucket� s z)BaseS3CLICommand.create_client_for_bucketc C sh | � ||� t|tj�r|�� �d�}| �||�}| �t|�t|�� ||kr2| � d||f � d S d S )Nr� z?Contents for %s/%s do not match (but they have the same length))
�wait_until_key_existsr� r r� r� rl �get_key_contentsr� �lenr� )r� �bucketr� �expected_contents�actual_contentsr r r �assert_key_contents_equal� s
��z*BaseS3CLICommand.assert_key_contents_equalNc C s> |s| j }t| j||�}|| j|< | �| j|� | �|� |S r? )ra r^ r! rK �
addCleanup�
delete_bucket�wait_bucket_exists)r� r` ra rc r r r r^ � s
z BaseS3CLICommand.create_bucketr: c s~ � � |�}|||d�}|d ur|�|� |jdi |��}� �� j||� i }|r5t� fdd�|�� D ��}� j|||d� |S )N)rU �Key�Bodyc 3 s& � | ]\}}|� j v r||fV qd S r? )�_PUT_HEAD_SHARED_EXTRAS)r� �k�vr r r � <genexpr> s �
��z.BaseS3CLICommand.put_object.<locals>.<genexpr>)� extra_paramsr )rT �update�
put_objectr\ �
delete_key�dict�itemsrU ) r� rc �key_namer� �
extra_argsrb � call_argsr_ �extra_head_paramsr r r rg � s&
�
��zBaseS3CLICommand.put_objectr? c C s� | � |� | �|�}|} |d8 }z|j|d� W n |jjy5 | �|�r)Y n|dkr.� t�|� Y nw q
| j� |d � d S )NTr� �rU r )
�remove_all_objectsrT r] �
exceptions� NoSuchBucket�bucket_not_exists�time�sleeprK r� )r� rc �attempts�delayrb �attempts_remainingr r r r] s"
��
z BaseS3CLICommand.delete_bucketc C sb | � |�}|�d�}|j|d�}g }|D ]}|dd� |�dg �D �7 }q|D ]}| �||� q&d S )N� list_objectsro c S s g | ]}|d �qS )r_ r )r� �objr r r r� 0 s z7BaseS3CLICommand.remove_all_objects.<locals>.<listcomp>�Contents)rT �
get_paginator�paginater< rh )r� rc rb � paginator�pages� key_names�pagerk r r r rp * s
�z#BaseS3CLICommand.remove_all_objectsc C s | � |�}|j||d�}d S �N�rU r_ )rT �
delete_object�r� rc rk rb r_ r r r rh 4 r� zBaseS3CLICommand.delete_keyc C s6 | � ||� | �|�}|j||d�}|d �� �d�S )Nr� r` r� )rU rT �
get_object�readrl r� r r r rV 8 s
z!BaseS3CLICommand.get_key_contents� c s8 | � � �}|�d��t|dd�}|�� �fdd�� d S )N�
bucket_existsT)�
min_successes�delay_initial_pollc s �j � d�d u S )Nro )�waitr �rc �waiterr r r� D s z5BaseS3CLICommand.wait_bucket_exists.<locals>.<lambda>)rT �
get_waiter�ConsistencyWaiterr� )r� rc r� rb �consistency_waiterr r� r r^ > s
� �z#BaseS3CLICommand.wait_bucket_existsc
C sV | � |�}z |j|d� W dS ty* } z|j�d�dkr%W Y d }~dS � d }~ww )Nro TrZ �404F)rT � head_bucketr r_ r< )r� rc rb �errorr r r rs G s
��z"BaseS3CLICommand.bucket_not_existsc C �0 z | j |||d� W dS ttfy Y dS w �N)r� TF)rU r r
�r� rc rk r� r r r �
key_existsQ � ��zBaseS3CLICommand.key_existsc C r� r� )�wait_until_key_not_existsr r
r� r r r �key_not_existsY r� z BaseS3CLICommand.key_not_existsc C s | j �� }|d S )N�Buckets)rb � list_buckets)r� r_ r r r r� a s
z BaseS3CLICommand.list_bucketsc C s | � ||�}|d S )N� ContentType)� head_object)r� rc rk �parsedr r r �content_type_for_keye s z%BaseS3CLICommand.content_type_for_keyc C s | � |�}|j||d�}|S r� )rT r� r� r r r r� i s
z BaseS3CLICommand.head_objectc C � | j ||||dd� d S )NT�r ��
_wait_for_key�r� rc rk re r� r r r rU n �
�z&BaseS3CLICommand.wait_until_key_existsc C r� )NFr� r� r� r r r r� s r� z*BaseS3CLICommand.wait_until_key_not_existsTc
C sb | � |�}|r
|�d�}n|�d�}||d�}|d ur |�|� t|�D ]
} |jdi |�� q$d S )N�
object_exists�object_not_existsr� r )rT r� rf �ranger� )
r� rc rk re r� r rb r� rd �_r r r r� x s
�z BaseS3CLICommand._wait_for_keyc C s^ | � |jdd|j|j|j f � | �d|j� | �d|j� | �d|j� | �d|j� d S )Nr z Non zero rc (%s) received: %szError:zfailed:z client errorz server error)r� r� r� r� � assertNotIn)r� rB r r r �assert_no_errors� s �z!BaseS3CLICommand.assert_no_errors�NN)r: N)r? r? )r� )Nr� )Nr� T) r� r� r� r� ra r� rL r� rO rS rT r[ r^ rg r] rp rh rV r^ rs r� r� r� r� r� rU r� r� r� r r r r rC � sB
�
�
�
rC c @ r� )�StringIOWithFileNoc C s dS r� r r r r r �fileno� s zStringIOWithFileNo.filenoN)r� r� r� r� r r r r r� � r� r� c @ s* e Zd Zddd�Zedd� �Zdd� ZdS ) �TestEventHandlerNc C s || _ d| _d| _d S r� )�_handler�_called�__test__)r� �handlerr r r r� � s
zTestEventHandler.__init__c C s | j S r? )r� r r r r �called� s zTestEventHandler.calledc K s( d| _ | jd ur| jdi |�� d S d S )NTr )r� r� )r� r� r r r r� � s
�zTestEventHandler.handlerr? )r� r� r� r� r r� r� r r r r r� � s
r� c @ r )�ConsistencyWaiterExceptionNr r r r r r� � r r� c @ s. e Zd ZdZ d
dd�Zdd � Zd
d � Zd S )r� a�
A waiter class for some check to reach a consistent state.
:type min_successes: int
:param min_successes: The minimum number of successful check calls to
treat the check as stable. Default of 1 success.
:type max_attempts: int
:param min_successes: The maximum number of times to attempt calling
the check. Default of 20 attempts.
:type delay: int
:param delay: The number of seconds to delay the next API call after a
failed check call. Default of 5 seconds.
r� � r? Fc C s || _ || _|| _|| _d S r? )r� � max_attemptsrw r� )r� r� r� rw r� r r r r� � s
zConsistencyWaiter.__init__c O s| d}d}| j r
t�| j� || jk r4|d7 }||i |��r)|d7 }|| jkr(dS nt�| j� || jk s| �||�}t|��)a�
Wait until the check succeeds the configured number of times
:type check: callable
:param check: A callable that returns True or False to indicate
if the check succeeded or failed.
:type args: list
:param args: Any ordered arguments to be passed to the check.
:type kwargs: dict
:param kwargs: Any keyword arguments to be passed to the check.
r r� N)r� rt ru rw r� r� �
_fail_messager� )r� �checkr� r� rv � successes�fail_msgr r r r� � s
�
� zConsistencyWaiter.waitc C s ||f}d| S )Nz/Failed after %s attempts, only had %s successesr )r� rv r� � format_argsr r r r� � s z ConsistencyWaiter._fail_messageN)r� r� r? F)r� r� r� r� r� r� r� r r r r r� � s
� r� r� )rn ro )r� )FNTNN)Rr� r$ r8 r� rI rt r r� rC r �
contextlib�stringrh �pprintr �
subprocessr r r r �
awscli.compatr r �botocore.sessionr �botocore.exceptionsr r
�botocore.loadersrI �botocore.awsrequestr �awscli.clidriverr �
awscli.pluginr r
r �PY2�TestCase�assertItemsEqual�assertCountEqual�loaders�Loaderr& � getLoggerr r- r r r8 r4 r@ �contextmanagerrP r^ rF r] rr r� �objectr� r� r� r� r� r� r� r � Exceptionr r r r, r' r+ r: rC r� r� r� r� r r r r �<module> s�
3
t 5;
�E V