pokerogue/src/test/localization/terrain.test.ts

import { beforeAll, describe, beforeEach, afterEach, expect, it, vi } from "vitest";
import Phaser from "phaser";
import GameManager from "#app/test/utils/gameManager";
import Overrides from "#app/overrides";
import { Species } from "#enums/species";
import { TerrainType, getTerrainName } from "#app/data/terrain";
import { getTerrainStartMessage, getTerrainClearMessage, getTerrainBlockMessage } from "#app/data/weather";
import i18next from "i18next";
import { mockI18next } from "../utils/testUtils";

describe("terrain", () => {
  let phaserGame: Phaser.Game;
  let game: GameManager;

  beforeAll(() => {
    phaserGame = new Phaser.Game({
      type: Phaser.HEADLESS,
    });
    i18next.init();
  });

  beforeEach(() => {
    game = new GameManager(phaserGame);
    vi.spyOn(Overrides, "BATTLE_TYPE_OVERRIDE", "get").mockReturnValue("single");
  });

  describe("NONE", () => {
    const terrainType = TerrainType.NONE;

    it("should return the obtain text", () => {
      mockI18next();

      const text = getTerrainName(terrainType);
      expect(text).toBe("");
    });

    it("should return the start text", () => {
      mockI18next();

      const text = getTerrainStartMessage(terrainType);
      expect(text).toBe(undefined);
    });

    it("should return the clear text", () => {
      mockI18next();
      const text = getTerrainClearMessage(terrainType);
      expect(text).toBe(undefined);
    });

    it("should return the block text", async () => {
      await game.startBattle([Species.MAGIKARP]);
      const pokemon = game.scene.getPlayerPokemon();
      mockI18next();
      const text = getTerrainBlockMessage(pokemon, terrainType);
      expect(text).toBe("terrain:defaultBlockMessage");
    });
  });

  describe("MISTY", () => {
    const terrainType = TerrainType.MISTY;

    it("should return the obtain text", () => {
      mockI18next();

      const text = getTerrainName(terrainType);
      expect(text).toBe("terrain:misty");
    });

    it("should return the start text", () => {
      mockI18next();

      const text = getTerrainStartMessage(terrainType);
      expect(text).toBe("terrain:mistyStartMessage");
    });

    it("should return the clear text", () => {
      mockI18next();
      const text = getTerrainClearMessage(terrainType);
      expect(text).toBe("terrain:mistyClearMessage");
    });

    it("should return the block text", async () => {
      await game.startBattle([Species.MAGIKARP]);
      const pokemon = game.scene.getPlayerPokemon();
      mockI18next();
      const text = getTerrainBlockMessage(pokemon, terrainType);
      expect(text).toBe("terrain:mistyBlockMessage");
    });
  });

  describe("ELECTRIC", () => {
    const terrainType = TerrainType.ELECTRIC;

    it("should return the obtain text", () => {
      mockI18next();

      const text = getTerrainName(terrainType);
      expect(text).toBe("terrain:electric");
    });

    it("should return the start text", () => {
      mockI18next();

      const text = getTerrainStartMessage(terrainType);
      expect(text).toBe("terrain:electricStartMessage");
    });

    it("should return the clear text", () => {
      mockI18next();
      const text = getTerrainClearMessage(terrainType);
      expect(text).toBe("terrain:electricClearMessage");
    });

    it("should return the block text", async () => {
      await game.startBattle([Species.MAGIKARP]);
      const pokemon = game.scene.getPlayerPokemon();
      mockI18next();
      const text = getTerrainBlockMessage(pokemon, terrainType);
      expect(text).toBe("terrain:defaultBlockMessage");
    });
  });

  describe("GRASSY", () => {
    const terrainType = TerrainType.GRASSY;

    it("should return the obtain text", () => {
      mockI18next();

      const text = getTerrainName(terrainType);
      expect(text).toBe("terrain:grassy");
    });

    it("should return the start text", () => {
      mockI18next();

      const text = getTerrainStartMessage(terrainType);
      expect(text).toBe("terrain:grassyStartMessage");
    });

    it("should return the clear text", () => {
      mockI18next();
      const text = getTerrainClearMessage(terrainType);
      expect(text).toBe("terrain:grassyClearMessage");
    });

    it("should return the block text", async () => {
      await game.startBattle([Species.MAGIKARP]);
      const pokemon = game.scene.getPlayerPokemon();
      mockI18next();
      const text = getTerrainBlockMessage(pokemon, terrainType);
      expect(text).toBe("terrain:defaultBlockMessage");
    });
  });

  describe("PSYCHIC", () => {
    const terrainType = TerrainType.PSYCHIC;

    it("should return the obtain text", () => {
      mockI18next();

      const text = getTerrainName(terrainType);
      expect(text).toBe("terrain:psychic");
    });

    it("should return the start text", () => {
      mockI18next();

      const text = getTerrainStartMessage(terrainType);
      expect(text).toBe("terrain:psychicStartMessage");
    });

    it("should return the clear text", () => {
      mockI18next();
      const text = getTerrainClearMessage(terrainType);
      expect(text).toBe("terrain:psychicClearMessage");
    });

    it("should return the block text", async () => {
      await game.startBattle([Species.MAGIKARP]);
      const pokemon = game.scene.getPlayerPokemon();
      mockI18next();
      const text = getTerrainBlockMessage(pokemon, terrainType);
      expect(text).toBe("terrain:defaultBlockMessage");
    });
  });


  afterEach(() => {
    game.phaseInterceptor.restoreOg();
    vi.resetAllMocks();
  });
});