46 lines
1.2 KiB
TypeScript
46 lines
1.2 KiB
TypeScript
import { S3Client, PutObjectCommand, GetObjectCommand } from '@aws-sdk/client-s3';
|
|
import { randomUUID } from 'crypto';
|
|
import { expect, test } from 'vitest';
|
|
import 'dotenv/config';
|
|
|
|
const s3 = new S3Client({
|
|
region: 'us-east-1',
|
|
endpoint: process.env.S3_ENDPOINT,
|
|
credentials: {
|
|
accessKeyId: process.env.S3_ACCESS_KEY!,
|
|
secretAccessKey: process.env.S3_SECRET_KEY!,
|
|
},
|
|
forcePathStyle: true,
|
|
});
|
|
|
|
const BUCKET = process.env.S3_BUCKET!;
|
|
|
|
test('can upload and download a file from S3', async () => {
|
|
const key = `test-file-${randomUUID()}.txt`;
|
|
const content = 'hello s3 world!';
|
|
|
|
// Upload
|
|
await s3.send(new PutObjectCommand({
|
|
Bucket: BUCKET,
|
|
Key: key,
|
|
Body: content,
|
|
}));
|
|
|
|
// Download
|
|
const getObj = await s3.send(new GetObjectCommand({
|
|
Bucket: BUCKET,
|
|
Key: key,
|
|
}));
|
|
const downloaded = await streamToString(getObj.Body);
|
|
|
|
expect(downloaded).toBe(content);
|
|
});
|
|
|
|
function streamToString(stream: any): Promise<string> {
|
|
return new Promise((resolve, reject) => {
|
|
const chunks: any[] = [];
|
|
stream.on('data', (chunk: any) => chunks.push(chunk));
|
|
stream.on('error', reject);
|
|
stream.on('end', () => resolve(Buffer.concat(chunks).toString('utf-8')));
|
|
});
|
|
}
|