Server/project/tests/CustomEnvironment.ts

import "reflect-metadata";

import path from "node:path";
import { Container } from "@spt/di/Container";
import { IDatabaseTables } from "@spt/models/spt/server/IDatabaseTables";
import { DatabaseServer } from "@spt/servers/DatabaseServer";
import { ImporterUtil } from "@spt/utils/ImporterUtil";
import { DependencyContainer, Lifecycle, container } from "tsyringe";
import type { Environment } from "vitest";

// Manually mock the logger.
import { WinstonLogger } from "@tests/__mocks__/WinstonLogger.mock";

export default (<Environment>{
    name: "spt-server",
    transformMode: "ssr",
    async setup() {
        // Register all of the dependencies in the container.
        Container.registerTypes(container);
        Container.registerListTypes(container);

        // Override registration to the container.
        container.register<WinstonLogger>("WinstonLogger", WinstonLogger, { lifecycle: Lifecycle.Singleton });

        // Import the database.
        await importDatabase(container);

        return {
            async teardown() {},
        };
    },
});

/**
 * Reads the database JSON files and imports them into memory.
 *
 * @param container The dependency container.
 * @returns A void promise.
 */
async function importDatabase(container: DependencyContainer): Promise<void> {
    const importerUtil = container.resolve<ImporterUtil>("ImporterUtil");
    const databaseServer = container.resolve<DatabaseServer>("DatabaseServer");

    // Read the data from the JSON files.
    const databaseDir = path.resolve("./assets/database");
    const dataToImport = await importerUtil.loadAsync<IDatabaseTables>(`${databaseDir}/`);

    // Save the data to memory.
    databaseServer.setTables(dataToImport);
}