[API] Fix bug

This commit is contained in:
syuilo 2017-02-14 06:42:01 +09:00
parent 2012477c82
commit a529b51c2d
2 changed files with 22 additions and 6 deletions

View File

@ -3,7 +3,7 @@
/**
* Module dependencies
*/
import DriveFile from './models/drive-file';
import DriveFile from '../models/drive-file';
/**
* Get drive information
@ -16,11 +16,11 @@ module.exports = (params, user) =>
new Promise(async (res, rej) =>
{
// Fetch all files to calculate drive usage
const files = await DriveFile
.find({ user_id: user._id }, {
datasize: true,
_id: false
});
const files = await DriveFile
.find({ user_id: user._id }, {
datasize: true,
_id: false
});
// Calculate drive usage (in byte)
const usage = files.map(file => file.datasize).reduce((x, y) => x + y, 0);

View File

@ -763,6 +763,22 @@ describe('API', () => {
}));
});
describe('drive', () => {
it('ドライブ情報を取得できる', () => new Promise(async (done) => {
const me = await insertSakurako();
const file = await insertDriveFile({
user_id: me._id,
datasize: 1024
});
request('/drive', {}, me).then(res => {
res.should.have.status(200);
res.body.should.be.a('object');
res.body.should.have.property('usage').eql(1024);
done();
});
}));
});
describe('drive/files/create', () => {
it('ファイルを作成できる', () => new Promise(async (done) => {
const me = await insertSakurako();