From 0758aa4f81c33c7f21fbdb8eb1b4d18de80cc743 Mon Sep 17 00:00:00 2001
From: Tres Seaver <tseaver@palladion.com>
Date: Thu, 28 Jan 2016 15:15:21 -0500
Subject: [PATCH] Defend against missing 'datasets' key in datasets/list
 response.

Fixes #1424.
---
 gcloud/bigquery/client.py      |  2 +-
 gcloud/bigquery/test_client.py | 26 +++-----------------------
 2 files changed, 4 insertions(+), 24 deletions(-)

diff --git a/gcloud/bigquery/client.py b/gcloud/bigquery/client.py
index 0927a03cacc7..f9110db429ec 100644
--- a/gcloud/bigquery/client.py
+++ b/gcloud/bigquery/client.py
@@ -88,7 +88,7 @@ def list_datasets(self, include_all=False, max_results=None,
         resp = self.connection.api_request(method='GET', path=path,
                                            query_params=params)
         datasets = [Dataset.from_api_repr(resource, self)
-                    for resource in resp['datasets']]
+                    for resource in resp.get('datasets', ())]
         return datasets, resp.get('nextPageToken')
 
     def dataset(self, dataset_name):
diff --git a/gcloud/bigquery/test_client.py b/gcloud/bigquery/test_client.py
index b9c0605bf634..c8a652a4df9b 100644
--- a/gcloud/bigquery/test_client.py
+++ b/gcloud/bigquery/test_client.py
@@ -74,27 +74,11 @@ def test_list_datasets_defaults(self):
         self.assertEqual(req['method'], 'GET')
         self.assertEqual(req['path'], '/%s' % PATH)
 
-    def test_list_datasets_explicit(self):
-        from gcloud.bigquery.dataset import Dataset
+    def test_list_datasets_explicit_response_missing_datasets_key(self):
         PROJECT = 'PROJECT'
-        DATASET_1 = 'dataset_one'
-        DATASET_2 = 'dataset_two'
         PATH = 'projects/%s/datasets' % PROJECT
         TOKEN = 'TOKEN'
-        DATA = {
-            'datasets': [
-                {'kind': 'bigquery#dataset',
-                 'id': '%s:%s' % (PROJECT, DATASET_1),
-                 'datasetReference': {'datasetId': DATASET_1,
-                                      'projectId': PROJECT},
-                 'friendlyName': None},
-                {'kind': 'bigquery#dataset',
-                 'id': '%s:%s' % (PROJECT, DATASET_2),
-                 'datasetReference': {'datasetId': DATASET_2,
-                                      'projectId': PROJECT},
-                 'friendlyName': 'Two'},
-            ]
-        }
+        DATA = {}
         creds = _Credentials()
         client = self._makeOne(PROJECT, creds)
         conn = client.connection = _Connection(DATA)
@@ -102,11 +86,7 @@ def test_list_datasets_explicit(self):
         datasets, token = client.list_datasets(
             include_all=True, max_results=3, page_token=TOKEN)
 
-        self.assertEqual(len(datasets), len(DATA['datasets']))
-        for found, expected in zip(datasets, DATA['datasets']):
-            self.assertTrue(isinstance(found, Dataset))
-            self.assertEqual(found.dataset_id, expected['id'])
-            self.assertEqual(found.friendly_name, expected['friendlyName'])
+        self.assertEqual(len(datasets), 0)
         self.assertEqual(token, None)
 
         self.assertEqual(len(conn._requested), 1)