add retry to address flakiness (#419)

Co-authored-by: Bryan Apellanes <[email protected]>
okta · Aug 20, 2020 · c53be9c · c53be9c
1 parent 3a8cf61
commit c53be9c
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/Okta.Sdk.IntegrationTests/UserScenarios.cs b/src/Okta.Sdk.IntegrationTests/UserScenarios.cs
@@ -785,9 +785,15 @@ public async Task ForgotPasswordGenerateOneTimeToken()
 
             try
             {
-                var forgotPasswordResponse = await createdUser.ForgotPasswordGenerateOneTimeTokenAsync(false);
-                forgotPasswordResponse.Should().NotBeNull();
-                forgotPasswordResponse.ResetPasswordUrl.Should().NotBeNullOrEmpty();
+                var policy = Polly.Policy
+                    .Handle<Exception>()
+                    .WaitAndRetryAsync(4, attemptNumber => TimeSpan.FromSeconds(Math.Pow(5, attemptNumber - 1)))
+                    .ExecuteAsync(async () =>
+                    {
+                        var forgotPasswordResponse = await createdUser.ForgotPasswordGenerateOneTimeTokenAsync(false);
+                        forgotPasswordResponse.Should().NotBeNull();
+                        forgotPasswordResponse.ResetPasswordUrl.Should().NotBeNullOrEmpty();
+                    });
             }
             finally
             {