Skip to content

Commit 22a037e

Browse files
authored
fix: allow subset for shorter response (GoogleCloudPlatform#9819)
## Description Fixes test project has so many assets that test was too slow.
1 parent 1cf94ed commit 22a037e

File tree

2 files changed

+16
-3
lines changed

2 files changed

+16
-3
lines changed

asset/snippets/quickstart_exportassets.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818
import argparse
1919

2020

21-
def export_assets(project_id, dump_file_path):
21+
def export_assets(project_id, dump_file_path, content_type=None):
2222
# [START asset_quickstart_export_assets]
2323
from google.cloud import asset_v1
2424

@@ -29,8 +29,16 @@ def export_assets(project_id, dump_file_path):
2929
parent = "projects/{}".format(project_id)
3030
output_config = asset_v1.OutputConfig()
3131
output_config.gcs_destination.uri = dump_file_path
32+
request_options = {
33+
"parent": parent,
34+
"output_config": output_config
35+
}
36+
37+
if content_type is not None:
38+
request_options["content_type"] = content_type
39+
3240
response = client.export_assets(
33-
request={"parent": parent, "output_config": output_config}
41+
request=request_options
3442
)
3543
print(response.result())
3644
# [END asset_quickstart_export_assets]

asset/snippets/quickstart_exportassets_test.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,8 +66,13 @@ def dataset(bigquery_client):
6666

6767

6868
def test_export_assets(asset_bucket, dataset, capsys):
69+
content_type = asset_v1.ContentType.IAM_POLICY
6970
dump_file_path = "gs://{}/assets-dump.txt".format(asset_bucket)
70-
quickstart_exportassets.export_assets(PROJECT, dump_file_path)
71+
quickstart_exportassets.export_assets(
72+
PROJECT,
73+
dump_file_path,
74+
content_type=content_type
75+
)
7176
out, _ = capsys.readouterr()
7277
assert dump_file_path in out
7378

0 commit comments

Comments
 (0)