spt/server
spt/server
0
0
Fork 0
mirror of https://github.com/sp-tarkov/server.git synced 2025-02-13 09:50:43 -05:00
Code

Add HealthController.healthTreatment() tests

Browse Source
This commit is contained in:
Dev 2023-11-18 15:01:51 +00:00
parent 0d205ed50c
commit 60ead44b30
1 changed files with 219 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

219
project/tests/controllers/HealthController.test.ts Normal file
View File

@ -0,0 +1,219 @@
/* eslint-disable @typescript-eslint/naming-convention */
import "reflect-metadata";
import { container } from "tsyringe";
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
import { HealthController } from "@spt-aki/controllers/HealthController";
import { IPmcData } from "@spt-aki/models/eft/common/IPmcData";
import { IHealthTreatmentRequestData } from "@spt-aki/models/eft/health/IHealthTreatmentRequestData";
describe("HealthController", () =>
{
let healthController: HealthController; // Using "any" to access private/protected methods without type errors.
beforeEach(() =>
{
healthController = container.resolve<HealthController>("HealthController");
});
afterEach(() =>
{
vi.restoreAllMocks();
});
describe("healthTreatment", () =>
{
it("Should Heal Players heavy bleed and heal chest to full hp", () =>
{
const maxHealth = 100
const pmcData = {
Health: {
BodyParts: {
Chest: {
Health: {
Current: 50, // Has damage
Maximum: maxHealth
},
Effects: {HeavyBleeding: {
Time: 20
}
}
}
}
}
};
const bleedRemovalAndLimbHealRequest = {
Action: "RestoreHealth",
trader: "54cb57776803fa99248b456e", // Therapist
difference: {
BodyParts: {
Chest: {
Health: 23, // > 0 value means it will heal
Effects: ["HeavyBleeding"] // non-null means it will remove effect from player
}
}
}
};
const sessionId = "12345";
// Mock output generation
vi.spyOn((healthController as any).eventOutputHolder, "getOutput").mockReturnValue(
{
warnings: {},
profileChanges: {
"12345": {
health: {}
}
}
});
// Mock payment
vi.spyOn((healthController as any).paymentService, "payMoney").mockReturnValue(
{
warnings: {},
profileChanges: {
"12345": {
health: {}
}
}
});
const result = healthController.healthTreatment(pmcData as unknown as IPmcData, bleedRemovalAndLimbHealRequest as IHealthTreatmentRequestData, sessionId);
// Has healed chest to full
expect(result.profileChanges[sessionId].health.BodyParts.Chest.Health.Current).equals(maxHealth);
// Has removed Heavy bleed effect from chest
expect(result.profileChanges[sessionId].health.BodyParts.Chest).not.toHaveProperty("Effects");
});
it("Should Heal Players heavy bleed and leave limb health at existing value", () =>
{
const maxHealth = 100
const pmcData = {
Health: {
BodyParts: {
Chest: {
Health: {
Current: 50, // Has damage
Maximum: maxHealth
},
Effects: {HeavyBleeding: {
Time: 20
}
}
}
}
}
};
const limbOnlyHealRequest = {
Action: "RestoreHealth",
trader: "54cb57776803fa99248b456e", // Therapist
difference: {
BodyParts: {
Chest: {
Health: 23, // > 0 value means it will heal limb to full
Effects: null // null means no healing of effects
}
}
}
};
const sessionId = "12345";
// Mock output generation
vi.spyOn((healthController as any).eventOutputHolder, "getOutput").mockReturnValue(
{
warnings: {},
profileChanges: {
"12345": {
health: {}
}
}
});
// Mock payment
vi.spyOn((healthController as any).paymentService, "payMoney").mockReturnValue(
{
warnings: {},
profileChanges: {
"12345": {
health: {}
}
}
});
const result = healthController.healthTreatment(pmcData as unknown as IPmcData, limbOnlyHealRequest as IHealthTreatmentRequestData, sessionId);
// Has healed chest to full
expect(result.profileChanges[sessionId].health.BodyParts.Chest.Health.Current).equals(maxHealth);
// Has not removed Heavy bleed effect from chest
expect(result.profileChanges[sessionId].health.BodyParts.Chest).toHaveProperty("Effects");
});
it("Should Heal Players heavy bleed and leave limb health at existing value", () =>
{
const maxHealth = 100
const currentHealth = 50;
const pmcData = {
Health: {
BodyParts: {
Chest: {
Health: {
Current: currentHealth, // Has damage
Maximum: maxHealth
},
Effects: {HeavyBleeding: {
Time: 20
}
}
}
}
}
};
const limbOnlyHealRequest = {
Action: "RestoreHealth",
trader: "54cb57776803fa99248b456e", // Therapist
difference: {
BodyParts: {
Chest: {
Health: 0, // 0 value means it will not heal and damage
Effects: null // null means no healing of effects
}
}
}
};
const sessionId = "12345";
// Mock output generation
vi.spyOn((healthController as any).eventOutputHolder, "getOutput").mockReturnValue(
{
warnings: {},
profileChanges: {
"12345": {
health: {}
}
}
});
// Mock payment
vi.spyOn((healthController as any).paymentService, "payMoney").mockReturnValue(
{
warnings: {},
profileChanges: {
"12345": {
health: {}
}
}
});
const result = healthController.healthTreatment(pmcData as unknown as IPmcData, limbOnlyHealRequest as IHealthTreatmentRequestData, sessionId);
// Has not healed chest to full
expect(result.profileChanges[sessionId].health.BodyParts.Chest.Health.Current).equals(currentHealth);
// Has not removed Heavy bleed effect from chest
expect(result.profileChanges[sessionId].health.BodyParts.Chest).toHaveProperty("Effects");
});
});
});