zoriya/flood
1
0
Fork 0
mirror of https://github.com/zoriya/flood.git synced 2026-06-07 04:15:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

server: tests: test torrent completion in create torrent case

Browse Source
This commit is contained in:
Jesse Chan
2020-10-21 13:31:43 +08:00
parent 3f8172d43c
commit 99bab36241
2 changed files with 107 additions and 91 deletions
Download Patch File Download Diff File Expand all files Collapse all files
server/routes/api/torrents.test.ts
+106 -90
View File
@@ -82,8 +82,6 @@ describe('POST /api/torrents/add-urls', () => {
start: false,
};
const torrentAdded = watchTorrentList('add');
it('Adds torrents to disallowed path via URLs', (done) => {
request
.post('/api/torrents/add-urls')
@@ -102,6 +100,7 @@ describe('POST /api/torrents/add-urls', () => {
});
it('Adds torrents via URLs', (done) => {
const torrentAdded = watchTorrentList('add');
request
.post('/api/torrents/add-urls')
.send(addTorrentByURLOptions)
@@ -112,47 +111,47 @@ describe('POST /api/torrents/add-urls', () => {
.end((err, _res) => {
if (err) done(err);
done();
torrentAdded.then(() => {
done();
});
});
});
it('GET /api/torrents to verify torrents are added via URLs', (done) => {
torrentAdded.then(() => {
request
.get('/api/torrents')
.send()
.set('Cookie', [authToken])
.set('Accept', 'application/json')
.expect(200)
.expect('Content-Type', /json/)
.end(async (err, res) => {
if (err) done(err);
request
.get('/api/torrents')
.send()
.set('Cookie', [authToken])
.set('Accept', 'application/json')
.expect(200)
.expect('Content-Type', /json/)
.end(async (err, res) => {
if (err) done(err);
expect(res.body.torrents).not.toBeNull();
const torrentList: TorrentList = res.body.torrents;
expect(res.body.torrents).not.toBeNull();
const torrentList: TorrentList = res.body.torrents;
const addedTorrents = Object.values(torrentList).filter((torrent) =>
addTorrentByURLOptions.tags?.every((tag) => torrent.tags.includes(tag)),
);
const addedTorrents = Object.values(torrentList).filter((torrent) =>
addTorrentByURLOptions.tags?.every((tag) => torrent.tags.includes(tag)),
);
expect(addedTorrents).toHaveLength(addTorrentByURLOptions.urls.length);
expect(addedTorrents).toHaveLength(addTorrentByURLOptions.urls.length);
await Promise.all(
addedTorrents.map(async (torrent) => {
expect(torrent.baseDirectory).toBe(addTorrentByURLOptions.destination);
await Promise.all(
addedTorrents.map(async (torrent) => {
expect(torrent.baseDirectory).toBe(addTorrentByURLOptions.destination);
const expectedStatuses: Array<TorrentStatus> = addTorrentByURLOptions.start
? ['downloading']
: ['stopped', 'inactive'];
expect(torrent.status).toEqual(expect.arrayContaining(expectedStatuses));
const expectedStatuses: Array<TorrentStatus> = addTorrentByURLOptions.start
? ['downloading']
: ['stopped', 'inactive'];
expect(torrent.status).toEqual(expect.arrayContaining(expectedStatuses));
torrentHash = torrent.hash;
}),
);
torrentHash = torrent.hash;
}),
);
done();
});
});
done();
});
});
});
@@ -165,8 +164,6 @@ describe('POST /api/torrents/add-files', () => {
start: false,
};
const torrentAdded = watchTorrentList('add');
it('Adds torrents to disallowed path via files', (done) => {
request
.post('/api/torrents/add-urls')
@@ -185,6 +182,7 @@ describe('POST /api/torrents/add-files', () => {
});
it('Adds torrents via files', (done) => {
const torrentAdded = watchTorrentList('add');
request
.post('/api/torrents/add-files')
.send(addTorrentByFileOptions)
@@ -195,40 +193,40 @@ describe('POST /api/torrents/add-files', () => {
.end((err, _res) => {
if (err) done(err);
done();
torrentAdded.then(() => {
done();
});
});
});
it('GET /api/torrents to verify torrents are added via files', (done) => {
torrentAdded.then(() => {
request
.get('/api/torrents')
.send()
.set('Cookie', [authToken])
.set('Accept', 'application/json')
.expect(200)
.expect('Content-Type', /json/)
.end(async (err, res) => {
if (err) done(err);
request
.get('/api/torrents')
.send()
.set('Cookie', [authToken])
.set('Accept', 'application/json')
.expect(200)
.expect('Content-Type', /json/)
.end(async (err, res) => {
if (err) done(err);
expect(res.body.torrents == null).toBe(false);
const torrentList: TorrentList = res.body.torrents;
expect(res.body.torrents == null).toBe(false);
const torrentList: TorrentList = res.body.torrents;
const addedTorrents = Object.values(torrentList).filter((torrent) =>
addTorrentByFileOptions.tags?.every((tag) => torrent.tags.includes(tag)),
);
const addedTorrents = Object.values(torrentList).filter((torrent) =>
addTorrentByFileOptions.tags?.every((tag) => torrent.tags.includes(tag)),
);
expect(addedTorrents).toHaveLength(addTorrentByFileOptions.files.length);
expect(addedTorrents).toHaveLength(addTorrentByFileOptions.files.length);
await Promise.all(
addedTorrents.map(async (torrent) => {
expect(torrent.directory.startsWith(addTorrentByFileOptions.destination)).toBe(true);
}),
);
await Promise.all(
addedTorrents.map(async (torrent) => {
expect(torrent.directory.startsWith(addTorrentByFileOptions.destination)).toBe(true);
}),
);
done();
});
});
done();
});
});
});
@@ -237,14 +235,15 @@ describe('POST /api/torrents/create', () => {
sourcePath: tempDirectory,
tags: ['testCreate'],
trackers: testTrackers,
start: false,
start: true,
isPrivate: false,
};
const torrentAdded = watchTorrentList('add');
const dummyFilePath = path.join(tempDirectory, 'dummy');
fs.writeFileSync(dummyFilePath, 'test');
it('Creates a torrent', (done) => {
fs.writeFileSync(path.join(tempDirectory, 'dummy'), 'test');
it('Creates a multi-file torrent', (done) => {
const torrentAdded = watchTorrentList('add');
request
.post('/api/torrents/create')
.send(createTorrentOptions)
@@ -261,36 +260,53 @@ describe('POST /api/torrents/create', () => {
});
});
it('GET /api/torrents to verify torrents are added via creation', (done) => {
torrentAdded.then(() => {
request
.get('/api/torrents')
.send()
.set('Cookie', [authToken])
.set('Accept', 'application/json')
.expect(200)
.expect('Content-Type', /json/)
.end(async (err, res) => {
if (err) done(err);
expect(res.body.torrents == null).toBe(false);
const torrentList: TorrentList = res.body.torrents;
const addedTorrents = Object.values(torrentList).filter((torrent) =>
createTorrentOptions.tags?.every((tag) => torrent.tags.includes(tag)),
);
expect(addedTorrents).toHaveLength(1);
await Promise.all(
addedTorrents.map(async (torrent) => {
expect(torrent.isPrivate).toBe(createTorrentOptions.isPrivate);
}),
);
it('Creates a single-file torrent', (done) => {
const torrentAdded = watchTorrentList('add');
request
.post('/api/torrents/create')
.send({...createTorrentOptions, sourcePath: dummyFilePath})
.set('Cookie', [authToken])
.set('Accept', 'application/x-bittorrent')
.expect(200)
.expect('Content-Type', /x-bittorrent/)
.end((err, _res) => {
if (err) done(err);
torrentAdded.then(() => {
done();
});
});
});
});
it('GET /api/torrents to verify torrents are added via creation', (done) => {
request
.get('/api/torrents')
.send()
.set('Cookie', [authToken])
.set('Accept', 'application/json')
.expect(200)
.expect('Content-Type', /json/)
.end(async (err, res) => {
if (err) done(err);
expect(res.body.torrents == null).toBe(false);
const torrentList: TorrentList = res.body.torrents;
const addedTorrents = Object.values(torrentList).filter((torrent) =>
createTorrentOptions.tags?.every((tag) => torrent.tags.includes(tag)),
);
expect(addedTorrents).toHaveLength(2);
await Promise.all(
addedTorrents.map(async (torrent) => {
expect(torrent.isPrivate).toBe(createTorrentOptions.isPrivate);
expect(torrent.percentComplete).toBe(100);
}),
);
done();
});
});
});
server/routes/api/torrents.ts
+1 -1
View File
@@ -153,7 +153,7 @@ router.post<unknown, unknown, CreateTorrentOptions>('/create', async (req, res)
req.services?.clientGatewayService
?.addTorrentsByFile({
files: [torrent.toString('base64')],
destination: sourcePath,
destination: fs.lstatSync(sanitizedPath).isDirectory() ? sanitizedPath : path.dirname(sanitizedPath),
tags,
isBasePath: true,
isCompleted: true,