feature: add user database tests (#6)

src/lib/__test__/database.test.ts

@@ -2,32 +2,42 @@ import {Firestore} from "@google-cloud/firestore";
 import {Database} from "../database";
 import {User} from "../../models/User";
 
+const directDatabaseConnectionForTestReset = new Firestore({
+  projectId: "birds-of-paradise",
+  // keyFilename: '/path/to/keyfile.json',
+});
+
 describe("database tests", () => {
   let fs: Database;
 
   beforeAll(async () => {
+    await directDatabaseConnectionForTestReset.collection('users').doc('Bob').delete()
     fs = new Database();
   })
 
-  it.skip("should pass (just a placeholder)", () => {
-    expect(true).toBeTruthy();
-  });
+  it("should add and get a user", async () => {
+    await fs.setUser({userId: 'Bob'});
+    const user = await fs.getUser({userId: 'Bob'});
 
-  it("should add a doc", async () => {
+    expect(user).toEqual({ userId: "Bob", completedMissions: [] });
+  });
 
-    const docRef = fs.db.collection('users').doc("bob");
+  it("should add completed missions", async () => {
+    await fs.setUser({userId: 'Bob'});
+    await fs.addCompletedMission({userId: 'Bob', missionId: 'Mission0001aweifjwek'});
+    const firstUserResponse = await fs.getUser({userId: 'Bob'});
 
-    await docRef.set({
-      userId: "1",
-      email: "bob@example.com",
+    expect(firstUserResponse).toEqual({
+      "userId": "Bob",
+      completedMissions: ['Mission0001aweifjwek']
     });
 
-    const snapshot = await fs.db.collection('users').get();
-    snapshot.forEach((doc) => {
-      console.log(doc.id, '=>', doc.data());
-    });
+    await fs.addCompletedMission({userId: 'Bob', missionId: 'Mission0002aweifjwek'});
+    const secondUserResponse = await fs.getUser({userId: 'Bob'});
 
-    expect(snapshot.size).toEqual(1);
-    await fs.db.collection("users").doc("bob").delete()
+    expect(secondUserResponse).toEqual({
+      "userId": "Bob",
+      completedMissions: ['Mission0001aweifjwek', 'Mission0002aweifjwek']
+    });
   });
-});
+});

src/lib/database.ts

@@ -1,12 +1,41 @@
 import {Firestore} from "@google-cloud/firestore";
+import { User } from "src/models/User";
 
 export class Database {
-  db: Firestore;
+  private db: Firestore;
 
   constructor() {
     this.db = new Firestore({
       projectId: "birds-of-paradise",
       // keyFilename: '/path/to/keyfile.json',
     });
   }
-}
+
+
+  async setUser({userId} :any): Promise<any> {
+    const userDoc = this.db.collection('users').doc(userId);
+
+    return userDoc.set({
+      userId,
+    }, {merge: true});
+  }
+
+  async getUser({userId}: any): Promise<User> {
+    const userDoc = this.db.collection('users').doc(userId);
+    const snapshot = await userDoc.get();
+    const completedMissions = snapshot.data()?.completedMissions || [];
+
+    return { userId, completedMissions }
+  }
+
+  async addCompletedMission({userId, missionId} :any): Promise<any> {
+    const userDoc = this.db.collection('users').doc(userId);
+    const { completedMissions } = await this.getUser({userId});
+    const updatedMissions = [ ...completedMissions, missionId ]
+
+
+    return userDoc.set({
+      completedMissions: updatedMissions,
+    }, {merge: true});
+  }
+}

src/models/LearningResource.ts

@@ -0,0 +1,4 @@
+export class LearningResource {
+  title!: string;
+  link!: string;
+}

src/models/Mission.ts

@@ -0,0 +1,11 @@
+import { LearningResource } from "./LearningResource";
+
+export class Mission {
+  id!: string;
+  title!: string;
+  technologies!: string[];
+  learningResources!: LearningResource[];
+
+  // status is not stored in the database, it is calculated from the user object
+  status!: string;
+}

src/models/User.ts

@@ -1,15 +1,4 @@
-// TODO
-export type Email = string;
-export type Timestamp = number;
-export type SessionId = string;
-export type SessionResult = object | SessionId;
-
 export class User {
-  userId?: string;
-
-  email?: Email;
-
-  results?: SessionResult[];
-
-  lastSession?: Timestamp;
-}
+  userId!: string;
+  completedMissions!: string[];
+}