/*
 * 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();
		});
	});
});