seeding csv

Author: pyramation

packages/rls-demo/__tests__/seeding/data/seed-data.sql

@@ -2,30 +2,14 @@
 -- This file can be used to populate the database with test data
 
 -- Insert test users
-INSERT INTO rls_test.user_profiles (id, email, name) VALUES 
-  ('550e8400-e29b-41d4-a716-446655440001', '[email protected]', 'Alice Johnson'),
-  ('550e8400-e29b-41d4-a716-446655440002', '[email protected]', 'Bob Smith'),
-  ('550e8400-e29b-41d4-a716-446655440003', '[email protected]', 'Charlie Brown'),
-  ('550e8400-e29b-41d4-a716-446655440004', '[email protected]', 'Diana Prince');
+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 products
-INSERT INTO rls_test.products (id, name, description, price, owner_id) VALUES 
-  ('660e8400-e29b-41d4-a716-446655440001', 'Laptop Pro', 'High-performance laptop for developers', 1299.99, '550e8400-e29b-41d4-a716-446655440001'),
-  ('660e8400-e29b-41d4-a716-446655440002', 'Wireless Mouse', 'Ergonomic wireless mouse', 49.99, '550e8400-e29b-41d4-a716-446655440001'),
-  ('660e8400-e29b-41d4-a716-446655440003', 'Mechanical Keyboard', 'RGB mechanical keyboard', 149.99, '550e8400-e29b-41d4-a716-446655440002'),
-  ('660e8400-e29b-41d4-a716-446655440004', 'Monitor 4K', '27-inch 4K monitor', 399.99, '550e8400-e29b-41d4-a716-446655440002'),
-  ('660e8400-e29b-41d4-a716-446655440005', 'Webcam HD', '1080p webcam for video calls', 89.99, '550e8400-e29b-41d4-a716-446655440003'),
-  ('660e8400-e29b-41d4-a716-446655440006', 'Headphones', 'Noise-cancelling headphones', 199.99, '550e8400-e29b-41d4-a716-446655440003'),
-  ('660e8400-e29b-41d4-a716-446655440007', 'Standing Desk', 'Adjustable height standing desk', 599.99, '550e8400-e29b-41d4-a716-446655440004'),
-  ('660e8400-e29b-41d4-a716-446655440008', 'Desk Lamp', 'LED desk lamp with USB charging', 79.99, '550e8400-e29b-41d4-a716-446655440004');
-
--- Verify the data
-SELECT 
-  u.name as user_name,
-  u.email,
-  COUNT(p.id) as product_count,
-  COALESCE(SUM(p.price), 0) as total_value
-FROM rls_test.user_profiles u
-LEFT JOIN rls_test.products p ON u.id = p.owner_id
-GROUP BY u.id, u.name, u.email
-ORDER BY u.name;
+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');

packages/rls-demo/__tests__/seeding/seeding.sql.test.ts

@@ -11,10 +11,11 @@ const cwd = path.resolve(__dirname, '../../');
 beforeAll(async () => {
   ({ db, teardown } = await getConnections(
     {}, [
-      seed.launchql(cwd),
-      seed.sqlfile([
-        sql('seed-data.sql'),
-      ])    ]
+    seed.launchql(cwd),
+    seed.sqlfile([
+      sql('seed-data.sql'),
+    ])
+  ]
   ));
 });
 
@@ -30,55 +31,45 @@ afterEach(async () => {
   await db.afterEach();
 });
 
+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'
+  }
+];
+
 describe('tutorial: testing with sql file seeding', () => {
   it('should work with sql file seed function', async () => {
 
-    db.setContext({ role: 'service_role' });
-
-    const user = await db.one(
-      `INSERT INTO rls_test.user_profiles (email, name) 
-       VALUES ($1, $2) 
-       RETURNING id`,
-      ['[email protected]', 'SQL Seed User 1']
-    );
-
     db.setContext({
       role: 'authenticated',
-      'request.jwt.claim.sub': user.id
+      'request.jwt.claim.sub': users[0].id
     });
 
-    await db.one(
-      `INSERT INTO rls_test.products (name, description, price, owner_id) 
-       VALUES ($1, $2, $3, $4) 
-       RETURNING id`,
-      ['Secret Product', 'Should not be visible', 100.00, user.id]
+    const verifiedPets = await db.any(
+      `SELECT id FROM rls_test.pets WHERE user_id = $1`,
+      [users[0].id]
     );
-
-    const verifiedUsers = await db.any(
-      `SELECT id FROM rls_test.user_profiles WHERE id = $1`,
-      [user.id]
-    );
-    expect(verifiedUsers.length).toBe(1);
-
-    const verifiedProducts = await db.any(
-      `SELECT id FROM rls_test.products WHERE owner_id = $1`,
-      [user.id]
-    );
-    expect(verifiedProducts.length).toBe(1);
+    expect(verifiedPets.length).toBe(1);
 
     db.clearContext();
-    
-    const anonUsers = await db.any(
-      `SELECT id FROM rls_test.user_profiles WHERE id = $1`,
-      [user.id]
-    );
-    expect(anonUsers.length).toBe(0);
 
-    const anonProducts = await db.any(
-      `SELECT id FROM rls_test.products WHERE owner_id = $1`,
-      [user.id]
+    const anonPets = await db.any(
+      `SELECT id FROM rls_test.pets WHERE user_id = $1`,
+      [users[0].id]
     );
-    expect(anonProducts.length).toBe(0);
+    expect(anonPets.length).toBe(0);
 
   });