add a test

This commit is contained in:
Emily 2018-07-12 16:07:18 -07:00
parent 5ff92c6452
commit 527e9f09c9
3 changed files with 29 additions and 5 deletions

View File

@ -160,6 +160,7 @@ function download(id, keychain, onprogress, canceller) {
resolve(this.result);
};
});
xhr.addEventListener('progress', function(event) {
if (event.lengthComputable && event.target.status === 200) {
onprogress([event.loaded, event.total]);
@ -168,8 +169,10 @@ function download(id, keychain, onprogress, canceller) {
const auth = await keychain.authHeader();
xhr.open('get', `/api/download/${id}`);
xhr.setRequestHeader('Authorization', auth);
xhr.setRequestHeader('Connection', 'close');
xhr.responseType = 'blob';
xhr.send();
onprogress([0, 1]);
});
}

View File

@ -14,14 +14,14 @@ module.exports = async function(req, res) {
'WWW-Authenticate': `send-v1 ${req.nonce}`
});
const file_stream = storage.get(id);
let sentBytes = 0;
let cancelled = false;
req.on('close', () => (cancelled = true));
req.on('close', () => {
cancelled = true;
file_stream.destroy();
});
file_stream.on('data', c => (sentBytes += c.length));
file_stream.on('end', async () => {
file_stream.on('close', async () => {
if (cancelled) {
return;
}

View File

@ -134,6 +134,27 @@ describe('Upload / Download flow', function() {
}
});
it('can cancel and not increase download count', async function() {
const fs = new FileSender(blob);
const file = await fs.upload();
const fr = new FileReceiver({
secretKey: file.toJSON().secretKey,
id: file.id,
nonce: file.keychain.nonce,
requiresPassword: false
});
await fr.getMetadata();
fr.once('progress', () => fr.cancel());
try {
await fr.download(noSave);
assert.fail('not cancelled');
} catch (e) {
await file.updateDownloadCount();
assert.equal(file.dtotal, 0);
}
});
it('can allow multiple downloads', async function() {
const fs = new FileSender(blob);
const file = await fs.upload();