100 lines
2.8 KiB
TypeScript
100 lines
2.8 KiB
TypeScript
/*
|
|
* SPDX-FileCopyrightText: syuilo and misskey-project
|
|
* SPDX-License-Identifier: AGPL-3.0-only
|
|
*/
|
|
|
|
process.env.NODE_ENV = 'test';
|
|
|
|
import { Test } from '@nestjs/testing';
|
|
import {
|
|
CompleteMultipartUploadCommand,
|
|
CreateMultipartUploadCommand,
|
|
PutObjectCommand,
|
|
S3Client,
|
|
UploadPartCommand,
|
|
} from '@aws-sdk/client-s3';
|
|
import { mockClient } from 'aws-sdk-client-mock';
|
|
import { MockInternalEventService } from '../misc/MockInternalEventService.js';
|
|
import type { TestingModule } from '@nestjs/testing';
|
|
import { GlobalModule } from '@/GlobalModule.js';
|
|
import { CoreModule } from '@/core/CoreModule.js';
|
|
import { S3Service } from '@/core/S3Service.js';
|
|
import { MiMeta } from '@/models/_.js';
|
|
import { HttpRequestService } from '@/core/HttpRequestService.js';
|
|
import { InternalEventService } from '@/global/InternalEventService.js';
|
|
import { DI } from '@/di-symbols.js';
|
|
|
|
describe('S3Service', () => {
|
|
let app: TestingModule;
|
|
let s3Service: S3Service;
|
|
let fakeMeta: MiMeta;
|
|
const s3Mock = mockClient(S3Client);
|
|
|
|
beforeAll(async () => {
|
|
app = await Test.createTestingModule({
|
|
imports: [GlobalModule, CoreModule],
|
|
})
|
|
.overrideProvider(InternalEventService).useClass(MockInternalEventService)
|
|
.compile();
|
|
|
|
await app.init();
|
|
app.enableShutdownHooks();
|
|
});
|
|
|
|
beforeEach(async () => {
|
|
s3Mock.reset();
|
|
|
|
fakeMeta = Object.create(app.get<MiMeta>(DI.meta));
|
|
s3Service = new S3Service(fakeMeta, app.get(HttpRequestService), app.get(InternalEventService));
|
|
});
|
|
|
|
afterAll(async () => {
|
|
await app.close();
|
|
});
|
|
|
|
describe('upload', () => {
|
|
test('upload a file', async () => {
|
|
s3Mock.on(PutObjectCommand).resolves({});
|
|
fakeMeta.objectStorageRegion = 'us-east-1';
|
|
|
|
await s3Service.upload({
|
|
Bucket: 'fake',
|
|
Key: 'fake',
|
|
Body: 'x',
|
|
});
|
|
});
|
|
|
|
test('upload a large file', async () => {
|
|
s3Mock.on(CreateMultipartUploadCommand).resolves({ UploadId: '1' });
|
|
s3Mock.on(UploadPartCommand).resolves({ ETag: '1' });
|
|
s3Mock.on(CompleteMultipartUploadCommand).resolves({ Bucket: 'fake', Key: 'fake' });
|
|
|
|
await s3Service.upload({
|
|
Bucket: 'fake',
|
|
Key: 'fake',
|
|
Body: 'x'.repeat(8 * 1024 * 1024 + 1), // デフォルトpartSizeにしている 8 * 1024 * 1024 を越えるサイズ
|
|
});
|
|
});
|
|
|
|
test('upload a file error', async () => {
|
|
s3Mock.on(PutObjectCommand).rejects({ name: 'Fake Error' });
|
|
fakeMeta.objectStorageRegion = 'us-east-1';
|
|
|
|
await expect(s3Service.upload({
|
|
Bucket: 'fake',
|
|
Key: 'fake',
|
|
Body: 'x',
|
|
})).rejects.toThrow();
|
|
});
|
|
|
|
test('upload a large file error', async () => {
|
|
s3Mock.on(UploadPartCommand).rejects();
|
|
|
|
await expect(s3Service.upload({
|
|
Bucket: 'fake',
|
|
Key: 'fake',
|
|
Body: 'x'.repeat(8 * 1024 * 1024 + 1), // デフォルトpartSizeにしている 8 * 1024 * 1024 を越えるサイズ
|
|
})).rejects.toThrow();
|
|
});
|
|
});
|
|
});
|