Creates a temporary testing account and profile for each test suite. This is really slow... May have to think this one through.

This commit is contained in:
Refringe 2023-10-29 22:21:21 -04:00
parent cc39c1847c
commit 4e33befaa0
No known key found for this signature in database
GPG Key ID: 64E03E5F892C6F9E

View File

@ -4,35 +4,56 @@ import path from "node:path";
// For the Vitest Custom Environment. // For the Vitest Custom Environment.
import type { Environment } from "vitest"; import type { Environment } from "vitest";
import { populateGlobal } from "vitest/environments";
// Import everything we intend to test so we can register it in the Jest custom environment.
import { Container } from "@spt-aki/di/Container"; import { Container } from "@spt-aki/di/Container";
// Required for importing the database.
import { IDatabaseTables } from "@spt-aki/models/spt/server/IDatabaseTables"; import { IDatabaseTables } from "@spt-aki/models/spt/server/IDatabaseTables";
import { DatabaseServer } from "@spt-aki/servers/DatabaseServer"; import { DatabaseServer } from "@spt-aki/servers/DatabaseServer";
import { ImporterUtil } from "@spt-aki/utils/ImporterUtil"; import { ImporterUtil } from "@spt-aki/utils/ImporterUtil";
// Required for creating a temporary test account/profile.
import { SaveServer } from "@spt-aki/servers/SaveServer";
import { LauncherController } from "@spt-aki/controllers/LauncherController";
import { IRegisterData } from "@spt-aki/models/eft/launcher/IRegisterData";
import { RandomUtil } from "@spt-aki/utils/RandomUtil";
import { ProfileController } from "@spt-aki/controllers/ProfileController";
import { IProfileCreateRequestData } from "@spt-aki/models/eft/profile/IProfileCreateRequestData";
export default <Environment> { export default <Environment> {
name: "spt-aki-server", name: "spt-aki-server",
transformMode: "ssr", transformMode: "ssr",
async setup() async setup()
{ {
// Register all of the dependencies in the container.
Container.registerTypes(container); Container.registerTypes(container);
Container.registerListTypes(container); Container.registerListTypes(container);
// Import the database. // Import the database.
await importDatabase(container); await importDatabase(container);
// TODO: Create test account/profile // Create a temporary test account/profile.
const sessionId = await createTestAccount();
// Populate the global scope with the session ID of the test account/profile.
populateGlobal(global, { sessionId: sessionId });
return { return {
teardown() async teardown()
{ {
// TODO: Delete test account/profile // Delete the temporary test account/profile.
await deleteTestAccount(sessionId);
} }
}; };
} }
}; };
/**
* 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> async function importDatabase(container: DependencyContainer): Promise<void>
{ {
const importerUtil = container.resolve<ImporterUtil>("ImporterUtil"); const importerUtil = container.resolve<ImporterUtil>("ImporterUtil");
@ -45,3 +66,55 @@ async function importDatabase(container: DependencyContainer): Promise<void>
// Save the data to memory. // Save the data to memory.
databaseServer.setTables(dataToImport); databaseServer.setTables(dataToImport);
} }
/**
* Creates a temporary test account/profile for testing purposes. Is deleted after testing is completed. The account is
* created with a random username, and the profile is created with the same username.
*
* @returns The session id of the created profile.
*/
async function createTestAccount(): Promise<string>
{
const launcherController = container.resolve<LauncherController>("LauncherController");
const randomUtil = container.resolve<RandomUtil>("RandomUtil");
const profileController = container.resolve<ProfileController>("ProfileController");
const username = `DarkHelmet-Test-${randomUtil.randInt(999999)}`;
// Register a new account.
const registerData: IRegisterData = {
edition: "Standard",
username: username,
password: "12345" // Get it? Heh
};
const sessionId = launcherController.register(registerData);
// It *could* happen. I guess.
if (!sessionId)
{
throw new Error("Failed to register a test account. Account already exists with this random username. Crazy.");
}
// Create a new profile for the account.
const profileCreateRequestData: IProfileCreateRequestData = {
side: "Usec",
nickname: username,
headId: "5cde96047d6c8b20b577f016",
voiceId: "5fc1223595572123ae7384a3"
};
profileController.createProfile(profileCreateRequestData, sessionId);
return sessionId;
}
/**
* Deletes the temporary test account/profile.
*
* @param $sessionId Session id of the profile to delete.
* @returns true when deleted, false when profile not found.
*/
async function deleteTestAccount($sessionId: string): Promise<boolean>
{
const saveServer = container.resolve<SaveServer>("SaveServer");
return saveServer.removeProfile($sessionId);
}