如何清除/重置Vitest中的模拟

Question

我有一个简单的可组合useRoles，我需要测试它

import { computed } from "vue";
import { useStore } from "./store";

export default function useRoles() {
  const store = useStore();

  const isLearner = computed(() => store.state.profile.currentRole === "learner");

  return {
    isLearner
  };
}

我的测试方法如下

import { afterEach, expect } from "vitest";
import useRoles from "./useRoles";

describe("useRoles", () => {
  afterEach(() => {
    vi.clearAllMocks();
    vi.resetAllMocks();
  });

  it("should verify values when is:Learner", () => { // works
    vi.mock("./store", () => ({
      useStore: () => ({
        state: {
          profile: {
            currentRole: "learner"
          },
        },
      }),
    }));

    const { isLearner } = useRoles();

    expect(isLearner.value).toBeTruthy();
  });

  it("should verify values when is:!Learner", () => { //fails
    vi.mock("./store", () => ({
      useStore: () => ({
        state: {
          profile: {
            currentRole: "admin"
          },
        },
      }),
    }));

    const { isLearner } = useRoles(); // Values are from prev mock

    expect(isLearner.value).toBeFalsy();
  });
});

useStore只是我想要模拟的一个简单的函数。

export function useStore() {
  return {/**/};
}

第一个测试成功运行，它拥有我实现的所有模拟值，但问题是它没有为每个测试重置(根本没有重置)。第二个测试具有以前模拟的旧值。

我用过

vi.clearAllMocks();
vi.resetAllMocks();

但出于某种原因，没有明确或重新设置。

如何清除每个测试的vi.mock值？

vitest = v0.23.0

Answer 1

我遇到了同样的问题，并找到了解决办法。在您的例子中，应该是这样的：

import { afterEach, expect } from "vitest";
import useRoles from "./useRoles";

vi.mock("./store");

describe("useRoles", () => {
  it("should verify values when is:Learner", async () => {
    const store = await import("./store");
    store.useStore = await vi.fn().mockReturnValueOnce({
      useStore: () => ({
        state: {
          profile: {
            currentRole: "learner"
          },
        },
      }),
    });

    const { isLearner } = useRoles();

    expect(isLearner.value).toBeTruthy();
  });

  it("should verify values when is:!Learner", async () => {
    const store = await import("./store");
    store.useStore = await vi.fn().mockReturnValueOnce({
      useStore: () => ({
        state: {
          profile: {
            currentRole: "admin"
          },
        },
      }),
    });

    const { isLearner } = useRoles(); // Values are from prev mock

    expect(isLearner.value).toBeFalsy();
  });
});