Function SnapshotPlugin

packages/vitest/src/integrations/snapshot/chai.ts:78–179

(chai, utils)

}

export const SnapshotPlugin: ChaiPlugin = (chai, utils) => {
  for (const key of ['matchSnapshot', 'toMatchSnapshot']) {
    utils.addMethod(
      chai.Assertion.prototype,
      key,
      wrapAssertion(utils, key, function (
        this,
        propertiesOrHint?: object | string,
        hint?: string,
      ) {
        const result = toMatchSnapshotImpl({
          assertion: this,
          received: utils.flag(this, 'object'),
          ...normalizeArguments(propertiesOrHint, hint),
        })
        return assertMatchResult(result, chai.util.flag(this, 'message'))
      }),
    )
  }

  utils.addMethod(
    chai.Assertion.prototype,
    'toMatchFileSnapshot',
    function (this: Chai.Assertion, filepath: string, hint?: string) {
      // set name manually since it's not wrapped by wrapAssertion
      utils.flag(this, '_name', 'toMatchFileSnapshot')
      // validate early synchronously just not to break some existing tests
      validateAssertion(this)
      const resultPromise = toMatchFileSnapshotImpl({
        assertion: this,
        received: utils.flag(this, 'object'),
        filepath,
        hint,
      })
      const assertPromise = resultPromise.then(result =>
        assertMatchResult(result, chai.util.flag(this, 'message')),
      )
      return recordAsyncExpect(
        getTest(this),
        assertPromise,
        createAssertionMessage(utils, this, true),
        new Error('resolves'),
        utils.flag(this, 'soft'),
      )
    },
  )

  utils.addMethod(
    chai.Assertion.prototype,
    'toMatchInlineSnapshot',
    wrapAssertion(utils, 'toMatchInlineSnapshot', function __INLINE_SNAPSHOT_OFFSET_3__(
      this,
      propertiesOrInlineSnapshot?: object | string,
      inlineSnapshotOrHint?: string,
      hint?: string,
    ) {
      const result = toMatchSnapshotImpl({
        assertion: this,