breaking: bring your own sinon instead of bundling, to resolve leaks when versions...

[squeep-test-helper] / lib / stub-database.js

/* eslint-disable class-methods-use-this */
'use strict';

class StubDatabase {
  constructor(sinon) {
    this._sinon = sinon;
    this._implementation.forEach((fn) => {
      if (!(fn in this)) {
        this[fn] = async () => undefined; // eslint-disable-line security/detect-object-injection
      }
    });
  }

  get _implementation() {
    return [
      ...this._spyFns,
      ...this._stubFns,
    ];
  }

  get _spyFns() {
    return [
      'context',
      'transaction',
    ];
  }

  get _stubFns() {
    return [
      'initialize',
      'healthCheck',
    ];
  }

  _reset() {
    this._spyFns.forEach((fn) => {
      this._sinon.spy(this, fn);
    });
    this._stubFns.forEach((fn) => {
      this._sinon.stub(this, fn);
    });
  }

  async context(fn) {
    await fn({});
  }

  async transaction(dbCtx, fn) {
    await fn(dbCtx);
  }

}

module.exports = StubDatabase;