From 82979d291abf1ea044092fc94bc5aeb5b3eeae74 Mon Sep 17 00:00:00 2001
From: zkldi <ddhitone@gmail.com>
Date: Sun, 4 Apr 2021 17:56:51 +0100
Subject: [PATCH] CreateBetaKey/AddBetaKey tests

---
 server/src/core/auth-core.test.ts | 43 +++++++++++++++++++++++++++++--
 1 file changed, 41 insertions(+), 2 deletions(-)

diff --git a/server/src/core/auth-core.test.ts b/server/src/core/auth-core.test.ts
index b424d05b0..fdf2eb2fb 100644
--- a/server/src/core/auth-core.test.ts
+++ b/server/src/core/auth-core.test.ts
@@ -1,6 +1,12 @@
-import { AddNewUser, AddNewUserAPIKey, CreateAPIKey, ReinstateBetakey } from "./auth-core";
+import {
+    AddNewUserAPIKey,
+    CreateAPIKey,
+    CreateBetaKey,
+    AddNewBetakey,
+    ReinstateBetakey,
+} from "./auth-core";
 import t from "tap";
-import db, { CloseConnection, ReOpenConnection } from "../db";
+import db, { CloseConnection } from "../db";
 import { BetaKeyDocument, PrivateUserDocument } from "kamaitachi-common";
 import prAssert from "../test-utils/prassert";
 import Prudence from "prudence";
@@ -80,4 +86,37 @@ t.test("#ReinstateBetaKey", (t) => {
     t.end();
 });
 
+t.test("#CreateBetaKey", (t) => {
+    t.match(CreateBetaKey(), /^[0-9a-f]{40}$/, "Betakey should be a 40 character hex string.");
+
+    t.end();
+});
+
+t.test("#AddNewBetakey", (t) => {
+    t.beforeEach(ResetDBState);
+
+    t.test("Should create a new beta key for a given user", async (t) => {
+        let userDoc = await db.get("users").findOne({ id: 1 });
+
+        let result = await AddNewBetakey(userDoc);
+
+        t.is(result.createdBy, userDoc.id, "Betakey should be created by the requesting user.");
+        t.is(result.consumed, false, "Betakey should not be consumed.");
+
+        // was created +/- 6 seconds from now. This is perhaps too lenient, but we're only really testing its just around now ish.
+        t.assert(
+            Math.abs(result.createdOn - Date.now()) <= 6000,
+            "Betakey was created roughly now."
+        );
+
+        t.match(
+            result.betakey,
+            /^[0-9a-f]{40}$/,
+            "Betakey should have contents of a 40 character hex string."
+        );
+    });
+
+    t.end();
+});
+
 t.teardown(CloseConnection);