grpc-promise/test/tools/grpc-mock/ServerStreamMock.js

GRPC promisify module for all Request/Response types: standard and stream

const stream = require('stream');
const util = require('util');

const ServerStreamMock = function (options) {
  if (options == null) {
    options = {};
  }
  stream.Readable.call(this, {objectMode: true});
  this.response = options.response;
};
util.inherits(ServerStreamMock, stream.Readable);

ServerStreamMock.prototype._read = function readBytes () {
  const self = this;
  if (!this.response) {
    this.emit('error', 'some_error');
    return;
  }
  while (this.response.length) {
    const chunk = this.response.shift();
    this.push(chunk);
    if (!this.response.length) {
      this.emit('end');
      return;
    }
  }
  setTimeout(readBytes.bind(self), 1);
};

module.exports = ServerStreamMock;