iceshrimp-legacy/packages/backend/test/misc/mock-resolver.ts
2022-05-21 22:21:41 +09:00

36 lines
844 B
TypeScript

import Resolver from '../../src/remote/activitypub/resolver.js';
import { IObject } from '../../src/remote/activitypub/type.js';
type MockResponse = {
type: string;
content: string;
};
export class MockResolver extends Resolver {
private _rs = new Map<string, MockResponse>();
public async _register(uri: string, content: string | Record<string, any>, type = 'application/activity+json') {
this._rs.set(uri, {
type,
content: typeof content === 'string' ? content : JSON.stringify(content),
});
}
public async resolve(value: string | IObject): Promise<IObject> {
if (typeof value !== 'string') return value;
const r = this._rs.get(value);
if (!r) {
throw {
name: 'StatusError',
statusCode: 404,
message: 'Not registed for mock',
};
}
const object = JSON.parse(r.content);
return object;
}
}