Skip to content
Snippets Groups Projects
Commit 16e7f582 authored by Ken Payson's avatar Ken Payson
Browse files

Updated exit_test for new grpc.Server API

This should have been part of a6b2a5a0
parent 7126163b
No related branches found
No related tags found
No related merge requests found
...@@ -184,11 +184,11 @@ if __name__ == '__main__': ...@@ -184,11 +184,11 @@ if __name__ == '__main__':
args = parser.parse_args() args = parser.parse_args()
if args.scenario == UNSTARTED_SERVER: if args.scenario == UNSTARTED_SERVER:
server = grpc.server((), DaemonPool()) server = grpc.server(DaemonPool())
if args.wait_for_interrupt: if args.wait_for_interrupt:
time.sleep(WAIT_TIME) time.sleep(WAIT_TIME)
elif args.scenario == RUNNING_SERVER: elif args.scenario == RUNNING_SERVER:
server = grpc.server((), DaemonPool()) server = grpc.server(DaemonPool())
port = server.add_insecure_port('[::]:0') port = server.add_insecure_port('[::]:0')
server.start() server.start()
if args.wait_for_interrupt: if args.wait_for_interrupt:
...@@ -203,7 +203,7 @@ if __name__ == '__main__': ...@@ -203,7 +203,7 @@ if __name__ == '__main__':
if args.wait_for_interrupt: if args.wait_for_interrupt:
time.sleep(WAIT_TIME) time.sleep(WAIT_TIME)
elif args.scenario == POLL_CONNECTIVITY: elif args.scenario == POLL_CONNECTIVITY:
server = grpc.server((), DaemonPool()) server = grpc.server(DaemonPool())
port = server.add_insecure_port('[::]:0') port = server.add_insecure_port('[::]:0')
server.start() server.start()
channel = grpc.insecure_channel('localhost:%d' % port) channel = grpc.insecure_channel('localhost:%d' % port)
...@@ -217,7 +217,7 @@ if __name__ == '__main__': ...@@ -217,7 +217,7 @@ if __name__ == '__main__':
else: else:
handler = GenericHandler() handler = GenericHandler()
server = grpc.server((), DaemonPool()) server = grpc.server(DaemonPool())
port = server.add_insecure_port('[::]:0') port = server.add_insecure_port('[::]:0')
server.add_generic_rpc_handlers((handler,)) server.add_generic_rpc_handlers((handler,))
server.start() server.start()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment