Make tests pass running with postgres

src/databases/IDatabase.ts

@@ -1,5 +1,5 @@
 export interface IDatabase {
-    init(): void;
+    async init(): Promise<void>;
 
     prepare(type: QueryType, query: string, params?: any[]): Promise<any | any[] | void>;
 }

src/databases/Mysql.ts

@@ -9,7 +9,7 @@ export class Mysql implements IDatabase {
     constructor(private config: any) {
     }
 
-    init(): void {
+    async init(): Promise<void> {
         this.connection = new MysqlInterface(this.config);
     }
 

src/databases/Postgres.ts

@@ -1,6 +1,6 @@
 import { Logger } from '../utils/logger';
 import { IDatabase, QueryType } from './IDatabase';
-import { Pool } from 'pg';
+import { Client, Pool } from 'pg';
 
 import fs from "fs";
 
@@ -13,6 +13,10 @@ export class Postgres implements IDatabase {
         this.pool = new Pool(this.config.postgres);
 
         if (!this.config.readOnly) {
+            if (this.config.createDbIfNotExists) {
+                await this.createDB();
+            }
+
             if (this.config.createDbIfNotExists && !this.config.readOnly && fs.existsSync(this.config.dbSchemaFileName)) {
                 await this.pool.query(this.processUpgradeQuery(fs.readFileSync(this.config.dbSchemaFileName).toString()));
             }
@@ -74,6 +78,29 @@ export class Postgres implements IDatabase {
         }
     }
 
+    private async createDB() {
+        const client = new Client({
+            ...this.config.postgres,
+            database: "postgres"
+        });
+
+        await client.connect();
+        
+        if ((await client.query(`SELECT * FROM pg_database WHERE datname = '${this.config.postgres.database}'`)).rowCount == 0) {
+            await client.query(`CREATE DATABASE "${this.config.postgres.database}"
+                                WITH 
+                                OWNER = ${this.config.postgres.user}
+                                ENCODING = 'UTF8'
+                                LC_COLLATE = 'en_US.utf8'
+                                LC_CTYPE = 'en_US.utf8'
+                                TABLESPACE = pg_default
+                                CONNECTION LIMIT = -1;`
+            );
+        }
+
+        client.end();
+    }
+
     private async upgradeDB(fileNamePrefix: string, schemaFolder: string) {
         const versionCodeInfo = await this.pool.query("SELECT value FROM config WHERE key = 'version'");
         let versionCode = versionCodeInfo.rows[0] ? versionCodeInfo.rows[0].value : 0;

src/databases/Sqlite.ts

@@ -30,7 +30,7 @@ export class Sqlite implements IDatabase {
         }
     }
 
-    init() {
+    async init() {
         // Make dirs if required
         if (!fs.existsSync(path.join(this.config.dbPath, "../"))) {
             fs.mkdirSync(path.join(this.config.dbPath, "../"));

src/databases/databases.ts

@@ -60,9 +60,9 @@ if (config.mysql) {
         enableWalCheckpointNumber: false
     });
 }
-function initDb() {
-    db.init();
-    privateDB.init();
+async function initDb() {
+    await db.init();
+    await privateDB.init();
 
     if (db instanceof Sqlite) {
         // Attach private db to main db

src/index.ts

@@ -3,7 +3,12 @@ import {initDb} from './databases/databases';
 import {createServer} from "./app";
 import {Logger} from "./utils/logger";
 
-initDb();
-createServer(() => {
-  Logger.info("Server started on port " + config.port + ".");
-});
+async function init() {
+    await initDb();
+
+    createServer(() => {
+        Logger.info("Server started on port " + config.port + ".");
+    });
+}
+ 
+init();

src/routes/shadowBanUser.ts

@@ -52,7 +52,7 @@ export async function shadowBanUser(req: Request, res: Response) {
 
             //find all previous submissions and unhide them
             if (unHideOldSubmissions) {
-                let segmentsToIgnore = (await db.prepare('all', `SELECT UUID FROM "sponsorTimes" st
+                let segmentsToIgnore = (await db.prepare('all', `SELECT "UUID" FROM "sponsorTimes" st
                                 JOIN "noSegments" ns on "st"."videoID" = "ns"."videoID" AND st.category = ns.category WHERE "st"."userID" = ?`
                                     , [userID])).map((item: {UUID: string}) => item.UUID);
                 let allSegments = (await db.prepare('all', `SELECT "UUID" FROM "sponsorTimes" st WHERE "st"."userID" = ?`, [userID]))

src/routes/voteOnSponsorTime.ts

@@ -357,7 +357,7 @@ export async function voteOnSponsorTime(req: Request, res: Response) {
         // Only change the database if they have made a submission before and haven't voted recently
         const ableToVote = isVIP
             || ((await db.prepare("get", `SELECT "userID" FROM "sponsorTimes" WHERE "userID" = ?`, [nonAnonUserID])) !== undefined
-                && (await privateDB.prepare("get", `SELECT userID FROM "shadowBannedUsers" WHERE "userID" = ?`, [nonAnonUserID])) === undefined
+                && (await privateDB.prepare("get", `SELECT "userID" FROM "shadowBannedUsers" WHERE "userID" = ?`, [nonAnonUserID])) === undefined
                 && (await privateDB.prepare("get", `SELECT "UUID" FROM "votes" WHERE "UUID" = ? AND "hashedIP" = ? AND "userID" != ?`, [UUID, hashedIP, userID])) === undefined);
 
         if (ableToVote) {