Server/project/tests/RepairHelper.test.ts

import "reflect-metadata";

import { RepairHelper } from "@spt-aki/helpers/RepairHelper";
import { Item } from "@spt-aki/models/eft/common/tables/IItem";
import { ITemplateItem } from "@spt-aki/models/eft/common/tables/ITemplateItem";
import { ILogger } from "@spt-aki/models/spt/utils/ILogger";
import { ConfigServer } from "@spt-aki/servers/ConfigServer";
import { DatabaseServer } from "@spt-aki/servers/DatabaseServer";
import { JsonUtil } from "@spt-aki/utils/JsonUtil";
import { RandomUtil } from "@spt-aki/utils/RandomUtil";
import { TestHelper } from "./common/TestHelper";

let testHelper: TestHelper;
let logger: ILogger;
let jsonUtil: JsonUtil;
let randomUtil: RandomUtil;
let configServer: ConfigServer;
let databaseServer: DatabaseServer;
let helper: RepairHelper;

describe("BotHelper", () => {

    beforeAll(async () => {
        testHelper = await TestHelper.fetchTestHelper();
        logger = testHelper.getTestLogger();
        jsonUtil = testHelper.getTestJsonUtil();
        randomUtil = testHelper.getTestRandomUtil();
        configServer = testHelper.getTestConfigServer();
        databaseServer = testHelper.getTestDatabaseServer();
    })

    beforeEach(() => {
        helper = new RepairHelper(
            logger,
            jsonUtil,
            randomUtil,
            databaseServer,
            configServer);
    });

    it("RepairHelper type check", () =>
    {
        expect(helper).toBeInstanceOf(RepairHelper);
    });

    it("updateItemDurability() repairkit with slick armor with max dura degradation", () =>
    {
        const slickTpl = "5e4abb5086f77406975c9342";
        const itemToRepair: Item = {
            _id: "12345",
            _tpl: slickTpl,
            upd: {
                Repairable: {
                    Durability: 60,
                    MaxDurability: 80
                }
            }
        };
        const itemToRepairDetails = <ITemplateItem>databaseServer.getTables().templates?.items[slickTpl];
        const isArmor = true;
        const useRepairKit = true;
        const useDegridation = true;

        helper.updateItemDurability(itemToRepair, itemToRepairDetails, isArmor, 5, useRepairKit, 1, useDegridation);

        expect(itemToRepair.upd?.Repairable?.Durability).toBeLessThan(80);
    });

    it("updateItemDurability() trader with slick armor with max dura degradation - partial repair", () =>
    {
        const slickTpl = "5e4abb5086f77406975c9342";
        const itemToRepair: Item = {
            _id: "12345",
            _tpl: slickTpl,
            upd: {
                Repairable: {
                    Durability: 60,
                    MaxDurability: 80
                }
            }
        };
        const itemToRepairDetails = <ITemplateItem>databaseServer.getTables().templates?.items[slickTpl];
        const isArmor = true;
        const useRepairKit = false;
        const useDegridation = true;

        helper.updateItemDurability(itemToRepair, itemToRepairDetails, isArmor, 5, useRepairKit, 1.2, useDegridation);

        expect(itemToRepair.upd?.Repairable?.Durability).toBeLessThan(80);
        expect(itemToRepair.upd?.Repairable?.Durability).toBeLessThanOrEqual(itemToRepair.upd?.Repairable?.MaxDurability);
    });

    it("updateItemDurability() trader with slick armor, no dura degradation", () =>
    {
        const slickTpl = "5e4abb5086f77406975c9342";
        const itemToRepair: Item = {
            _id: "12345",
            _tpl: slickTpl,
            upd: {
                Repairable: {
                    Durability: 60,
                    MaxDurability: 80
                }
            }
        };
        const itemToRepairDetails = <ITemplateItem>databaseServer.getTables().templates?.items[slickTpl];
        const isArmor = true;
        const useRepairKit = false;
        const useDegradation = false;

        helper.updateItemDurability(itemToRepair, itemToRepairDetails, isArmor, 20, useRepairKit, 1.2, useDegradation);

        expect(itemToRepair.upd?.Repairable?.Durability).toBe(80);
        expect(itemToRepair.upd?.Repairable?.Durability).toBeLessThanOrEqual(itemToRepair.upd?.Repairable?.MaxDurability);
    });

    it("updateItemDurability() repairkit with g36 with max dura degradation - Full repair", () =>
    {
        const itemCurrentDura = 68;
        const itemCurrentMaxDura = 100;
        const duraDifference = itemCurrentMaxDura - itemCurrentDura;
        const g36Tpl = "623063e994fc3f7b302a9696";
        const itemToRepair: Item = {
            _id: "12345",
            _tpl: g36Tpl,
            upd: {
                Repairable: {
                    Durability: itemCurrentDura,
                    MaxDurability: itemCurrentMaxDura
                }
            }
        };
        const itemToRepairDetails = <ITemplateItem>databaseServer.getTables().templates?.items[g36Tpl];
        const isArmor = false;
        const useRepairKit = true;
        const useDegradation = true;

        helper.updateItemDurability(itemToRepair, itemToRepairDetails, isArmor, duraDifference, useRepairKit, 1, useDegradation);

        expect(itemToRepair.upd?.Repairable?.Durability).toBeLessThan(100);
        expect(itemToRepair.upd?.Repairable?.Durability).toBeLessThanOrEqual(itemToRepair.upd?.Repairable?.MaxDurability);
    });

    it("updateItemDurability() trader with g36 with max dura degradation - Full repair", () =>
    {
        const itemCurrentDura = 68;
        const itemCurrentMaxDura = 100;
        const duraDifference = itemCurrentMaxDura - itemCurrentDura;
        const g36Tpl = "623063e994fc3f7b302a9696";
        const itemToRepair: Item = {
            _id: "12345",
            _tpl: g36Tpl,
            upd: {
                Repairable: {
                    Durability: itemCurrentDura,
                    MaxDurability: itemCurrentMaxDura
                }
            }
        };
        const itemToRepairDetails = <ITemplateItem>databaseServer.getTables().templates?.items[g36Tpl];
        const isArmor = false;
        const useRepairKit = false;
        const useDegradation = true;

        helper.updateItemDurability(itemToRepair, itemToRepairDetails, isArmor, duraDifference, useRepairKit, 1.2, useDegradation);

        expect(itemToRepair.upd?.Repairable?.Durability).toBeLessThan(100);
        expect(itemToRepair.upd?.Repairable?.Durability).toBeLessThanOrEqual(itemToRepair.upd?.Repairable?.MaxDurability);
    });

    it("updateItemDurability() faceshield broken use repairkit with max dura degradation", () =>
    {
        const twExfilBallisticFaceShieldTpl = "5e00cdd986f7747473332240";
        const itemToRepair: Item = {
            _id: "12345",
            _tpl: twExfilBallisticFaceShieldTpl,
            upd: {
                Repairable: {
                    Durability: 30,
                    MaxDurability: 45
                },
                FaceShield: {
                    Hits: 2
                }
            }
        };

        const itemToRepairDetails = <ITemplateItem>databaseServer.getTables().templates?.items[twExfilBallisticFaceShieldTpl];
        const isArmor = true;
        const useRepairKit = true;
        const useDegradation = true;

        helper.updateItemDurability(itemToRepair, itemToRepairDetails, isArmor, 5, useRepairKit, 1, useDegradation);

        expect(itemToRepair.upd?.FaceShield?.Hits).toBe(0);
        expect(itemToRepair.upd?.Repairable?.Durability).toBeLessThan(45);
        expect(itemToRepair.upd?.Repairable?.Durability).toBeLessThanOrEqual(itemToRepair.upd?.Repairable?.MaxDurability);
    });

    it("isWeaponTemplate() g36 weapon", () =>
    {
        const result = helper.isWeaponTemplate("623063e994fc3f7b302a9696");

        expect(result).toBe(true);
    });

    it("isWeaponTemplate() slick armor", () =>
    {
        const result = helper.isWeaponTemplate("5e4abb5086f77406975c9342");

        expect(result).toBe(false);
    });
});