def _stub(args): if args.oauth_scope: metadata_transformer = lambda x: [('Authorization', 'Bearer %s' % _oauth_access_token(args))] else: metadata_transformer = lambda x: [] if args.use_tls: if args.use_test_ca: root_certificates = resources.test_root_certificates() else: root_certificates = resources.prod_root_certificates() stub = implementations.stub( methods.SERVICE_NAME, methods.CLIENT_METHODS, args.server_host, args.server_port, metadata_transformer=metadata_transformer, secure=True, root_certificates=root_certificates, server_host_override=args.server_host_override) else: stub = implementations.stub(methods.SERVICE_NAME, methods.CLIENT_METHODS, args.server_host, args.server_port, secure=False) return stub
def _stub(args): if args.oauth_scope: if args.test_case == 'oauth2_auth_token': access_token = _oauth_access_token(args) metadata_transformer = lambda x: [ ('Authorization', 'Bearer %s' % access_token)] else: metadata_transformer = lambda x: [ ('Authorization', 'Bearer %s' % _oauth_access_token(args))] else: metadata_transformer = lambda x: [] if args.use_tls: if args.use_test_ca: root_certificates = resources.test_root_certificates() else: root_certificates = resources.prod_root_certificates() stub = implementations.stub( methods.SERVICE_NAME, methods.CLIENT_METHODS, args.server_host, args.server_port, metadata_transformer=metadata_transformer, secure=True, root_certificates=root_certificates, server_host_override=args.server_host_override) else: stub = implementations.stub( methods.SERVICE_NAME, methods.CLIENT_METHODS, args.server_host, args.server_port, secure=False) return stub
def _stub(args): if args.oauth_scope: if args.test_case == 'oauth2_auth_token': # TODO(jtattermusch): This testcase sets the auth metadata key-value # manually, which also means that the user would need to do the same # thing every time he/she would like to use and out of band oauth token. # The transformer function that produces the metadata key-value from # the access token should be provided by gRPC auth library. access_token = _oauth_access_token(args) metadata_transformer = lambda x: [ ('authorization', 'Bearer %s' % access_token)] else: metadata_transformer = lambda x: [ ('authorization', 'Bearer %s' % _oauth_access_token(args))] else: metadata_transformer = lambda x: [] if args.use_tls: if args.use_test_ca: root_certificates = resources.test_root_certificates() else: root_certificates = resources.prod_root_certificates() channel = test_utilities.not_really_secure_channel( args.server_host, args.server_port, implementations.ssl_client_credentials(root_certificates, None, None), args.server_host_override) stub = test_pb2.beta_create_TestService_stub( channel, metadata_transformer=metadata_transformer) else: channel = implementations.insecure_channel( args.server_host, args.server_port) stub = test_pb2.beta_create_TestService_stub(channel) return stub
def _stub(args): if args.oauth_scope: if args.test_case == 'oauth2_auth_token': # TODO(jtattermusch): This testcase sets the auth metadata key-value # manually, which also means that the user would need to do the same # thing every time he/she would like to use and out of band oauth token. # The transformer function that produces the metadata key-value from # the access token should be provided by gRPC auth library. access_token = _oauth_access_token(args) metadata_transformer = lambda x: [('authorization', 'Bearer %s' % access_token)] else: metadata_transformer = lambda x: [('authorization', 'Bearer %s' % _oauth_access_token(args))] else: metadata_transformer = lambda x: [] if args.use_tls: if args.use_test_ca: root_certificates = resources.test_root_certificates() else: root_certificates = resources.prod_root_certificates() channel = test_utilities.not_really_secure_channel( args.server_host, args.server_port, implementations.ssl_client_credentials(root_certificates, None, None), args.server_host_override) stub = test_pb2.beta_create_TestService_stub( channel, metadata_transformer=metadata_transformer) else: channel = implementations.insecure_channel(args.server_host, args.server_port) stub = test_pb2.beta_create_TestService_stub(channel) return stub
def _stub(args): if args.oauth_scope: if args.test_case == 'oauth2_auth_token': # TODO(jtattermusch): This testcase sets the auth metadata key-value # manually, which also means that the user would need to do the same # thing every time he/she would like to use and out of band oauth token. # The transformer function that produces the metadata key-value from # the access token should be provided by gRPC auth library. access_token = _oauth_access_token(args) metadata_transformer = lambda x: [ ('authorization', 'Bearer %s' % access_token)] else: metadata_transformer = lambda x: [ ('authorization', 'Bearer %s' % _oauth_access_token(args))] else: metadata_transformer = lambda x: [] if args.use_tls: if args.use_test_ca: root_certificates = resources.test_root_certificates() else: root_certificates = resources.prod_root_certificates() stub = implementations.stub( methods.SERVICE_NAME, methods.CLIENT_METHODS, args.server_host, args.server_port, metadata_transformer=metadata_transformer, secure=True, root_certificates=root_certificates, server_host_override=args.server_host_override) else: stub = implementations.stub( methods.SERVICE_NAME, methods.CLIENT_METHODS, args.server_host, args.server_port, secure=False) return stub
def setUp(self): self.server = implementations.server( methods.SERVICE_NAME, methods.SERVER_METHODS, 0, private_key=resources.private_key(), certificate_chain=resources.certificate_chain()) self.server.start() port = self.server.port() self.stub = implementations.stub( methods.SERVICE_NAME, methods.CLIENT_METHODS, 'localhost', port, secure=True, root_certificates=resources.test_root_certificates(), server_host_override=_SERVER_HOST_OVERRIDE)
def setUp(self): self.server = test_pb2.beta_create_TestService_server(methods.TestService()) port = self.server.add_secure_port( '[::]:0', implementations.ssl_server_credentials( [(resources.private_key(), resources.certificate_chain())])) self.server.start() self.stub = test_pb2.beta_create_TestService_stub( test_utilities.not_really_secure_channel( '[::]', port, implementations.ssl_client_credentials( resources.test_root_certificates(), None, None), _SERVER_HOST_OVERRIDE))