2.data-mining.test.js

const pokemons = require("./pokeData");
const trainers = require("./trainerData");
const gyms = require("./gymData");

const {
  getTrainerPokemons,
  getTrainersPokemons,
  getGymLeader,
  getBigGyms,
} = require("./2.data-mining");

test("getGymleader: gets the gymleader belonging to a gym", () => {
  const fuchsiaCity = gyms.find((gym) => gym.city === "Fuchsia City");
  const gymLeader = getGymLeader(fuchsiaCity);
  expect(gymLeader).toEqual(
    expect.objectContaining({
      id: expect.any(Number),
      name: expect.any(String),
      pokemonIds: expect.any(Array),
    })
  );
  expect(gymLeader.name).toBe("Koga");
});

test("getTrainerPokemons: gets the pokemons belonging to a trainer", () => {
  const ash = trainers.find((trainer) => trainer.name === "Ash");
  const teamAsh = getTrainerPokemons(ash);

  expect(teamAsh).toEqual(expect.any(Array));
  expect(teamAsh.map((pokemon) => pokemon.name)).toEqual(
    expect.arrayContaining([
      "Pikachu",
      "Bulbasaur",
      "Squirtle",
      "Charizard",
      "Pidgeotto",
      "Butterfree",
    ])
  );
});

test(`getTrainersPokemons: replaces pokemonIds with 
        the pokemons belonging to a trainer for an array of trainers`, () => {
  const trainersWithPokemons = getTrainersPokemons();

  expect(trainersWithPokemons).toEqual(expect.any(Array));

  trainersWithPokemons.forEach((trainer) => {
    expect(trainer).toEqual(
      expect.objectContaining({
        id: expect.any(Number),
        name: expect.any(String),
        pokemons: expect.any(Array),
      })
    );

    trainer.pokemons.forEach((pokemon) => {
      expect(pokemon).toEqual(
        expect.objectContaining({
          id: expect.any(Number),
          name: expect.any(String),
          avg_spawns: expect.any(Number),
        })
      );
    });
  });
});

test("getBigGyms: gets the city names with gym leaders who have 4 pokemons or more", () => {
  const bigGymCities = getBigGyms();
  expect(bigGymCities).toEqual(
    expect.arrayContaining([
      "Saffron City",
      "Fuchsia City",
      "Cinnabar Island",
      "Viridian City",
    ])
  );
});