finish lost-at-sea-encounter.test.ts (for now)

src/test/mystery-encounter/encounterTestUtils.ts

@@ -1,6 +1,6 @@
 import { Button } from "#app/enums/buttons";
 import { MessagePhase } from "#app/phases";
-import { MysteryEncounterOptionSelectedPhase, MysteryEncounterPhase, MysteryEncounterRewardsPhase } from "#app/phases/mystery-encounter-phase";
+import { MysteryEncounterPhase, MysteryEncounterRewardsPhase } from "#app/phases/mystery-encounter-phase";
 import MysteryEncounterUiHandler from "#app/ui/mystery-encounter-ui-handler";
 import { Mode } from "#app/ui/ui";
 import GameManager from "../utils/gameManager";
@@ -50,7 +50,5 @@ export async function runSelectMysteryEncounterOption(game: GameManager, optionN
     const uiHandler = game.scene.ui.getHandler<MysteryEncounterUiHandler>();
     uiHandler.processInput(Button.ACTION);
   });
-  await game.phaseInterceptor.run(MysteryEncounterOptionSelectedPhase);
-
   await game.phaseInterceptor.to(MysteryEncounterRewardsPhase);
 }

src/test/mystery-encounter/encounters/lost-at-sea-encounter.test.ts

@@ -28,7 +28,9 @@ describe("Lost at Sea - Mystery Encounter", () => {
     game = new GameManager(phaserGame);
     game.override.mysteryEncounterChance(100);
     game.override.startingBiome(Biome.SEA);
-    vi.spyOn(MysteryEncounters, "allMysteryEncounters", "get").mockReturnValue({ [MysteryEncounterType.LOST_AT_SEA]: LostAtSeaEncounter });
+    vi.spyOn(MysteryEncounters, "allMysteryEncounters", "get").mockReturnValue({
+      [MysteryEncounterType.LOST_AT_SEA]: LostAtSeaEncounter,
+    });
   });
 
   afterEach(() => {
@@ -47,14 +49,16 @@ describe("Lost at Sea - Mystery Encounter", () => {
     expect(LostAtSeaEncounter.options.length).toBe(3);
   });
 
+  it("should not run outside of sea biome", async () => {
+    // TODO: a little tricky
+  });
+
   it("should not run below wave 11", async () => {
     game.override.startingWave(10);
 
     await game.runToMysteryEncounter();
 
-    const { currentBattle } = game.scene;
+    expect(game.scene.currentBattle.mysteryEncounter).toBeUndefined();
-    expect(currentBattle).toBeDefined();
-    expect(currentBattle.mysteryEncounter).toBeUndefined();
   });
 
   it("should not run above wave 179", async () => {
@@ -62,9 +66,7 @@ describe("Lost at Sea - Mystery Encounter", () => {
 
     await game.runToMysteryEncounter();
 
-    const { currentBattle } = game.scene;
+    expect(game.scene.currentBattle.mysteryEncounter).toBeUndefined();
-    expect(currentBattle).toBeDefined();
-    expect(currentBattle.mysteryEncounter).toBeUndefined();
   });
 
   it("should set the correct dialog tokens during initialization", () => {
@@ -99,6 +101,37 @@ describe("Lost at Sea - Mystery Encounter", () => {
         ],
       });
     });
+
+    it("should award exp to surfable PKM (Blastoise)", async () => {
+      const laprasBaseExp = 187;
+      const wave = 33;
+      game.override.startingWave(wave);
+
+      await workaround_reInitSceneWithOverrides(game);
+      await game.runToMysteryEncounter(defaultParty);
+      const party = game.scene.getParty();
+      const blastoise = party.find((pkm) => pkm.species.speciesId === Species.PIDGEOT);
+      const expBefore = blastoise.exp;
+
+      await runSelectMysteryEncounterOption(game, 2);
+
+      expect(blastoise.exp).toBe(expBefore + laprasBaseExp * wave);
+    });
+
+    it("should leave encounter without battle", async () => {
+      game.override.startingWave(33);
+      const leaveEncounterWithoutBattleSpy = vi.spyOn(EncounterPhaseUtils, "leaveEncounterWithoutBattle");
+
+      await workaround_reInitSceneWithOverrides(game);
+      await game.runToMysteryEncounter(defaultParty);
+      await runSelectMysteryEncounterOption(game, 1);
+
+      expect(leaveEncounterWithoutBattleSpy).toBeCalled();
+    });
+
+    it("should be disabled if no surfable PKM is in party", async () => {
+      // TODO
+    });
   });
 
   describe("Option 2 - Fly", () => {
@@ -120,22 +153,35 @@ describe("Lost at Sea - Mystery Encounter", () => {
       });
     });
 
-    it("should award exp to surfable pokemone (Blastoise)", async () => {
+    it("should award exp to flyable PKM (Pidgeot)", async () => {
-      game.override.startingWave(33);
+      const laprasBaseExp = 187;
-      const setEncounterExpSpy = vi.spyOn(EncounterPhaseUtils, "setEncounterExp");
+      const wave = 33;
+      game.override.startingWave(wave);
 
-      workaround_reInitSceneWithOverrides(game);
+      await workaround_reInitSceneWithOverrides(game);
       await game.runToMysteryEncounter(defaultParty);
-      let blastoise = game.scene.getParty().find((pkm) => pkm.species.speciesId === Species.BLASTOISE);
+      const party = game.scene.getParty();
-      console.log("BLASTOISE EXP BEFORE: ", blastoise.exp);
+      const pidgeot = party.find((pkm) => pkm.species.speciesId === Species.PIDGEOT);
+      const expBefore = pidgeot.exp;
 
       await runSelectMysteryEncounterOption(game, 2);
 
-      blastoise = game.scene.getParty().find((pkm) => pkm.species.speciesId === Species.BLASTOISE);
+      expect(pidgeot.exp).toBe(expBefore + laprasBaseExp * wave);
-      console.log("BLASTOISE EXP AFTER: ", blastoise.exp);
+    });
 
-      expect(blastoise.exp).toBe(128);
+    it("should leave encounter without battle", async () => {
-      expect(setEncounterExpSpy).toHaveBeenCalledWith(expect.anything(), 128, expect.anything(), true);
+      game.override.startingWave(33);
+      const leaveEncounterWithoutBattleSpy = vi.spyOn(EncounterPhaseUtils, "leaveEncounterWithoutBattle");
+
+      await workaround_reInitSceneWithOverrides(game);
+      await game.runToMysteryEncounter(defaultParty);
+      await runSelectMysteryEncounterOption(game, 2);
+
+      expect(leaveEncounterWithoutBattleSpy).toBeCalled();
+    });
+
+    it("should be disabled if no flyable PKM is in party", async () => {
+      // TODO
     });
   });
 
@@ -156,20 +202,25 @@ describe("Lost at Sea - Mystery Encounter", () => {
       });
     });
 
-    it("should damage all party pokemon by 25%", async () => {
+    it("should damage all (allowed in battle) party PKM by 25%", async () => {
       game.override.startingWave(33);
 
-      workaround_reInitSceneWithOverrides(game);
+      await workaround_reInitSceneWithOverrides(game);
       await game.runToMysteryEncounter(defaultParty);
+
+      const party = game.scene.getParty();
+      const abra = party.find((pkm) => pkm.species.speciesId === Species.ABRA);
+      vi.spyOn(abra, "isAllowedInBattle").mockReturnValue(false);
+
       await runSelectMysteryEncounterOption(game, 3);
 
-      game.scene
+      const allowedPkm = party.filter((pkm) => pkm.isAllowedInBattle());
-        .getParty()
+      const notAllowedPkm = party.filter((pkm) => !pkm.isAllowedInBattle());
-        .forEach((pkm) =>
+      allowedPkm.forEach((pkm) =>
-          expect(pkm.hp, `${pkm.name} should have receivd 25% damage: ${pkm.hp} / ${pkm.getMaxHp()} HP`).toBe(
+        expect(pkm.hp, `${pkm.name} should have receivd 25% damage: ${pkm.hp} / ${pkm.getMaxHp()} HP`).toBe(pkm.getMaxHp() - Math.floor(pkm.getMaxHp() * 0.25))
-            pkm.getMaxHp() - Math.floor(pkm.getMaxHp() * 0.25)
-          )
       );
+
+      notAllowedPkm.forEach((pkm) => expect(pkm.hp, `${pkm.name} should be full hp: ${pkm.hp} / ${pkm.getMaxHp()} HP`).toBe(pkm.getMaxHp()));
     });
 
     it("should leave encounter without battle", async () => {

src/test/utils/phaseInterceptor.ts

@@ -8,6 +8,7 @@ import {
   EncounterPhase,
   EnemyCommandPhase,
   FaintPhase,
+  LearnMovePhase,
   LoginPhase,
   MessagePhase,
   MoveEffectPhase,
@@ -41,6 +42,7 @@ import {
   MysteryEncounterBattlePhase,
   MysteryEncounterOptionSelectedPhase,
   MysteryEncounterPhase,
+  MysteryEncounterRewardsPhase,
   PostMysteryEncounterPhase
 } from "#app/phases/mystery-encounter-phase";
 
@@ -100,11 +102,13 @@ export default class PhaseInterceptor {
     [MysteryEncounterPhase, this.startPhase],
     [MysteryEncounterOptionSelectedPhase, this.startPhase],
     [MysteryEncounterBattlePhase, this.startPhase],
-    [PostMysteryEncounterPhase, this.startPhase]
+    [MysteryEncounterRewardsPhase, this.startPhase],
+    [PostMysteryEncounterPhase, this.startPhase],
+    [LearnMovePhase, this.startPhase]
   ];
 
   private endBySetMode = [
-    TitlePhase, SelectGenderPhase, CommandPhase, SelectModifierPhase
+    TitlePhase, SelectGenderPhase, CommandPhase, SelectModifierPhase, PostMysteryEncounterPhase
   ];
 
   /**