allow setting Mongo location using MONGO_URI

Author: darthfiddler

src/server/db.js

@@ -66,6 +66,7 @@ export class Mongo {
    */
   constructor({ url, dbname, cacheSize, mockClient }) {
     if (cacheSize === undefined) cacheSize = 1000;
+    if (dbname === undefined) dbname = 'bgio';
 
     this.client = mockClient || MongoClient;
     this.url = url;

src/server/db.test.js

@@ -76,7 +76,8 @@ test('Mongo', async () => {
   }
 
   {
-    const db = new Mongo({ url: 'a' });
+    const db = new Mongo({ url: 'a', dbname: 'test' });
     expect(db.client).toBeDefined();
+    expect(db.dbname).toBe('test');
   }
 });

src/server/index.js

@@ -9,7 +9,7 @@
 const Koa = require('koa');
 const IO = require('koa-socket');
 const Redux = require('redux');
-import { InMemory } from './db';
+import { InMemory, Mongo } from './db';
 import { createGameReducer } from '../core/reducer';
 const PING_TIMEOUT = 20 * 1e3;
 const PING_INTERVAL = 10 * 1e3;
@@ -26,7 +26,11 @@ export function Server({ games, db, _clientInfo, _roomInfo }) {
   io.attach(app);
 
   if (db === undefined) {
-    db = new InMemory();
+    if (process.env.MONGO_URI) {
+      db = new Mongo({ url: process.env.MONGO_URI });
+    } else {
+      db = new InMemory();
+    }
   }
 
   const clientInfo = _clientInfo || new Map();
@@ -131,6 +135,7 @@ export function Server({ games, db, _clientInfo, _roomInfo }) {
 
   return {
     app,
+    db,
     run: async (port, callback) => {
       await db.connect();
       app.listen(port, callback);

src/server/index.test.js

@@ -11,6 +11,10 @@ import Game from '../core/game';
 import * as ActionCreators from '../core/action-creators';
 import * as Redux from 'redux';
 
+beforeEach(() => {
+  jest.resetModules();
+});
+
 jest.mock('koa-socket', () => {
   class MockSocket {
     constructor() {
@@ -278,3 +282,10 @@ test('custom db implementation', async () => {
   await io.socket.receive('sync', 'gameID');
   expect(getId).toBe('gameID');
 });
+
+test('MONGO_URI', () => {
+  process.env.MONGO_URI = 'test';
+  const server = Server({ games: [game] });
+  expect(server.db.url).toBe('test');
+  delete process.env.MONGO_URI;
+});