Add routing to micro tests

apollographql · Jun 15, 2017 · 387633c · 387633c
1 parent 7f7d24e
commit 387633c
Show file tree

Hide file tree

Showing 5 changed files with 62 additions and 13 deletions.
diff --git a/packages/graphql-server-integration-testsuite/src/index.ts b/packages/graphql-server-integration-testsuite/src/index.ts
@@ -817,5 +817,20 @@ export default (createApp: CreateAppFunc, destroyApp?: DestroyAppFunc) => {
           });
       });
     });
+
+    describe('server setup', () => {
+      it('throws error on 404 routes', () => {
+          app = createApp();
+
+          const query = {
+              query: '{ testString }',
+          };
+          const req = request(app)
+              .get(`/bogus-route?${querystring.stringify(query)}`);
+          return req.then((res) => {
+              expect(res.status).to.equal(404);
+          });
+      });
+    });
   });
 };
diff --git a/packages/graphql-server-lambda/src/lambdaApollo.test.ts b/packages/graphql-server-lambda/src/lambdaApollo.test.ts
@@ -6,19 +6,28 @@ import 'mocha';
 import * as url from 'url';
 
 function createLambda(options: CreateAppOptions = {}) {
-  let handler,
+  let route,
+    handler,
     callback,
     event,
     context;
 
   options.graphqlOptions = options.graphqlOptions || { schema: Schema };
   if (options.graphiqlOptions ) {
+    route = '/graphiql';
     handler = graphiqlLambda( options.graphiqlOptions );
   } else {
+    route = '/graphql';
     handler = graphqlLambda( options.graphqlOptions );
   }
 
   return function(req, res) {
+    if (!req.url.startsWith(route)) {
+      res.statusCode = 404;
+      res.end();
+      return;
+    }
+
     let body = '';
     req.on('data', function (chunk) {
       body += chunk;

diff --git a/packages/graphql-server-micro/README.md b/packages/graphql-server-micro/README.md
@@ -10,14 +10,15 @@ import micro, { send } from "micro";
 import { get, post, router } from "microrouter";
 import schema from "./schema";
 
-const graphqlHandler: microGraphql({ schema });
+const graphqlHandler = microGraphql({ schema });
+const graphiqlHandler = microGraphiql({ endpointURL: "/graphql" });
 
 const server = micro(
   router(
     get("/graphql", graphqlHandler),
     post("/graphql", graphqlHandler),
-    get("/graphiql", microGraphiql({ endpointURL: "/graphql" })),
-    (req, res) => send(res, 404, "404 Not found."),
+    get("/graphiql", graphiqlHandler),
+    (req, res) => send(res, 404, "not found"),
   ),
 );
 

diff --git a/packages/graphql-server-micro/package.json b/packages/graphql-server-micro/package.json
@@ -30,7 +30,8 @@
   },
   "devDependencies": {
     "graphql-server-integration-testsuite": "^0.8.4",
-    "micro": "^7.3.3"
+    "micro": "^7.3.3",
+    "microrouter": "^2.1.1"
   },
   "peerDependencies": {
     "graphql": "^0.9.0 || ^0.10.1",

diff --git a/packages/graphql-server-micro/src/microApollo.test.ts b/packages/graphql-server-micro/src/microApollo.test.ts
@@ -1,16 +1,39 @@
 import { microGraphql, microGraphiql } from './microApollo';
 import 'mocha';
 
-import micro from 'micro';
-import testSuite, { schema as Schema, CreateAppOptions  } from 'graphql-server-integration-testsuite';
+import micro, { send } from 'micro';
+import { router, get, post, put, patch, del, head, options as opts } from 'microrouter';
+import testSuite, { schema, CreateAppOptions  } from 'graphql-server-integration-testsuite';
+
 
 function createApp(options: CreateAppOptions) {
-    if (options && options.graphiqlOptions ) {
-        return micro(microGraphiql(options.graphiqlOptions));
-    } else {
-        const graphqlOptions = (options && options.graphqlOptions) || { schema: Schema };
-        return micro(microGraphql(graphqlOptions));
-    }
+    const graphqlOptions = (options && options.graphqlOptions) || { schema };
+    const graphiqlOptions = (options && options.graphiqlOptions) || { endpointURL: '/graphql' };
+
+    const graphqlHandler = microGraphql(graphqlOptions);
+    const graphiqlHandler = microGraphiql(graphiqlOptions);
+
+    return micro(
+        router(
+            get('/graphql', graphqlHandler),
+            post('/graphql', graphqlHandler),
+            put('/graphql', graphqlHandler),
+            patch('/graphql', graphqlHandler),
+            del('/graphql', graphqlHandler),
+            head('/graphql', graphqlHandler),
+            opts('/graphql', graphqlHandler),
+
+            get('/graphiql', graphiqlHandler),
+            post('/graphiql', graphiqlHandler),
+            put('/graphiql', graphiqlHandler),
+            patch('/graphiql', graphiqlHandler),
+            del('/graphiql', graphiqlHandler),
+            head('/graphiql', graphiqlHandler),
+            opts('/graphiql', graphiqlHandler),
+
+            (req, res) => send(res, 404, 'not found'),
+        ),
+    );
 }
 
 describe('integration:Micro', () => {