From d2af1b9464b151f0013179a06baae21307b0e318 Mon Sep 17 00:00:00 2001
From: LukeFZ <17146677+LukeFZ@users.noreply.github.com>
Date: Sat, 12 Aug 2023 21:27:24 +0200
Subject: [PATCH] fix tests 2

---
 DragaliaAPI.Test/Services/DragonServiceTest.cs | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/DragaliaAPI.Test/Services/DragonServiceTest.cs b/DragaliaAPI.Test/Services/DragonServiceTest.cs
index 5fd7e118c..6f4981353 100644
--- a/DragaliaAPI.Test/Services/DragonServiceTest.cs
+++ b/DragaliaAPI.Test/Services/DragonServiceTest.cs
@@ -94,7 +94,13 @@ out List<DbPlayerStoryState> stories
         dragonRels.Add(DbPlayerDragonReliabilityFactory.Create(DeviceAccountId, Dragons.Garuda));
 
         mockMissionProgressionService.Setup(
-            x => x.OnDragonBondLevelUp(Dragons.Garuda, UnitElement.Wind, 3, 4)
+            x =>
+                x.OnDragonBondLevelUp(
+                    Dragons.Garuda,
+                    UnitElement.Wind,
+                    It.IsIn(3, 6),
+                    It.IsIn(4, 10)
+                )
         );
 
         mockMissionProgressionService.Setup(
@@ -235,7 +241,9 @@ out List<DbPlayerStoryState> stories
             x => x.OnDragonBondLevelUp(dragon, element, expectedLvl - 1, expectedLvl)
         );
 
-        mockMissionProgressionService.Setup(x => x.OnDragonGiftSent(dragon, gift, element, 1, 0));
+        mockMissionProgressionService.Setup(
+            x => x.OnDragonGiftSent(dragon, gift, element, usedQuantity, 0)
+        );
 
         DragonSendGiftMultipleData responseData = await dragonService.DoDragonSendGiftMultiple(
             new DragonSendGiftMultipleRequest()