@autofleet/node-common 1.1.9 → 1.1.10

package/delorean-client/index.js

@@ -0,0 +1,52 @@
+const tk = require('timekeeper');
+const WebSocket = require('ws');
+const Logger = require('../logger');
+
+const logger = Logger();
+
+const createWs = (host, resolve) => {
+  const ws = new WebSocket(`ws://${host}`);
+
+  ws.on('open', () => {
+    ws.send(JSON.stringify({ type: 'ping' }));
+    if (resolve) {
+      resolve(true);
+    }
+  });
+
+  ws.on('error', (error) => {
+    logger.error('ERROR', error);
+  });
+
+  ws.on('message', (data) => {
+    const jsonMsg = JSON.parse(data);
+
+    if (jsonMsg.type === 'freeze') {
+      logger.info('I know you just sent me back to the future, but I\'m back.I\'m back from the future.', { goingBackTo: jsonMsg.time });
+      tk.freeze(new Date(jsonMsg.time));
+    } else if (jsonMsg.type === 'travel') {
+      tk.travel(jsonMsg.time);
+    }
+  });
+
+  ws.on('close', () => {
+    setTimeout(() => {
+      if (process.env.TIME_TRAVEL_MS_SERVICE_HOST === host) {
+        createWs(host, null);
+      }
+    }, 500);
+  });
+};
+
+module.exports = () => new Promise((resolve) => {
+  const { TIME_TRAVEL_MS_SERVICE_HOST, SHOULD_TIME_TRAVEL } = process.env;
+  if (TIME_TRAVEL_MS_SERVICE_HOST && SHOULD_TIME_TRAVEL) {
+    logger.info('If my calculations are correct, when this baby hits 88 miles per hour... you\'re gonna see some serious shit.', {
+      TIME_TRAVEL_MS_SERVICE_HOST,
+    });
+
+    createWs(TIME_TRAVEL_MS_SERVICE_HOST, resolve);
+  } else {
+    resolve(false);
+  }
+});

package/delorean-client/index.test.js

@@ -0,0 +1,105 @@
+const tk = require('timekeeper');
+const deloreanClient = require('./index');
+const portfinder = require('portfinder');
+const WebSocket = require('ws');
+
+const { env } = process;
+
+const createUrl = async () => {
+  const port = await portfinder.getPortPromise();
+  return {
+    url: `127.0.0.1:${port}`,
+    port,
+  };
+};
+
+const mockServer = (port, time, type) => {
+  const wss = new WebSocket.Server({ port });
+  let gotPing = false;
+
+  wss.on('connection', (ws) => {
+    ws.on('message', (message) => {
+      const json = JSON.parse(message);
+      if (json.type === 'ping') {
+        gotPing = true;
+      }
+    });
+
+    ws.send(JSON.stringify({
+      type,
+      time,
+    }));
+  });
+
+  return {
+    close: () => wss.close(),
+    gotPing: () => gotPing,
+  };
+};
+
+describe('Logger', () => {
+  beforeEach(() => {
+    tk.reset();
+    env.SHOULD_TIME_TRAVEL = true;
+    env.TIME_TRAVEL_MS_SERVICE_HOST = '';
+  });
+
+  afterAll(() => {
+    env.SHOULD_TIME_TRAVEL = false;
+    env.TIME_TRAVEL_MS_SERVICE_HOST = '0';
+  });
+
+  test('Conects to server, sends ping an freeze time', async (done) => {
+    const { url, port } = await createUrl();
+    env.TIME_TRAVEL_MS_SERVICE_HOST = url;
+    const mockedDate = new Date('2013-10-08 12:32:50 +0530').getTime();
+    const { close, gotPing } = mockServer(port, mockedDate, 'freeze');
+    await deloreanClient();
+
+    setTimeout(() => {
+      expect(Date.now()).toEqual(mockedDate);
+      expect(gotPing()).toBeTruthy();
+      close();
+      done();
+    }, 100);
+  });
+
+  it('Conects to server, sends ping an freeze time', async (done) => {
+    const { url, port } = await createUrl();
+    env.TIME_TRAVEL_MS_SERVICE_HOST = url;
+    const mockedDate = new Date('2013-10-08 12:32:50 +0530').getTime();
+    const { close, gotPing } = mockServer(port, mockedDate, 'travel');
+    await deloreanClient();
+
+    setTimeout(() => {
+      expect(Date.now()).toBeGreaterThan(mockedDate);
+      expect(Date.now()).toBeLessThan(new Date('2013-10-08 13:32:50 +0530').getTime());
+      expect(gotPing()).toBeTruthy();
+      close();
+      done();
+    }, 100);
+  });
+
+  it('Wont start without TIME_TRAVEL_MS_SERVICE_HOST', async () => {
+    const { port } = await createUrl();
+    delete env.TIME_TRAVEL_MS_SERVICE_HOST;
+    const mockedDate = new Date('2013-10-08 12:32:50 +0530').getTime();
+    const { close } = mockServer(port, mockedDate, 'travel');
+    const result = await deloreanClient();
+
+    expect(result).toBeFalsy();
+    close();
+  });
+
+  it('Wont start without SHOULD_TIME_TRAVEL', async () => {
+    const { url, port } = await createUrl();
+    env.TIME_TRAVEL_MS_SERVICE_HOST = url;
+    delete env.SHOULD_TIME_TRAVEL;
+    const mockedDate = new Date('2013-10-08 12:32:50 +0530').getTime();
+    const { close } = mockServer(port, mockedDate, 'travel');
+    const result = await deloreanClient();
+
+    expect(result).toBeFalsy();
+    close();
+  });
+});

package/index.js

@@ -1,9 +1,13 @@
 const Network = require('./network');
 const Logger = require('./logger');
 const Settings = require('./settings');
+const client = require('./delorean-client');
 
 module.exports = {
   Network,
   Logger,
   Settings,
+  Delorean: {
+    client,
+  },
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@autofleet/node-common",
-  "version": "1.1.9",
+  "version": "1.1.10",
   "description": "",
   "scripts": {
     "coverage": "jest --coverage --forceExit --runInBand",
@@ -30,9 +30,9 @@
     "node-cache": "^4.2.0",
     "portfinder": "^1.0.13",
     "qs": "^6.5.2",
+    "timekeeper": "^2.1.2",
     "winston": "^3.0.0-rc4",
-    "ws": "^5.2.0",
-    "timekeeper": "^2.1.2"
+    "ws": "^5.2.1"
   },
   "devDependencies": {
     "eslint": "^4.19.1",