bitburner-src/test/jest/Netscript/Bladeburner.test.ts

import { currentNodeMults } from "../../../src/BitNode/BitNodeMultipliers";
import { Skill } from "../../../src/Bladeburner/Skill";
import { BladeburnerSkillName } from "../../../src/Enums";
import { PositiveInteger, isPositiveInteger, isPositiveNumber } from "../../../src/types";
import { randomInRange } from "../../../src/utils/helpers/randomInRange";
import { getRandomIntInclusive } from "../../../src/utils/helpers/getRandomIntInclusive";

const skill = new Skill({
  name: BladeburnerSkillName.Hyperdrive,
  desc: "",
  baseCost: 1,
  costInc: 1,
  mults: {},
});

describe("Test calculateMaxUpgradeCount", function () {
  test("errorCount", () => {
    let testCaseCount = 0;
    let errorCount = 0;
    const test1Errors = [];
    const test2Errors = [];
    for (let i = 0; i < 10; ++i) {
      skill.baseCost = getRandomIntInclusive(1, 1000);
      for (let j = 0; j < 10; ++j) {
        skill.costInc = randomInRange(1, 1000);
        for (let k = 0; k < 10; ++k) {
          currentNodeMults.BladeburnerSkillCost = randomInRange(1, 1000);
          for (let m = 0; m < 1e4; ++m) {
            const currentLevel = getRandomIntInclusive(0, 1e9);
            let count = 0;
            let cost = 0;

            // Test 1
            ++testCaseCount;
            const expectedCount = getRandomIntInclusive(1, 1e9);
            if (!isPositiveInteger(expectedCount)) {
              throw new Error(`Invalid expectedCount: ${expectedCount}`);
            }
            cost = skill.calculateCost(currentLevel, expectedCount);
            if (!isPositiveNumber(cost)) {
              throw new Error(`Invalid cost: ${cost}`);
            }
            count = skill.calculateMaxUpgradeCount(currentLevel, cost);
            if (expectedCount !== count) {
              ++errorCount;
              test1Errors.push({
                baseCost: skill.baseCost,
                costInc: skill.costInc,
                mult: currentNodeMults.BladeburnerSkillCost,
                currentLevel,
                cost,
                count,
                expectedCount,
              });
            }

            // Test 2
            ++testCaseCount;
            const budget = randomInRange(1e9, 1e50);
            if (!isPositiveNumber(budget)) {
              throw new Error(`Invalid budget: ${budget}`);
            }
            count = skill.calculateMaxUpgradeCount(currentLevel, budget);
            if (!isPositiveInteger(count)) {
              throw new Error(`Invalid count: ${count}`);
            }
            cost = skill.calculateCost(currentLevel, count);
            const costOfCountPlus1 = skill.calculateCost(currentLevel, (count + 1) as PositiveInteger);
            if (count !== count + 1 && (budget < cost || budget >= costOfCountPlus1)) {
              ++errorCount;
              test2Errors.push({
                baseCost: skill.baseCost,
                costInc: skill.costInc,
                mult: currentNodeMults.BladeburnerSkillCost,
                currentLevel,
                count,
                budget,
                cost,
                costOfCountPlus1,
              });
            }
          }
        }
      }
    }
    if (errorCount !== 0) {
      // There may be hundreds of thousands or even millions of failed test cases, so we only show a limited number of them.
      console.error(
        `testCaseCount: ${testCaseCount}. errorCount: ${errorCount}. test1Errors: ${JSON.stringify(
          test1Errors.slice(0, 1000),
        )}. test2Errors: ${JSON.stringify(test2Errors.slice(0, 1000))}`,
      );
    }

    expect(errorCount).toBe(0);
  });
});