mirror of https://github.com/grpc/grpc.git
The C based gRPC (C++, Python, Ruby, Objective-C, PHP, C#)
https://grpc.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
81 lines
2.6 KiB
81 lines
2.6 KiB
# Copyright 2019 the gRPC authors. |
|
# |
|
# Licensed under the Apache License, Version 2.0 (the "License"); |
|
# you may not use this file except in compliance with the License. |
|
# You may obtain a copy of the License at |
|
# |
|
# http://www.apache.org/licenses/LICENSE-2.0 |
|
# |
|
# Unless required by applicable law or agreed to in writing, software |
|
# distributed under the License is distributed on an "AS IS" BASIS, |
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|
# See the License for the specific language governing permissions and |
|
# limitations under the License. |
|
"""The Python implementation of the GRPC helloworld.Greeter client.""" |
|
|
|
from concurrent import futures |
|
import contextlib |
|
import datetime |
|
import logging |
|
import unittest |
|
|
|
# TODO(https://github.com/grpc/grpc/issues/29284) |
|
# isort: off |
|
import grpc |
|
from google.protobuf import duration_pb2 |
|
from google.protobuf import timestamp_pb2 |
|
from google.cloud import helloworld_pb2 |
|
from google.cloud import helloworld_pb2_grpc |
|
|
|
# isort: on |
|
|
|
_HOST = "localhost" |
|
_SERVER_ADDRESS = "{}:0".format(_HOST) |
|
|
|
|
|
class Greeter(helloworld_pb2_grpc.GreeterServicer): |
|
def SayHello(self, request, context): |
|
request_in_flight = ( |
|
datetime.datetime.now() - request.request_initiation.ToDatetime() |
|
) |
|
request_duration = duration_pb2.Duration() |
|
request_duration.FromTimedelta(request_in_flight) |
|
return helloworld_pb2.HelloReply( |
|
message="Hello, %s!" % request.name, |
|
request_duration=request_duration, |
|
) |
|
|
|
|
|
@contextlib.contextmanager |
|
def _listening_server(): |
|
server = grpc.server(futures.ThreadPoolExecutor()) |
|
helloworld_pb2_grpc.add_GreeterServicer_to_server(Greeter(), server) |
|
port = server.add_insecure_port(_SERVER_ADDRESS) |
|
server.start() |
|
try: |
|
yield port |
|
finally: |
|
server.stop(0) |
|
|
|
|
|
class ImportTest(unittest.TestCase): |
|
def test_import(self): |
|
with _listening_server() as port: |
|
with grpc.insecure_channel("{}:{}".format(_HOST, port)) as channel: |
|
stub = helloworld_pb2_grpc.GreeterStub(channel) |
|
request_timestamp = timestamp_pb2.Timestamp() |
|
request_timestamp.GetCurrentTime() |
|
response = stub.SayHello( |
|
helloworld_pb2.HelloRequest( |
|
name="you", |
|
request_initiation=request_timestamp, |
|
), |
|
wait_for_ready=True, |
|
) |
|
self.assertEqual(response.message, "Hello, you!") |
|
self.assertGreater(response.request_duration.nanos, 0) |
|
|
|
|
|
if __name__ == "__main__": |
|
logging.basicConfig() |
|
unittest.main()
|
|
|