simplified adding routes to swagger test server

agco · Feb 22, 2016 · 1fd59f4 · 1fd59f4
1 parent 224d8ce
commit 1fd59f4
Showing 1 changed file with 3 additions and 18 deletions.
diff --git a/test/swagger.spec.js b/test/swagger.spec.js
@@ -72,24 +72,9 @@ describe('Swagger docs', function () {
                 { register: HapiSwagger, options: swaggerOptions }
             ], () => {
                 server.start(() => {
-                    _.forEach(schemas, function (schema) {
-                        [
-                            'get',
-                            'getById',
-                            'getChangesStreaming',
-                            'post',
-                            'patch',
-                            'delete'
-                        ].forEach(function (verb) {
-                            const route = server.plugins['hapi-harvester'].routes[verb](schema)
-                            if (_.isArray(route)) {
-                                _.forEach(route, function (route) {
-                                    server.route(route)
-                                });
-                            } else {
-                                server.route(route)
-                            }
-                        })
+                    const harvester = server.plugins['hapi-harvester']
+                    _.each(schemas, (schema) => {
+                        _.each(harvester.routes.all(schema), (route) => server.route(route))
                     })
                     resolve(server)
                 })