9SrSSKrSSKrSSKrSSKJr SSKJr SSKJr SSKJ r "SS\ 5r S r "S S \ 5r "S S \ 5r"SS\5r"SS\5rSr"SS\5rg)aThe mock module allows easy mocking of apitools clients. This module allows you to mock out the constructor of a particular apitools client, for a specific API and version. Then, when the client is created, it will be run against an expected session that you define. This way code that is not aware of the testing framework can construct new clients as normal, as long as it's all done within the context of a mock. N)messages)base_api)encoding) exceptionsc\rSrSrSrSrg)Error$zExceptions for this module.N)__name__ __module__ __qualname____firstlineno____doc____static_attributes__r 0lib/third_party/apitools/base/py/testing/mock.pyrr$s%rrc[U[5(aI[U[5(a4[U5[U5:wag[S[ X555$[U[ R 5(a[U[ R 5(dX:H$UR5H?n[XR5n[XR5n[X45(aM? g g)aCompare two protorpc messages for equality. Using python's == operator does not work in all cases, specifically when there is a list involved. Args: msg1: protorpc.messages.Message or [protorpc.messages.Message] or number or string, One of the messages to compare. msg2: protorpc.messages.Message or [protorpc.messages.Message] or number or string, One of the messages to compare. Returns: If the messages are isomorphic. Fc3<# UHup[X5v M g7fN)_MessagesEqual).0xys r !_MessagesEqual..;sDODA>!''OsT) isinstancelistlenallziprMessage all_fieldsgetattrnamer)msg1msg2fieldfield1field2s rrr)s$*T4"8"8 t9D !DCODDD tX-- . .4!1!122|"zz*zz*f-- # rc(^\rSrSrU4SjrSrU=r$)UnexpectedRequestExceptionHc>Uup4UupV[R"USS9n[R"USS9nUR5n UR5n [R"X5n SR U 5n X5:wa5SR S5R UUUUS9n [[U]'U 5 gSR S5R UUUU S9n [[U]'U 5 g)NT multiline )z,expected: {expected_key}({expected_request})z,received: {received_key}({received_request})) expected_keyexpected_request received_keyreceived_request)zfor request to {key},zexpected: {expected_request}zreceived: {received_request}z diff: {diff}r1)keyr3r5diff) r MessageToRepr splitlinesdifflib unified_diffjoinformatsuperr+__init__)self received_call expected_callr2r3r4r5 expected_repr received_reprexpected_linesreceived_lines diff_linesr7msg __class__s rr?#UnexpectedRequestException.__init__Js )6& )6&  .. .  .. . '113&113)).I yy$  '))v)!.)!. 0  ,d SnUH+up4USRU[R"USS9S9- nM- [[U]U5 g)Nz expected: z{key}({request}) Tr.)r6request)r=rr8r>rOr?)r@expected_callsrHr6rRrIs rr?"ExpectedRequestsException.__init__usX,NS '.. ..w$G/I IC- '7>44>@ @ Z :3C3CDD4479 9#$rcUR$r)rYr@s rr6_ExpectedRequestResponse.keys zzrcUR$r)rZrbs rrR _ExpectedRequestResponse.requests ~~rc XR:wd*URU:Xd=[X R5(d#[X4URUR45eUR(a UReUR $)aValidate that key and request match expectations, and respond if so. Args: key: str, Actual key to compare against expectations. request: protorpc.messages.Message or [protorpc.messages.Message] or number or string, Actual request to compare againt expectations Raises: UnexpectedRequestException: If key or request dont match expectations. apitools_base.Error: If a non-None exception is specified to be thrown. Returns: The response that was specified to be returned. )rYrZrr+r]r\)r@r6rRs rValidateAndRespond+_ExpectedRequestResponse.ValidateAndRespondsl$ ** T^^w%>%3G^^%L%L,c^.2jj$..-IK K   "" "r) __exception__key __request __responseNN) r r r rrr?propertyr6rRrgrr rrrVrV~s7J%"rrVc8\rSrSrSrSrSrS SjrSrSr g) _MockedMethodzA mocked API service method.cHURUlXlX lX0lURUlUR 5n[ URR UR5Ul[ URR UR5Ul gr) r _MockedMethod__key_MockedMethod__mocked_client_MockedMethod__real_method method_configr#MESSAGES_MODULErequest_type_name_MockedMethod__request_typeresponse_type_name_MockedMethod__response_type)r@r6 mocked_client real_methodconfigs rr?_MockedMethod.__init__s#,,  ,((66##%%d&:&:&J&J&,&>&>@&t';';'K'K'-'@'@ Brc U(aSnURnOSnURn[X5(d:[R"SR X0R U[U555eg)aKEnsure the given message is of the expected type of this method. Args: msg: The message instance to check. is_request: True to validate against the expected request type, False to validate against the expected response type. Raises: exceptions.ConfigurationValueError: If the type of the message was not correct. rRr^z[Expected {} is not of the correct type for method [{}]. Required: [{}] Given: [{}]N)ryr{rrr[r=rstype)r@rH is_requestmode real_types r _TypeCheck_MockedMethod._TypeChecksh D++ID,,I#))44$$*F**ic%<= =*rNc U(a'URUSS9 U(aURUSS9 URRR[ UR UUUS95 g)aAdd an expectation on the mocked method. Exactly one of response and exception should be specified. Args: request: The request that should be expected response: The response that should be returned or None if exception is provided. exception: An exception that should be thrown, or None. enable_type_checking: When true, the message type of the request and response (if provided) will be checked against the types required by this method. T)rF)r^r_N)rrt_request_responsesappendrVrs)r@rRr^r_enable_type_checking unused_kwargss rExpect_MockedMethod.Expects]* OOGO 5U; //66 $TZZ%,.6/8 : ;rc |URR(a&URRRS5nO[URU4S5eUR URU5nUcCUR (a2UR U5n[[R"USSS95 U$U$)NrrmT)r/ shortstrings) rtrpopr+rsrgruprintrr8)r@rRrrequest_responser^s r__call___MockedMethod.__call__s    2 2#33FFJJ1M ,W%|5 5$66tzz7K   2 2))'2H (((Dt= >Or)rj__mocked_clientr __real_method__request_type__response_typerv)NNT) r r r rrr?rrrrr rrrprps#& B=48<$( ;Frrpc "SS[R5nUR5H;nSnU(a [XF5n[ UU[ US-U-S-U-UU55 M= U$)Nc\rSrSrSrg))_MakeMockedService..MockedServicei"r Nr r r rrr rr MockedServicer"s rr.)rBaseApiServiceGetMethodsListr#setattrrp)api_namecollection_name mock_clientservice real_servicermethodr}s r_MakeMockedServicer ss // ((* !,7K hn>DvM))+ , + rcD\rSrSrSrS SjrSrSrSrS Sjr S r S r g) Clienti1zMock an apitools client.NcU(dU"SS9nURUlXl0UlX l/UlSUlg)aMock an apitools API, given its class. Args: client_class: The class for the API. eg, if you from apis.sqladmin import v1beta3 then you can pass v1beta3.SqladminV1beta3 to this class and anything within its context will use your mocked version. real_client: apitools Client, The client to make requests against when the expected response is None. Fget_credentialsN)rI_Client__orig_class_Client__client_class_Client__real_service_classes_Client__real_clientr_Client__real_include_fields)r@ client_class real_clients rr?Client.__init__5sA&u=K NN*&(#("$%)"rc"UR5$r)Mockrbs r __enter__Client.__enter__Nsyy{rc NUR=(d URSS9n"SSURUR5nX l[UR5Hn[ URU5n[ U[ 5(dM0[U[R5(dMQX@RU'URnURR<SURR<3n[XeUUUR(aU"U5OS5n[URX75 [XU"U55 M URR UlUR URlUR$UlUR&UlU$)z/Stub out the client class with mocked services.Frc\rSrSrSrg)Client.Mock..PatchediVr Nrr rrPatchedrVs rr_N)rrrIdirr#rr issubclassrrr_NAME_PACKAGE _URL_VERSIONrr IncludeFieldsr_url_http)r@clientrr$ service_classrrmocked_service_classs rr Client.MockQs_###t':':!(;(# dnnd&9&9  ++,D#D$7$7>MmT22mX-D-DEE0= ' ' -+11O"&"5"5">">"&"5"5"B"BDH#54)-);); f%$G D'' D D+?+E F'-*&*%8%8%F%F",0,>,>)KK \\  rcfUSLnURUS9 U(a[R"XU5 g)N)suppressT)Unmocksixreraise)r@exc_typevalue tracebackis_active_exceptions r__exit__Client.__exit__xs0#4/ 0 1  KK 3rcURUlURR5H0up#[ UR X#5 [ XR5 M2 0UlU?U? URUR l SUl URVs/sHnURUR4PM nn/Ul U(a-U(d%[R "5Sc [#U5egggs snf)N)rrIritemsrrdelattrrrrrrrr6rRsysexc_inforO)r@rr$rrq_rsrequestss rr Client.Unmocks**#'#>#>#D#D#F D D'' = D-- .$G')# I J,0,F,F)%)""&!8!8:!8YY .!8 :"$ H):)B+H5 5*CH8 :s"C;c^UR(aURURU5$gr)rr)r@include_fieldss rrClient.IncludeFieldss.   --d.@.@.<> > r) rI__client_class __orig_class __real_client__real_include_fields__real_service_classesrrrr)F) r r r rrr?rrrrrrr rrrr1s$"*2%N6&>rr)rr:rrapitools.base.protorpcliterapitools.base.pyrrr Exceptionrrr+rOobjectrVrprrr rrrs|  /%%'&I& >(B(BV==8v8vdFdN"d>Vd>r