@geekmidas/testkit 0.0.7 → 0.0.8

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@geekmidas/testkit",
-  "version": "0.0.7",
+  "version": "0.0.8",
   "private": false,
   "type": "module",
   "exports": {

package/src/VitestObjectionTransactionIsolator.ts

@@ -60,7 +60,8 @@ export class VitestObjectionTransactionIsolator extends VitestPostgresTransactio
     level: IsolationLevel,
     fn: (trx: Knex.Transaction) => Promise<void>,
   ): Promise<void> {
-    const isolationLevel = level.toUpperCase() as Lowercase<IsolationLevel>;
+    const isolationLevel = level.toLowerCase() as Lowercase<IsolationLevel>;
+
     await conn.transaction(
       async (trx) => {
         await fn(trx);

package/src/__tests__/VitestObjectionTransactionIsolator.spec.ts

@@ -0,0 +1,144 @@
+import { Model } from 'objection';
+import { it as base, describe, expect } from 'vitest';
+
+import { createKnexDb, createTestTablesKnex } from '../../test/helpers';
+import { wrapVitestObjectionTransaction } from '../objection';
+
+// Define Objection models for testing
+class User extends Model {
+  static get tableName() {
+    return 'users';
+  }
+
+  id!: number;
+  name!: string;
+  email!: string;
+  role?: string;
+  createdAt!: Date;
+  updatedAt?: Date;
+
+  static get relationMappings() {
+    return {
+      posts: {
+        relation: Model.HasManyRelation,
+        modelClass: Post,
+        join: {
+          from: 'users.id',
+          to: 'posts.user_id',
+        },
+      },
+      comments: {
+        relation: Model.HasManyRelation,
+        modelClass: Comment,
+        join: {
+          from: 'users.id',
+          to: 'comments.user_id',
+        },
+      },
+    };
+  }
+}
+
+class Post extends Model {
+  static get tableName() {
+    return 'posts';
+  }
+
+  id!: number;
+  title!: string;
+  content!: string;
+  userId!: number;
+  published?: boolean;
+  createdAt!: Date;
+  updatedAt?: Date;
+
+  static get relationMappings() {
+    return {
+      user: {
+        relation: Model.BelongsToOneRelation,
+        modelClass: User,
+        join: {
+          from: 'posts.user_id',
+          to: 'users.id',
+        },
+      },
+      comments: {
+        relation: Model.HasManyRelation,
+        modelClass: Comment,
+        join: {
+          from: 'posts.id',
+          to: 'comments.post_id',
+        },
+      },
+    };
+  }
+}
+
+class Comment extends Model {
+  static get tableName() {
+    return 'comments';
+  }
+
+  id!: number;
+  content!: string;
+  postId!: number;
+  userId!: number;
+  createdAt!: Date;
+
+  static get relationMappings() {
+    return {
+      post: {
+        relation: Model.BelongsToOneRelation,
+        modelClass: Post,
+        join: {
+          from: 'comments.post_id',
+          to: 'posts.id',
+        },
+      },
+      user: {
+        relation: Model.BelongsToOneRelation,
+        modelClass: User,
+        join: {
+          from: 'comments.user_id',
+          to: 'users.id',
+        },
+      },
+    };
+  }
+}
+
+// Create database connection
+const knex = createKnexDb();
+
+// Bind models to Knex instance
+Model.knex(knex);
+
+// Create wrapped test with transaction isolation
+const it = wrapVitestObjectionTransaction(base, knex, async (trx) => {
+  // Create tables in the transaction
+  await createTestTablesKnex(trx);
+});
+
+describe('VitestObjectionTransactionIsolator', () => {
+  describe('Transaction Isolation', () => {
+    it('should rollback data after test completes', async ({ trx }) => {
+      // Create a user within the transaction
+      const user = await User.query(trx).insert({
+        name: 'Test User',
+        email: 'test@example.com',
+        role: 'user',
+      });
+
+      expect(user).toBeDefined();
+      expect(user.id).toBeDefined();
+      expect(user.name).toBe('Test User');
+
+      // Verify user exists in transaction
+      const foundUser = await User.query(trx).findById(user.id);
+      expect(foundUser).toBeDefined();
+      expect(foundUser?.email).toBe(user.email);
+
+      // Data will be rolled back after this test
+    });
+  });
+});

package/test/helpers.ts

@@ -162,7 +162,9 @@ export async function createTestTables(
 /**
  * Creates test tables using Knex
  */
-async function createTestTablesKnex(trx: Knex.Transaction): Promise<void> {
+export async function createTestTablesKnex(
+  trx: Knex.Transaction,
+): Promise<void> {
   // Create users table
   await trx.schema.createTable('users', (table) => {
     table.bigIncrements('id').primary();