From 25941ebd34c0bec4407343cf9927b3658b7f2db5 Mon Sep 17 00:00:00 2001
From: Shishant Biswas <shishantbiswas@gmail.com>
Date: Wed, 22 Apr 2026 11:07:27 +0530
Subject: [PATCH] revert test case, refactor `getPrimaryField` in Entity

---
 app/__test__/data/specs/fields/PrimaryField.spec.ts | 6 +++---
 app/src/data/entities/Entity.ts                     | 2 +-
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/app/__test__/data/specs/fields/PrimaryField.spec.ts b/app/__test__/data/specs/fields/PrimaryField.spec.ts
index de642e03..459be4f6 100644
--- a/app/__test__/data/specs/fields/PrimaryField.spec.ts
+++ b/app/__test__/data/specs/fields/PrimaryField.spec.ts
@@ -2,7 +2,7 @@ import { describe, expect, test } from "bun:test";
 import { PrimaryField } from "data/fields";
 
 describe("[data] PrimaryField", async () => {
-   const field = new PrimaryField("primary");
+   const field = new PrimaryField("primary", { fillable: false });
 
    test("name", async () => {
       expect(field.name).toBe("primary");
@@ -24,7 +24,7 @@ describe("[data] PrimaryField", async () => {
    });
 
    test("isFillable", async () => {
-      expect(field.isFillable()).toBe(true);
+      expect(field.isFillable()).toBe(false);
    });
 
    test("isHidden", async () => {
@@ -36,7 +36,7 @@ describe("[data] PrimaryField", async () => {
    });
 
    test("transformPersist/Retrieve", async () => {
-      expect(field.transformPersist(1)).resolves.toBe(1);
+      expect(field.transformPersist(1)).rejects.toThrow();
       expect(field.transformRetrieve(1)).toBe(1);
    });
 
diff --git a/app/src/data/entities/Entity.ts b/app/src/data/entities/Entity.ts
index d616bd23..0d686a8a 100644
--- a/app/src/data/entities/Entity.ts
+++ b/app/src/data/entities/Entity.ts
@@ -171,7 +171,7 @@ export class Entity<
    }
 
    getPrimaryField(): PrimaryField {
-      return this.fields[0] as PrimaryField;
+      return this.fields.find((field) => field.type === "primary") as PrimaryField;
    }
 
    id(): PrimaryField {