diff --git a/src/api/endpoints/drive.js b/src/api/endpoints/drive.js index f39019e6e..00f9be780 100644 --- a/src/api/endpoints/drive.js +++ b/src/api/endpoints/drive.js @@ -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); diff --git a/test/api.js b/test/api.js index c55fae716..31983bfe2 100644 --- a/test/api.js +++ b/test/api.js @@ -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();