try waiting for connection state

airbytehq · Mar 30, 2022 · 6cb3181 · 6cb3181
1 parent 2e90a21
commit 6cb3181
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/airbyte-tests/src/acceptanceTests/java/io/airbyte/test/acceptance/AcceptanceTests.java b/airbyte-tests/src/acceptanceTests/java/io/airbyte/test/acceptance/AcceptanceTests.java
@@ -1153,11 +1153,15 @@ public void testCancelSyncWhenCancelledWhenWorkerIsNotRunning() throws Exception
   }
 
   @Test
-  @Order(22)
+  @Order(-3)
   @DisabledIfEnvironmentVariable(named = "KUBE",
                                  matches = "true")
   public void testActionsWhenTemporalIsInTerminalState() throws Exception {
-    final WorkflowServiceStubs temporalService = TemporalUtils.createTemporalService("localhost:7233");
+    String temporalHost = "localhost:7233";
+    if (!USE_EXTERNAL_DEPLOYMENT) {
+      temporalHost = "airbyte-temporal:7233";
+    }
+    final WorkflowServiceStubs temporalService = TemporalUtils.createTemporalService(temporalHost);
     final WorkflowClient workflowCLient = WorkflowClient.newInstance(temporalService);
 
     final String connectionName = "test-connection";
@@ -1172,8 +1176,11 @@ public void testActionsWhenTemporalIsInTerminalState() throws Exception {
         .cursorField(List.of(COLUMN_ID))
         .destinationSyncMode(destinationSyncMode)
         .primaryKey(List.of(List.of(COLUMN_NAME))));
+
     final UUID connectionId =
         createConnection(connectionName, sourceId, destinationId, List.of(operationId), catalog, null).getConnectionId();
+    waitForConnectionState(apiClient, connectionId);
+
     // check if temporal workflow is reachable
     final ConnectionManagerWorkflow connectionManagerWorkflow =
         workflowCLient.newWorkflowStub(ConnectionManagerWorkflow.class, "connection_manager_" + connectionId);