Fix issues related to console spies in createSlice.test.ts

Author: aryaemami59

packages/toolkit/src/tests/createSlice.test.ts

@@ -1,3 +1,4 @@
+import { noop } from '@internal/listenerMiddleware/utils'
 import type { PayloadAction, WithSlice } from '@reduxjs/toolkit'
 import {
   asyncThunkCreator,
@@ -7,19 +8,18 @@ import {
   createAction,
   createSlice,
 } from '@reduxjs/toolkit'
-import {
-  createConsole,
-  getLog,
-  mockConsole,
-} from 'console-testing-library/pure'
 
 type CreateSlice = typeof createSlice
 
 describe('createSlice', () => {
-  let restore: () => void
+  const consoleErrorSpy = vi.spyOn(console, 'error').mockImplementation(noop)
 
   beforeEach(() => {
-    restore = mockConsole(createConsole())
+    vi.clearAllMocks()
+  })
+
+  afterAll(() => {
+    vi.restoreAllMocks()
   })
 
   describe('when slice is undefined', () => {
@@ -57,8 +57,10 @@ describe('createSlice', () => {
   describe('when initial state is undefined', () => {
     beforeEach(() => {
       vi.stubEnv('NODE_ENV', 'development')
+    })
 
-      return vi.unstubAllEnvs
+    afterEach(() => {
+      vi.unstubAllEnvs()
     })
 
     it('should throw an error', () => {
@@ -68,7 +70,9 @@ describe('createSlice', () => {
         initialState: undefined,
       })
 
-      expect(getLog().log).toBe(
+      expect(consoleErrorSpy).toHaveBeenCalledOnce()
+
+      expect(consoleErrorSpy).toHaveBeenLastCalledWith(
         'You must provide an `initialState` value that is not `undefined`. You may have misspelled `initialState`',
       )
     })
@@ -397,7 +401,6 @@ describe('createSlice', () => {
 
     beforeEach(() => {
       vi.resetModules()
-      restore = mockConsole(createConsole())
     })
 
     afterEach(() => {