Adding more directory-related unit tests. Several more bug fixes and QoL improvements

src/PersonObjects/Player/PlayerObject.js

@@ -1,3 +1,4 @@
+import * as augmentationMethods from "./PlayerObjectAugmentationMethods";
 import * as bladeburnerMethods from "./PlayerObjectBladeburnerMethods";
 import * as corporationMethods from "./PlayerObjectCorporationMethods";
 import * as gangMethods from "./PlayerObjectGangMethods";
@@ -209,7 +210,8 @@ Object.assign(
     serverMethods,
     bladeburnerMethods,
     corporationMethods,
-    gangMethods
+    gangMethods,
+    augmentationMethods
 );
 
 PlayerObject.prototype.toJSON = function() {

src/PersonObjects/Player/PlayerObjectAugmentationMethods.ts

@@ -0,0 +1,20 @@
+/**
+ * Augmentation-related methods for the Player class (PlayerObject)
+ */
+import { IPlayer } from "../IPlayer";
+
+import { Augmentation } from "../../Augmentation/Augmentation";
+
+export function hasAugmentation(this: IPlayer, aug: string | Augmentation): boolean {
+    const augName: string = (aug instanceof Augmentation) ? aug.name : aug;
+
+    for (const owned of this.augmentations) {
+        if (owned.name === augName) { return true; }
+    }
+
+    for (const owned of this.queuedAugmentations) {
+        if (owned.name === augName) { return true; }
+    }
+
+    return false;
+}

src/PersonObjects/Player/PlayerObjectGeneralMethods.js

@@ -219,6 +219,11 @@ export function prestigeSourceFile() {
         }
     }
 
+    const characterMenuHeader = document.getElementById("character-menu-header");
+    if (characterMenuHeader instanceof HTMLElement) {
+        characterMenuHeader.click(); characterMenuHeader.click();
+    }
+
     this.isWorking = false;
     this.currentWorkFactionName = "";
     this.currentWorkFactionDescription = "";
@@ -1567,7 +1572,9 @@ export function hospitalize() {
         );
     }
 
-    this.loseMoney(this.max_hp * CONSTANTS.HospitalCostPerHp);
+    const cost = this.max_hp * CONSTANTS.HospitalCostPerHp
+    this.loseMoney(cost);
+    this.recordMoneySource(-1 * cost, "hospitalization");
     this.hp = this.max_hp;
 }