From a0655a9da5bf7d921b26a4d1a0be59d42b2d7808 Mon Sep 17 00:00:00 2001
From: Michael Li <michael5530@outlook.com>
Date: Sun, 13 Oct 2024 00:56:42 -0400
Subject: [PATCH] Tests: Added checks for correct number of Pokemon on field

---
 src/test/moves/dragon_tail.test.ts | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/test/moves/dragon_tail.test.ts b/src/test/moves/dragon_tail.test.ts
index f648d01caf2..cf801eb42c1 100644
--- a/src/test/moves/dragon_tail.test.ts
+++ b/src/test/moves/dragon_tail.test.ts
@@ -154,9 +154,10 @@ describe("Moves - Dragon Tail", () => {
     expect(enemy).toBeDefined();
     expect(enemy.isFullHp()).toBe(true);
 
-    // Make sure the enemy has a fainted Pokemon in their party
+    // Make sure the enemy has a fainted Pokemon in their party and not on the field
     const faintedEnemy = game.scene.getEnemyParty().find(p => !p.isAllowedInBattle());
     expect(faintedEnemy).toBeDefined();
+    expect(game.scene.getEnemyField().length).toBe(1);
   });
 
   it("should not cause a softlock when activating an opponent trainer's reviver seed", async () => {
@@ -173,6 +174,7 @@ describe("Moves - Dragon Tail", () => {
     const enemy = game.scene.getEnemyPokemon()!;
     expect(enemy).toBeDefined();
     expect(enemy.hp).toBe(Math.floor(enemy.getMaxHp() / 2));
+    expect(game.scene.getEnemyField().length).toBe(1);
   });
 
   it("should not cause a softlock when activating a player's reviver seed", async () => {
@@ -189,5 +191,6 @@ describe("Moves - Dragon Tail", () => {
     const dratini = game.scene.getPlayerPokemon()!;
     expect(dratini).toBeDefined();
     expect(dratini.hp).toBe(Math.floor(dratini.getMaxHp() / 2));
+    expect(game.scene.getPlayerField().length).toBe(1);
   });
 });