From 7ce52f9655b47e323fe4eef39a844f7c69db527e Mon Sep 17 00:00:00 2001
From: muscode13 <mmustafafareed@hotmail.com>
Date: Wed, 9 Oct 2024 19:40:15 -0600
Subject: [PATCH] changed documentation wording, added test for stuff cheeks

---
 src/data/ability.ts                 |  2 +-
 src/test/abilities/unburden.test.ts | 19 +++++++++++++++----
 2 files changed, 16 insertions(+), 5 deletions(-)

diff --git a/src/data/ability.ts b/src/data/ability.ts
index 69df0d06dfd..dd927573dce 100644
--- a/src/data/ability.ts
+++ b/src/data/ability.ts
@@ -3793,7 +3793,7 @@ export class PostDancingMoveAbAttr extends PostMoveUsedAbAttr {
 }
 
 /**
- * Ability attribute for Unburden, triggers upon berry consumption
+ * Ability attribute for Unburden, triggers when a Pokemon consumes a berry they are holding
  * @extends PostTurnAbAttr
  * @see {@linkcode applyPostTurn}
  * @see {@linkcode getCondition}
diff --git a/src/test/abilities/unburden.test.ts b/src/test/abilities/unburden.test.ts
index f01501de021..c467c2495fe 100644
--- a/src/test/abilities/unburden.test.ts
+++ b/src/test/abilities/unburden.test.ts
@@ -29,7 +29,7 @@ describe("Abilities - Unburden", () => {
       .battleType("single")
       .starterSpecies(Species.TREECKO)
       .startingLevel(1)
-      .moveset([Moves.POPULATION_BOMB, Moves.KNOCK_OFF, Moves.PLUCK, Moves.THIEF])
+      .moveset([ Moves.POPULATION_BOMB, Moves.KNOCK_OFF, Moves.PLUCK, Moves.THIEF ])
       .ability(Abilities.UNBURDEN)
       .startingHeldItems([
         { name: "BERRY", count: 1, type: BerryType.SITRUS },
@@ -38,7 +38,7 @@ describe("Abilities - Unburden", () => {
       ])
       .enemySpecies(Species.NINJASK)
       .enemyLevel(100)
-      .enemyMoveset([Moves.FALSE_SWIPE])
+      .enemyMoveset([ Moves.FALSE_SWIPE ])
       .enemyAbility(Abilities.UNBURDEN)
       .enemyHeldItems([
         { name: "BERRY", type: BerryType.SITRUS, count: 1 },
@@ -99,7 +99,7 @@ describe("Abilities - Unburden", () => {
       .enemyAbility(Abilities.PICKPOCKET)
       .startingHeldItems([
         { name: "MULTI_LENS", count: 3 },
-        { name: "SOUL_DEW", count: 1},
+        { name: "SOUL_DEW", count: 1 },
         { name: "LUCKY_EGG", count: 1 },
       ]);
     await game.classicMode.startBattle();
@@ -113,7 +113,7 @@ describe("Abilities - Unburden", () => {
     expect(playerPokemon.getEffectiveStat(Stat.SPD)).toBeCloseTo(initialPlayerSpeed * 2);
   });
   it("should activate when an item is stolen via move", async () => {
-    vi.spyOn(allMoves[Moves.THIEF], "attrs", "get").mockReturnValue([new StealHeldItemChanceAttr(1.0)]); // give Thief 100% steal rate
+    vi.spyOn(allMoves[Moves.THIEF], "attrs", "get").mockReturnValue([ new StealHeldItemChanceAttr(1.0) ]); // give Thief 100% steal rate
     game.override.startingHeldItems([
       { name: "MULTI_LENS", count: 3 },
     ]);
@@ -166,4 +166,15 @@ describe("Abilities - Unburden", () => {
     expect(playerPokemon.getHeldItems().length).toBeLessThan(playerHeldItems);
     expect(playerPokemon.getEffectiveStat(Stat.SPD)).toBeCloseTo(initialPlayerSpeed);
   });
+  it("should activate when a move that consumes a berry is used", async () => {
+    game.override.enemyMoveset([ Moves.STUFF_CHEEKS ]);
+    await game.classicMode.startBattle();
+    const enemyPokemon = game.scene.getEnemyPokemon()!;
+    const enemyHeldItemCt = enemyPokemon.getHeldItems().length;
+    const initialEnemySpeed = enemyPokemon.getStat(Stat.SPD);
+    game.move.select(Moves.STUFF_CHEEKS);
+    await game.toNextTurn();
+    expect(enemyPokemon.getHeldItems().length).toBeLessThan(enemyHeldItemCt);
+    expect(enemyPokemon.getEffectiveStat(Stat.SPD)).toBeCloseTo(initialEnemySpeed * 2);
+  });
 });