From 9cd7c144f4d06f4d415f3effafa72f4df7681183 Mon Sep 17 00:00:00 2001
From: mike12345567 <me@michaeldrury.co.uk>
Date: Fri, 7 Jun 2024 16:57:46 +0100
Subject: [PATCH] Adding test case.

---
 .../src/integration-test/postgres.spec.ts     | 44 +++++++++++++++++++
 1 file changed, 44 insertions(+)

diff --git a/packages/server/src/integration-test/postgres.spec.ts b/packages/server/src/integration-test/postgres.spec.ts
index ccf63d0820..9da56034d1 100644
--- a/packages/server/src/integration-test/postgres.spec.ts
+++ b/packages/server/src/integration-test/postgres.spec.ts
@@ -1265,4 +1265,48 @@ describe("postgres integrations", () => {
       expect(JSON.parse(decoded)[0]).toBe("1111")
     })
   })
+
+  describe("check fetching null/not null table", () => {
+    beforeAll(async () => {
+      await rawQuery(
+        rawDatasource,
+        `CREATE TABLE nullableTable (
+          order_id SERIAL PRIMARY KEY,
+          order_number INT NOT NULL
+        );
+      `
+      )
+    })
+
+    it("should be able to change the table to allow nullable and refetch this", async () => {
+      const response = await makeRequest(
+        "post",
+        `/api/datasources/${datasource._id}/schema`
+      )
+      const entities = response.body.datasource.entities
+      expect(entities).toBeDefined()
+      const nullableTable = entities["nullabletable"]
+      expect(nullableTable).toBeDefined()
+      expect(nullableTable.schema["order_number"].constraints.presence).toEqual(
+        true
+      )
+      await rawQuery(
+        rawDatasource,
+        `ALTER TABLE nullableTable
+             ALTER COLUMN order_number DROP NOT NULL;
+        `
+      )
+      const responseAfter = await makeRequest(
+        "post",
+        `/api/datasources/${datasource._id}/schema`
+      )
+      const entitiesAfter = responseAfter.body.datasource.entities
+      expect(entitiesAfter).toBeDefined()
+      const nullableTableAfter = entitiesAfter["nullabletable"]
+      expect(nullableTableAfter).toBeDefined()
+      expect(
+        nullableTableAfter.schema["order_number"].constraints.presence
+      ).toEqual(false)
+    })
+  })
 })