141 lines
3.9 KiB
TypeScript
141 lines
3.9 KiB
TypeScript
import type { Payload } from 'payload'
|
|
|
|
import * as AWS from '@aws-sdk/client-s3'
|
|
import path from 'path'
|
|
import { fileURLToPath } from 'url'
|
|
|
|
import { describeIfInCIOrHasLocalstack } from '../helpers.js'
|
|
import { initPayloadInt } from '../helpers/initPayloadInt.js'
|
|
import configPromise from './config.js'
|
|
import { mediaSlug, mediaWithPrefixSlug, prefix } from './shared.js'
|
|
import { clearTestBucket, createTestBucket } from './utils.js'
|
|
|
|
const filename = fileURLToPath(import.meta.url)
|
|
const dirname = path.dirname(filename)
|
|
|
|
let payload: Payload
|
|
|
|
describe('@payloadcms/plugin-cloud-storage', () => {
|
|
beforeAll(async () => {
|
|
;({ payload } = await initPayloadInt(configPromise))
|
|
})
|
|
|
|
afterAll(async () => {
|
|
if (typeof payload.db.destroy === 'function') {
|
|
await payload.db.destroy()
|
|
}
|
|
})
|
|
|
|
const TEST_BUCKET = process.env.S3_BUCKET
|
|
|
|
let client: AWS.S3Client
|
|
describeIfInCIOrHasLocalstack()('plugin-cloud-storage', () => {
|
|
describe('S3', () => {
|
|
beforeAll(async () => {
|
|
client = new AWS.S3({
|
|
endpoint: process.env.S3_ENDPOINT,
|
|
forcePathStyle: process.env.S3_FORCE_PATH_STYLE === 'true',
|
|
region: process.env.S3_REGION,
|
|
credentials: {
|
|
accessKeyId: process.env.S3_ACCESS_KEY_ID,
|
|
secretAccessKey: process.env.S3_SECRET_ACCESS_KEY,
|
|
},
|
|
})
|
|
|
|
await createTestBucket()
|
|
await clearTestBucket(client)
|
|
})
|
|
|
|
afterEach(async () => {
|
|
await clearTestBucket(client)
|
|
})
|
|
|
|
it('can upload', async () => {
|
|
const upload = await payload.create({
|
|
collection: mediaSlug,
|
|
data: {},
|
|
filePath: path.resolve(dirname, '../uploads/image.png'),
|
|
})
|
|
|
|
expect(upload.id).toBeTruthy()
|
|
|
|
await verifyUploads({
|
|
collectionSlug: mediaSlug,
|
|
uploadId: upload.id,
|
|
})
|
|
|
|
expect(upload.url).toEqual(`/api/${mediaSlug}/file/${String(upload.filename)}`)
|
|
})
|
|
|
|
it('can upload with prefix', async () => {
|
|
const upload = await payload.create({
|
|
collection: mediaWithPrefixSlug,
|
|
data: {},
|
|
filePath: path.resolve(dirname, '../uploads/image.png'),
|
|
})
|
|
|
|
expect(upload.id).toBeTruthy()
|
|
|
|
await verifyUploads({
|
|
collectionSlug: mediaWithPrefixSlug,
|
|
uploadId: upload.id,
|
|
prefix,
|
|
})
|
|
expect(upload.url).toEqual(`/api/${mediaWithPrefixSlug}/file/${String(upload.filename)}`)
|
|
})
|
|
})
|
|
})
|
|
|
|
describe('Azure', () => {
|
|
it.todo('can upload')
|
|
})
|
|
|
|
describe('GCS', () => {
|
|
it.todo('can upload')
|
|
})
|
|
|
|
describe('R2', () => {
|
|
it.todo('can upload')
|
|
})
|
|
|
|
async function verifyUploads({
|
|
collectionSlug,
|
|
uploadId,
|
|
prefix = '',
|
|
}: {
|
|
collectionSlug: string
|
|
prefix?: string
|
|
uploadId: number | string
|
|
}) {
|
|
const uploadData = (await payload.findByID({
|
|
collection: collectionSlug,
|
|
id: uploadId,
|
|
})) as unknown as { filename: string; sizes: Record<string, { filename: string }> }
|
|
|
|
const fileKeys = Object.keys(uploadData.sizes || {}).map((key) => {
|
|
const rawFilename = uploadData.sizes[key].filename
|
|
return prefix ? `${prefix}/${rawFilename}` : rawFilename
|
|
})
|
|
|
|
fileKeys.push(`${prefix ? `${prefix}/` : ''}${uploadData.filename}`)
|
|
try {
|
|
for (const key of fileKeys) {
|
|
const { $metadata } = await client.send(
|
|
new AWS.HeadObjectCommand({ Bucket: TEST_BUCKET, Key: key }),
|
|
)
|
|
|
|
if ($metadata.httpStatusCode !== 200) {
|
|
console.error('Error verifying uploads', key, $metadata)
|
|
throw new Error(`Error verifying uploads: ${key}, ${$metadata.httpStatusCode}`)
|
|
}
|
|
|
|
// Verify each size was properly uploaded
|
|
expect($metadata.httpStatusCode).toBe(200)
|
|
}
|
|
} catch (error: unknown) {
|
|
console.error('Error verifying uploads:', fileKeys, error)
|
|
throw error
|
|
}
|
|
}
|
|
})
|