From e2362604c71c1782996f077560bd643301a80663 Mon Sep 17 00:00:00 2001
From: Jesse Wierzbinski <contact@cpluspatch.com>
Date: Thu, 16 May 2024 22:34:42 -1000
Subject: [PATCH] test(api): :white_check_mark: Add tests for password reset

---
 server/api/api/auth/reset/index.test.ts | 152 ++++++++++++++++++++++++
 1 file changed, 152 insertions(+)
 create mode 100644 server/api/api/auth/reset/index.test.ts

diff --git a/server/api/api/auth/reset/index.test.ts b/server/api/api/auth/reset/index.test.ts
new file mode 100644
index 00000000..228feb22
--- /dev/null
+++ b/server/api/api/auth/reset/index.test.ts
@@ -0,0 +1,152 @@
+import { afterAll, describe, expect, test } from "bun:test";
+import { randomBytes } from "node:crypto";
+import { eq } from "drizzle-orm";
+import { db } from "~drizzle/db";
+import { Applications } from "~drizzle/schema";
+import { config } from "~packages/config-manager";
+import {
+    deleteOldTestUsers,
+    getTestUsers,
+    sendTestRequest,
+} from "~tests/utils";
+import { meta } from "./index";
+
+await deleteOldTestUsers();
+
+const { users, deleteUsers, passwords } = await getTestUsers(1);
+const token = randomBytes(32).toString("hex");
+const newPassword = randomBytes(16).toString("hex");
+
+// Create application
+const application = (
+    await db
+        .insert(Applications)
+        .values({
+            name: "Test Application",
+            clientId: randomBytes(32).toString("hex"),
+            secret: "test",
+            redirectUri: "https://example.com",
+            scopes: "read write",
+        })
+        .returning()
+)[0];
+
+afterAll(async () => {
+    await deleteUsers();
+    await db.delete(Applications).where(eq(Applications.id, application.id));
+});
+
+// /api/auth/reset
+describe(meta.route, () => {
+    test("should login with normal password", async () => {
+        const formData = new FormData();
+
+        formData.append("identifier", users[0]?.getUser().username ?? "");
+        formData.append("password", passwords[0]);
+
+        const response = await sendTestRequest(
+            new Request(
+                new URL(
+                    `/api/auth/login?client_id=${application.clientId}&redirect_uri=https://example.com&response_type=code&scope=read+write`,
+                    config.http.base_url,
+                ),
+                {
+                    method: "POST",
+                    body: formData,
+                },
+            ),
+        );
+
+        expect(response.status).toBe(302);
+        expect(response.headers.get("location")).toBeDefined();
+    });
+
+    test("should reset password and refuse login with old password", async () => {
+        await users[0]?.update({
+            passwordResetToken: token,
+        });
+
+        const formData = new FormData();
+
+        formData.append("identifier", users[0]?.getUser().username ?? "");
+        formData.append("password", passwords[0]);
+
+        const response = await sendTestRequest(
+            new Request(
+                new URL(
+                    `/api/auth/login?client_id=${application.clientId}&redirect_uri=https://example.com&response_type=code&scope=read+write`,
+                    config.http.base_url,
+                ),
+                {
+                    method: "POST",
+                    body: formData,
+                },
+            ),
+        );
+
+        expect(response.status).toBe(302);
+        expect(response.headers.get("location")).toBeDefined();
+        const locationHeader = new URL(
+            response.headers.get("Location") ?? "",
+            "",
+        );
+
+        expect(locationHeader.pathname).toBe("/oauth/reset");
+        expect(locationHeader.searchParams.get("token")).toBe(token);
+    });
+
+    test("should reset password and login with new password", async () => {
+        const formData = new FormData();
+
+        formData.append("token", token);
+        formData.append("password", newPassword);
+        formData.append("password2", newPassword);
+
+        const response = await sendTestRequest(
+            new Request(new URL("/api/auth/reset", config.http.base_url), {
+                method: "POST",
+                body: formData,
+            }),
+        );
+
+        expect(response.status).toBe(302);
+        expect(response.headers.get("location")).toBeDefined();
+
+        const loginFormData = new FormData();
+
+        loginFormData.append("identifier", users[0]?.getUser().username ?? "");
+        loginFormData.append("password", newPassword);
+
+        const loginResponse = await sendTestRequest(
+            new Request(
+                new URL(
+                    `/api/auth/login?client_id=${application.clientId}&redirect_uri=https://example.com&response_type=code&scope=read+write`,
+                    config.http.base_url,
+                ),
+                {
+                    method: "POST",
+                    body: loginFormData,
+                },
+            ),
+        );
+
+        expect(loginResponse.status).toBe(302);
+        expect(loginResponse.headers.get("location")).toBeDefined();
+        const locationHeader = new URL(
+            loginResponse.headers.get("Location") ?? "",
+            "",
+        );
+
+        expect(locationHeader.pathname).toBe("/oauth/consent");
+        expect(locationHeader.searchParams.get("client_id")).toBe(
+            application.clientId,
+        );
+        expect(locationHeader.searchParams.get("redirect_uri")).toBe(
+            "https://example.com",
+        );
+        expect(locationHeader.searchParams.get("response_type")).toBe("code");
+        expect(locationHeader.searchParams.get("scope")).toBe("read write");
+
+        expect(loginResponse.headers.get("Set-Cookie")).toMatch(/jwt=[^;]+;/);
+    });
+});