def UnUn(self, request, context):
if _application_common.UNARY_UNARY_REQUEST == request:
return _application_common.UNARY_UNARY_RESPONSE
else:
context.set_code(grpc.StatusCode.INVALID_ARGUMENT)
context.set_details('Something is wrong with your request!')
return services_pb2.Down()
def UnUn(self, request, context):
if request == _application_common.UNARY_UNARY_REQUEST:
return _application_common.UNARY_UNARY_RESPONSE
elif request == _application_common.ABORT_REQUEST:
with self._abort_lock:
try:
context.abort(grpc.StatusCode.PERMISSION_DENIED,
"Denying permission to test abort.")
except Exception as e: # pylint: disable=broad-except
self._abort_response = _application_common.ABORT_SUCCESS_RESPONSE
else:
self._abort_status = _application_common.ABORT_FAILURE_RESPONSE
return None # NOTE: For the linter.
elif request == _application_common.ABORT_SUCCESS_QUERY:
with self._abort_lock:
return self._abort_response
else:
context.set_code(grpc.StatusCode.INVALID_ARGUMENT)
context.set_details('Something is wrong with your request!')
return services_pb2.Down()
# See the License for the specific language governing permissions and # limitations under the License. """An example gRPC Python-using application's common code elements.""" from tests.testing.proto import requests_pb2 from tests.testing.proto import services_pb2 SERVICE_NAME = 'tests_of_grpc_testing.FirstService' UNARY_UNARY_METHOD_NAME = 'UnUn' UNARY_STREAM_METHOD_NAME = 'UnStre' STREAM_UNARY_METHOD_NAME = 'StreUn' STREAM_STREAM_METHOD_NAME = 'StreStre' UNARY_UNARY_REQUEST = requests_pb2.Up(first_up_field=2) ERRONEOUS_UNARY_UNARY_REQUEST = requests_pb2.Up(first_up_field=3) UNARY_UNARY_RESPONSE = services_pb2.Down(first_down_field=5) ERRONEOUS_UNARY_UNARY_RESPONSE = services_pb2.Down(first_down_field=7) UNARY_STREAM_REQUEST = requests_pb2.Charm(first_charm_field=11) STREAM_UNARY_REQUEST = requests_pb2.Charm(first_charm_field=13) STREAM_UNARY_RESPONSE = services_pb2.Strange(first_strange_field=17) STREAM_STREAM_REQUEST = requests_pb2.Top(first_top_field=19) STREAM_STREAM_RESPONSE = services_pb2.Bottom(first_bottom_field=23) TWO_STREAM_STREAM_RESPONSES = (STREAM_STREAM_RESPONSE, ) * 2 ABORT_REQUEST = requests_pb2.Up(first_up_field=42) ABORT_SUCCESS_QUERY = requests_pb2.Up(first_up_field=43) ABORT_NO_STATUS_RESPONSE = services_pb2.Down(first_down_field=50) ABORT_SUCCESS_RESPONSE = services_pb2.Down(first_down_field=51) ABORT_FAILURE_RESPONSE = services_pb2.Down(first_down_field=52) STREAM_STREAM_MUTATING_REQUEST = requests_pb2.Top(first_top_field=24601) STREAM_STREAM_MUTATING_COUNT = 2
