seed function

Author: pyramation

packages/rls-demo/__tests__/hello-world.test.ts

@@ -9,11 +9,29 @@ let user1: any;
 let user2: any;
 let user3: any;
 
+const users = [
+  {
+    id: '550e8400-e29b-41d4-a716-446655440001',
+    email: '[email protected]',
+    name: 'Tutorial User 1'
+  },
+  {
+    id: '550e8400-e29b-41d4-a716-446655440002',
+    email: '[email protected]',
+    name: 'Tutorial User 2'
+  },
+  {
+    id: '550e8400-e29b-41d4-a716-446655440003',
+    email: '[email protected]',
+    name: 'Tutorial User 3'
+  }
+];
+
 beforeAll(async () => {
   ({ pg, db, teardown } = await getConnections());
-  user1 = await insertUser(pg, '[email protected]');
-  user2 = await insertUser(pg, '[email protected]');
-  user3 = await insertUser(pg, '[email protected]');
+  user1 = await insertUser(pg, users[0].email, users[0].id);
+  user2 = await insertUser(pg, users[1].email, users[1].id);
+  user3 = await insertUser(pg, users[2].email, users[2].id);
 });
 
 afterAll(async () => {

packages/rls-demo/test-utils/index.ts

@@ -4,32 +4,26 @@ import { PgTestClient } from "supabase-test";
  * Helper function to insert a new user into auth.users
  * @param client - The PgTestClient to use (pg or db)
  * @param email - The user's email address
- * @param options - Optional additional fields and return options
- * @returns The inserted user object
+ * @param id - Optional user ID (UUID). If not provided, a random UUID will be generated.
+ * @returns The inserted user object with id and email
  */
 export async function insertUser(
     client: PgTestClient,
     email: string,
-    options?: {
-      id?: string;
-      returnFields?: string[];
-    }
-  ): Promise<{ id: string; email: string; [key: string]: any }> {
-    const returnFields = options?.returnFields || ['id', 'email'];
-    const fields = returnFields.join(', ');
-    
-    if (options?.id) {
+    id?: string
+  ): Promise<{ id: string; email: string }> {
+    if (id) {
       return await client.one(
         `INSERT INTO auth.users (id, email) 
          VALUES ($1, $2) 
-         RETURNING ${fields}`,
-        [options.id, email]
+         RETURNING id, email`,
+        [id, email]
       );
     } else {
       return await client.one(
         `INSERT INTO auth.users (id, email) 
          VALUES (gen_random_uuid(), $1) 
-         RETURNING ${fields}`,
+         RETURNING id, email`,
         [email]
       );
     }