test(koa): cover Koa middleware with tests

package.json

@@ -4,7 +4,7 @@
   "description": "",
   "main": "index.js",
   "scripts": {
-    "test": "NODE_ENV=test mocha $(find src -name \"*.spec.js\")",
+    "test": "NODE_ENV=test mocha --require co-mocha $(find src -name \"*.spec.js\")",
     "eslint": "eslint src"
   },
   "repository": {
@@ -21,6 +21,7 @@
     "babel-eslint": "^4.0.0",
     "chai": "^3.2.0",
     "chai-subset": "^1.0.1",
+    "co-mocha": "^1.1.2",
     "eslint": "1.0.0-rc-3",
     "express": "^4.13.1",
     "hapi": "^8.8.0",

src/koa/index.js

@@ -20,7 +20,7 @@ function create(options) {
 
       if (isGet(this) && isPrefixed(this, prefix)) {
         this.body = yield graphql(schema, query);
-        return;
+        return this.body;
       }
 
       yield next;

src/koa/index.spec.js

@@ -0,0 +1,120 @@
+var expect = require('chai').expect;
+
+describe('graffiti koa', function() {
+
+  var koa = require('./');
+
+  describe('checks for required options', function() {
+
+    it('throws an error if not all met', function() {
+
+      var mwFactory = koa.create({
+        graphql: {}
+      });
+
+      try {
+        mwFactory({
+          prefix: '/graphql'
+        });
+      } catch (ex) {
+        expect(ex).to.be.ok;
+        return;
+      }
+
+      throw new Error('Error should have been thrown');
+
+    });
+
+  });
+
+  it('creates the schema', function() {
+    var mwFactory = koa.create({
+      graphql: {}
+    });
+    var getSchemaSpy = this.sandbox.spy();
+    var models = [{
+      name: 'User'
+    }];
+
+    mwFactory({
+      prefix: '/graphql',
+      models: models,
+      adapter: {
+        getSchema: getSchemaSpy
+      }
+    });
+
+    expect(getSchemaSpy).to.be.calledWith(models);
+  });
+
+  describe('requested url starts with prefix', function() {
+
+    it('returns with proper results', function *() {
+
+      var result = {
+        data: 1
+      };
+
+      var mwFactory = koa.create({
+        graphql: function() {
+          return Promise.resolve(result);
+        }
+      });
+
+      var mw = mwFactory({
+        prefix: '/graphql',
+        models: [],
+        adapter: {
+          getSchema: function() {}
+        }
+      });
+
+      var request = {
+        method: 'GET',
+        path: '/graphql',
+        query: {
+          q: '{__type}'
+        }
+      };
+
+      var res = yield mw.call(request);
+
+      expect(res).to.eql(result);
+    });
+
+  });
+
+  describe('requested url does not start with prefix', function() {
+
+    it('calls the next middleware', function *(done) {
+
+      var mwFactory = koa.create({
+        graphql: {}
+      });
+
+      var mw = mwFactory({
+        prefix: '/graphql',
+        models: [],
+        adapter: {
+          getSchema: function() {}
+        }
+      });
+
+      var request = {
+        method: 'GET',
+        path: '/not-good',
+        query: {
+          q: '{__type}'
+        }
+      };
+
+      yield mw.call(request, function *() {
+        yield done();
+      });
+
+      throw new Error('next should have been called');
+    });
+
+  });
+
+});