begin adding tests

Author: akintner

query-connector/src/app/tests/integration/user-queries.test.ts

@@ -0,0 +1,106 @@
+import { addUserIfNotExists } from "@/app/backend/user-management";
+import {
+  getUsersWithGroupStatus,
+  getQueriesWithGroupStatus,
+  addQueriesToGroup,
+  saveQueryGroupMembership,
+} from "@/app/backend/usergroup-management";
+import { auth } from "@/auth";
+import { getDbClient } from "@/app/backend/dbClient";
+import { User, Query } from "@/app/models/entities/users";
+import { suppressConsoleLogs } from "./fixtures";
+
+const dbClient = getDbClient();
+
+jest.mock("@/app/utils/auth", () => {
+  return {
+    superAdminAccessCheck: jest.fn(() => Promise.resolve(true)),
+    adminAccessCheck: jest.fn(() => Promise.resolve(true)),
+  };
+});
+
+const TEST_USER = {
+  id: "13e1efb2-5889-4157-8f34-78d7f02dbf84",
+  username: "Q the omnipotent",
+  email: "worf_security@[email protected]",
+  firstName: "Q",
+  lastName: "Q178652",
+};
+const TEST_GROUP_ID = "00000000-0000-0000-0000-000000000001";
+const TEST_QUERY_ID = "00000000-0000-0000-0000-000000000004";
+
+(auth as jest.Mock).mockResolvedValue(TEST_USER);
+
+describe("User Queries Integration Tests", () => {
+  beforeAll(async () => {
+    suppressConsoleLogs();
+    await dbClient.query("BEGIN");
+
+   // Insert test group
+   const insertGroupQuery = `
+   INSERT INTO usergroup (id, name)
+   VALUES ($1, 'Q Continuum Membership List');
+ `;
+    await dbClient.query(insertGroupQuery, [TEST_GROUP_ID]);
+
+  // Insert test query
+  const insertQueryQuery = `
+    INSERT INTO query (id, query_name)
+    VALUES ($1, 'What is the nature of the universe?');
+  `;
+    await dbClient.query(insertQueryQuery, [TEST_QUERY_ID]);
+  });
+
+  afterAll(async () => {
+    await dbClient.query("ROLLBACK");
+  });
+
+  test("should find user and know their role", async () => {
+    const result = await addUserIfNotExists(TEST_USER);
+    const createdUser: User[] = await getUsersWithGroupStatus(TEST_GROUP_ID);
+
+    expect(Array.isArray(result)).toBe(true);
+    expect(result[0]).toHaveProperty("id");
+    expect(result[0]).toHaveProperty("username");
+    expect(result[0]).toHaveProperty("userGroupMemberships");
+
+    const membership = result[0].userGroupMemberships?.find(
+      (m: { usergroup_id: string; group_name: string }) =>
+        m.usergroup_id === TEST_GROUP_ID,
+      (m: { usergroup_id: string; group_name: string }) =>
+        m.group_name === "Q Continuum Membership List",
+    );
+    expect(membership).toBeDefined();
+    expect(membership?.is_member).toBeDefined();
+  });
+
+  test("should know queries belong to a group", async () => {
+    const result = await addQueriesToGroup(TEST_GROUP_ID, [TEST_QUERY_ID]);
+    expect(result).toContain(TEST_QUERY_ID);
+    expect(result.length).toBe(1);
+
+    const updatedQueries: Query[] =
+      await getQueriesWithGroupStatus(TEST_GROUP_ID);
+    const members = updatedQueries.filter((query) =>
+      query.userGroupMemberships?.some((m) => m.is_member),
+    );
+
+    expect(members.length).toBe(1);
+    expect(members.some((query) => query.id === TEST_QUERY_ID)).toBe(true);
+    expect(members.some((query) => query.name === "What is the nature of the universe?")).toBe(true);
+  });
+
+  test("should know query group memberships", async () => {
+    const updatedQueries = await saveQueryGroupMembership(TEST_GROUP_ID, [
+      TEST_QUERY_ID,
+    ]);
+
+    expect(
+      updatedQueries.some(
+        (query) =>
+          query.id === TEST_QUERY_ID &&
+          query.userGroupMemberships?.some((m) => m.is_member),
+      ),
+    ).toBe(true);
+  });
+});