From 4e33befaa02649c4601610ee318bc01a399a64a8 Mon Sep 17 00:00:00 2001
From: Refringe <brownelltyler@gmail.com>
Date: Sun, 29 Oct 2023 22:21:21 -0400
Subject: [PATCH] Creates a temporary testing account and profile for each test
 suite. This is really slow... May have to think this one through.

---
 project/tests/CustomEnvironment.ts | 83 ++++++++++++++++++++++++++++--
 1 file changed, 78 insertions(+), 5 deletions(-)

diff --git a/project/tests/CustomEnvironment.ts b/project/tests/CustomEnvironment.ts
index 88f00c0f..cbc55fcc 100644
--- a/project/tests/CustomEnvironment.ts
+++ b/project/tests/CustomEnvironment.ts
@@ -4,35 +4,56 @@ import path from "node:path";
 
 // For the Vitest Custom Environment.
 import type { Environment } from "vitest";
-
-// Import everything we intend to test so we can register it in the Jest custom environment.
+import { populateGlobal } from "vitest/environments";
 import { Container } from "@spt-aki/di/Container";
+
+// Required for importing the database.
 import { IDatabaseTables } from "@spt-aki/models/spt/server/IDatabaseTables";
 import { DatabaseServer } from "@spt-aki/servers/DatabaseServer";
 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> {
     name: "spt-aki-server",
     transformMode: "ssr",
     async setup()
     {
+        // Register all of the dependencies in the container.
         Container.registerTypes(container);
         Container.registerListTypes(container);
 
         // Import the database.
         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 {
-            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>
 {
     const importerUtil = container.resolve<ImporterUtil>("ImporterUtil");
@@ -45,3 +66,55 @@ async function importDatabase(container: DependencyContainer): Promise<void>
     // Save the data to memory.
     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);
+}