reserve/mock/Response.js

'use strict'

const { Duplex } = require('stream')
const headersFactory = require('./headers')

module.exports = class Response extends Duplex {
  _read () {
    this.push(this.toString())
    this.push(null)
  }

  _write (chunk, encoding, onwrite) {
    this._headersSent = true
    this._buffer.push(chunk.toString())
    if (this._asynchronous) {
      setTimeout(onwrite, 0)
    } else {
      onwrite()
    }
  }

  setHeader (name, value) {
    this._headers[name] = value
  }

  writeHead (statusCode, headers = {}) {
    this._statusCode = statusCode
    Object.keys(headers).forEach(header => {
      this._headers[header] = headers[header]
    })
  }

  flushHeaders () {
    this._headersSent = true
  }

  end () {
    this._headersSent = true
    return super.end.apply(this, arguments)
  }

  constructor (options) {
    super(options)
    this._buffer = []
    this._headers = headersFactory()
    this._headersSent = false
    let resolver
    this._waitForFinish = new Promise(resolve => {
      resolver = resolve
    })
    this.on('finish', () => resolver(this))
  }

  get headers () {
    return this._headers
  }

  get statusCode () {
    return this._statusCode
  }

  get headersSent () {
    return this._headersSent
  }

  toString () {
    return this._buffer.join('')
  }

  waitForFinish () {
    return this._waitForFinish
  }

  isInitial () {
    return this._buffer.length === 0 &&
      !this._headersSent &&
      Object.keys(this._headers).length === 0
  }

  setAsynchronous () {
    this._asynchronous = true
  }
}