def test_batch_annotate_images(self, mock_create_stub):
# Mock gRPC layer
grpc_stub = mock.Mock()
mock_create_stub.return_value = grpc_stub
client = image_annotator_client.ImageAnnotatorClient()
# Mock request
requests = []
# Mock response
expected_response = image_annotator_pb2.BatchAnnotateImagesResponse()
grpc_stub.BatchAnnotateImages.return_value = expected_response
response = client.batch_annotate_images(requests)
self.assertEqual(expected_response, response)
grpc_stub.BatchAnnotateImages.assert_called_once()
args, kwargs = grpc_stub.BatchAnnotateImages.call_args
self.assertEqual(len(args), 2)
self.assertEqual(len(kwargs), 1)
self.assertIn('metadata', kwargs)
actual_request = args[0]
expected_request = image_annotator_pb2.BatchAnnotateImagesRequest(
requests=requests)
self.assertEqual(expected_request, actual_request)
def test_batch_annotate_images_exception(self, mock_create_stub):
# Mock gRPC layer
grpc_stub = mock.Mock()
mock_create_stub.return_value = grpc_stub
client = image_annotator_client.ImageAnnotatorClient()
# Mock request
requests = []
# Mock exception response
grpc_stub.BatchAnnotateImages.side_effect = CustomException()
self.assertRaises(errors.GaxError, client.batch_annotate_images,
requests)
def __init__(self, client=None):
self._client = client
self._annotator_client = image_annotator_client.ImageAnnotatorClient(
credentials=client._credentials,
lib_name='gccl',
lib_version=__version__)
def __init__(self, client=None):
self._client = client
self._annotator_client = image_annotator_client.ImageAnnotatorClient()
