@budibase/server/src/sdk/tests/attachments.spec.ts

Budibase Web Server

import TestConfig from "../../tests/utilities/TestConfiguration"
import { db as dbCore } from "@budibase/backend-core"
import sdk from "../index"
import {
  FieldType,
  INTERNAL_TABLE_SOURCE_ID,
  TableSourceType,
} from "@budibase/types"
import { FIND_LIMIT } from "../app/rows/attachments"

const attachment = {
  size: 73479,
  name: "2022-12-14 11_11_44-.png",
  url: "/prod-budi-app-assets/app_bbb/attachments/a.png",
  extension: "png",
  key: "app_bbb/attachments/a.png",
}

describe("should be able to re-write attachment URLs", () => {
  const config = new TestConfig()

  beforeAll(async () => {
    await config.init()
  })

  const coreBehaviour = async (tblSchema: any, row: any) => {
    const table = await config.api.table.save({
      name: "photos",
      type: "table",
      sourceId: INTERNAL_TABLE_SOURCE_ID,
      sourceType: TableSourceType.INTERNAL,
      schema: tblSchema,
    })

    for (let i = 0; i < FIND_LIMIT * 4; i++) {
      await config.api.row.save(table._id!, {
        ...row,
      })
    }

    const db = dbCore.getDB(config.getAppId())
    await config.doInContext(config.getAppId(), () =>
      sdk.backups.updateAttachmentColumns(db.name, db)
    )

    return {
      db,
      rows: (await sdk.rows.getAllInternalRows(db.name)).filter(
        row => row.tableId === table._id
      ),
    }
  }

  it("Attachment field, should update URLs on a number of rows over the limit", async () => {
    const { rows, db } = await coreBehaviour(
      {
        photo: {
          type: FieldType.ATTACHMENT_SINGLE,
          name: "photo",
        },
        gallery: {
          type: FieldType.ATTACHMENTS,
          name: "gallery",
        },
        otherCol: {
          type: FieldType.STRING,
          name: "otherCol",
        },
      },
      {
        photo: { ...attachment },
        gallery: [{ ...attachment }, { ...attachment }],
        otherCol: "string",
      }
    )
    for (const row of rows) {
      expect(row.otherCol).toBe("string")
      expect(row.photo.url).toBe("")
      expect(row.photo.key).toBe(`${db.name}/attachments/a.png`)
      expect(row.gallery[0].url).toBe("")
      expect(row.gallery[0].key).toBe(`${db.name}/attachments/a.png`)
      expect(row.gallery[1].url).toBe("")
      expect(row.gallery[1].key).toBe(`${db.name}/attachments/a.png`)
    }
  })
  it("Signature field, should update URLs on a number of rows over the limit", async () => {
    const { rows, db } = await coreBehaviour(
      {
        signature: {
          type: FieldType.SIGNATURE_SINGLE,
          name: "signature",
        },
        otherCol: {
          type: FieldType.STRING,
          name: "otherCol",
        },
      },
      {
        signature: { ...attachment },
        otherCol: "string",
      }
    )
    for (const row of rows) {
      expect(row.otherCol).toBe("string")
      expect(row.signature.url).toBe("")
      expect(row.signature.key).toBe(`${db.name}/attachments/a.png`)
    }
  })
})