Add tunnel field to splitgraph.yml external object

splitgraph · Aug 4, 2022 · c60b233 · c60b233
1 parent 9499135
commit c60b233
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 0 deletions.
diff --git a/splitgraph/cloud/project/models.py b/splitgraph/cloud/project/models.py
@@ -35,6 +35,7 @@ class External(BaseModel):
     tables: Dict[str, Table]
     is_live: bool = True
     schedule: Optional[IngestionSchedule]
+    tunnel: bool = False
 
 
 class Source(BaseModel):

diff --git a/splitgraph/commandline/cloud.py b/splitgraph/commandline/cloud.py
@@ -683,6 +683,7 @@ def load_c(
                     repository.external,
                     credential_map=credential_map,
                     initial_private=initial_private,
+                    tunnel=repository.external.tunnel,
                 )
                 external_repositories.append(external_repository)
         rest_client.bulk_upsert_external(

diff --git a/test/splitgraph/commandline/http_fixtures.py b/test/splitgraph/commandline/http_fixtures.py
@@ -423,6 +423,7 @@ def cb(request, uri, response_headers):
                 },
                 "schedule": None,
                 "initial_private": initial_private,
+                "tunnel": False,
             },
             {
                 "namespace": "someuser",
@@ -434,6 +435,7 @@ def cb(request, uri, response_headers):
                 "credential_id": "123e4567-e89b-12d3-a456-426655440000",
                 "schedule": None,
                 "initial_private": initial_private,
+                "tunnel": False,
             },
             {
                 "namespace": "someuser",
@@ -445,6 +447,7 @@ def cb(request, uri, response_headers):
                 "credential_id": "00000000-0000-0000-0000-000000000000",
                 "schedule": None,
                 "initial_private": initial_private,
+                "tunnel": False,
             },
         ]