
    |                     r    d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d Zd Z G d dej                        Z
y)    N)
gen_client)
test_utilsc                     t        j                  j                  t         j                  j                  t              | dz   g| S )N_sample)ospathjoindirname__file__)api_namer   s     Cplatform/gsutil/third_party/apitools/samples/uptodate_check_test.pyGetSampleClientPathr      s.    77<<18i3GO$OO    c                 d    t        |       5 }|j                         cd d d        S # 1 sw Y   y xY w)N)openread)	file_pathfs     r   _GetContentr      s    	iAvvx 
s   &/c                   <    e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
y	)
ClientGenCliTestc                     ||k7  rLt        j                  |j                         |j                               }t        dj	                  |            y)z?Like unittest.assertEqual with a diff in the exception message.
N)difflibunified_diff
splitlinesAssertionErrorr	   )selfexpectedactualr   s       r   AssertDiffEqualz ClientGenCliTest.AssertDiffEqual$   sJ    v"//##%v'8'8':<L <!899 r   c                 *   |dz   |z   }t        j                         5 }t        j                  t        j                  dddt        ||dz         d|ddd	j                  ||      d
g       t        |dz   |dz   dg      }| j                  |t        t        j                  |                   |D ]O  }| j                  t        t        |||            t        t        j                  j                  ||                   Q 	 d d d        y # 1 sw Y   y xY w)N_z--init-fileemptyz--infilez.jsonz--outdirz--overwritez--root_packagezsamples.{0}_sample.{0}_{1}clientz
_client.pyz_messages.pyz__init__.py)r   TempDirr   mainr   r   formatsetassertEqualr   listdirr!   r   r   r	   )r   r   api_versionprefixtmp_dir_pathexpected_filesexpected_files          r   _CheckGeneratedFilesz%ClientGenCliTest._CheckGeneratedFiles+   s   C+-!\OO##w#Hfw.>?L ,33HkJ
 
 Vl*n,"$ %  ^SL1I-JK!/$$ 3 &-!9 :\= IJL "0# "!!s   C"D		Dc                 (    | j                  dd       y )Nbigqueryv2r1   r   s    r   testGenClient_BigqueryDocz*ClientGenCliTest.testGenClient_BigqueryDocD   s    !!*d3r   c                 (    | j                  dd       y )Ndnsv1r5   r6   s    r   testGenClient_DnsDocz%ClientGenCliTest.testGenClient_DnsDocG       !!%.r   c                 (    | j                  dd       y )Nfusiontablesr:   r5   r6   s    r   testGenClient_FusiontablesDocz.ClientGenCliTest.testGenClient_FusiontablesDocJ   s    !!.$7r   c                 (    | j                  dd       y )Niamr:   r5   r6   s    r   testGenClient_IamDocz%ClientGenCliTest.testGenClient_IamDocM   r<   r   c                 (    | j                  dd       y )Nservicemanagementr:   r5   r6   s    r   "testGenClient_ServicemanagementDocz3ClientGenCliTest.testGenClient_ServicemanagementDocP   s    !!"5t<r   c                 (    | j                  dd       y )Nstorager:   r5   r6   s    r   testGenClient_StorageDocz)ClientGenCliTest.testGenClient_StorageDocS   s    !!)T2r   N)__name__
__module____qualname__r!   r1   r7   r;   r?   rB   rE   rH    r   r   r   r   "   s+    :L24/8/=3r   r   )r   r   unittestsixapitools.genr   r   r   r   TestCaser   rL   r   r   <module>rQ      s6    
   
 # #P
23x(( 23r   