updates tutorial to add datasets to to_delete before running (GoogleCloudPlatform#1579)

alixhami · web-flow · commit f83bb369fe8c · 2018-07-17T16:22:27.000-07:00
diff --git a/bigquery/cloud-client/authorized_view_tutorial.py b/bigquery/cloud-client/authorized_view_tutorial.py
@@ -101,7 +101,6 @@ def run_authorized_view_tutorial():
         source_dataset, ['access_entries'])  # API request
     # [END bigquery_avt_source_dataset_access]
     # [END bigquery_authorized_view_tutorial]
-    return (source_dataset, shared_dataset)
 
 
 if __name__ == '__main__':
diff --git a/bigquery/cloud-client/authorized_view_tutorial_test.py b/bigquery/cloud-client/authorized_view_tutorial_test.py
@@ -37,10 +37,14 @@ def to_delete(client):
 
 
 def test_authorized_view_tutorial(client, to_delete):
-    source_dataset, shared_dataset = (
-        authorized_view_tutorial.run_authorized_view_tutorial())
-    to_delete.extend([source_dataset, shared_dataset])
+    source_dataset_ref = client.dataset('github_source_data')
+    shared_dataset_ref = client.dataset('shared_views')
+    to_delete.extend([source_dataset_ref, shared_dataset_ref])
 
+    authorized_view_tutorial.run_authorized_view_tutorial()
+
+    source_dataset = client.get_dataset(source_dataset_ref)
+    shared_dataset = client.get_dataset(shared_dataset_ref)
     analyst_email = 'example-analyst-group@google.com'
     analyst_entries = [entry for entry in shared_dataset.access_entries
                        if entry.entity_id == analyst_email]
