From e0443988668ce62c58f30c785644a33f4ffb2430 Mon Sep 17 00:00:00 2001
From: Stuart Ferguson <stuart_ferguson1@btinternet.com>
Date: Mon, 6 Sep 2021 18:31:29 +0100
Subject: [PATCH] Add retry on balance check

---
 .../Shared/SharedSteps.cs                          | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/TransactionProcessorACL.IntegrationTests/Shared/SharedSteps.cs b/TransactionProcessorACL.IntegrationTests/Shared/SharedSteps.cs
index e9ed4c1..2a6abbb 100644
--- a/TransactionProcessorACL.IntegrationTests/Shared/SharedSteps.cs
+++ b/TransactionProcessorACL.IntegrationTests/Shared/SharedSteps.cs
@@ -262,10 +262,16 @@ public async Task GivenIMakeTheFollowingManualMerchantDeposits(Table table)
                 this.TestingContext.Logger.LogInformation($"Deposit Reference {makeMerchantDepositRequest.Reference} made for Merchant {merchantName}");
 
                 // Check the merchant balance
-                MerchantBalanceResponse currentMerchantBalance =
-                    await this.TestingContext.DockerHelper.EstateClient.GetMerchantBalance(token, estateDetails.EstateId, merchantId, CancellationToken.None);
-
-                currentMerchantBalance.AvailableBalance.ShouldBe(previousMerchantBalance.AvailableBalance + makeMerchantDepositRequest.Amount);
+                await Retry.For(async () =>
+                                {
+                                    MerchantBalanceResponse currentMerchantBalance =
+                                        await this.TestingContext.DockerHelper.EstateClient.GetMerchantBalance(token,
+                                                                                                               estateDetails.EstateId,
+                                                                                                               merchantId,
+                                                                                                               CancellationToken.None);
+
+                                    currentMerchantBalance.AvailableBalance.ShouldBe(previousMerchantBalance.AvailableBalance + makeMerchantDepositRequest.Amount);
+                                });
             }
         }