Fix tests.

packages/server/src/utilities/rowProcessor/index.ts

@@ -4,6 +4,7 @@ import {
   cache,
   context,
   db,
+  features,
   HTTPError,
   objectStore,
   utils,
@@ -350,7 +351,7 @@ export async function outputProcessing<T extends Row[] | Row>(
   }
   // remove null properties to match internal API
   const isExternal = isExternalTableID(table._id!)
-  if (isExternal || db.isSqsEnabledForTenant()) {
+  if (isExternal || (await features.flags.isEnabled("SQS"))) {
     for (const row of enriched) {
       for (const key of Object.keys(row)) {
         if (row[key] === null) {

packages/server/src/utilities/rowProcessor/tests/outputProcessing.spec.ts

@@ -8,15 +8,9 @@ import {
 } from "@budibase/types"
 import { outputProcessing } from ".."
 import { generator, structures } from "@budibase/backend-core/tests"
+import { setEnv as setCoreEnv } from "@budibase/backend-core"
 import * as bbReferenceProcessor from "../bbReferenceProcessor"
-
-jest.mock("@budibase/backend-core", () => ({
-  ...jest.requireActual("@budibase/backend-core"),
-  db: {
-    ...jest.requireActual("@budibase/backend-core").db,
-    isSqsEnabledForTenant: () => true,
-  },
-}))
+import TestConfiguration from "../../../tests/utilities/TestConfiguration"
 
 jest.mock("../bbReferenceProcessor", (): typeof bbReferenceProcessor => ({
   processInputBBReference: jest.fn(),
@@ -26,8 +20,24 @@ jest.mock("../bbReferenceProcessor", (): typeof bbReferenceProcessor => ({
 }))
 
 describe("rowProcessor - outputProcessing", () => {
+  const config = new TestConfiguration()
+  let cleanupEnv: () => void = () => {}
+
+  beforeAll(async () => {
+    await config.init()
+  })
+
+  afterAll(async () => {
+    config.end()
+  })
+
   beforeEach(() => {
     jest.resetAllMocks()
+    cleanupEnv = setCoreEnv({ TENANT_FEATURE_FLAGS: "*SQS" })
+  })
+
+  afterEach(() => {
+    cleanupEnv()
   })
 
   const processOutputBBReferenceMock =
@@ -36,6 +46,7 @@ describe("rowProcessor - outputProcessing", () => {
     bbReferenceProcessor.processOutputBBReferences as jest.Mock
 
   it("fetches single user references given a populated field", async () => {
+    await config.doInContext(config.getAppId(), async () => {
       const table: Table = {
         _id: generator.guid(),
         name: "TestTable",
@@ -75,16 +86,17 @@ describe("rowProcessor - outputProcessing", () => {
 
       expect(result).toEqual({ name: "Jack", user })
 
-    expect(bbReferenceProcessor.processOutputBBReference).toHaveBeenCalledTimes(
-      1
-    )
-    expect(bbReferenceProcessor.processOutputBBReference).toHaveBeenCalledWith(
-      "123",
-      BBReferenceFieldSubType.USER
-    )
+      expect(
+        bbReferenceProcessor.processOutputBBReference
+      ).toHaveBeenCalledTimes(1)
+      expect(
+        bbReferenceProcessor.processOutputBBReference
+      ).toHaveBeenCalledWith("123", BBReferenceFieldSubType.USER)
+    })
   })
 
   it("fetches users references given a populated field", async () => {
+    await config.doInContext(config.getAppId(), async () => {
       const table: Table = {
         _id: generator.guid(),
         name: "TestTable",
@@ -127,13 +139,14 @@ describe("rowProcessor - outputProcessing", () => {
       expect(
         bbReferenceProcessor.processOutputBBReferences
       ).toHaveBeenCalledTimes(1)
-    expect(bbReferenceProcessor.processOutputBBReferences).toHaveBeenCalledWith(
-      "123",
-      BBReferenceFieldSubType.USER
-    )
+      expect(
+        bbReferenceProcessor.processOutputBBReferences
+      ).toHaveBeenCalledWith("123", BBReferenceFieldSubType.USER)
+    })
   })
 
   it("should handle attachment list correctly", async () => {
+    await config.doInContext(config.getAppId(), async () => {
       const table: Table = {
         _id: generator.guid(),
         name: "TestTable",
@@ -175,8 +188,10 @@ describe("rowProcessor - outputProcessing", () => {
       const output3 = await outputProcessing(table, row, { squash: false })
       expect(output3.attach[0].url).toBe("aaaa")
     })
+  })
 
   it("should handle single attachment correctly", async () => {
+    await config.doInContext(config.getAppId(), async () => {
       const table: Table = {
         _id: generator.guid(),
         name: "TestTable",
@@ -216,8 +231,10 @@ describe("rowProcessor - outputProcessing", () => {
       const output3 = await outputProcessing(table, row, { squash: false })
       expect(output3.attach.url).toBe("aaaa")
     })
+  })
 
   it("process output even when the field is not empty", async () => {
+    await config.doInContext(config.getAppId(), async () => {
       const table: Table = {
         _id: generator.guid(),
         name: "TestTable",
@@ -257,8 +274,10 @@ describe("rowProcessor - outputProcessing", () => {
         bbReferenceProcessor.processOutputBBReferences
       ).toHaveBeenCalledTimes(1)
     })
+  })
 
   it("does not fetch bb references when not in the schema", async () => {
+    await config.doInContext(config.getAppId(), async () => {
       const table: Table = {
         _id: generator.guid(),
         name: "TestTable",
@@ -299,3 +318,4 @@ describe("rowProcessor - outputProcessing", () => {
       ).not.toHaveBeenCalled()
     })
   })
+})