micro.js

/*
 * JavaScript tracker for Snowplow: tests/functional/helpers.js
 *
 * Significant portions copyright 2010 Anthon Pang. Remainder copyright
 * 2012-2016 Snowplow Analytics Ltd. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are
 * met:
 *
 * * Redistributions of source code must retain the above copyright
 *   notice, this list of conditions and the following disclaimer.
 *
 * * Redistributions in binary form must reproduce the above copyright
 *   notice, this list of conditions and the following disclaimer in the
 *   documentation and/or other materials provided with the distribution.
 *
 * * Neither the name of Anthon Pang nor Snowplow Analytics Ltd nor the
 *   names of their contributors may be used to endorse or promote products
 *   derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

const http = require('http')
const Docker = require('dockerode')
const { Writable } = require('stream')

const docker = new Docker()

export const start = () => {
  return docker
    .createContainer({
      Image: 'snowplow/snowplow-micro:latest',
      AttachStdin: false,
      AttachStdout: true,
      AttachStderr: true,
      Tty: true,
      Cmd: [
        '--collector-config',
        '/config/micro.conf',
        '--iglu',
        '/config/iglu.json',
      ],
      OpenStdin: false,
      StdinOnce: false,
      Hostconfig: {
        Binds: [`${process.cwd()}/tests/micro-config:/config`],
        PortBindings: {
          '9090/tcp': [
            {
              HostIp: '',
              HostPort: '',
            },
          ],
        },
      },
      ExposedPorts: {
        '9090/tcp': {},
      },
    })
    .then(c => {
      return c.start().then(() => {
        const outs = new Writable({
          write(chunk, encoding, callback) {
            let found = chunk.toString().includes('REST interface bound')
            if (found) this.end()
            callback()
          },
        })

        c.attach(
          { stream: true, stdout: true, stderr: true },
          (err, stream) => {
            stream.pipe(process.stdout)
            stream.pipe(outs)
          }
        )

        return new Promise(r => {
          outs.on('finish', () => r(c))
        })
      })
    })
}

export const stop = container => container.stop().then(() => container.remove())

const createMicroCall = url => () =>
  new Promise((resolve, reject) => {
    const req = http.request(url, res => {
      let body = ''
      res.on('data', chunk => {
        body += chunk
      })
      res.on('end', () => {
        resolve(body)
      })
    })
    req.on('error', reject)
    req.end()
  })

export const fetchResults = containerUrl =>
  createMicroCall(`http://${containerUrl}/micro/good`)().then(good =>
    JSON.parse(good)
  )