From ab6cb23bab938a23d5e5671e62cb8676fcbbfec2 Mon Sep 17 00:00:00 2001
From: MokaStitcher <millennium.stitcher@gmail.com>
Date: Mon, 30 Sep 2024 12:01:01 +0200
Subject: [PATCH] [test] add  some sandstorm and hail tests

---
 src/test/arena/weather_hail.test.ts      | 14 ++++++++++
 src/test/arena/weather_sandstorm.test.ts | 33 ++++++++++++++++++++++++
 2 files changed, 47 insertions(+)

diff --git a/src/test/arena/weather_hail.test.ts b/src/test/arena/weather_hail.test.ts
index 8a1d660912d..2070e40dbcc 100644
--- a/src/test/arena/weather_hail.test.ts
+++ b/src/test/arena/weather_hail.test.ts
@@ -58,4 +58,18 @@ describe("Weather - Hail", () => {
     expect(playerPokemon.hp).toBe(playerPokemon.getMaxHp());
     expect(enemyPokemon.hp).toBe(enemyPokemon.getMaxHp() - Math.max(Math.floor(enemyPokemon.getMaxHp() / 16), 1));
   });
+
+  it("does not inflict damage to Ice type Pokemon", async () => {
+    await game.classicMode.startBattle([Species.CLOYSTER]);
+
+    game.move.select(Moves.SPLASH);
+
+    await game.phaseInterceptor.to("TurnEndPhase");
+
+    const playerPokemon = game.scene.getPlayerPokemon()!;
+    const enemyPokemon = game.scene.getEnemyPokemon()!;
+
+    expect(playerPokemon.hp).toBe(playerPokemon.getMaxHp());
+    expect(enemyPokemon.hp).toBe(enemyPokemon.getMaxHp() - Math.max(Math.floor(enemyPokemon.getMaxHp() / 16), 1));
+  });
 });
diff --git a/src/test/arena/weather_sandstorm.test.ts b/src/test/arena/weather_sandstorm.test.ts
index 0b64266f8c0..2419ca11b70 100644
--- a/src/test/arena/weather_sandstorm.test.ts
+++ b/src/test/arena/weather_sandstorm.test.ts
@@ -1,4 +1,6 @@
 import { WeatherType } from "#app/data/weather";
+import { Abilities } from "#app/enums/abilities";
+import { Stat } from "#app/enums/stat";
 import { Moves } from "#enums/moves";
 import { Species } from "#enums/species";
 import GameManager from "#test/utils/gameManager";
@@ -55,4 +57,35 @@ describe("Weather - Sandstorm", () => {
     expect(playerPokemon.hp).toBe(playerPokemon.getMaxHp());
     expect(enemyPokemon.hp).toBe(enemyPokemon.getMaxHp() - Math.max(Math.floor(enemyPokemon.getMaxHp() / 16), 1));
   });
+
+  it("does not inflict damage to Rock, Ground and Steel type Pokemon", async () => {
+    game.override
+      .battleType("double")
+      .enemySpecies(Species.SANDSHREW)
+      .ability(Abilities.BALL_FETCH)
+      .enemyAbility(Abilities.BALL_FETCH);
+
+    await game.classicMode.startBattle([Species.ROCKRUFF, Species.KLINK]);
+
+    game.move.select(Moves.SPLASH, 0);
+    game.move.select(Moves.SPLASH, 1);
+
+    await game.phaseInterceptor.to("TurnEndPhase");
+
+    game.scene.getField(true).forEach(pokemon => {
+      expect(pokemon.hp).toBe(pokemon.getMaxHp());
+    });
+  });
+
+  it("increases Rock type Pokemon Sp.Def by 50%", async () => {
+    await game.classicMode.startBattle([Species.ROCKRUFF]);
+
+    const playerPokemon = game.scene.getPlayerPokemon()!;
+    const playerSpdef = playerPokemon.getStat(Stat.SPDEF);
+    expect(playerPokemon.getEffectiveStat(Stat.SPDEF)).toBe(Math.floor(playerSpdef * 1.5));
+
+    const enemyPokemon = game.scene.getEnemyPokemon()!;
+    const enemySpdef = enemyPokemon.getStat(Stat.SPDEF);
+    expect(enemyPokemon.getEffectiveStat(Stat.SPDEF)).toBe(enemySpdef);
+  });
 });