From 6f11e3de34e17cf21855fcf233c4a76587b7685b Mon Sep 17 00:00:00 2001
From: Dev <dev@dev.sp-tarkov.com>
Date: Sun, 5 Nov 2023 16:44:08 +0000
Subject: [PATCH] Add `ItemBaseClassService` tests

---
 .../services/ItemBaseClassService.test.ts     | 149 ++++++++++++++++++
 1 file changed, 149 insertions(+)
 create mode 100644 project/tests/services/ItemBaseClassService.test.ts

diff --git a/project/tests/services/ItemBaseClassService.test.ts b/project/tests/services/ItemBaseClassService.test.ts
new file mode 100644
index 00000000..716d6d53
--- /dev/null
+++ b/project/tests/services/ItemBaseClassService.test.ts
@@ -0,0 +1,149 @@
+/* eslint-disable @typescript-eslint/naming-convention */
+import "reflect-metadata";
+import { container } from "tsyringe";
+import { vi, beforeAll, afterEach, describe, expect, it } from "vitest";
+import { ItemBaseClassService } from "@spt-aki/services/ItemBaseClassService";
+
+describe("ItemBaseClassService", () =>
+{
+    let itemBaseClassService: any;
+
+    beforeAll(() =>
+    {
+        itemBaseClassService = container.resolve<ItemBaseClassService>("ItemBaseClassService");
+    });
+
+    afterEach(() =>
+    {
+        vi.restoreAllMocks();
+    });
+
+    describe("hydrateItemBaseClassCache", () =>
+    {
+        it("should hydrate itemBaseClassesCache dictionary property", () =>
+        {
+            itemBaseClassService.cacheGenerated = false;
+            itemBaseClassService.itemBaseClassesCache = {};
+            itemBaseClassService.hydrateItemBaseClassCache();
+
+            expect(Object.keys(itemBaseClassService.itemBaseClassesCache).length).greaterThan(100);
+        });
+    });
+
+    describe("itemHasBaseClass", () =>
+    {
+        it("should return false when undefined is passed in", () =>
+        {
+            const result = itemBaseClassService.itemHasBaseClass(undefined, []);
+
+            expect(result).toBe(false);
+        });
+
+        it("should return false when the base item type is passed in", () =>
+        {
+            // Remove item from base cache
+            const result = itemBaseClassService.itemHasBaseClass("54009119af1c881c07000029", []); // "Base item"
+
+            expect(result).toBe(false);
+        });
+
+        it("should return true when an item is passed in", () =>
+        {
+            const salewaTpl = "544fb45d4bdc2dee738b4568";
+
+            // Remove item from base cache
+            delete itemBaseClassService.itemBaseClassesCache[salewaTpl];
+            const result = itemBaseClassService.itemHasBaseClass(salewaTpl, ["543be5664bdc2dd4348b4569"]); // "Meds" type
+
+            expect(result).toBe(true);
+        });
+
+        it("should return true when an item and 2 matching base classes are passed in", () =>
+        {
+            const salewaTpl = "544fb45d4bdc2dee738b4568";
+
+            // Remove item from base cache
+            delete itemBaseClassService.itemBaseClassesCache[salewaTpl];
+            const result = itemBaseClassService.itemHasBaseClass(salewaTpl, ["543be5664bdc2dd4348b4569", "54009119af1c881c07000029"]); // "Meds" and "Item" type
+
+            expect(result).toBe(true);
+        });
+
+        it("should return true when an item is passed in and cache has not been generated", () =>
+        {
+            // Set cache to false
+            itemBaseClassService.cacheGenerated = false;
+            
+            const hydrateItemBaseClassCacheSpy = vi.spyOn(itemBaseClassService, "hydrateItemBaseClassCache");
+            
+            // Remove item from base cache
+            const salewaTpl = "544fb45d4bdc2dee738b4568";
+            delete itemBaseClassService.itemBaseClassesCache[salewaTpl];
+
+            // Perform check
+            const result = itemBaseClassService.itemHasBaseClass(salewaTpl, ["543be5664bdc2dd4348b4569"]); // "Meds" type
+
+            expect(result).toBe(true);
+            expect(hydrateItemBaseClassCacheSpy).toHaveBeenCalled();
+        });
+
+        it("should throw an exception when an invalid item is passed in", () =>
+        {
+            expect(() => itemBaseClassService.itemHasBaseClass("fakeTpl", ["543be5664bdc2dd4348b4569"])).toThrowError();
+        });
+    });
+
+    describe("getItemBaseClasses", () =>
+    {
+        it("should return empty array when undefined is passed in", () =>
+        {
+            const result = itemBaseClassService.getItemBaseClasses(undefined);
+
+            expect(result).toStrictEqual([]);
+        });
+
+        it("should return empty array when the base item type is passed in", () =>
+        {
+            // Remove item from base cache
+            const result = itemBaseClassService.getItemBaseClasses("54009119af1c881c07000029");
+
+            expect(result).toStrictEqual([]);
+        });
+
+        it("should return array of 3 items when an item is passed in", () =>
+        {
+            const salewaTpl = "544fb45d4bdc2dee738b4568";
+
+            const result = itemBaseClassService.getItemBaseClasses(salewaTpl);
+
+            expect(result.length).toBe(3);
+        });
+
+        it("should return array of 3 items when an item is passed in and cache has not been generated", () =>
+        {
+            itemBaseClassService.cacheGenerated = false;
+            const hydrateItemBaseClassCacheSpy = vi.spyOn(itemBaseClassService, "hydrateItemBaseClassCache");
+
+            const salewaTpl = "544fb45d4bdc2dee738b4568";
+            const result = itemBaseClassService.getItemBaseClasses(salewaTpl);
+
+            expect(result.length).toBe(3);
+            expect(hydrateItemBaseClassCacheSpy).toHaveBeenCalled();
+        });
+
+        it("should return base item type when an item is passed in", () =>
+        {
+            const salewaTpl = "544fb45d4bdc2dee738b4568";
+            const result = itemBaseClassService.getItemBaseClasses(salewaTpl);
+
+            expect(result).toContain("54009119af1c881c07000029");
+            expect(result).toContain("5448f39d4bdc2d0a728b4568");
+        });
+
+        it("should return empty array when an invalid item is passed in", () =>
+        {
+            const result = itemBaseClassService.getItemBaseClasses("fakeTpl");
+            expect(result).toStrictEqual([]);
+        });
+    });
+});