elasticsearch-js/test/utils/MockConnection.js

// Licensed to Elasticsearch B.V under one or more agreements.
// Elasticsearch B.V licenses this file to you under the Apache 2.0 License.
// See the LICENSE file in the project root for more information

'use strict'

const assert = require('assert')
const { Connection } = require('../../index')
const {
  ConnectionError,
  RequestAbortedError,
  TimeoutError
} = require('../../lib/errors')
const intoStream = require('into-stream')

class MockConnection extends Connection {
  request (params, callback) {
    var aborted = false
    const stream = intoStream(JSON.stringify({ hello: 'world' }))
    stream.statusCode = setStatusCode(params.path)
    stream.headers = {
      'content-type': 'application/json;utf=8',
      date: new Date().toISOString(),
      connection: 'keep-alive',
      'content-length': '17'
    }
    process.nextTick(() => {
      if (!aborted) {
        callback(null, stream)
      } else {
        callback(new RequestAbortedError(), null)
      }
    })
    return {
      abort: () => { aborted = true }
    }
  }
}

class MockConnectionTimeout extends Connection {
  request (params, callback) {
    var aborted = false
    process.nextTick(() => {
      if (!aborted) {
        callback(new TimeoutError('Request timed out', params), null)
      } else {
        callback(new RequestAbortedError(), null)
      }
    })
    return {
      abort: () => { aborted = true }
    }
  }
}

class MockConnectionError extends Connection {
  request (params, callback) {
    var aborted = false
    process.nextTick(() => {
      if (!aborted) {
        callback(new ConnectionError('Kaboom'), null)
      } else {
        callback(new RequestAbortedError(), null)
      }
    })
    return {
      abort: () => { aborted = true }
    }
  }
}

class MockConnectionSniff extends Connection {
  request (params, callback) {
    var aborted = false
    const sniffResult = {
      nodes: {
        'node-1': {
          http: {
            publish_address: 'localhost:9200'
          },
          roles: ['master', 'data', 'ingest']
        },
        'node-2': {
          http: {
            publish_address: 'localhost:9201'
          },
          roles: ['master', 'data', 'ingest']
        }
      }
    }
    const stream = intoStream(JSON.stringify(sniffResult))
    stream.statusCode = setStatusCode(params.path)
    stream.headers = {
      'content-type': 'application/json;utf=8',
      date: new Date().toISOString(),
      connection: 'keep-alive',
      'content-length': '205'
    }
    process.nextTick(() => {
      if (!aborted) {
        if (params.headers.timeout) {
          callback(new TimeoutError('Request timed out', params), null)
        } else {
          callback(null, stream)
        }
      } else {
        callback(new RequestAbortedError(), null)
      }
    })
    return {
      abort: () => { aborted = true }
    }
  }
}

function buildMockConnection (opts) {
  assert(opts.onRequest, 'Missing required onRequest option')

  class MockConnection extends Connection {
    request (params, callback) {
      var { body, statusCode } = opts.onRequest(params)
      if (typeof body !== 'string') {
        body = JSON.stringify(body)
      }
      var aborted = false
      const stream = intoStream(body)
      stream.statusCode = statusCode || 200
      stream.headers = {
        'content-type': 'application/json;utf=8',
        date: new Date().toISOString(),
        connection: 'keep-alive',
        'content-length': Buffer.byteLength(body)
      }
      process.nextTick(() => {
        if (!aborted) {
          callback(null, stream)
        } else {
          callback(new RequestAbortedError(), null)
        }
      })
      return {
        abort: () => { aborted = true }
      }
    }
  }

  return MockConnection
}

function setStatusCode (path) {
  const statusCode = Number(path.slice(1))
  if (Number.isInteger(statusCode)) {
    return statusCode
  }
  return 200
}

module.exports = {
  MockConnection,
  MockConnectionTimeout,
  MockConnectionError,
  MockConnectionSniff,
  buildMockConnection
}