5. x d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlZddlZddlm Z ddl
m Z ddl
Z
ddlZdd lmZ dd
lmZ dd lmZ ddlmc mZ dd lmZ dd
lmZ ddlmZ ddl m Z ddl m Z dZ!dZ"dZ#dZ$d Z% ed G d dejL Z' ed G d dejL Z(y) $Integration tests for label command. )absolute_import)print_function)division)unicode_literalsN) parseString) _exceptions)handler)Tags)CommandException) SkipForGS) SkipForS3) ObjectToURI)Retry)UTF8key_onekey_two value_one value_twoc J | rdj | S dj | S )Nz Updating {}z%Setting label configuration on {}/...)format)using_gcloud_storage
bucket_uris )platform/gsutil/gslib/tests/test_label.py_get_label_setting_outputr . s) /-396*3EF
166*3EF z#Tests use S3-style XML passthrough.c e Zd ZdZ edez dz ez dz dz ez dz ez dz dz j d Z
fd Z d
Z d Z
d Zd
Z xZS ) TestLabelS3zDS3-specific tests. Most other test cases are covered in TestLabelGS.zz
zz
zz )indentc t t | | j | j j t | _ y Ncontents)superr setUpCreateTempFile
_label_xmlencoder xml_fpathself __class__s r r% zTestLabelS3.setUp= s4 +t"$(($//2H2H2N(ODNr c n t j r| j || y | j || y )N)sixPY2assertItemsEqualassertCountEqual)r+ item1item2s r DoAssertItemsMatchz TestLabelS3.DoAssertItemsMatchA s+
ww
E5) E5)r c
i }t }t j |d } t j j || |D ]" }|D ] }|j ||j < $ |S # t j $ rC}t d|j d|j d|j d }~ww xY w)Nz,Requested labels/tagging config is invalid: z at line z , column )r r
XmlHandlerxmlsaxr
SaxExceptionsSAXParseExceptionr
getMessage
getLineNumbergetColumnNumbervaluekey)r+ xml_str
label_dict tags_listhe tagset_listtags r _LabelDictFromXmlStringz#TestLabelS3._LabelDictFromXmlStringH s JI9d+AK gg'1%
! # !ii
377 ! * * K ,,.!//"3Q5F5F5H J
K KKs A- -C >B>>Cc | j j dd j t gd }t j
t } j
r j
|| n j |j | t t dd fd } | y ) NlabelsetT
return_stderr tries timeout_secsc j ddt gd } j j | j j y NrI getT
return_stdout RunGsUtilsurir4 rG r' stdoutr r+ s r _Check1z*TestLabelS3.testSetAndGet.._Check1e U ~~wtJ/? @,0 2f
d : :6 B " : :4?? KMr ) CreateBucketrX r) rY r _use_gcloud_storageassertIn assertEqualstripr AssertionError)r+ stderrexpected_outputr\ r s ` @r
testSetAndGetzTestLabelS3.testSetAndGetW s ""$J
^^WeT^^ !* - /*. 0F 00H0H04Z0@BO
mmOV,
v||~7 >3M 4M Ir c
j j dddt dt dt dt
t
g t t dd fd } | j ddd t dd
d d t
g t t
d d
it t dd fd } | y )NrI ch-l:rM rN rO c j ddt gd } j j | j j y rS rW rZ s r r\ z#TestLabelS3.testCh.._Check1y r] r -dznew_key:new_valueznonexistent-keynew_key new_valuec j ddt gd } j j | y rS )rX rY r4 rG )r[ r
expected_dictr+ s r _Check2z#TestLabelS3.testCh.._Check2 sE ~~wtJ/? @,0 2f
d : :6 B +-r ) r^ rX KEY1VALUE1KEY2VALUE2rY r rc )r+ r\ rq r rp s ` @@r testChzTestLabelS3.testChn s ""$JNNt $ Z >3M 4M I NNtT4)3- 4- Ir )__name__
__module__ __qualname____doc__r rr rs rt ru toprettyxmlr' r% r4 rG rf rv
__classcell__r, s @r r r 3 s L=D* +-3 46F G' (*. /1@ ACI J , , 1 1 2 3>+V+2L
P*
.$r r zTests use GS-style c X e Zd ZdZeeeeiZ fdZ d Z
d Z d Z d Z
d Zd Zd Z xZS )
TestLabelGSr c t t | | j t j
| j j t | _ y r! )
r$ r r% r& jsondumps _label_dictr( r
json_fpathr* s r r% zTestLabelGS.setUp sD +t"$))D , ,-44T: * *. 0FF+T-=-=>r c | j }| j }| j dd| j t | t | gd }t |j }t t
| j t | t
| j t | g }| j r+| j t |D cg c] }||v c} y | j || y c c}w )NrI rJ TrK ) r^ rX r rY rJ
splitlinesr r_
assertTrueallassertSetEqual)r+ r bucket2_urird actualexpectedxs r %testSetOnMultipleBucketsInSameCommandz1TestLabelGS.testSetOnMultipleBucketsInSameCommand s ""$J##%K ^^ %
j
k F
""$
%F!$":":D*. 0FF+X6r c
| j }dt dt dt dt g}| j ddg|z t
| gz d }t | j t
| }| j r| j || n | j |j | | j ddt
| gd }| j t j | | j d }d
}| j ddd|d|d t t
| g | j ddt
| gd }t j | }t t ||i} | j || y ) Nri rj rI rh TrK rT rU znew-keyz new-valuerl )r^ rr rs rt ru rX rY r r_ r` ra rb r r r r )
r+ r
ch_subargsrd re r[ rm rn r r s
r testInitialAndSubsequentChz&TestLabelGS.testInitialAndSubsequentCh sr ""$J v& J ^^WdOj 8D*. 0FF+T-=-=> GINNtI&d Z
^^WeT*-= >*. 0F
ZZ
Ffgy1H*r c t | j t | j g}dt dt dt dt
g}| j
ddg|z |z d }t |j }t |D cg c] }t | j | c} }| j r+| j t |D cg c] }||v c} n| j || |D ]G }| j
dd|gd }| j t j | | j" I y c c}w c c}w ) Nri rj rI rh TrK rT rU )rY r^ rr rs rt ru rX rJ r r r_ r r r r r r r ) r+ bucket_surisr rd r bucket_surir r r[ s r #testChAppliesChangesToAllBucketArgsz/TestLabelGS.testChAppliesChangesToAllBucketArgs s= **,-tD4E4E4G/HIL v& J ^^WdOj 8< G*. 0F
""$
%F''K "$":":KH' H
ooc911;9:;
&(+ $ ~~w{ ;4~Pf
4::f-t/?/?@ $
:s E
E c | j }| j ddddt | g | j ddt | gd }| j dt | z | y ) NrI rh rl z dummy-keyrT TrU z %s/ has no label configuration.)r^ rX rY r` )r+ r r[ s r &testChMinusDWorksWithoutExistingLabelsz2TestLabelGS.testChMinusDWorksWithoutExistingLabels sl ""$JNNGT4d:6FGH
^^WeT*-= >*. 0FMM3d:6FF r c dgddgg dddgddgg df}|D ]( }| j |dd }| j d
| * | j g d dd }| j d | y
)z3Ensures label commands fail with too few arguments.rI rJ )rI rJ filenamerT rh )rI rh ri zkey:valTrN )rL expected_statuszcommand requires at least)rI rh z(gs://some-nonexistent-foobar-bucket-namez(Please specify at least one label changeN)rX r` )r+ invocations_missing_argsarg_listrd s r testTooFewArgumentsFailsz$TestLabelGS.testTooFewArgumentsFails" s
%$ % $(
* -~~hdA~Nf
mm/8 -
^^C F MMr s + & % '
' 0
, ' ' ? ? 0 ( & F
01^(44 ^ 2^B !bF(44 bF "bFr