pokemon-showdown/test/server/chat-plugins/datasearch.js

/**
 * Tests for the Data Search chat plugin
 */

'use strict';

const assert = require('../../assert').strict;

describe("Datasearch Plugin", () => {
	let datasearch = null;
	before(() => {
		datasearch = require('../../../dist/server/chat-plugins/datasearch');
	});

	it('should return pokemon with pivot moves', async () => {
		const cmd = 'ds';
		const target = 'pivot|batonpass, mod=gen8';
		const dexSearch = datasearch.testables.runDexsearch(target, cmd, `/${cmd} ${target}`, true);
		assert(dexSearch.results.includes('Absol'));
	});

	it('should return pokemon with pivot moves, but not baton pass', async () => {
		const cmd = 'ds';
		const target = 'pivot, mod=gen8';
		const dexSearch = datasearch.testables.runDexsearch(target, cmd, `/${cmd} ${target}`, true);
		assert.false(dexSearch.results.includes('Absol'));
		assert(dexSearch.results.includes('Abra'));
	});

	it('should return pivot moves', async () => {
		const cmd = 'ms';
		const target = 'pivot';
		const moveSearch = datasearch.testables.runMovesearch(target, cmd, `/${cmd} ${target}`, true);
		assert(moveSearch.results.includes('U-turn'));
	});

	it('should not return pivot moves', async () => {
		const cmd = 'ms';
		const target = '!pivot';
		const moveSearch = datasearch.testables.runMovesearch(target, cmd, `/${cmd} ${target}`, true);
		assert.false(moveSearch.results.includes('U-turn'));
	});

	it('should error', async () => {
		const cmd = 'ms';
		const target = 'pivot|!pivot';
		const moveSearch = datasearch.testables.runMovesearch(target, cmd, `/${cmd} ${target}`, true);
		assert(moveSearch.error);
	});

	it('should return recoil moves', async () => {
		const cmd = 'ms';
		const target = 'recoil';
		const moveSearch = datasearch.testables.runMovesearch(target, cmd, `/${cmd} ${target}`, true);
		assert(moveSearch.results.includes('Brave Bird'));
	});

	it('should not return recoil moves', async () => {
		const cmd = 'ms';
		const target = '!recoil';
		const moveSearch = datasearch.testables.runMovesearch(target, cmd, `/${cmd} ${target}`, true);
		assert.false(moveSearch.results.includes('Brave Bird'));
	});

	it('should return recovery moves', async () => {
		const cmd = 'ms';
		const target = 'recovery';
		const moveSearch = datasearch.testables.runMovesearch(target, cmd, `/${cmd} ${target}`, true);
		assert(moveSearch.results.includes('Absorb'));
	});

	it('should not return recovery moves', async () => {
		const cmd = 'ms';
		const target = '!recovery';
		const moveSearch = datasearch.testables.runMovesearch(target, cmd, `/${cmd} ${target}`, true);
		assert.false(moveSearch.results.includes('Absorb'));
	});

	it('should return zrecovery moves', async () => {
		const cmd = 'ms';
		const target = 'zrecovery';
		const moveSearch = datasearch.testables.runMovesearch(target, cmd, `/${cmd} ${target}`, true);
		assert(moveSearch.results.includes('Belly Drum'));
	});

	it('should not return zrecovery moves', async () => {
		const cmd = 'ms';
		const target = '!zrecovery';
		const moveSearch = datasearch.testables.runMovesearch(target, cmd, `/${cmd} ${target}`, true);
		assert.false(moveSearch.results.includes('Belly Drum'));
	});

	it('should include result where query string in ability is adjacent to special character', () => {
		const cmd = 'as';
		const target = 'water';
		const abilitySearch = datasearch.testables.runAbilitysearch(target, cmd, `/${cmd} ${target}`);
		assert(abilitySearch.reply.includes('Steam Engine'));
	});

	it('should exclude formes where the base Pokemon is included', () => {
		const cmd = 'ds';
		const target = 'ice, monotype';
		const search = datasearch.testables.runDexsearch(target, cmd, `/${cmd} ${target}`);
		assert.false(search.reply.includes('Eiscue-Noice'));
	});

	it('should include formes if a sort differentiates them from the base Pokemon', () => {
		const cmd = 'ds';
		let target = 'ice, monotype, spe desc';
		let search = datasearch.testables.runDexsearch(target, cmd, `/${cmd} ${target}`);
		assert(search.reply.includes('Eiscue-Noice'));

		target = 'ice, monotype, hp desc';
		search = datasearch.testables.runDexsearch(target, cmd, `/${cmd} ${target}`);
		assert.false(search.reply.includes('Eiscue-Noice'));
	});

	it('should include Pokemon capable of learning normally unobtainable moves under the provided rulsets', () => {
		const cmd = 'ds';
		const target = 'mod=gen8, rule=stabmonsmovelegality, rule=alphabetcupmovelegality, calm mind, boomburst, steel';
		const search = datasearch.testables.runDexsearch(target, cmd, `/${cmd} ${target}`);
		assert(search.reply.includes('Metagross'));
	});

	it('should exclude Pokemon capable of learning moves under normal circumstances or the provided rulesets', () => {
		const cmd = 'ds';
		const target = 'mod=gen9, rule=stabmonsmovelegality, !wish';
		const search = datasearch.testables.runDexsearch(target, cmd, `/${cmd} ${target}`);
		assert.false(search.reply.includes('Alomomola'));
		assert.false(search.reply.includes('Altaria'));
	});

	it('should include only Pokemon allowed by the provided rulesets', () => {
		const cmd = 'ds';
		const target = 'mod=gen8, rule=kalospokedex, rule=hoennpokedex, water, ground';
		const search = datasearch.testables.runDexsearch(target, cmd, `/${cmd} ${target}`);
		assert(search.reply.includes('Whiscash'));
		assert.false(search.reply.includes('Swampert'));
		assert.false(search.reply.includes('Quagsire'));
		assert.false(search.reply.includes('Gastrodon'));
	});

	it('should sort Pokemon with modified stats based on rulesets', () => {
		const cmd = 'ds';
		const target = 'mod=gen8, rule=reevolutionmod, water, bst desc';
		const search = datasearch.testables.runDexsearch(target, cmd, `/${cmd} ${target}`);
		assert(search.reply.includes('Milotic'));
	});

	it('should collapse results by default when broadcasting a large datasearch with all as a parameter', () => {
		let cmd = 'ds';
		let target = 'mod=gen8, water, all';
		let search = datasearch.testables.runDexsearch(target, cmd, `!${cmd} ${target}`);
		assert.false(search.reply.includes('<details class="details" open>'));

		cmd = 'ms';
		target = 'mod=gen8, water, all';
		search = datasearch.testables.runDexsearch(target, cmd, `!${cmd} ${target}`);
		assert.false(search.reply.includes('<details class="details" open>'));
	});

	it('should expand results by default when not broadcasting a large datasearch with all as a parameter', () => {
		let cmd = 'ds';
		let target = 'mod=gen8, water, all';
		let search = datasearch.testables.runDexsearch(target, cmd, `/${cmd} ${target}`);
		assert(search.reply.includes('<details class="details" open>'));

		cmd = 'ms';
		target = 'mod=gen8, water, all';
		search = datasearch.testables.runDexsearch(target, cmd, `/${cmd} ${target}`);
		assert(search.reply.includes('<details class="details" open>'));
	});
});