From 8d624459d46dd5aa674ecc446059178156aa404c Mon Sep 17 00:00:00 2001
From: Dick Choi <fluke8259@gmail.com>
Date: Sun, 28 Aug 2016 00:20:08 +0900
Subject: [PATCH] add assertions to addStorage test

check version and folders attribute from resolved data and json
---
 browser/main/lib/dataApi/addStorage.js | 3 ++-
 tests/dataApi/addStorage.js            | 9 +++++++++
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/browser/main/lib/dataApi/addStorage.js b/browser/main/lib/dataApi/addStorage.js
index 8d760b12..f7fbf6ef 100644
--- a/browser/main/lib/dataApi/addStorage.js
+++ b/browser/main/lib/dataApi/addStorage.js
@@ -40,7 +40,8 @@ function addStorage (input) {
 
   return Promise.resolve(newStorage)
     .then(resolveStorageData)
-    .then(function saveMetadataToLocalStorage () {
+    .then(function saveMetadataToLocalStorage (resolvedStorage) {
+      newStorage = resolvedStorage
       rawStorages.push({
         key: newStorage.key,
         type: newStorage.type,
diff --git a/tests/dataApi/addStorage.js b/tests/dataApi/addStorage.js
index 001595b9..1ca10c54 100644
--- a/tests/dataApi/addStorage.js
+++ b/tests/dataApi/addStorage.js
@@ -12,6 +12,7 @@ const TestDummy = require('../fixtures/TestDummy')
 const sander = require('sander')
 const _ = require('lodash')
 const os = require('os')
+const CSON = require('season')
 
 const v1StoragePath = path.join(os.tmpdir(), 'test/addStorage-v1-storage')
 // const legacyStoragePath = path.join(os.tmpdir(), 'test/addStorage-legacy-storage')
@@ -42,6 +43,8 @@ test.serial('Add Storage', (t) => {
       t.is(storage.name, input.name)
       t.is(storage.type, input.type)
       t.is(storage.path, input.path)
+      t.is(storage.version, '1.0')
+      t.is(storage.folders.length, t.context.v1StorageData.json.folders.length)
 
       // Check data.notes
       t.is(notes.length, t.context.v1StorageData.notes.length)
@@ -54,6 +57,12 @@ test.serial('Add Storage', (t) => {
       t.is(cacheData.name, input.name)
       t.is(cacheData.type, input.type)
       t.is(cacheData.path, input.path)
+
+      // Check boostnote.json
+      let jsonData = CSON.readFileSync(path.join(storage.path, 'boostnote.json'))
+      t.true(_.isArray(jsonData.folders))
+      t.is(jsonData.version, '1.0')
+      t.is(jsonData.folders.length, t.context.v1StorageData.json.folders.length)
     })
 })