Refactor repository tests to use DomainUser interface, enhance test setup with seed data function, and improve cleanup logic. Update test cases for consistency and clarity.

Author: kingRayhan

__tests__/repository/count.test.ts

@@ -1,16 +1,10 @@
-import { Repository } from "../../src/repository/repository";
-import { eq, gt, and, like } from "../../src";
-import { executor, setupTestTables, cleanupTestData } from "../../test-setup";
+import {Repository} from "../../src/repository/repository";
+import {and, eq, gt, like} from "../../src";
+import {cleanupTestData, DomainUser, executor, setupTestTables} from "../../test-setup";
 
-interface User {
-  id: number;
-  name: string;
-  email: string;
-  age?: number;
-}
 
 describe("Repository count", () => {
-  let repository: Repository<User>;
+  let repository: Repository<DomainUser>;
 
   beforeAll(async () => {
     await setupTestTables();
@@ -20,6 +14,9 @@ describe("Repository count", () => {
     repository = new Repository("users", executor);
     await cleanupTestData();
   });
+  afterAll(async () => {
+    await cleanupTestData();
+  });
 
   it("should count all rows without condition", async () => {
     // Insert test data
@@ -29,7 +26,6 @@ describe("Repository count", () => {
     );
 
     const result = await repository.count(like("name", "%Doe%"));
-
     expect(result).toBe(2);
   });
 

__tests__/repository/delete.test.ts

@@ -1,29 +1,23 @@
 import { SqlExecutor, QueryResult } from "../../src";
 import { Repository } from "../../src/repository/repository";
 import { eq, gt, and, like } from "../../src";
+import {DomainUser} from "../../test-setup";
 
 // Mock SqlExecutor
 const mockExecutor: jest.Mocked<SqlExecutor> = {
   executeSQL: jest.fn(),
 };
 
-interface User {
-  id: string;
-  name: string;
-  email: string;
-  age?: number;
-}
-
 describe("Repository delete", () => {
-  let repository: Repository<User>;
+  let repository: Repository<DomainUser>;
 
   beforeEach(() => {
     repository = new Repository("users", mockExecutor);
     jest.clearAllMocks();
   });
 
   it("should delete a record and return it", async () => {
-    const mockUser: User = {
+    const mockUser: Omit<DomainUser, 'created_at'> = {
       id: "1",
       name: "John Doe",
       email: "[email protected]",
@@ -43,7 +37,7 @@ describe("Repository delete", () => {
   });
 
   it("should delete a record with specific returning columns", async () => {
-    const mockUser: Partial<User> = {
+    const mockUser: Partial<DomainUser> = {
       id: "1",
       name: "John Doe",
     };
@@ -74,7 +68,7 @@ describe("Repository delete", () => {
   });
 
   it("should handle complex where conditions", async () => {
-    const mockUser: User = {
+    const mockUser: DomainUser = {
       id: "1",
       name: "John Doe",
       email: "[email protected]",

__tests__/repository/find-row.test.ts

@@ -1,23 +1,19 @@
 import { Repository } from "../../src/repository/repository";
 import { eq, and, gt, like } from "../../src";
-import { executor, setupTestTables, cleanupTestData } from "../../test-setup";
+import {executor, setupTestTables, cleanupTestData, seedTestData, DomainUser} from "../../test-setup";
+
 
-interface User {
-  id: number;
-  name: string;
-  email: string;
-  age?: number;
-}
 
 describe("Repository findRow", () => {
-  let repository: Repository<User>;
+  let repository: Repository<DomainUser>;
 
   beforeAll(async () => {
     await setupTestTables();
+    await seedTestData();
+    repository = new Repository("users", executor);
   });
 
-  beforeEach(async () => {
-    repository = new Repository("users", executor);
+  afterAll(async () => {
     await cleanupTestData();
   });
 
@@ -38,8 +34,7 @@ describe("Repository findRow", () => {
   });
 
   it("should return null when no row is found", async () => {
-    const result = await repository.findRow(eq("id", 999));
-
+    const result = await repository.findRow(eq("email", '[email protected]'));
     expect(result).toBeNull();
   });