3
SษLj
ใ @ s| d Z dZy ddlZW n ek
r0 ddlZY nX ddlmZ ddlmZ ddlmZ G dd dej Z
e d krxej dS )
z6Tests for google.protobuf.internal.service_reflection.z petar@google.com (Petar Petrov)้ N)ฺ unittest_pb2)ฺservice_reflection)ฺservicec @ s e Zd Zdd Zdd ZdS )ฺ FooUnitTestc
s G dd dt j}G dd dt j}d _G dd 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 G d d
d
tj} | }d |_ |j|tj | jd |j jd|j d |_ |j |j d
|tj
| jd |j jd|j d S )Nc @ s e Zd Zdd ZdS )z/FooUnitTest.testService..MockRpcChannelc S s || _ || _|| _|| d S )N)ฺmethodฺ
controllerฺrequest)ฺselfr r r ฺresponseฺcallbackฉ r ๚-/usr/lib/python3.6/service_reflection_test.pyฺ
CallMethod3 s z:FooUnitTest.testService..MockRpcChannel.CallMethodN)ฺ__name__ฺ
__module__ฺ __qualname__r r r r r
ฺMockRpcChannel2 s r c @ s e Zd Zdd ZdS )z2FooUnitTest.testService..MockRpcControllerc S s
|| _ d S )N)ฺfailure_message)r ฺmsgr r r
ฺ SetFailed: s z.MockRpcController.SetFailedN)r r r r r r r r
ฺMockRpcController9 s r c @ s e Zd ZdS )z*FooUnitTest.testService..MyServiceN)r r r r r r r
ฺ MyService? s r c s
| _ d S )N)ฺcallback_response)r
)r r r
ฺ
MyCallbackD s z+FooUnitTest.testService..MyCallbackzMethod Foo not implemented.้ zMethod Bar not implemented.c @ s e Zd Zdd Zdd ZdS )z.FooUnitTest.testService..MyServiceImplc S s
d| _ d S )NT)ฺ
foo_called)r ฺrpc_controllerr ฺdoner r r
ฺFoo] s z2FooUnitTest.testService..MyServiceImpl.Fooc S s
d| _ d S )NT)ฺ
bar_called)r r r r r r r
ฺBar_ s z2FooUnitTest.testService..MyServiceImpl.BarN)r r r r r r r r r
ฺ
MyServiceImpl\ s r! T)r ฺ
RpcChannelZ
RpcControllerr r Z TestServicer Z
FooRequestฺ assertEqualr ฺ
GetDescriptorr ฺmethodsZ
BarRequestZ
assertTrueZGetRequestClassZGetResponseClassZ BarResponser r )
r r r r r r ฺchannelZsrvcZservice_descriptorr! r )r r
ฺ testService1 sF
zFooUnitTest.testServicec sจ G dd dt j}d _ fdd}| }tj|}d}d} jtjj |j |j||| j jtj j||j
j||j j|j j d |j
d S )Nc @ s e Zd Zdd ZdS )z3FooUnitTest.testServiceStub..MockRpcChannelc S s || _ || _|| _|| d S )N)r r r )r r r r Zresponse_classr r r r
r p s z>FooUnitTest.testServiceStub..MockRpcChannel.CallMethodN)r r r r r r r r
r o s r c s
| _ d S )N)r )r
)r r r
r y s z/FooUnitTest.testServiceStub..MyCallbackr r r )r r" r r ZTestService_Stubr# r$ r ZassertIsInstanceZ FooResponser r r% r )r r r r& Zstubr r r )r r
ฺtestServiceStubn s
zFooUnitTest.testServiceStubN)r r r r' r( r r r r
r / s =r ฺ__main__)
ฺ__doc__ฺ
__author__Z unittest2Zunittestฺ ImportErrorZgoogle.protobufr r r ZTestCaser r ฺmainr r r r
ฺ s _