fix unit test about Vampiric logging

planetarium · Dec 6, 2023 · 29cfc65 · 29cfc65
1 parent 6448d0c
commit 29cfc65
Showing 1 changed file with 10 additions and 9 deletions.
diff --git a/.Lib9c.Tests/Model/PlayerTest.cs b/.Lib9c.Tests/Model/PlayerTest.cs
@@ -629,15 +629,16 @@ public void Vampiric()
                 if (currLog is Nekoyume.Model.BattleStatus.NormalAttack)
                 {
                     var nextLog = logList[i + 1];
-
-                    Assert.True(nextLog is TickDamage);
-                }
-                else if (currLog is TickDamage)
-                {
-                    var nextLog = logList[i + 1];
-                    Assert.True(currLog.Character.CurrentHP > nextLog.Character.CurrentHP);
+                    if (currLog.Character.ActionBuffs.Any(actionBuff => actionBuff is Vampiric))
+                    {
+                        Assert.True(nextLog is Tick);
+                    }
+                    else
+                    {
+                        Assert.True(nextLog is TickDamage);
+                    }
                 }
-                else if (currLog is Nekoyume.Model.BattleStatus.HealSkill healSkill)
+                else if (currLog is Tick healSkill)
                 {
                     Assert.Equal(vampiric.RowData.Id, healSkill.SkillId);
                     var healInfo = healSkill.SkillInfos.First();
@@ -653,7 +654,7 @@ public void Vampiric()
             Assert.Contains(logList, e => e is Nekoyume.Model.BattleStatus.NormalAttack);
             Assert.Contains(logList, e => e is TickDamage);
             Assert.Contains(logList, e => e is RemoveBuffs);
-            Assert.Contains(logList, e => e is Nekoyume.Model.BattleStatus.HealSkill);
+            Assert.Contains(logList, e => e is Tick);
         }
     }
 }