stub-helpers.js

const { install: installClock } = require('lolex')
const { stub } = require('sinon')

const dist = require('./dist')

const { default: Timers } = require(dist('lib/Timers'))

function stubLog () {
  const log = stub({
    _lastIndex () {},
    get lastIndex () { return this._lastIndex() },
    _lastTerm () {},
    get lastTerm () { return this._lastTerm() },
    appendValue () {},
    commit () {},
    getEntry () {},
    getTerm () {},
    mergeEntries () {},
    getEntriesSince () {},
    checkOutdated () {}
  })
  log._lastIndex.returns(0)
  log.getEntry.returns(undefined)
  log.mergeEntries.returns(Promise.resolve())

  log.appendValue.throws(new Error('appendValue() stub must be customized'))
  log.getTerm.throws(new Error('getTerm() stub must be customized'))
  log.getEntriesSince.throws(new Error('getEntriesSince() stub must be customized'))

  log.checkOutdated.returns(false)

  return log
}

function stubMessages () {
  const messages = stub({ canTake () {}, take () {}, await () {} })
  messages.canTake.returns(false)
  messages.take.returns(null)
  messages.await.returns(new Promise(() => {}))
  return messages
}

let peerCount = 0
function stubPeer () {
  return stub({ messages: stubMessages(), send () {}, id: ++peerCount })
}

function stubState () {
  const state = stub({
    _currentTerm () {},
    get currentTerm () { return this._currentTerm() },
    _votedFor () {},
    get votedFor () { return this._votedFor() },
    nextTerm () {},
    replace () {},
    setTerm () {},
    setTermAndVote () {}
  })
  state._currentTerm.returns(0)
  state._votedFor.returns(null)
  state.nextTerm.returns(Promise.resolve())
  state.setTerm.returns(Promise.resolve())
  state.setTermAndVote.returns(Promise.resolve())
  return state
}

function stubTimers () {
  const timers = new Timers()
  const clock = installClock(timers, 0, ['clearInterval', 'setInterval', 'clearTimeout', 'setTimeout'])
  return { clock, timers }
}

Object.assign(exports, {
  stubLog,
  stubMessages,
  stubPeer,
  stubState,
  stubTimers
})