test: fix unexpected warning check when asserting same message for bo…

…th conditions
vuejs · Feb 9, 2024 · 5d48844 · 5d48844
1 parent 77a804b
commit 5d48844
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/scripts/setup-vitest.ts b/scripts/setup-vitest.ts
@@ -4,9 +4,9 @@ vi.stubGlobal('MathMLElement', class MathMLElement {})
 
 expect.extend({
  toHaveBeenWarned(received: string) {
- asserted.add(received)
  const passed = warn.mock.calls.some(args => args[0].includes(received))
  if (passed) {
+ asserted.add(received)
  return {
  pass: true,
  message: () => `expected "${received}" not to have been warned.`,
@@ -25,10 +25,10 @@ expect.extend({
  },
 
  toHaveBeenWarnedLast(received: string) {
- asserted.add(received)
  const passed =
  warn.mock.calls[warn.mock.calls.length - 1][0].includes(received)
  if (passed) {
+ asserted.add(received)
  return {
  pass: true,
  message: () => `expected "${received}" not to have been warned last.`,
@@ -44,7 +44,6 @@ expect.extend({
  },
 
  toHaveBeenWarnedTimes(received: string, n: number) {
- asserted.add(received)
  let found = 0
  warn.mock.calls.forEach(args => {
  if (args[0].includes(received)) {
@@ -53,6 +52,7 @@ expect.extend({
  })
 
  if (found === n) {
+ asserted.add(received)
  return {
  pass: true,
  message: () => `expected "${received}" to have been warned ${n} times.`,