From e7ac5b4bd7c6465d83cde2e4f6881ff952c05e1f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?No=C3=ABl=20Mugnier?= <noel.mugnier@outlook.com>
Date: Mon, 9 Nov 2020 22:40:33 +0100
Subject: [PATCH] Fix - Correction de l'update d'un transfer si c'est une
 donation

---
 .../Commands/TransferCommandsHandler.cs       | 77 ++++++++++++++-----
 1 file changed, 57 insertions(+), 20 deletions(-)

diff --git a/Sheaft.Application.Handlers/Commands/TransferCommandsHandler.cs b/Sheaft.Application.Handlers/Commands/TransferCommandsHandler.cs
index b946cdec7..874e4894b 100644
--- a/Sheaft.Application.Handlers/Commands/TransferCommandsHandler.cs
+++ b/Sheaft.Application.Handlers/Commands/TransferCommandsHandler.cs
@@ -85,28 +85,15 @@ public async Task<Result<TransactionStatus>> Handle(RefreshTransferStatusCommand
         {
             return await ExecuteAsync(request, async () =>
             {
-                var transfer = await _context.GetSingleAsync<Transfer>(c => c.Identifier == request.Identifier, token);
-                if (transfer.Status == TransactionStatus.Succeeded || transfer.Status == TransactionStatus.Failed)
-                    return Ok(transfer.Status);
+                var transfer = await _context.FindSingleAsync<Transfer>(c => c.Identifier == request.Identifier, token);
+                if(transfer != null)
+                    return await HandleTransferStatusAsync(request, transfer, token);
 
-                var pspResult = await _pspService.GetTransferAsync(transfer.Identifier, token);
-                if (!pspResult.Success)
-                    return Failed<TransactionStatus>(pspResult.Exception);
+                var donation = await _context.FindSingleAsync<Donation>(c => c.Identifier == request.Identifier, token);
+                if(donation != null)
+                    return await HandleDonationStatusAsync(request, donation, token);
 
-                transfer.SetStatus(pspResult.Data.Status);
-                transfer.SetResult(pspResult.Data.ResultCode, pspResult.Data.ResultMessage);
-                transfer.SetExecutedOn(pspResult.Data.ProcessedOn);
-
-                await _context.SaveChangesAsync(token);
-
-                switch (transfer.Status)
-                {
-                    case TransactionStatus.Failed:
-                        _mediatr.Post(new TransferFailedEvent(request.RequestUser) { TransferId = transfer.Id });
-                        break;
-                }
-
-                return Ok(transfer.Status);
+                return NotFound<TransactionStatus>();
             });
         }
 
@@ -155,6 +142,56 @@ public async Task<Result<Guid>> Handle(CreateTransferCommand request, Cancellati
             });
         }
 
+        private async Task<Result<TransactionStatus>> HandleTransferStatusAsync(RefreshTransferStatusCommand request, Transfer transfer, CancellationToken token)
+        {
+            if (transfer.Status == TransactionStatus.Succeeded || transfer.Status == TransactionStatus.Failed)
+                return Ok(transfer.Status);
+
+            var pspResult = await _pspService.GetTransferAsync(transfer.Identifier, token);
+            if (!pspResult.Success)
+                return Failed<TransactionStatus>(pspResult.Exception);
+
+            transfer.SetStatus(pspResult.Data.Status);
+            transfer.SetResult(pspResult.Data.ResultCode, pspResult.Data.ResultMessage);
+            transfer.SetExecutedOn(pspResult.Data.ProcessedOn);
+
+            await _context.SaveChangesAsync(token);
+
+            switch (transfer.Status)
+            {
+                case TransactionStatus.Failed:
+                    _mediatr.Post(new TransferFailedEvent(request.RequestUser) { TransferId = transfer.Id });
+                    break;
+            }
+
+            return Ok(transfer.Status);
+        }
+
+        private async Task<Result<TransactionStatus>> HandleDonationStatusAsync(RefreshTransferStatusCommand request, Donation donation, CancellationToken token)
+        {
+            if (donation.Status == TransactionStatus.Succeeded || donation.Status == TransactionStatus.Failed)
+                return Ok(donation.Status);
+
+            var pspResult = await _pspService.GetTransferAsync(donation.Identifier, token);
+            if (!pspResult.Success)
+                return Failed<TransactionStatus>(pspResult.Exception);
+
+            donation.SetStatus(pspResult.Data.Status);
+            donation.SetResult(pspResult.Data.ResultCode, pspResult.Data.ResultMessage);
+            donation.SetExecutedOn(pspResult.Data.ProcessedOn);
+
+            await _context.SaveChangesAsync(token);
+
+            switch (donation.Status)
+            {
+                case TransactionStatus.Failed:
+                    _mediatr.Post(new DonationFailedEvent(request.RequestUser) { DonationId = donation.Id });
+                    break;
+            }
+
+            return Ok(donation.Status);
+        }
+
         private async Task<IEnumerable<Guid>> GetNextTransferIdsAsync(int skip, int take, CancellationToken token)
         {
             return await _context.Transfers