kissmetrics-export/test/event_stream.js

Kissmetrics exporter in node.js.

const { expect } = require('chai');
const from = require('from2');

const config = {
  Bucket: process.env.s3_bucket,
  accessKeyId: process.env.s3_access_key,
  secretAccessKey: process.env.s3_secret_access_key,
};

describe('EventStream', function () {
  const S3 = require('../lib/s3');
  const s3 = new S3(config);
  const EventStream = require('../lib/event_stream');
  const eventStream = new EventStream(s3);
  it('should read the files coming from an object stream', function(done) {
    this.timeout(60000);
    // mock an objectStream
    const files = ['revisions/2.json','revisions/3.json'];
    const objectStream = from.obj(function(size, next) {
      if (files.length <= 0) {
        return this.push(null);
      }
      const chunk = files.slice(0,1)[0];
        files = files.slice(1);
      next(null, chunk);
    });
    const stream = objectStream.pipe(eventStream);
    eventStream.on('data', function(data) {
      expect(data).to.be.not.null;
    });
    stream.on('error', function(err) {
      console.log(err.stack);
      expect(0).to.equal(1);
    });
    stream.on('end', function() {
      console.log('done');
      done();
    });
  });
});