vas3k-TaxHacker/lib/uploads.test.ts

import { afterAll, describe, expect, it, vi } from "vitest"
import { mkdtemp, readFile, rm } from "fs/promises"
import { tmpdir } from "os"
import path from "path"

const tmpRoot = await mkdtemp(path.join(tmpdir(), "th-uploads-"))
process.env.UPLOAD_PATH = tmpRoot
process.env.SELF_HOSTED_MODE = "true"

const created: any[] = []
vi.mock("./config", () => ({
  default: {
    upload: { images: { maxWidth: 1920, maxHeight: 1080, quality: 80 } },
    selfHosted: { isEnabled: true },
  },
}))
vi.mock("@/models/files", () => ({
  createFile: vi.fn(async (userId: string, data: any) => {
    const row = { ...data, userId }
    created.push(row)
    return row
  }),
}))

const { ingestUnsortedFile } = await import("./uploads")

const user = { id: "user-1", email: "u@example.com", storageUsed: 0, storageLimit: -1 } as any

afterAll(async () => {
  await rm(tmpRoot, { recursive: true, force: true })
})

describe("ingestUnsortedFile", () => {
  it("writes the buffer under the user's unsorted dir and creates an unsorted File row", async () => {
    const buffer = Buffer.from("%PDF-1.4 fake invoice")
    const file = await ingestUnsortedFile(user, {
      buffer,
      filename: "invoice.pdf",
      mimetype: "application/pdf",
      metadata: { source: "email" },
    })

    expect(file.filename).toBe("invoice.pdf")
    expect(file.mimetype).toBe("application/pdf")
    expect(file.path).toMatch(/^unsorted\/.+\.pdf$/)
    expect((file.metadata as any).source).toBe("email")

    const onDisk = await readFile(path.join(tmpRoot, user.email, file.path))
    expect(onDisk.equals(buffer)).toBe(true)
    expect(created).toHaveLength(1)
  })
})