def test_update_feed(self): # Setup Expected Response name = "name3373707" expected_response = {"name": name} expected_response = asset_service_pb2.Feed(**expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = asset_v1p2beta1.AssetServiceClient() # Setup Request feed = {} update_mask = {} response = client.update_feed(feed, update_mask) assert expected_response == response assert len(channel.requests) == 1 expected_request = asset_service_pb2.UpdateFeedRequest( feed=feed, update_mask=update_mask) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_create_feed(self): # Setup Expected Response name = "name3373707" expected_response = {"name": name} expected_response = asset_service_pb2.Feed(**expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = asset_v1p2beta1.AssetServiceClient() # Setup Request parent = "parent-995424086" feed_id = "feedId-976011428" feed = {} response = client.create_feed(parent, feed_id, feed) assert expected_response == response assert len(channel.requests) == 1 expected_request = asset_service_pb2.CreateFeedRequest(parent=parent, feed_id=feed_id, feed=feed) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_batch_get_assets_history(self): # Setup Expected Response expected_response = {} expected_response = asset_service_pb2.BatchGetAssetsHistoryResponse( **expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = asset_v1p2beta1.AssetServiceClient() # Setup Request parent = "parent-995424086" asset_names = [] content_type = enums.ContentType.CONTENT_TYPE_UNSPECIFIED response = client.batch_get_assets_history(parent, asset_names, content_type) assert expected_response == response assert len(channel.requests) == 1 expected_request = asset_service_pb2.BatchGetAssetsHistoryRequest( parent=parent, asset_names=asset_names, content_type=content_type) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_export_assets(self): # Setup Expected Response expected_response = {} expected_response = asset_service_pb2.ExportAssetsResponse( **expected_response) operation = operations_pb2.Operation( name="operations/test_export_assets", done=True) operation.response.Pack(expected_response) # Mock the API response channel = ChannelStub(responses=[operation]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = asset_v1p2beta1.AssetServiceClient() # Setup Request parent = "parent-995424086" output_config = {} response = client.export_assets(parent, output_config) result = response.result() assert expected_response == result assert len(channel.requests) == 1 expected_request = asset_service_pb2.ExportAssetsRequest( parent=parent, output_config=output_config) actual_request = channel.requests[0][1] assert expected_request == actual_request
def sample_delete_feed(): # Create a client client = asset_v1p2beta1.AssetServiceClient() # Initialize request argument(s) request = asset_v1p2beta1.DeleteFeedRequest(name="name_value", ) # Make the request client.delete_feed(request=request)
def list_feeds(parent_resource): # [START asset_quickstart_list_feeds] from google.cloud import asset_v1p2beta1 # TODO parent_resource = 'Parent resource you want to list all feeds' client = asset_v1p2beta1.AssetServiceClient() response = client.list_feeds(parent_resource) print('feeds: {}'.format(response.feeds))
def delete_feed(feed_name): # [START asset_quickstart_delete_feed] from google.cloud import asset_v1p2beta1 # TODO feed_name = 'Feed name you want to delete' client = asset_v1p2beta1.AssetServiceClient() client.delete_feed(feed_name) print('deleted_feed')
def get_feed(feed_name): # [START asset_quickstart_get_feed] from google.cloud import asset_v1p2beta1 # TODO feed_name = 'Feed Name you want to get' client = asset_v1p2beta1.AssetServiceClient() response = client.get_feed(feed_name) print('gotten_feed: {}'.format(response))
def sample_get_feed(): # Create a client client = asset_v1p2beta1.AssetServiceClient() # Initialize request argument(s) request = asset_v1p2beta1.GetFeedRequest(name="name_value", ) # Make the request response = client.get_feed(request=request) # Handle the response print(response)
def sample_list_feeds(): # Create a client client = asset_v1p2beta1.AssetServiceClient() # Initialize request argument(s) request = asset_v1p2beta1.ListFeedsRequest(parent="parent_value", ) # Make the request response = client.list_feeds(request=request) # Handle the response print(response)
def test_delete_feed_exception(self): # Mock the API response channel = ChannelStub(responses=[CustomException()]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = asset_v1p2beta1.AssetServiceClient() # Setup request name = client.feed_path("[PROJECT]", "[FEED]") with pytest.raises(CustomException): client.delete_feed(name)
def test_list_feeds_exception(self): # Mock the API response channel = ChannelStub(responses=[CustomException()]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = asset_v1p2beta1.AssetServiceClient() # Setup request parent = "parent-995424086" with pytest.raises(CustomException): client.list_feeds(parent)
def test_update_feed_exception(self): # Mock the API response channel = ChannelStub(responses=[CustomException()]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = asset_v1p2beta1.AssetServiceClient() # Setup request feed = {} update_mask = {} with pytest.raises(CustomException): client.update_feed(feed, update_mask)
def sample_update_feed(): # Create a client client = asset_v1p2beta1.AssetServiceClient() # Initialize request argument(s) feed = asset_v1p2beta1.Feed() feed.name = "name_value" request = asset_v1p2beta1.UpdateFeedRequest(feed=feed, ) # Make the request response = client.update_feed(request=request) # Handle the response print(response)
def test_batch_get_assets_history_exception(self): # Mock the API response channel = ChannelStub(responses=[CustomException()]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = asset_v1p2beta1.AssetServiceClient() # Setup request parent = "parent-995424086" asset_names = [] content_type = enums.ContentType.CONTENT_TYPE_UNSPECIFIED with pytest.raises(CustomException): client.batch_get_assets_history(parent, asset_names, content_type)
def test_delete_feed(self): channel = ChannelStub() patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = asset_v1p2beta1.AssetServiceClient() # Setup Request name = client.feed_path("[PROJECT]", "[FEED]") client.delete_feed(name) assert len(channel.requests) == 1 expected_request = asset_service_pb2.DeleteFeedRequest(name=name) actual_request = channel.requests[0][1] assert expected_request == actual_request
def create_feed(project_id, feed_id, asset_names, topic): # [START asset_quickstart_create_feed] from google.cloud import asset_v1p2beta1 from google.cloud.asset_v1p2beta1.proto import asset_service_pb2 # TODO project_id = 'Your Google Cloud Project ID' # TODO feed_id = 'Feed ID you want to create' # TODO asset_names = 'List of asset names the feed listen to' # TODO topic = "Topic name of the feed" client = asset_v1p2beta1.AssetServiceClient() parent = "projects/{}".format(project_id) feed = asset_service_pb2.Feed() feed.asset_names.extend(asset_names) feed.feed_output_config.pubsub_destination.topic = topic response = client.create_feed(parent, feed_id, feed) print('feed: {}'.format(response))
def update_feed(feed_name, topic): # [START asset_quickstart_update_feed] from google.cloud import asset_v1p2beta1 from google.cloud.asset_v1p2beta1.proto import asset_service_pb2 from google.protobuf import field_mask_pb2 # TODO feed_name = 'Feed Name you want to update' # TODO topic = "Topic name you want to update with" client = asset_v1p2beta1.AssetServiceClient() feed = asset_service_pb2.Feed() feed.name = feed_name feed.feed_output_config.pubsub_destination.topic = topic update_mask = field_mask_pb2.FieldMask() # In this example, we update topic of the feed update_mask.paths.append("feed_output_config.pubsub_destination.topic") response = client.update_feed(feed, update_mask) print('updated_feed: {}'.format(response))
def test_export_assets_exception(self): # Setup Response error = status_pb2.Status() operation = operations_pb2.Operation( name="operations/test_export_assets_exception", done=True) operation.error.CopyFrom(error) # Mock the API response channel = ChannelStub(responses=[operation]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = asset_v1p2beta1.AssetServiceClient() # Setup Request parent = "parent-995424086" output_config = {} response = client.export_assets(parent, output_config) exception = response.exception() assert exception.errors[0] == error
def test_get_feed(self): # Setup Expected Response name_2 = "name2-1052831874" expected_response = {"name": name_2} expected_response = asset_service_pb2.Feed(**expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = asset_v1p2beta1.AssetServiceClient() # Setup Request name = client.feed_path("[PROJECT]", "[FEED]") response = client.get_feed(name) assert expected_response == response assert len(channel.requests) == 1 expected_request = asset_service_pb2.GetFeedRequest(name=name) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_list_feeds(self): # Setup Expected Response expected_response = {} expected_response = asset_service_pb2.ListFeedsResponse( **expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = asset_v1p2beta1.AssetServiceClient() # Setup Request parent = "parent-995424086" response = client.list_feeds(parent) assert expected_response == response assert len(channel.requests) == 1 expected_request = asset_service_pb2.ListFeedsRequest(parent=parent) actual_request = channel.requests[0][1] assert expected_request == actual_request