2023-07-27 07:31:52 +02:00
|
|
|
/*
|
2024-02-12 03:37:45 +01:00
|
|
|
* SPDX-FileCopyrightText: syuilo and misskey-project
|
2023-07-27 07:31:52 +02:00
|
|
|
* SPDX-License-Identifier: AGPL-3.0-only
|
|
|
|
*/
|
|
|
|
|
2023-07-08 12:34:19 +02:00
|
|
|
process.env.NODE_ENV = 'test';
|
|
|
|
|
|
|
|
import { jest } from '@jest/globals';
|
|
|
|
import { Test } from '@nestjs/testing';
|
|
|
|
import { GlobalModule } from '@/GlobalModule.js';
|
|
|
|
import { FetchInstanceMetadataService } from '@/core/FetchInstanceMetadataService.js';
|
|
|
|
import { FederatedInstanceService } from '@/core/FederatedInstanceService.js';
|
|
|
|
import { HttpRequestService } from '@/core/HttpRequestService.js';
|
|
|
|
import { LoggerService } from '@/core/LoggerService.js';
|
|
|
|
import { UtilityService } from '@/core/UtilityService.js';
|
|
|
|
import { IdService } from '@/core/IdService.js';
|
|
|
|
import { DI } from '@/di-symbols.js';
|
|
|
|
import type { TestingModule } from '@nestjs/testing';
|
|
|
|
|
2024-03-15 05:08:54 +01:00
|
|
|
function mockRedisSetNX() {
|
|
|
|
const hash = {} as FIXME;
|
|
|
|
// eslint-disable-next-line @typescript-eslint/no-unused-vars
|
|
|
|
return jest.fn((key: string, value, secondsToken: 'EX', seconds: number, nx: 'NX', callback?): Promise<'OK' | null> => {
|
|
|
|
return new Promise(resolve => {
|
|
|
|
if (hash[key]) return resolve(null);
|
|
|
|
hash[key] = value;
|
|
|
|
resolve('OK');
|
|
|
|
});
|
2023-07-27 07:31:52 +02:00
|
|
|
});
|
2023-07-08 12:34:19 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
describe('FetchInstanceMetadataService', () => {
|
|
|
|
let app: TestingModule;
|
|
|
|
let fetchInstanceMetadataService: jest.Mocked<FetchInstanceMetadataService>;
|
|
|
|
let federatedInstanceService: jest.Mocked<FederatedInstanceService>;
|
|
|
|
let httpRequestService: jest.Mocked<HttpRequestService>;
|
|
|
|
|
2023-09-04 06:33:38 +02:00
|
|
|
beforeEach(async () => {
|
2023-07-08 12:34:19 +02:00
|
|
|
app = await Test
|
|
|
|
.createTestingModule({
|
|
|
|
imports: [
|
|
|
|
GlobalModule,
|
|
|
|
],
|
|
|
|
providers: [
|
|
|
|
FetchInstanceMetadataService,
|
|
|
|
LoggerService,
|
|
|
|
UtilityService,
|
|
|
|
IdService,
|
|
|
|
],
|
|
|
|
})
|
|
|
|
.useMocker((token) => {
|
|
|
|
if (token === HttpRequestService) {
|
2023-07-27 07:31:52 +02:00
|
|
|
return { getJson: jest.fn(), getHtml: jest.fn(), send: jest.fn() };
|
2023-07-08 12:34:19 +02:00
|
|
|
} else if (token === FederatedInstanceService) {
|
|
|
|
return { fetch: jest.fn() };
|
|
|
|
} else if (token === DI.redis) {
|
2024-03-15 05:08:54 +01:00
|
|
|
return { set: mockRedisSetNX() };
|
2024-01-08 09:43:52 +01:00
|
|
|
}
|
2024-03-04 05:48:57 +01:00
|
|
|
return null;
|
2024-01-08 09:43:52 +01:00
|
|
|
})
|
2023-07-08 12:34:19 +02:00
|
|
|
.compile();
|
|
|
|
|
|
|
|
app.enableShutdownHooks();
|
|
|
|
|
2024-03-03 12:15:35 +01:00
|
|
|
fetchInstanceMetadataService = app.get<FetchInstanceMetadataService>(FetchInstanceMetadataService) as jest.Mocked<FetchInstanceMetadataService>;
|
2023-07-08 12:34:19 +02:00
|
|
|
federatedInstanceService = app.get<FederatedInstanceService>(FederatedInstanceService) as jest.Mocked<FederatedInstanceService>;
|
|
|
|
httpRequestService = app.get<HttpRequestService>(HttpRequestService) as jest.Mocked<HttpRequestService>;
|
|
|
|
});
|
|
|
|
|
2023-09-04 06:33:38 +02:00
|
|
|
afterEach(async () => {
|
2023-07-08 12:34:19 +02:00
|
|
|
await app.close();
|
|
|
|
});
|
|
|
|
|
|
|
|
test('Lock and update', async () => {
|
|
|
|
const now = Date.now();
|
2024-03-15 05:08:54 +01:00
|
|
|
federatedInstanceService.fetch.mockResolvedValue({ infoUpdatedAt: { getTime: () => { return now - 10 * 1000 * 60 * 60 * 24; } } } as FIXME);
|
2023-07-08 12:34:19 +02:00
|
|
|
httpRequestService.getJson.mockImplementation(() => { throw Error(); });
|
|
|
|
const tryLockSpy = jest.spyOn(fetchInstanceMetadataService, 'tryLock');
|
|
|
|
const unlockSpy = jest.spyOn(fetchInstanceMetadataService, 'unlock');
|
2024-03-04 05:48:57 +01:00
|
|
|
|
2024-03-15 05:08:54 +01:00
|
|
|
await fetchInstanceMetadataService.fetchInstanceMetadata({ host: 'example.com' } as FIXME);
|
2023-07-08 12:34:19 +02:00
|
|
|
expect(tryLockSpy).toHaveBeenCalledTimes(1);
|
|
|
|
expect(unlockSpy).toHaveBeenCalledTimes(1);
|
|
|
|
expect(federatedInstanceService.fetch).toHaveBeenCalledTimes(1);
|
|
|
|
expect(httpRequestService.getJson).toHaveBeenCalled();
|
|
|
|
});
|
2023-09-04 06:33:38 +02:00
|
|
|
|
2023-07-27 07:31:52 +02:00
|
|
|
test('Lock and don\'t update', async () => {
|
2023-07-08 12:34:19 +02:00
|
|
|
const now = Date.now();
|
2024-03-15 05:08:54 +01:00
|
|
|
federatedInstanceService.fetch.mockResolvedValue({ infoUpdatedAt: { getTime: () => now } } as FIXME);
|
2023-07-08 12:34:19 +02:00
|
|
|
httpRequestService.getJson.mockImplementation(() => { throw Error(); });
|
|
|
|
const tryLockSpy = jest.spyOn(fetchInstanceMetadataService, 'tryLock');
|
|
|
|
const unlockSpy = jest.spyOn(fetchInstanceMetadataService, 'unlock');
|
2024-03-04 05:48:57 +01:00
|
|
|
|
2024-03-15 05:08:54 +01:00
|
|
|
await fetchInstanceMetadataService.fetchInstanceMetadata({ host: 'example.com' } as FIXME);
|
2023-07-08 12:34:19 +02:00
|
|
|
expect(tryLockSpy).toHaveBeenCalledTimes(1);
|
|
|
|
expect(unlockSpy).toHaveBeenCalledTimes(1);
|
|
|
|
expect(federatedInstanceService.fetch).toHaveBeenCalledTimes(1);
|
|
|
|
expect(httpRequestService.getJson).toHaveBeenCalledTimes(0);
|
|
|
|
});
|
2023-09-04 06:33:38 +02:00
|
|
|
|
2023-07-08 12:34:19 +02:00
|
|
|
test('Do nothing when lock not acquired', async () => {
|
2024-03-03 12:15:35 +01:00
|
|
|
const now = Date.now();
|
2024-03-15 05:08:54 +01:00
|
|
|
federatedInstanceService.fetch.mockResolvedValue({ infoUpdatedAt: { getTime: () => now - 10 * 1000 * 60 * 60 * 24 } } as FIXME);
|
2023-07-08 12:34:19 +02:00
|
|
|
httpRequestService.getJson.mockImplementation(() => { throw Error(); });
|
2024-03-04 05:48:57 +01:00
|
|
|
await fetchInstanceMetadataService.tryLock('example.com');
|
2023-07-08 12:34:19 +02:00
|
|
|
const tryLockSpy = jest.spyOn(fetchInstanceMetadataService, 'tryLock');
|
|
|
|
const unlockSpy = jest.spyOn(fetchInstanceMetadataService, 'unlock');
|
2024-03-04 05:48:57 +01:00
|
|
|
|
2024-03-15 05:08:54 +01:00
|
|
|
await fetchInstanceMetadataService.fetchInstanceMetadata({ host: 'example.com' } as FIXME);
|
2024-03-04 05:48:57 +01:00
|
|
|
expect(tryLockSpy).toHaveBeenCalledTimes(1);
|
2023-07-08 12:34:19 +02:00
|
|
|
expect(unlockSpy).toHaveBeenCalledTimes(0);
|
|
|
|
expect(federatedInstanceService.fetch).toHaveBeenCalledTimes(0);
|
|
|
|
expect(httpRequestService.getJson).toHaveBeenCalledTimes(0);
|
|
|
|
});
|
2024-03-04 05:48:57 +01:00
|
|
|
|
|
|
|
test('Do when lock not acquired but forced', async () => {
|
|
|
|
const now = Date.now();
|
2024-03-15 05:08:54 +01:00
|
|
|
federatedInstanceService.fetch.mockResolvedValue({ infoUpdatedAt: { getTime: () => now - 10 * 1000 * 60 * 60 * 24 } } as FIXME);
|
2024-03-04 05:48:57 +01:00
|
|
|
httpRequestService.getJson.mockImplementation(() => { throw Error(); });
|
|
|
|
await fetchInstanceMetadataService.tryLock('example.com');
|
|
|
|
const tryLockSpy = jest.spyOn(fetchInstanceMetadataService, 'tryLock');
|
|
|
|
const unlockSpy = jest.spyOn(fetchInstanceMetadataService, 'unlock');
|
|
|
|
|
2024-03-15 05:08:54 +01:00
|
|
|
await fetchInstanceMetadataService.fetchInstanceMetadata({ host: 'example.com' } as FIXME, true);
|
2024-03-05 18:56:00 +01:00
|
|
|
expect(tryLockSpy).toHaveBeenCalledTimes(1);
|
2024-03-04 05:48:57 +01:00
|
|
|
expect(unlockSpy).toHaveBeenCalledTimes(1);
|
|
|
|
expect(federatedInstanceService.fetch).toHaveBeenCalledTimes(0);
|
|
|
|
expect(httpRequestService.getJson).toHaveBeenCalled();
|
|
|
|
});
|
2023-07-08 12:34:19 +02:00
|
|
|
});
|