blueberry/grpc/blueberry_test_client.py

"""Blueberry Test Client.

Simple gRPC client to test the Blueberry Mock server.
"""

from absl import app
from absl import flags

import grpc

# Internal import
from blueberry.grpc.proto import blueberry_device_controller_pb2
from blueberry.grpc.proto import blueberry_device_controller_pb2_grpc

FLAGS = flags.FLAGS
flags.DEFINE_string('server', 'dns:///[::1]:10000', 'server address')


def _UpdateDiscoveryMode(stub, request):
  try:
    print('try SetDiscoverableMode')
    response = stub.SetDiscoverableMode(request)
    print('complete response')
    print(response)
    return 0
  except grpc.RpcError as rpc_error:
    print(rpc_error)
    return -1


def main(unused_argv):
  channel_creds = loas2.loas2_channel_credentials()
  with grpc.secure_channel(FLAGS.server, channel_creds) as channel:
    grpc.channel_ready_future(channel).result()
    stub = blueberry_device_controller_pb2_grpc.BlueberryDeviceControllerStub(
        channel)

    print('request grpc')
    request = blueberry_device_controller_pb2.DiscoverableMode(
        mode=True)
    print('Call _UpdateDiscoveryMode')
    return _UpdateDiscoveryMode(stub, request)


if __name__ == '__main__':
  app.run(main)