3 SÉLj ã @s|dZdZy ddlZWnek r0ddlZYnXddlmZddlmZddlmZGdd„dej ƒZ e d krxej ƒdS) z6Tests for google.protobuf.internal.service_reflection.zpetar@google.com (Petar Petrov)éN)Ú unittest_pb2)Úservice_reflection)Úservicec@seZdZdd„Zdd„ZdS)Ú FooUnitTestc s˜Gdd„dtjƒ}Gdd„dtjƒ}dˆ_Gdd„dtjƒ}dˆ_‡fdd„}|ƒ}|ƒ}|ƒ}|j|tjƒ|ƒˆjd |j ƒˆjdˆjƒd|_ tjj ƒ}|j |j d |tj ƒ|ƒˆj|j|j d ƒtj kƒˆj|j|j d ƒtjkƒˆjd |j ƒˆjdˆjƒGd d „d tjƒ} | ƒ}d|_ |j|tjƒ|ƒˆjd|j ƒˆjd|jƒd|_ |j |j d |tj ƒ|ƒˆjd|j ƒˆjd|jƒdS)Nc@seZdZdd„ZdS)z/FooUnitTest.testService..MockRpcChannelcSs||_||_||_||ƒdS)N)ÚmethodÚ controllerÚrequest)ÚselfrrrÚresponseÚcallback©r ú-/usr/lib/python3.6/service_reflection_test.pyÚ CallMethod3sz:FooUnitTest.testService..MockRpcChannel.CallMethodN)Ú__name__Ú __module__Ú __qualname__rr r r r ÚMockRpcChannel2src@seZdZdd„ZdS)z2FooUnitTest.testService..MockRpcControllercSs ||_dS)N)Úfailure_message)r Úmsgr r r Ú SetFailed:sz.MockRpcController.SetFailedN)rrrrr r r r ÚMockRpcController9src@s eZdZdS)z*FooUnitTest.testService..MyServiceN)rrrr r r r Ú MyService?srcs |ˆ_dS)N)Úcallback_response)r )r r r Ú MyCallbackDsz+FooUnitTest.testService..MyCallbackzMethod Foo not implemented.ézMethod Bar not implemented.c@seZdZdd„Zdd„ZdS)z.FooUnitTest.testService..MyServiceImplcSs d|_dS)NT)Ú foo_called)r Úrpc_controllerrÚdoner r r ÚFoo]sz2FooUnitTest.testService..MyServiceImpl.FoocSs d|_dS)NT)Ú bar_called)r rrrr r r ÚBar_sz2FooUnitTest.testService..MyServiceImpl.BarN)rrrrr r r r r Ú MyServiceImpl\sr!T)rÚ RpcChannelZ RpcControllerrrZ TestServicerZ FooRequestÚ assertEqualrÚ GetDescriptorrÚmethodsZ BarRequestZ assertTrueZGetRequestClassZGetResponseClassZ BarResponserr) r rrrrrÚchannelZsrvcZservice_descriptorr!r )r r Ú testService1sF      zFooUnitTest.testServicecs¨Gdd„dtjƒ}dˆ_‡fdd„}|ƒ}tj|ƒ}d}d}ˆjtjjƒ|jƒƒ|j|||ƒˆjˆjtj ƒˆj||j ƒˆj||j ƒˆj|jƒj d|j ƒdS)Nc@seZdZdd„ZdS)z3FooUnitTest.testServiceStub..MockRpcChannelcSs ||_||_||_||ƒƒdS)N)rrr)r rrrZresponse_classr r r r rpsz>FooUnitTest.testServiceStub..MockRpcChannel.CallMethodN)rrrrr r r r rosrcs |ˆ_dS)N)r)r )r r r rysz/FooUnitTest.testServiceStub..MyCallbackrrr)rr"rrZTestService_Stubr#r$rZassertIsInstanceZ FooResponserrr%r)r rrr&Zstubrrr )r r ÚtestServiceStubns    zFooUnitTest.testServiceStubN)rrrr'r(r r r r r/s=rÚ__main__) Ú__doc__Ú __author__Z unittest2ZunittestÚ ImportErrorZgoogle.protobufrrrZTestCaserrÚmainr r r r Ú s    _