From 168ed1cc55468863107edda6af375284a7cd133a Mon Sep 17 00:00:00 2001
From: Eduardo Quezada <eduardo602002@gmail.com>
Date: Wed, 22 Mar 2023 07:33:13 -0300
Subject: [PATCH] Xerneas/Zacian/Zamazenta form change tests

---
 test/form_change.c | 81 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 81 insertions(+)
 create mode 100644 test/form_change.c

diff --git a/test/form_change.c b/test/form_change.c
new file mode 100644
index 000000000..8d761bb12
--- /dev/null
+++ b/test/form_change.c
@@ -0,0 +1,81 @@
+#include "global.h"
+#include "test_battle.h"
+
+SINGLE_BATTLE_TEST("Xerneas changes into Active Form upon battle start")
+{
+    GIVEN {
+        ASSUME(P_GEN_6_POKEMON == TRUE);
+        PLAYER(SPECIES_XERNEAS);
+        OPPONENT(SPECIES_WOBBUFFET);
+    } WHEN {
+        TURN { MOVE(player, MOVE_CELEBRATE); }
+    } THEN {
+        EXPECT_EQ(player->species, SPECIES_XERNEAS_ACTIVE);
+    }
+}
+
+SINGLE_BATTLE_TEST("Zacian changes into its Crowned Form when holding the Rusted Sword upon battle start")
+{
+    u16 item;
+    PARAMETRIZE { item = ITEM_NONE; }
+    PARAMETRIZE { item = ITEM_RUSTED_SWORD; }
+    GIVEN {
+        ASSUME(P_GEN_8_POKEMON == TRUE);
+        PLAYER(SPECIES_ZACIAN) { Item(item); };
+        OPPONENT(SPECIES_WOBBUFFET);
+    } WHEN {
+        TURN { MOVE(player, MOVE_CELEBRATE); }
+    } THEN {
+        if (item == ITEM_NONE)
+            EXPECT_EQ(player->species, SPECIES_ZACIAN);
+        else
+            EXPECT_EQ(player->species, SPECIES_ZACIAN_CROWNED_SWORD);
+    }
+}
+
+SINGLE_BATTLE_TEST("Zacian's Iron Head becomes Behemoth Blade upon form change")
+{
+    GIVEN {
+        ASSUME(P_GEN_8_POKEMON == TRUE);
+        PLAYER(SPECIES_ZACIAN) { Item(ITEM_RUSTED_SWORD); Moves(MOVE_IRON_HEAD, MOVE_CELEBRATE); };
+        OPPONENT(SPECIES_WOBBUFFET);
+    } WHEN {
+        TURN { MOVE(player, MOVE_CELEBRATE); }
+    } THEN {
+        ASSUME(player->species == SPECIES_ZACIAN_CROWNED_SWORD); // Assumes form change worked.
+        EXPECT_EQ(player->moves[0], MOVE_BEHEMOTH_BLADE);
+    }
+}
+
+SINGLE_BATTLE_TEST("Zamazenta changes into its Crowned Form when holding the Rusted Shield upon battle start")
+{
+    u16 item;
+    PARAMETRIZE { item = ITEM_NONE; }
+    PARAMETRIZE { item = ITEM_RUSTED_SHIELD; }
+    GIVEN {
+        ASSUME(P_GEN_8_POKEMON == TRUE);
+        PLAYER(SPECIES_ZAMAZENTA) { Item(item); };
+        OPPONENT(SPECIES_WOBBUFFET);
+    } WHEN {
+        TURN { MOVE(player, MOVE_CELEBRATE); }
+    } THEN {
+        if (item == ITEM_NONE)
+            EXPECT_EQ(player->species, SPECIES_ZAMAZENTA);
+        else
+            EXPECT_EQ(player->species, SPECIES_ZAMAZENTA_CROWNED_SHIELD);
+    }
+}
+
+SINGLE_BATTLE_TEST("Zamazenta's Iron Head becomes Behemoth Bash upon form change")
+{
+    GIVEN {
+        ASSUME(P_GEN_8_POKEMON == TRUE);
+        PLAYER(SPECIES_ZAMAZENTA) { Item(ITEM_RUSTED_SHIELD); Moves(MOVE_IRON_HEAD, MOVE_CELEBRATE); };
+        OPPONENT(SPECIES_WOBBUFFET);
+    } WHEN {
+        TURN { MOVE(player, MOVE_CELEBRATE); }
+    } THEN {
+        ASSUME(player->species == SPECIES_ZAMAZENTA_CROWNED_SHIELD); // Assumes form change worked.
+        EXPECT_EQ(player->moves[0], MOVE_BEHEMOTH_BASH);
+    }
+}