94 lines
3.2 KiB
TypeScript
94 lines
3.2 KiB
TypeScript
/*
|
||
* SPDX-FileCopyrightText: syuilo and misskey-project
|
||
* SPDX-License-Identifier: AGPL-3.0-only
|
||
*/
|
||
|
||
process.env.NODE_ENV = 'test';
|
||
|
||
import * as assert from 'assert';
|
||
import { MiNote } from '@/models/Note.js';
|
||
import type { Packed } from '@/misc/json-schema.js';
|
||
import { api, initTestDb, makeStreamCatcher, post, signup, uploadFile } from '../utils.js';
|
||
import type * as misskey from 'misskey-js';
|
||
import type{ Repository } from 'typeorm';
|
||
|
||
describe('Drive', () => {
|
||
let Notes: Repository<MiNote>;
|
||
|
||
let alice: misskey.entities.SignupResponse;
|
||
let bob: misskey.entities.SignupResponse;
|
||
|
||
beforeAll(async () => {
|
||
const connection = await initTestDb(true);
|
||
Notes = connection.getRepository(MiNote);
|
||
alice = await signup({ username: 'alice' });
|
||
bob = await signup({ username: 'bob' });
|
||
}, 1000 * 60 * 2);
|
||
|
||
test('ファイルURLからアップロードできる', async () => {
|
||
// utils.js uploadUrl の処理だがAPIレスポンスも見るためここで同様の処理を書いている
|
||
|
||
const marker = Math.random().toString();
|
||
|
||
const url = 'https://raw.githubusercontent.com/MisskeyIO/misskey/io/packages/backend/test/resources/Lenna.jpg';
|
||
|
||
const catcher = makeStreamCatcher(
|
||
alice,
|
||
'main',
|
||
(msg) => msg.type === 'urlUploadFinished' && msg.body.marker === marker,
|
||
(msg) => msg.body.file as Packed<'DriveFile'>,
|
||
10 * 1000);
|
||
|
||
const res = await api('drive/files/upload-from-url', {
|
||
url,
|
||
marker,
|
||
force: true,
|
||
}, alice);
|
||
|
||
const file = await catcher;
|
||
|
||
assert.strictEqual(res.status, 204);
|
||
assert.strictEqual(file.name, 'Lenna.jpg');
|
||
assert.strictEqual(file.type, 'image/jpeg');
|
||
});
|
||
|
||
test('ローカルからアップロードできる', async () => {
|
||
// APIレスポンスを直接使用するので utils.js uploadFile が通過することで成功とする
|
||
|
||
const res = await uploadFile(alice, { path: 'Lenna.jpg', name: 'テスト画像' });
|
||
|
||
assert.strictEqual(res.body?.name, 'テスト画像.jpg');
|
||
assert.strictEqual(res.body.type, 'image/jpeg');
|
||
});
|
||
|
||
test('添付ノート一覧を取得できる', async () => {
|
||
const ids = (await Promise.all([uploadFile(alice), uploadFile(alice), uploadFile(alice)])).map(elm => elm.body!.id);
|
||
|
||
const note0 = await post(alice, { fileIds: [ids[0]] });
|
||
const note1 = await post(alice, { fileIds: [ids[0], ids[1]] });
|
||
|
||
const attached0 = await api('drive/files/attached-notes', { fileId: ids[0] }, alice);
|
||
assert.strictEqual(attached0.body.length, 2);
|
||
assert.strictEqual(attached0.body[0].id, note1.id);
|
||
assert.strictEqual(attached0.body[1].id, note0.id);
|
||
|
||
const attached1 = await api('drive/files/attached-notes', { fileId: ids[1] }, alice);
|
||
assert.strictEqual(attached1.body.length, 1);
|
||
assert.strictEqual(attached1.body[0].id, note1.id);
|
||
|
||
const attached2 = await api('drive/files/attached-notes', { fileId: ids[2] }, alice);
|
||
assert.strictEqual(attached2.body.length, 0);
|
||
});
|
||
|
||
test('添付ノート一覧は他の人から見えない', async () => {
|
||
const file = await uploadFile(alice);
|
||
|
||
await post(alice, { fileIds: [file.body!.id] });
|
||
|
||
const res = await api('drive/files/attached-notes', { fileId: file.body!.id }, bob);
|
||
assert.strictEqual(res.status, 400);
|
||
assert.strictEqual('error' in res.body, true);
|
||
});
|
||
});
|
||
|