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"])