Add tests for sse and argument parsing

Author: relekang

src/index.js

@@ -1,14 +1,11 @@
 const micro = require('micro')
 const SSE = require('sse')
-const args = require('args');
-
-const flags = args
-  .option(['p', 'port'], 'Port to listen on', process.env.PORT || 3000, Number)
-  .option(['H', 'host'], 'Host to listen on', '0.0.0.0')
-  .option(['a', 'adapter'], 'Database adapter used', process.env.DB_ADAPTER || 'flat-file-db')
-  .parse(process.argv, { name: 'micro-analytics' })
 
+const parseArgs = require('./parseArgs')
 const db = require('./db');
+
+const flags = parseArgs(process.argv)
+
 db.initDbAdapter(flags.adapter)
 
 const handler = require('./handler')

src/parseArgs.js

@@ -0,0 +1,10 @@
+
+const args = require('args');
+
+module.exports = function parseArgs(argv) {
+  return args
+    .option(['p', 'port'], 'Port to listen on', process.env.PORT || 3000, Number)
+    .option(['H', 'host'], 'Host to listen on', process.env.HOST ||'0.0.0.0')
+    .option(['a', 'adapter'], 'Database adapter used', process.env.DB_ADAPTER || 'flat-file-db')
+    .parse(argv, { name: 'micro-analytics' })
+}

tests/__snapshots__/parseArgs.test.js.snap

@@ -0,0 +1,12 @@
+exports[`parseArgs should have correct defaults 1`] = `
+Object {
+  "H": "0.0.0.0",
+  "a": "flat-file-db",
+  "adapter": "flat-file-db",
+  "host": "0.0.0.0",
+  "p": 3000,
+  "port": 3000,
+}
+`;
+
+exports[`parseArgs should use DB_ADAPTER environment variable as default if set 1`] = `"flat-file-db"`;

tests/parseArgs.test.js

@@ -0,0 +1,27 @@
+jest.mock('pkginfo', () => () => ({version: '1.0.0'}))
+const parseArgs = require('../src/parseArgs')
+
+
+describe('parseArgs', () => {
+  it("should have correct defaults", () => {
+    expect(parseArgs(['node', 'micro-analytics'])).toMatchSnapshot();
+  })
+
+  it("should use DB_ADAPTER environment variable as default if set", () => {
+    process.env.DB_ADAPTER = 'redis'
+    expect(parseArgs(['node', 'micro-analytics']).adapter).toEqual('redis');
+    delete process.env.DB_ADAPTER
+  })
+
+  it("should use PORT environment variable as default if set", () => {
+    process.env.PORT = '3000'
+    expect(parseArgs(['node', 'micro-analytics']).port).toEqual(3000);
+    delete process.env.PORT
+  })
+
+  it("should use HOST environment variable as default if set", () => {
+    process.env.HOST = 'localhost'
+    expect(parseArgs(['node', 'micro-analytics']).host).toEqual('localhost');
+    delete process.env.HOST
+  })
+})

tests/sse.test.js

@@ -0,0 +1,23 @@
+const { listen, mockDb } = require('./utils')
+jest.mock('flat-file-db', () => mockDb)
+const db = require('../src/db')
+const sseHandler = require('../src/sse')
+let url
+
+db.initDbAdapter('flat-file-db');
+
+beforeEach(() => {
+  mockDb._reset()
+})
+
+describe('sse', () => {
+  it("should publish events", () => {
+    const sendMock = jest.fn()
+    const eventListenerMock = jest.fn()
+    sseHandler({send: sendMock, on: eventListenerMock})
+
+    db.put('superview', { views: [{ time: 1490432584312 }] })
+
+    expect(sendMock).toHaveBeenCalled()
+  })
+})