add retry to poll result

pangeacyber · Jan 12, 2024 · ccbf2ff · ccbf2ff
1 parent ae5c4f1
commit ccbf2ff
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/pangea-sdk/v3/service/file_scan/integration_test.go b/pangea-sdk/v3/service/file_scan/integration_test.go
@@ -172,10 +172,20 @@ func Test_Integration_FileScan_NoRetry_reversinglabs(t *testing.T) {
 	assert.Nil(t, resp)
 	ae := err.(*pangea.AcceptedError)
 
-	// Wait until result should be ready
-	time.Sleep(time.Duration(40 * time.Second))
+	var pr *pangea.PangeaResponse[any]
+	i := 0
+
+	for i < 24 {
+		// Wait until result should be ready
+		time.Sleep(time.Duration(10 * time.Second))
+
+		pr, err = client.PollResultByError(ctx, *ae)
+		if err == nil {
+			break
+		}
+		i++
+	}
 
-	pr, err := client.PollResultByError(ctx, *ae)
 	assert.NoError(t, err)
 	assert.NotNil(t, pr)
 	assert.NotNil(t, pr.Result)