def test_get_set_dataset_iam_policy(capsys):
datasets.create_dataset(
service_account_json,
project_id,
cloud_region,
dataset_id)
get_response = datasets.get_dataset_iam_policy(
service_account_json,
project_id,
cloud_region,
dataset_id)
set_response = datasets.set_dataset_iam_policy(
service_account_json,
project_id,
cloud_region,
dataset_id,
'serviceAccount:[email protected]',
'roles/viewer')
# Clean up
datasets.delete_dataset(
service_account_json,
project_id,
cloud_region,
dataset_id)
out, _ = capsys.readouterr()
assert 'etag' in get_response
assert 'bindings' in set_response
assert len(set_response['bindings']) == 1
assert 'python-docs-samples-tests' in str(set_response['bindings'])
assert 'roles/viewer' in str(set_response['bindings'])
def test_get_set_dataset_iam_policy(capsys, test_dataset):
get_response = datasets.get_dataset_iam_policy(project_id, cloud_region, dataset_id)
set_response = datasets.set_dataset_iam_policy(
project_id,
cloud_region,
dataset_id,
"serviceAccount:[email protected]",
"roles/viewer",
)
out, _ = capsys.readouterr()
assert "etag" in get_response
assert "bindings" in set_response
assert len(set_response["bindings"]) == 1
assert "python-docs-samples-tests" in str(set_response["bindings"])
assert "roles/viewer" in str(set_response["bindings"])