copilot/portal-data/test/index.js

'use strict';

const pkg = require('../package.json');
const { expect } = require('code');
const { before, describe, it, script } = require('lab');
const { Server } = require('zerorpc');
const PortalData = require('../');

exports.lab = script();

const server = new Server({
  up: function(options, manifest, fn) {
    fn(null, {
      projectName: options.project_name
    });
  }
});

const internals = {
  options: { test: true, name: 'test', dockerHost: 'tcp://0.0.0.0:4242'}
};


before(() => {
  server.bind(internals.options.dockerHost);
});

after(() => {
  server.close();
});

describe('createDeployment()', () => {
  it('creates a deployment record in the deployment table', done => {
    const data = new PortalData(internals.options);
    const deployment = {
      name: 'User Services',
      datacenter: 'us-sw-1'
    };

    data.connect().then(() => {
      data.createDeployment({
        deployment
      }).then(deployment => {
        expect(deployment.id).to.exist();
        done();
      });
    });
  });
});

describe('getDeployment()', () => {
  it('will retrieve an existing deployment', done => {
    const data = new PortalData(internals.options);
    data.connect().then(() => {
      const deployment = {
        name: 'User Services',
        datacenter: 'us-sw-1'
      };

      data.createDeployment(deployment).then(deployment => {
        expect(deployment.id).to.exist();
        data.getDeployment(deployment.id).then(retrievedDeployment => {
          expect(deployment).to.equal(retrievedDeployment);
          done();
        });
      });
    });
  });
});

describe('updateService()', () => {
  it('will update the services for an existing deployment', done => {
    const data = new PortalData(internals.options);
    data.connect().then(() => {
      const deployment = {
        name: 'User Services',
        datacenter: 'us-sw-1'
      };
      const service = {
        name: 'consul',
        containers: [
          {
            server_id: '423e7432-b760-11e2-bf6c-002590c3f1a0',
            alias: 'nodejsexample_consul_1',
            image_id: '91b757b5-bd29-2126-5ff9-ae9235011ff5',
            owner_id: '30f62ec2-24a2-6f8e-8fad-d46b04c8a0b9',
            id: '81205d4a-92f4-c4d9-da8a-aafd689eeabb'
          }
        ],
        count: 1
      };

      data.createDeployment(deployment).then(deployment => {
        expect(deployment.id).to.exist();
        data.updateService(deployment.id, service).then(updatedService => {
          expect(updatedService).to.equal(service);
          done();
        });
      });
    });
  });
});

describe('deploymentChanges()', () => {
  it('will execute the handler when a deployment service changes', done => {
    const data = new PortalData(internals.options);
    data.connect().then(() => {
      const deployment = {
        name: 'User Services',
        datacenter: 'us-sw-1'
      };
      const service1 = {
        name: 'consul',
        containers: [
          {
            server_id: '423e7432-b760-11e2-bf6c-002590c3f1a0',
            alias: 'nodejsexample_consul_1',
            image_id: '91b757b5-bd29-2126-5ff9-ae9235011ff5',
            owner_id: '30f62ec2-24a2-6f8e-8fad-d46b04c8a0b9',
            id: '81205d4a-92f4-c4d9-da8a-aafd689eeabb'
          }
        ],
        count: 1
      };

      const service2 = {
        name: 'consul',
        containers: [
          {
            server_id: '423e7432-b760-11e2-bf6c-002590c3f1a0',
            alias: 'nodejsexample_consul_1',
            image_id: '91b757b5-bd29-2126-5ff9-ae9235011ff5',
            owner_id: '30f62ec2-24a2-6f8e-8fad-d46b04c8a0b9',
            id: '81205d4a-92f4-c4d9-da8a-aafd689eeabb'
          },
          {
            server_id: '423e7432-b760-11e2-bf6c-002590c3f1a0',
            alias: 'nodejsexample_consul_2',
            image_id: '91b757b5-bd29-2126-5ff9-ae9235011ff5',
            owner_id: '30f62ec2-24a2-6f8e-8fad-d46b04c8a0b9',
            id: '81205d4a-92f4-c4d9-da8a-aafd689eeabb'
          },
          {
            server_id: '423e7432-b760-11e2-bf6c-002590c3f1a0',
            alias: 'nodejsexample_consul_3',
            image_id: '91b757b5-bd29-2126-5ff9-ae9235011ff5',
            owner_id: '30f62ec2-24a2-6f8e-8fad-d46b04c8a0b9',
            id: '81205d4a-92f4-c4d9-da8a-aafd689eeabb'
          }
        ],
        count: 3
      };

      data.createDeployment(deployment).then(deployment => {
        expect(deployment.id).to.exist();
        data.updateService(deployment.id, service1).then(updatedService1 => {
          expect(updatedService1).to.equal(service1);

          let executed = false;
          data
            .deploymentChanges((err, changes) => {
              if (executed) {
                return;
              }

              expect(changes.before).to.exist();
              expect(changes.after).to.exist();
              done();
              executed = true;
            })
            .then(() => {
              data
                .updateService(deployment.id, service2)
                .then(updatedService2 => {
                  expect(updatedService2).to.equal(service2);
                });
            });
        });
      });
    });
  });
});

describe('insertMetrics()', () => {
  it("will add new metrics to a service and won't overwrite existing ones", done => {
    const data = new PortalData(internals.options);
    data.connect().then(() => {
      const containerId = '81205d4a-92f4-c4d9-da8a-aafd689eeabb';
      const metrics1 = [
        {
          timestamp: 1494360995851,
          cpu: 1.2,
          memory: 23344523,
          network: 5024
        }
      ];

      const metrics2 = [
        {
          timestamp: 1495360995851,
          cpu: 1.3,
          memory: 23344523,
          network: 4024
        }
      ];

      data.insertMetrics(containerId, metrics1).then(result1 => {
        expect(result1.id).to.equal(containerId);
        expect(result1.metrics).to.equal(metrics1);
        data.insertMetrics(containerId, metrics2).then(result2 => {
          expect(result2.id).to.equal(containerId);
          data.getMetrics(containerId).then(results => {
            expect(results.metrics.length).to.equal(2);
            done();
          });
        });
      });
    });
  });
});