Merge pull request tylertreat#105 from ruxandraburtica/method-getting-project-ids

tylertreat · web-flow · commit af4902aea2a9 · 2016-07-10T14:44:24.000-05:00
Add method getting projects
diff --git a/bigquery/client.py b/bigquery/client.py
@@ -148,6 +148,20 @@ def get_client(project_id=None, credentials=None,
     return BigQueryClient(bq_service, project_id, swallow_results)
 
 
+def get_projects(bq_service):
+    """Given the BigQuery service, return data about all projects."""
+    projects_request = bq_service.projects().list().execute()
+
+    projects = []
+    for project in projects_request.get('projects', []):
+        project_data = {
+            'id': project['id'],
+            'name': project['friendlyName']
+        }
+        projects.append(project_data)
+    return projects
+
+
 def _get_bq_service(credentials=None, service_url=None):
     """Construct an authorized BigQuery service object."""
 
diff --git a/bigquery/tests/test_client.py b/bigquery/tests/test_client.py
@@ -214,6 +214,39 @@ def test_initialize_json_key_file_without_project_id(self, mock_open, mock_build
         self.assertEquals(json_key['project_id'], bq_client.project_id)
 
 
+class TestGetProjectIds(unittest.TestCase):
+
+    def test_get_project_ids(self):
+        mock_bq_service = mock.Mock()
+        mock_bq_service.projects().list().execute.return_value = {
+            'kind': 'bigquery#projectList',
+            'projects': [
+                {
+                    'friendlyName': 'Big Query Test',
+                    'id': 'big-query-test',
+                    'kind': 'bigquery#project',
+                    'numericId': '1435372465',
+                    'projectReference': {'projectId': 'big-query-test'}
+                },
+                {
+                    'friendlyName': 'BQ Company project',
+                    'id': 'bq-project',
+                    'kind': 'bigquery#project',
+                    'numericId': '4263574685796',
+                    'projectReference': {'projectId': 'bq-project'}
+                }
+            ],
+            'totalItems': 2
+        }
+
+        projects = client.get_projects(mock_bq_service)
+        expected_projects_data = [
+            {'id': 'big-query-test', 'name': 'Big Query Test'},
+            {'id': 'bq-project', 'name': 'BQ Company project'}
+        ]
+        self.assertEqual(projects, expected_projects_data)
+
+
 class TestQuery(unittest.TestCase):
 
     def setUp(self):
