Battle-for-Tismo/tests/modals_test.js

import { TestCase } from "./framework/test_case.js";
import { setupDom, teardownDom } from "./framework/dom.js";
import { TrainModal, BuildingsModal, NationModal } from "../client/js/modals.js";
import { BUILDINGS, GOVERNMENTS, TECHNOLOGIES, PROTO_UNITS } from "../shared/data.js";

export class TrainModalTest extends TestCase {
  async test_lists_trainable_units_and_requests_one() {
    const env = await setupDom();
    try {
      const modal = new TrainModal();
      let requested = null;
      modal.onTrain = (cityId, protoIndex) => {
        requested = [cityId, protoIndex];
      };
      modal.show({ id: 5, name: "Paris" }, 1.0e9, [{ ...PROTO_UNITS[0], serverIndex: 0 }]);
      this.assertEqual(env.$("#train-title").text(), "Paris — Train");
      this.assertSize(env.$("#train-list button"), 1);
      this.assertTrue(env.$("#train-list button").text().includes("Modern infantry"));
      env.$("#train-list button").click();
      this.assertEqual(requested, [5, 0]);
      modal.close();
      this.assertTrue(env.$("#modal-train").hasClass("hidden"));
    } finally {
      teardownDom(env);
    }
  }

  async test_training_progress_states() {
    const env = await setupDom();
    try {
      const modal = new TrainModal();
      modal.show({ id: 1, name: "Lyon" }, 1.0e9, [{ ...PROTO_UNITS[0], serverIndex: 0 }]);
      modal.beginTraining("Modern infantry", 10, 2);
      this.assertFalse(env.$("#train-status").hasClass("hidden"));
      this.assertTrue(env.$("#train-list button").prop("disabled"));
      modal.setTrainingProgress("Modern infantry", 4, 10);
      this.assertTrue(env.$("#train-label").text().includes("6 h left"));
      modal.endTraining();
      this.assertEqual(env.$("#train-label").text(), "Training complete");
      this.assertFalse(env.$("#train-list button").prop("disabled"));
    } finally {
      teardownDom(env);
    }
  }
}

export class BuildingsModalTest extends TestCase {
  async test_prices_rows_and_requests_build_and_demolish() {
    const env = await setupDom();
    try {
      const modal = new BuildingsModal();
      let built = null;
      let demolished = null;
      modal.onBuild = (cityId, index) => {
        built = [cityId, index];
      };
      modal.onDemolish = (cityId, index) => {
        demolished = [cityId, index];
      };
      modal.show({ id: 3, name: "Lyon" }, 1.0e9, 1.0e15, BUILDINGS, {});
      this.assertEqual(env.$("#buildings-title").text(), "Lyon — Buildings");
      this.assertSize(env.$("#buildings-list .row-card"), BUILDINGS.length);
      const first = env.$("#buildings-list .row-card").first();
      this.assertTrue(first.find(".name").text().includes("Level 0"));
      this.assertFalse(first.find(".actions button").first().prop("disabled"));
      first.find(".actions button").first().click();
      this.assertEqual(built, [3, 0]);

      modal.refresh(1.0e9, 1.0e15, { 0: 1 });
      this.assertTrue(first.find(".name").text().includes("Level 1"));
      first.find(".actions button").eq(1).click();
      this.assertEqual(demolished, [3, 0]);
      modal.close();
      this.assertTrue(env.$("#modal-buildings").hasClass("hidden"));
    } finally {
      teardownDom(env);
    }
  }

  async test_unaffordable_upgrade_is_disabled() {
    const env = await setupDom();
    try {
      const modal = new BuildingsModal();
      modal.show({ id: 1, name: "X" }, 1.0e9, 0, BUILDINGS, {});
      const upgrade = env.$("#buildings-list .row-card").first().find(".actions button").first();
      this.assertTrue(upgrade.prop("disabled"));
    } finally {
      teardownDom(env);
    }
  }
}

export class NationModalTest extends TestCase {
  async test_government_and_technology_tabs() {
    const env = await setupDom();
    try {
      const modal = new NationModal();
      let government = null;
      let technology = null;
      modal.onGovernment = (index) => {
        government = index;
      };
      modal.onTechnology = (index) => {
        technology = index;
      };
      modal.show(GOVERNMENTS, 0, TECHNOLOGIES, new Set(), 1.0e9, 1.0e15, 1.0e6);
      this.assertSize(env.$("#government-list .row-card"), GOVERNMENTS.length);
      this.assertSize(env.$("#technology-list .row-card"), TECHNOLOGIES.length);

      const governmentRows = env.$("#government-list .row-card");
      this.assertTrue(governmentRows.eq(0).find(".actions button").prop("disabled"), "current government is locked");
      governmentRows.eq(1).find(".actions button").click();
      this.assertEqual(government, 1);

      // A prerequisite-gated technology starts locked.
      const method = TECHNOLOGIES.findIndex((t) => t.id === "scientific_method");
      const automation = TECHNOLOGIES.findIndex((t) => t.id === "industrial_automation");
      const techRows = env.$("#technology-list .row-card");
      this.assertEqual(techRows.eq(automation).find(".actions button").text(), "Locked");

      // Researching its prerequisite unlocks it.
      modal.refresh(0, new Set([method]), 1.0e9, 1.0e15, 1.0e6);
      this.assertTrue(techRows.eq(method).find(".actions button").text().startsWith("Researched"));
      this.assertTrue(techRows.eq(automation).find(".actions button").text().startsWith("Research"));
      techRows.eq(automation).find(".actions button").click();
      this.assertEqual(technology, automation);

      env.$("#modal-nation .tab[data-tab=technology]").click();
      this.assertTrue(env.$("#tab-technology").hasClass("active"));
      env.$("#modal-nation .tab[data-tab=government]").click();
      this.assertTrue(env.$("#tab-government").hasClass("active"));
      modal.close();
      this.assertTrue(env.$("#modal-nation").hasClass("hidden"));
    } finally {
      teardownDom(env);
    }
  }
}