Refactered flattening

Author: sergeysova

lib/flatten.js

@@ -3,16 +3,17 @@ function flattenize(routes, prefix, parent) {
 
   Object.keys(routes.local).forEach(method => {
     const localName = prefix === '/' ? '/' : `${prefix}/`
-    list[localName] = {
-      method,
-      listeners: [].concat(
-        parent.before,
-        routes.before,
-        routes.local[method],
-        routes.after,
-        parent.after
-      ),
+
+    if (!list[localName]) {
+      list[localName] = {}
     }
+    list[localName][method] = [].concat(
+      parent.before,
+      routes.before,
+      routes.local[method],
+      routes.after,
+      parent.after
+    )
   })
 
   Object.keys(routes.scoped).forEach(scope => {

lib/flatten.test.js

@@ -0,0 +1,98 @@
+import test from 'ava'
+import { createRest, flattenRoutes } from './index'
+
+/* eslint-disable no-shadow */
+
+const get = () => {}
+const post = () => {}
+const put = () => {}
+const patch = () => {}
+const destroy = () => {}
+const before = () => {}
+const after = () => {}
+
+const make = (method, listeners = []) => ({
+  method, listeners,
+})
+
+test('Empty', t => {
+  t.deepEqual(
+    flattenRoutes(createRest(r => {
+    })),
+    {
+    }
+  )
+})
+
+test('Local methods', t => {
+  t.deepEqual(
+    flattenRoutes(createRest(r => {
+      r.get(get)
+      r.post(post)
+      r.put(put)
+      r.patch(patch)
+      r.delete(destroy)
+    })),
+    {
+      '/': {
+        GET: [get],
+        POST: [post],
+        PUT: [put],
+        PATCH: [patch],
+        DELETE: [destroy],
+      }
+    }
+  )
+})
+
+test('Local and scoped methods', t => {
+  t.deepEqual(
+    flattenRoutes(createRest(r => {
+      r.get(get)
+      r.post(post)
+      r.get('/name', get)
+      r.post('/name', post)
+    })),
+    {
+      '/': {
+        GET: [get],
+        POST: [post],
+      },
+      '/name/': {
+        GET: [get],
+        POST: [post],
+      },
+    }
+  )
+})
+
+test('Local, scoped methods and methods in scope', t => {
+  t.deepEqual(
+    flattenRoutes(createRest(r => {
+      r.get(get)
+      r.post(post)
+      r.get('/name', get)
+      r.post('name', post)
+      r.scope('foo', r => {
+        r.get(get)
+        r.post('bar', post)
+      })
+    })),
+    {
+      '/': {
+        GET: [get],
+        POST: [post],
+      },
+      '/name/': {
+        GET: [get],
+        POST: [post],
+      },
+      '/foo/': {
+        GET: [get],
+      },
+      '/foo/bar/': {
+        POST: [post],
+      },
+    }
+  )
+})