File Explorer

1'use strict'2 3const { promisify } = require('node:util')4const Client = require('../dispatcher/client')5const { buildMockDispatch } = require('./mock-utils')6const {7  kDispatches,8  kMockAgent,9  kClose,10  kOriginalClose,11  kOrigin,12  kOriginalDispatch,13  kConnected,14  kIgnoreTrailingSlash15} = require('./mock-symbols')16const { MockInterceptor } = require('./mock-interceptor')17const Symbols = require('../core/symbols')18const { InvalidArgumentError } = require('../core/errors')19 20/**21 * MockClient provides an API that extends the Client to influence the mockDispatches.22 */23class MockClient extends Client {24  constructor (origin, opts) {25    if (!opts || !opts.agent || typeof opts.agent.dispatch !== 'function') {26      throw new InvalidArgumentError('Argument opts.agent must implement Agent')27    }28 29    super(origin, opts)30 31    this[kMockAgent] = opts.agent32    this[kOrigin] = origin33    this[kIgnoreTrailingSlash] = opts.ignoreTrailingSlash ?? false34    this[kDispatches] = []35    this[kConnected] = 136    this[kOriginalDispatch] = this.dispatch37    this[kOriginalClose] = this.close.bind(this)38 39    this.dispatch = buildMockDispatch.call(this)40    this.close = this[kClose]41  }42 43  get [Symbols.kConnected] () {44    return this[kConnected]45  }46 47  /**48   * Sets up the base interceptor for mocking replies from undici.49   */50  intercept (opts) {51    return new MockInterceptor(52      opts && { ignoreTrailingSlash: this[kIgnoreTrailingSlash], ...opts },53      this[kDispatches]54    )55  }56 57  cleanMocks () {58    this[kDispatches] = []59  }60 61  async [kClose] () {62    await promisify(this[kOriginalClose])()63    this[kConnected] = 064    this[kMockAgent][Symbols.kClients].delete(this[kOrigin])65  }66}67 68module.exports = MockClient69