1# Copyright 2019 the gRPC authors. 2# 3# Licensed under the Apache License, Version 2.0 (the "License"); 4# you may not use this file except in compliance with the License. 5# You may obtain a copy of the License at 6# 7# http://www.apache.org/licenses/LICENSE-2.0 8# 9# Unless required by applicable law or agreed to in writing, software 10# distributed under the License is distributed on an "AS IS" BASIS, 11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12# See the License for the specific language governing permissions and 13# limitations under the License. 14"""The Python implementation of the GRPC helloworld.Greeter client.""" 15 16from concurrent import futures 17import contextlib 18import datetime 19import logging 20import unittest 21 22from google.protobuf import duration_pb2 23from google.protobuf import timestamp_pb2 24import grpc 25import helloworld_pb2 26import helloworld_pb2_grpc 27 28_HOST = 'localhost' 29_SERVER_ADDRESS = '{}:0'.format(_HOST) 30 31 32class Greeter(helloworld_pb2_grpc.GreeterServicer): 33 34 def SayHello(self, request, context): 35 request_in_flight = datetime.datetime.now() - \ 36 request.request_initiation.ToDatetime() 37 request_duration = duration_pb2.Duration() 38 request_duration.FromTimedelta(request_in_flight) 39 return helloworld_pb2.HelloReply( 40 message='Hello, %s!' % request.name, 41 request_duration=request_duration, 42 ) 43 44 45@contextlib.contextmanager 46def _listening_server(): 47 server = grpc.server(futures.ThreadPoolExecutor()) 48 helloworld_pb2_grpc.add_GreeterServicer_to_server(Greeter(), server) 49 port = server.add_insecure_port(_SERVER_ADDRESS) 50 server.start() 51 try: 52 yield port 53 finally: 54 server.stop(0) 55 56 57class ImportTest(unittest.TestCase): 58 59 def test_import(self): 60 with _listening_server() as port: 61 with grpc.insecure_channel('{}:{}'.format(_HOST, port)) as channel: 62 stub = helloworld_pb2_grpc.GreeterStub(channel) 63 request_timestamp = timestamp_pb2.Timestamp() 64 request_timestamp.GetCurrentTime() 65 response = stub.SayHello(helloworld_pb2.HelloRequest( 66 name='you', 67 request_initiation=request_timestamp, 68 ), 69 wait_for_ready=True) 70 self.assertEqual(response.message, "Hello, you!") 71 self.assertGreater(response.request_duration.nanos, 0) 72 73 74if __name__ == '__main__': 75 logging.basicConfig() 76 unittest.main() 77