Mock flat-file-db instead of db.js

Author: relekang

src/db.js

@@ -14,4 +14,5 @@ module.exports = {
   getAll: adapter.getAll,
   put: adapter.put,
   has: adapter.has,
+  keys: adapter.keys,
 }

src/flat-file-adapter.js

@@ -3,17 +3,24 @@ const promise = require('promise')
 
 const db = flatfile.sync(process.env.DB_NAME || 'views.db')
 
-function getAll(options) {
-  const data = {};
-  const keys = db.keys()
-    .filter(key => String(options.filter) === 'false' ? true : key.startsWith(pathname));
-
-  return Promise.resolve(keys.map(key => db.get(key)));
-}
 
 module.exports = {
-  get: promise.denodeify(db.get.bind(db)),
   put: promise.denodeify(db.put.bind(db)),
-  has: promise.denodeify(db.has.bind(db)),
-  getAll: getAll,
+
+  has: (key) => Promise.resolve(db.has(key)),
+  get: (key) => Promise.resolve(db.get(key)),
+  keys: () => Promise.resolve(db.keys()),
+
+  getAll: async function getAll(options) {
+    const data = {}
+    const keys = await module.exports.keys()
+
+    keys
+      .filter(key => String(options.filter) === 'false' ? true : key.startsWith(options.pathname))
+      .forEach((key) => {
+        data[key] = db.get(key)
+      })
+
+    return data
+  }
 }

src/index.js

@@ -8,12 +8,17 @@ module.exports = async function (req, res) {
   const { pathname, query } = url.parse(req.url, /* parseQueryString */ true)
   // Send all views down if "?all" is true
   if (String(query.all) === 'true') {
-    const data = {
-      data: await db.getAll(),
-      time: Date.now()
+    try {
+      const data = {
+        data: await db.getAll({ pathname: pathname, filter: query.filter }),
+        time: Date.now()
+      }
+      send(res, 200, data)
+      return
+    } catch (err) {
+      console.log(err)
+      throw createError(500, 'Internal server error.')
     }
-    send(res, 200, data)
-    return
   }
   // Check that a page is provided
   if (pathname.length <= 1) {

tests/atomicity.test.js

@@ -1,7 +1,7 @@
 const request = require('request-promise')
 const { listen, mockDb } = require('./utils')
 
-jest.mock('../src/flat-file-adapter', () => mockDb)
+jest.mock('flat-file-db', () => mockDb)
 const service = require('../src')
 let url
 

tests/items.test.js

@@ -1,7 +1,7 @@
 const request = require('request-promise')
 const { listen, mockDb } = require('./utils')
 
-jest.mock('../src/flat-file-adapter', () => mockDb)
+jest.mock('flat-file-db', () => mockDb)
 const service = require('../src')
 let url
 

tests/utils.js

@@ -6,23 +6,17 @@ const DB = () => {
   let DELAY = 1
 
   return {
-    get: (key) => new Promise((res, rej) => {
-      setTimeout(() => {
-        res(data[key])
-      }, DELAY)
+    sync: () => ({
+      get: (key) => data[key],
+      put: (key, val, cb) => {
+        setTimeout(() => {
+          data[key] = val
+          cb()
+        }, DELAY)
+      },
+      has: (key) => ({}.hasOwnProperty.call(data, key)),
+      keys: () => Object.keys(data),
     }),
-    put: (key, val) => new Promise((res, rej) => {
-      setTimeout(() => {
-        data[key] = val
-        res()
-      }, DELAY)
-    }),
-    has: (key) => new Promise((res, rej) => {
-      setTimeout(() => {
-        res(!!data[key])
-      }, DELAY)
-    }),
-    getAll: (options) => Object.assign({}, data),
 
     // Custom methods used in tests
     _reset: () => { data = {} },