Cleanup route/metrics a bit

lib/db/client-metrics-store.js

@@ -22,7 +22,7 @@ class ClientMetricsStore {
         this.db(TABLE)
             .whereRaw('created_at < now() - interval \'1 hour\'')
             .del()
-            .then((res) => logger.info(`Delted ${res} metrics`));
+            .then((res) => logger.info(`Deleted ${res} metrics`));
     }
 
     // Insert new client metrics

lib/routes/metrics-schema.js

@@ -0,0 +1,24 @@
+const joi = require('joi');
+
+const clientMetricsSchema = joi.object().keys({
+    appName: joi.string().required(),
+    instanceId: joi.string().required(),
+    bucket: joi.object().required()
+    .keys({
+        start: joi.date().required(),
+        stop: joi.date().required(),
+        toggles: joi.object(),
+    }),
+});
+
+const clientRegisterSchema = joi.object().keys({
+    appName: joi.string().required(),
+    instanceId: joi.string().required(),
+    strategies: joi.array()
+        .required()
+        .items(joi.string(), joi.any().strip()),
+    started: joi.date().required(),
+    interval: joi.number().required(),
+});
+
+module.exports = { clientMetricsSchema, clientRegisterSchema }

lib/routes/metrics.js

@@ -4,6 +4,7 @@ const logger = require('../logger');
 const ClientMetrics = require('../client-metrics');
 const ClientMetricsService = require('../client-metrics/service');
 const joi = require('joi');
+const { clientMetricsSchema, clientRegisterSchema } = require('./metrics-schema');
 
 module.exports = function (app, config) {
     const {
@@ -11,6 +12,7 @@ module.exports = function (app, config) {
         clientStrategyStore,
         clientInstanceStore,
     } = config.stores;
+    
     const metrics = new ClientMetrics();
     const service = new ClientMetricsService(clientMetricsStore);
 
@@ -28,45 +30,17 @@ module.exports = function (app, config) {
         res.json(metrics.getTogglesMetrics());
     });
 
-    const clientMetricsSchema = joi.object().keys({
-        appName: joi.string().required(),
-        instanceId: joi.string().required(),
-        bucket: joi.object().required()
-        .keys({
-            start: joi.date().required(),
-            stop: joi.date().required(),
-            toggles: joi.object()
-                .required()
-                .unknown()
-                .min(1)
-                .max(1000),
-        }),
-    });
-
     app.post('/client/metrics', (req, res) => {
-        try {
-            const data = typeof req.body === 'string' ? JSON.parse(req.body) : req.body;
-            const result = joi.validate(data, clientMetricsSchema);
-            if (result.error) {
-                throw result.error;
+        const data = req.body;
+        joi.validate(data, clientMetricsSchema, (err, cleaned) => {
+            if (err) {
+                return res.status(400).json(err);
             }
-            service
-                .insert(result.value)
+            service.insert(cleaned)
                 .catch(e => logger.error('Error inserting metrics data', e));
-        } catch (e) {
-            logger.error('Error receiving metrics', e);
-        }
-        res.end();
-    });
-
-    const clientRegisterSchema = joi.object().keys({
-        appName: joi.string().required(),
-        instanceId: joi.string().required(),
-        strategies: joi.array()
-            .required()
-            .items(joi.string(), joi.any().strip()),
-        started: joi.date().required(),
-        interval: joi.number().required(),
+            
+            res.status(202).end();
+        });
     });
 
     app.post('/client/register', (req, res) => {
@@ -84,10 +58,10 @@ module.exports = function (app, config) {
                     instanceId: cleaned.instanceId,
                     clientIp,
                 }))
-                .then(() => console.log('new client registerd'))
+                .then(() => logger.info('New client registered!'))
                 .catch((error) => logger.error('Error registering client', error));
 
-            res.end();
+            res.status(202).end();
         });
     });
 
@@ -98,6 +72,6 @@ module.exports = function (app, config) {
     app.get('/client/instances', (req, res) => {
         clientInstanceStore.getAll()
             .then(data => res.json(data))
-            .catch(err => console.error(err));
+            .catch(err => logger.error(err));
     });
 };

test/unit/routes/feature.test.js

@@ -1,32 +1,23 @@
 'use strict';
 
-const clientMetricsStore = require('./mocks/fake-metrics-store');
-const featureToggleStore = require('./mocks/fake-feature-toggle-store');
-const strategyStore = require('./mocks/fake-strategies-store');
+const store = require('./mocks/store');
 
 const supertest = require('supertest');
 const assert = require('assert');
-const sinon = require('sinon');
+
 
 let request;
+let featureToggleStore;
 
 describe('Unit: The features api', () => {
     beforeEach(done => {
-        featureToggleStore.reset();
-
+        const stores = store.createStores(); 
         const app = require('../../../app')({
             baseUriPath: '',
-            stores: {
-                db: sinon.stub(),
-                eventStore: sinon.stub(),
-                featureToggleStore,
-                clientMetricsStore,
-                strategyStore,
-                clientStrategyStore: sinon.stub(),
-                clientInstanceStore: sinon.stub(),
-            },
+            stores: stores,
         });
 
+        featureToggleStore = stores.featureToggleStore;
         request = supertest(app);
         done();
     });

test/unit/routes/metrics.test.js

@@ -1,28 +1,17 @@
 'use strict';
 
-const clientMetricsStore = require('./mocks/fake-metrics-store');
-const clientStrategyStore = require('./mocks/fake-client-strategy-store');
-const clientInstanceStore = require('./mocks/fake-client-instance-store');
-
+const store = require('./mocks/store');
 const supertest = require('supertest');
 const assert = require('assert');
-const sinon = require('sinon');
 
 let request;
 
 describe('Unit: The metrics api', () => {
     beforeEach(done => {
+        const stores = store.createStores(); 
         const app = require('../../../app')({
             baseUriPath: '',
-            stores: {
-                db: sinon.stub(),
-                eventStore: sinon.stub(),
-                featureToggleStore: sinon.stub(),
-                clientMetricsStore,
-                strategyStore: sinon.stub(),
-                clientStrategyStore,
-                clientInstanceStore,
-            },
+            stores: stores,
         });
 
         request = supertest(app);
@@ -32,8 +21,14 @@ describe('Unit: The metrics api', () => {
     it('should register client', (done) => {
         request
             .post('/api/client/register')
-            .send({ appName: 'demo', instanceId: 'test', strategies: ['default'], started: Date.now(), interval: 10 })
-            .expect(200, done);
+            .send({ 
+                appName: 'demo',
+                instanceId: 'test',
+                strategies: ['default'],
+                started: Date.now(),
+                interval: 10 
+            })
+            .expect(202, done);
     });
 
     it('should require appName field', (done) => {
@@ -41,4 +36,33 @@ describe('Unit: The metrics api', () => {
             .post('/api/client/register')
             .expect(400, done)
     });
+
+    it('should require strategies field', (done) => {
+        request
+            .post('/api/client/register')
+            .send({ 
+                appName: 'demo',
+                instanceId: 'test',
+                //strategies: ['default'],
+                started: Date.now(),
+                interval: 10 
+            })
+            .expect(400, done)
+    });
+
+
+    it('should accept client metrics', (done) => {
+        request
+            .post('/api/client/metrics')
+            .send({ 
+                appName: 'demo',
+                instanceId: '1',
+                bucket: {
+                    start: Date.now(),
+                    stop: Date.now(),
+                    toggles: {}
+                }
+            })
+            .expect(202, done)
+    });
 });

test/unit/routes/mocks/fake-client-instance-store.js

@@ -1,5 +1,6 @@
 'use strict';
 
 module.exports = {
+    reset: () => {},
     insert: () => Promise.resolve(),
 };

test/unit/routes/mocks/fake-client-strategy-store.js

@@ -1,5 +1,6 @@
 'use strict';
 
 module.exports = {
+    reset: () => {},
     insert: () => Promise.resolve(),
 };

test/unit/routes/mocks/fake-metrics-store.js

@@ -1,5 +1,7 @@
 'use strict';
 
 module.exports = {
+    reset: () => {},
     getMetricsLastHour: () => Promise.resolve([]),
+    insert: () => Promise.resolve(),
 };

test/unit/routes/mocks/store.js

@@ -0,0 +1,27 @@
+const sinon = require('sinon');
+
+const clientMetricsStore = require('./fake-metrics-store');
+const clientStrategyStore = require('./fake-client-strategy-store');
+const clientInstanceStore = require('./fake-client-instance-store');
+const featureToggleStore = require('./fake-feature-toggle-store');
+const strategyStore = require('./fake-strategies-store');
+
+module.exports = {
+    createStores: () => {
+        clientMetricsStore.reset();
+        clientStrategyStore.reset();
+        clientInstanceStore.reset();
+        featureToggleStore.reset();
+        strategyStore.reset();
+
+        return {
+            db: sinon.stub(),
+            clientMetricsStore,
+            clientStrategyStore,
+            clientInstanceStore,
+            featureToggleStore,
+            strategyStore,
+        }
+
+    }
+};

test/unit/routes/strategies.test.js

@@ -1,32 +1,23 @@
 'use strict';
 
-const clientMetricsStore = require('./mocks/fake-metrics-store');
-const featureToggleStore = require('./mocks/fake-feature-toggle-store');
-const strategyStore = require('./mocks/fake-strategies-store');
+const store = require('./mocks/store');
+
 
 const supertest = require('supertest');
 const assert = require('assert');
 const sinon = require('sinon');
 
 let request;
+let strategyStore;
 
 describe('Unit: The strategies api', () => {
     beforeEach(done => {
-        strategyStore.reset();
-
+       const stores = store.createStores(); 
         const app = require('../../../app')({
             baseUriPath: '',
-            stores: {
-                db: sinon.stub(),
-                eventStore: sinon.stub(),
-                featureToggleStore,
-                clientMetricsStore,
-                strategyStore,
-                clientStrategyStore: sinon.stub(),
-                clientInstanceStore: sinon.stub(),
-            },
+            stores: stores,
         });
-
+        strategyStore = stores.strategyStore;
         request = supertest(app);
         done();
     });