Test: Improve Imprison tests

smogon · Mar 27, 2019 · b65e6f6 · b65e6f6
1 parent 60b65c5
commit b65e6f6
Showing 1 changed file with 28 additions and 10 deletions.
diff --git a/test/simulator/moves/imprison.js b/test/simulator/moves/imprison.js
@@ -10,7 +10,7 @@ describe('Imprison', function () {
 		battle.destroy();
 	});
 
-	it('should prevent foes from using moves that the user knows', function () {
+	it(`should prevent foes from using moves that the user knows`, function () {
 		battle = common.createBattle();
 		battle.setPlayer('p1', {team: [
 			{species: 'Abra', ability: 'prankster', moves: ['imprison', 'calmmind', 'batonpass']},
@@ -20,37 +20,55 @@ describe('Imprison', function () {
 			{species: 'Abra', ability: 'synchronize', moves: ['calmmind', 'gravity']},
 			{species: 'Kadabra', ability: 'prankster', moves: ['imprison', 'calmmind']},
 		]});
+
 		battle.makeChoices('move imprison', 'move calmmind');
-		assert.strictEqual(battle.p2.active[0].boosts['spa'], 0);
-		assert.strictEqual(battle.p2.active[0].boosts['spd'], 0);
+		assert.statStage(battle.p2.active[0], 'spa', 0);
+		assert.statStage(battle.p2.active[0], 'spd', 0);
 		assert.cantMove(() => battle.choose('p2', 'move calmmind'), 'Abra', 'Calm Mind');
 
 		// Imprison doesn't end when the foe switches
 		battle.makeChoices('default', 'switch 2');
 		battle.makeChoices('move calmmind', 'move calmmind');
-		assert.strictEqual(battle.p2.active[0].boosts['spa'], 0);
+		assert.statStage(battle.p2.active[0], 'spa', 0);
 
 		// Imprison is not passed by Baton Pass
 		battle.makeChoices('move batonpass', 'move calmmind');
-		assert.strictEqual(battle.p2.active[0].boosts['spa'], 0);
+		assert.statStage(battle.p2.active[0], 'spa', 0);
 		battle.makeChoices('switch 2', '');
-		assert.strictEqual(battle.p2.active[0].boosts['spa'], 0);
+		assert.statStage(battle.p2.active[0], 'spa', 0);
 		battle.makeChoices('move calmmind', 'move calmmind');
-		assert.strictEqual(battle.p2.active[0].boosts['spa'], 1);
+		assert.statStage(battle.p2.active[0], 'spa', 1);
 
 		// Imprison ends after user switches
 		battle.makeChoices('switch 2', 'move calmmind');
-		assert.strictEqual(battle.p2.active[0].boosts['spa'], 2);
+		assert.statStage(battle.p2.active[0], 'spa', 2);
 		battle.makeChoices('move calmmind', 'move calmmind');
-		assert.strictEqual(battle.p2.active[0].boosts['spa'], 3);
+		assert.statStage(battle.p2.active[0], 'spa', 3);
 	});
 
-	it('should not prevent foes from using Z-Powered Status moves', function () {
+	it(`should not prevent foes from using Z-Powered Status moves`, function () {
 		battle = common.createBattle();
 		battle.setPlayer('p1', {team: [{species: 'Sableye', ability: 'prankster', moves: ['imprison', 'sunnyday']}]});
 		battle.setPlayer('p2', {team: [{species: 'Charmander', ability: 'blaze', item: 'firiumz', moves: ['sunnyday']}]});
+
 		battle.makeChoices('move imprison', 'move sunnyday zmove');
 		assert.statStage(battle.p2.active[0], 'spe', 1);
 		assert.ok(battle.field.isWeather('sunnyday'));
 	});
+
+	it(`should not prevent the user from using moves that a foe knows`, function () {
+		battle = common.createBattle();
+		battle.setPlayer('p1', {team: [
+			{species: 'Abra', ability: 'prankster', moves: ['imprison', 'calmmind', 'batonpass']},
+		]});
+		battle.setPlayer('p2', {team: [
+			{species: 'Abra', ability: 'synchronize', moves: ['calmmind', 'gravity']},
+		]});
+		const imprisonUser = battle.p1.active[0];
+
+		battle.makeChoices('move imprison', 'auto');
+		battle.makeChoices('move calmmind', 'auto');
+		assert.statStage(imprisonUser, 'spa', 1);
+		assert.statStage(imprisonUser, 'spd', 1);
+	});
 });