Added missing Belch tests (#5881)

rh-hideout · Dec 27, 2024 · dccf263 · dccf263
1 parent 767e5a1
commit dccf263
Showing 1 changed file with 62 additions and 4 deletions.
diff --git a/test/battle/move_effect/belch.c b/test/battle/move_effect/belch.c
@@ -21,7 +21,7 @@ AI_SINGLE_BATTLE_TEST("AI: Belch has nonzero score after eating a berry")
         TURN { MOVE(player, MOVE_MUD_SHOT); EXPECT_MOVE(opponent, MOVE_TACKLE); }
         TURN { MOVE(player, MOVE_TACKLE); EXPECT_MOVE(opponent, MOVE_BELCH);}
     } SCENE {
-            ANIMATION(ANIM_TYPE_MOVE, MOVE_BELCH, opponent);
+        ANIMATION(ANIM_TYPE_MOVE, MOVE_BELCH, opponent);
     }
 }
 
@@ -53,6 +53,64 @@ SINGLE_BATTLE_TEST("Belch cannot be used if the user has not eaten a berry")
     }
 }
 
-TO_DO_BATTLE_TEST("Belch can still be used after switching out");
-TO_DO_BATTLE_TEST("Belch can still be used after fainting");
-TO_DO_BATTLE_TEST("Belch can still be used after restoring the consumed berry");
+SINGLE_BATTLE_TEST("Belch can still be used after switching out")
+{
+    GIVEN {
+        ASSUME(gMovesInfo[MOVE_STUFF_CHEEKS].effect == EFFECT_STUFF_CHEEKS);
+        PLAYER(SPECIES_GREEDENT) { Item(ITEM_ORAN_BERRY); }
+        PLAYER(SPECIES_SKWOVET);
+        OPPONENT(SPECIES_WOBBUFFET);
+    } WHEN {
+        TURN { MOVE(player, MOVE_STUFF_CHEEKS); }
+        TURN { SWITCH(player, 1); }
+        TURN { SWITCH(player, 0); }
+        TURN { MOVE(player, MOVE_BELCH); }
+    } SCENE {
+        ANIMATION(ANIM_TYPE_MOVE, MOVE_STUFF_CHEEKS, player);
+        SWITCH_OUT_MESSAGE("Greedent");
+        SWITCH_OUT_MESSAGE("Skwovet");
+        ANIMATION(ANIM_TYPE_MOVE, MOVE_BELCH, player);
+    }
+}
+
+SINGLE_BATTLE_TEST("Belch can still be used after fainting")
+{
+    GIVEN {
+        ASSUME(gMovesInfo[MOVE_STUFF_CHEEKS].effect == EFFECT_STUFF_CHEEKS);
+        ASSUME(gMovesInfo[MOVE_FISSURE].effect == EFFECT_OHKO);
+        ASSUME(gMovesInfo[MOVE_REVIVAL_BLESSING].effect == EFFECT_REVIVAL_BLESSING);
+        PLAYER(SPECIES_GREEDENT) { Item(ITEM_ORAN_BERRY); }
+        PLAYER(SPECIES_SKWOVET);
+        OPPONENT(SPECIES_WOBBUFFET);
+    } WHEN {
+        TURN { MOVE(player, MOVE_STUFF_CHEEKS); MOVE(opponent, MOVE_FISSURE); SEND_OUT(player, 1); }
+        TURN { MOVE(player, MOVE_REVIVAL_BLESSING, partyIndex: 0); }
+        TURN { SWITCH(player, 0); }
+        TURN { MOVE(player, MOVE_BELCH); }
+    } SCENE {
+        ANIMATION(ANIM_TYPE_MOVE, MOVE_STUFF_CHEEKS, player);
+        ANIMATION(ANIM_TYPE_MOVE, MOVE_FISSURE, opponent);
+        ANIMATION(ANIM_TYPE_MOVE, MOVE_REVIVAL_BLESSING, player);
+        SWITCH_OUT_MESSAGE("Skwovet");
+        ANIMATION(ANIM_TYPE_MOVE, MOVE_BELCH, player);
+    }
+}
+
+SINGLE_BATTLE_TEST("Belch can still be used after restoring the consumed berry")
+{
+    GIVEN {
+        ASSUME(gMovesInfo[MOVE_STUFF_CHEEKS].effect == EFFECT_STUFF_CHEEKS);
+        ASSUME(gMovesInfo[MOVE_RECYCLE].effect == EFFECT_RECYCLE);
+        PLAYER(SPECIES_GREEDENT) { Item(ITEM_ORAN_BERRY); }
+        PLAYER(SPECIES_SKWOVET);
+        OPPONENT(SPECIES_WOBBUFFET);
+    } WHEN {
+        TURN { MOVE(player, MOVE_STUFF_CHEEKS); }
+        TURN { MOVE(player, MOVE_RECYCLE); }
+        TURN { MOVE(player, MOVE_BELCH); }
+    } SCENE {
+        ANIMATION(ANIM_TYPE_MOVE, MOVE_STUFF_CHEEKS, player);
+        ANIMATION(ANIM_TYPE_MOVE, MOVE_RECYCLE, player);
+        ANIMATION(ANIM_TYPE_MOVE, MOVE_BELCH, player);
+    }
+}