From d93986e0259952257f2571f60719b52099c29c0c Mon Sep 17 00:00:00 2001
From: Tim Swast <swast@google.com>
Date: Mon, 22 Mar 2021 03:54:39 -0500
Subject: [PATCH] feat: add `ExternalConfig.connection_id` property to connect
 to external sources (#560)

* feat: add `ExternalConfig.connection_id` property to connect to external sources

* add tests

* fix unit tests
---
 google/cloud/bigquery/external_config.py | 17 +++++++++++++++++
 tests/unit/test_external_config.py       |  8 ++++++++
 2 files changed, 25 insertions(+)

diff --git a/google/cloud/bigquery/external_config.py b/google/cloud/bigquery/external_config.py
index 112dfdba4..59e4960f9 100644
--- a/google/cloud/bigquery/external_config.py
+++ b/google/cloud/bigquery/external_config.py
@@ -760,6 +760,23 @@ def schema(self):
         prop = self._properties.get("schema", {})
         return [SchemaField.from_api_repr(field) for field in prop.get("fields", [])]
 
+    @property
+    def connection_id(self):
+        """Optional[str]: [Experimental] ID of a BigQuery Connection API
+        resource.
+
+        .. WARNING::
+
+           This feature is experimental. Pre-GA features may have limited
+           support, and changes to pre-GA features may not be compatible with
+           other pre-GA versions.
+        """
+        return self._properties.get("connectionId")
+
+    @connection_id.setter
+    def connection_id(self, value):
+        self._properties["connectionId"] = value
+
     @schema.setter
     def schema(self, value):
         prop = value
diff --git a/tests/unit/test_external_config.py b/tests/unit/test_external_config.py
index 4ca2e9012..648a8717e 100644
--- a/tests/unit/test_external_config.py
+++ b/tests/unit/test_external_config.py
@@ -74,6 +74,7 @@ def test_to_api_repr_base(self):
         ec.autodetect = True
         ec.ignore_unknown_values = False
         ec.compression = "compression"
+        ec.connection_id = "path/to/connection"
         ec.schema = [schema.SchemaField("full_name", "STRING", mode="REQUIRED")]
 
         exp_schema = {
@@ -87,10 +88,17 @@ def test_to_api_repr_base(self):
             "autodetect": True,
             "ignoreUnknownValues": False,
             "compression": "compression",
+            "connectionId": "path/to/connection",
             "schema": exp_schema,
         }
         self.assertEqual(got_resource, exp_resource)
 
+    def test_connection_id(self):
+        ec = external_config.ExternalConfig("")
+        self.assertIsNone(ec.connection_id)
+        ec.connection_id = "path/to/connection"
+        self.assertEqual(ec.connection_id, "path/to/connection")
+
     def test_schema_None(self):
         ec = external_config.ExternalConfig("")
         ec.schema = None