examples

Author: pyramation

packages/hello-world/__tests__/seeding/seeding.csv.test.ts

@@ -1,20 +1,19 @@
-import { getConnections, PgTestClient, seed } from 'supabase-test';
 import path from 'path';
+import { getConnections, PgTestClient, seed } from 'supabase-test';
 import { users } from './data/seed-data';
 
 let pg: PgTestClient;
 let db: PgTestClient;
 let teardown: () => Promise<void>;
 
 const csv = (file: string) => path.resolve(__dirname, './data', file);
-const cwd = path.resolve(__dirname, '../../');
 
 beforeAll(async () => {
   ({ pg, db, teardown } = await getConnections(
     {},
     [
-      // create schema
-      seed.launchql(cwd),
+      // load schema and it's dependencies (supabase full schema)
+      seed.launchql(),
 
       // load from csv
       seed.csv({

packages/hello-world/__tests__/seeding/seeding.json.test.ts

@@ -1,23 +1,22 @@
 import { getConnections, PgTestClient, seed } from 'supabase-test';
-import path from 'path';
 import { pets, users } from './data/seed-data';
 
 let db: PgTestClient;
 let teardown: () => Promise<void>;
 
-const sql = (f: string) => path.join(__dirname, 'data', f);
-
-const cwd = path.resolve(__dirname, '../../');
-
 beforeAll(async () => {
   ({ db, teardown } = await getConnections(
-    {}, [
-    seed.launchql(cwd),
-    seed.json({
-      'auth.users': users,
-      'rls_test.pets': pets
-    })
-  ]
+    {},
+    [
+      // load schema and it's dependencies (supabase full schema)
+      seed.launchql(),
+
+      // load data from json files
+      seed.json({
+        'auth.users': users,
+        'rls_test.pets': pets
+      })
+    ]
   ));
 });
 
@@ -34,7 +33,7 @@ afterEach(async () => {
 });
 
 describe('tutorial: testing with sql file seeding', () => {
-  it('should work with sql file seed function', async () => {
+  it('data should be loaded into the database', async () => {
 
     db.setContext({
       role: 'authenticated',

packages/hello-world/__tests__/seeding/seeding.pg.test.ts

@@ -0,0 +1,102 @@
+import { getConnections, PgTestClient, seed } from 'supabase-test';
+import { pets, users } from './data/seed-data';
+
+let db: PgTestClient;
+let teardown: () => Promise<void>;
+
+beforeAll(async () => {
+  ({ db, teardown } = await getConnections(
+    {},
+    [
+      // load schema and it's dependencies (supabase full schema)
+      seed.launchql(),
+
+      // load data from json files
+      seed.fn(async ({ pg }) => {
+        await pg.query(`
+-- Insert test users
+INSERT INTO auth.users (id, email) VALUES 
+  ('550e8400-e29b-41d4-a716-446655440001', '[email protected]'),
+  ('550e8400-e29b-41d4-a716-446655440002', '[email protected]'),
+  ('550e8400-e29b-41d4-a716-446655440003', '[email protected]'),
+  ('550e8400-e29b-41d4-a716-446655440004', '[email protected]');
+
+-- Insert test pets
+INSERT INTO rls_test.pets (id, name, breed, user_id) VALUES 
+  ('660e8400-e29b-41d4-a716-446655440001', 'Fido', 'Labrador', '550e8400-e29b-41d4-a716-446655440001'),
+  ('660e8400-e29b-41d4-a716-446655440002', 'Buddy', 'Golden Retriever', '550e8400-e29b-41d4-a716-446655440002'),
+  ('660e8400-e29b-41d4-a716-446655440003', 'Rex', 'German Shepherd', '550e8400-e29b-41d4-a716-446655440003');
+        `);
+      }),
+    ]
+  ));
+});
+
+afterAll(async () => {
+  await teardown();
+});
+
+beforeEach(async () => {
+  await db.beforeEach();
+});
+
+afterEach(async () => {
+  await db.afterEach();
+});
+
+describe('tutorial: testing with pg function seeding', () => {
+  it('data should be loaded into the database', async () => {
+
+    db.setContext({
+      role: 'authenticated',
+      'request.jwt.claim.sub': users[0].id
+    });
+
+    const verifiedPets = await db.any(
+      `SELECT id FROM rls_test.pets WHERE user_id = $1`,
+      [users[0].id]
+    );
+    expect(verifiedPets.length).toBe(1);
+
+    db.clearContext();
+
+    const anonPets = await db.any(
+      `SELECT id FROM rls_test.pets WHERE user_id = $1`,
+      [users[0].id]
+    );
+    expect(anonPets.length).toBe(0);
+
+  });
+  it('should enforce RLS - users can only see their own pets', async () => {
+    // set context to first user
+    db.setContext({
+      role: 'authenticated',
+      'request.jwt.claim.sub': users[0].id
+    });
+
+    // user1 should only see their own pet (Fido)
+    const user1Pets = await db.many(
+      `SELECT id, name, breed, user_id FROM rls_test.pets ORDER BY name`
+    );
+
+    expect(user1Pets.length).toBe(1);
+    expect(user1Pets[0].user_id).toBe(users[0].id);
+    expect(user1Pets[0].name).toBe('Fido');
+
+    // set context to second user
+    db.setContext({
+      role: 'authenticated',
+      'request.jwt.claim.sub': users[1].id
+    });
+
+    // user2 should only see their own pet (Buddy)
+    const user2Pets = await db.many(
+      `SELECT id, name, breed, user_id FROM rls_test.pets ORDER BY name`
+    );
+
+    expect(user2Pets.length).toBe(1);
+    expect(user2Pets[0].user_id).toBe(users[1].id);
+    expect(user2Pets[0].name).toBe('Buddy');
+  });
+});
+

packages/hello-world/__tests__/seeding/seeding.sql.test.ts

@@ -7,16 +7,18 @@ let teardown: () => Promise<void>;
 
 const sql = (f: string) => path.join(__dirname, 'data', f);
 
-const cwd = path.resolve(__dirname, '../../');
-
 beforeAll(async () => {
   ({ db, teardown } = await getConnections(
-    {}, [
-    seed.launchql(cwd),
-    seed.sqlfile([
-      sql('seed-data.sql'),
-    ])
-  ]
+    {},
+    [
+      // load schema and it's dependencies (supabase full schema)
+      seed.launchql(),
+
+      // load data from sql file
+      seed.sqlfile([
+        sql('seed-data.sql'),
+      ])
+    ]
   ));
 });
 
@@ -33,7 +35,7 @@ afterEach(async () => {
 });
 
 describe('tutorial: testing with sql file seeding', () => {
-  it('should work with sql file seed function', async () => {
+  it('data should be loaded into the database', async () => {
 
     db.setContext({
       role: 'authenticated',