introduce another retry around an api request (#8571)

duffelhq · Aug 28, 2023 · 19a2545 · 19a2545
1 parent d94f714
commit 19a2545
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/airbyte-test-utils/src/main/java/io/airbyte/test/utils/AcceptanceTestHarness.java b/airbyte-test-utils/src/main/java/io/airbyte/test/utils/AcceptanceTestHarness.java
@@ -785,14 +785,16 @@ public SourceDefinitionRead createE2eSourceDefinition(final UUID workspaceId) {
     return sourceDefinitionRead;
   }
 
-  public DestinationDefinitionRead createE2eDestinationDefinition(final UUID workspaceId) throws ApiException {
-    return apiClient.getDestinationDefinitionApi().createCustomDestinationDefinition(new CustomDestinationDefinitionCreate()
-        .workspaceId(workspaceId)
-        .destinationDefinition(new DestinationDefinitionCreate()
-            .name("E2E Test Destination")
-            .dockerRepository("airbyte/destination-e2e-test")
-            .dockerImageTag(DESTINATION_E2E_TEST_CONNECTOR_VERSION)
-            .documentationUrl(URI.create("https://example.com"))));
+  public DestinationDefinitionRead createE2eDestinationDefinition(final UUID workspaceId) throws Exception {
+    return AirbyteApiClient.retryWithJitterThrows(() -> apiClient.getDestinationDefinitionApi()
+        .createCustomDestinationDefinition(new CustomDestinationDefinitionCreate()
+            .workspaceId(workspaceId)
+            .destinationDefinition(new DestinationDefinitionCreate()
+                .name("E2E Test Destination")
+                .dockerRepository("airbyte/destination-e2e-test")
+                .dockerImageTag(DESTINATION_E2E_TEST_CONNECTOR_VERSION)
+                .documentationUrl(URI.create("https://example.com")))),
+        "create destination definition", 10, 60, 3);
   }
 
   public SourceRead createPostgresSource() {