From 68cd6283c2e19a2d047dd22f8dc03e8581aaad47 Mon Sep 17 00:00:00 2001
From: syuilo <syuilotan@yahoo.co.jp>
Date: Tue, 14 Feb 2017 06:42:01 +0900
Subject: [PATCH] [API] Fix bug

---
 src/api/endpoints/drive.js | 12 ++++++------
 test/api.js                | 16 ++++++++++++++++
 2 files changed, 22 insertions(+), 6 deletions(-)

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();