Merge branch 'main' into 71-new-state-for-dbt-test-failures

DalgoT4D · Aug 26, 2023 · 4994a43 · 4994a43
2 parents be8eb89 + 385433a
commit 4994a43
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/proxy/service.py b/proxy/service.py
@@ -250,8 +250,7 @@ async def create_airbyte_connection_block(
         ) from exc
 
     connection_block = AirbyteConnection(
-        airbyte_server=serverblock,
-        connection_id=conninfo.connectionId,
+        airbyte_server=serverblock, connection_id=conninfo.connectionId, timeout=15
     )
     try:
         block_name_for_save = cleaned_name_for_prefectblock(

diff --git a/tests/test_service.py b/tests/test_service.py
@@ -487,9 +487,10 @@ def dict(self):
 
 
 class MockAirbyteConnection:
-    def __init__(self, airbyte_server, connection_id):
+    def __init__(self, airbyte_server, connection_id, timeout):
         self.airbyte_server = airbyte_server
         self.connection_id = connection_id
+        self.timeout = timeout
 
     async def save(self, block_name):
         if self.connection_id == "test_error_connection_id":