1+ import { afterEach , describe , expect , test , vi } from 'vitest' ;
12import browser from 'webextension-polyfill' ;
23
34import { BrowserStorageCache , MemoryStorageCache } from '../storage' ;
@@ -21,7 +22,8 @@ describe('StorageCache', () => {
2122
2223 describe ( 'set' , ( ) => {
2324 test ( 'setting the storage cache' , async ( ) => {
24- const setMock = jest . mocked ( browser . storage . local . set ) . mockResolvedValueOnce ( _void ) ;
25+ // eslint-disable-next-line jest/unbound-method -- The Jest ESLint plugin doesn't support Vitest
26+ const setMock = vi . mocked ( browser . storage . local . set ) . mockResolvedValueOnce ( _void ) ;
2527
2628 expect ( await BrowserStorageCache . set ( KEY , VALUE ) ) . toBe ( _void ) ;
2729 expect ( setMock ) . toHaveBeenCalledTimes ( 1 ) ;
@@ -31,7 +33,8 @@ describe('StorageCache', () => {
3133
3234 describe ( 'remove' , ( ) => {
3335 test ( 'removing from the storage cache' , async ( ) => {
34- const removeMock = jest . mocked ( browser . storage . local . remove ) . mockResolvedValueOnce ( _void ) ;
36+ // eslint-disable-next-line jest/unbound-method -- The Jest ESLint plugin doesn't support Vitest
37+ const removeMock = vi . mocked ( browser . storage . local . remove ) . mockResolvedValueOnce ( _void ) ;
3538
3639 expect ( await BrowserStorageCache . remove ( KEY ) ) . toBe ( _void ) ;
3740 expect ( removeMock ) . toHaveBeenCalledTimes ( 1 ) ;
@@ -40,16 +43,22 @@ describe('StorageCache', () => {
4043 } ) ;
4144
4245 describe ( 'get' , ( ) => {
46+ afterEach ( ( ) => {
47+ vi . restoreAllMocks ( ) ;
48+ } ) ;
49+
4350 test ( 'value missing' , async ( ) => {
44- const getMock = jest . mocked ( browser . storage . local . get ) . mockResolvedValue ( { } ) ;
51+ // eslint-disable-next-line jest/unbound-method -- The Jest ESLint plugin doesn't support Vitest
52+ const getMock = vi . mocked ( browser . storage . local . get ) . mockResolvedValue ( { } ) ;
4553
4654 expect ( await BrowserStorageCache . get ( KEY ) ) . toBeUndefined ( ) ;
4755 expect ( getMock ) . toHaveBeenCalledTimes ( 1 ) ;
4856 expect ( getMock ) . toHaveBeenCalledWith ( KEY ) ;
4957 } ) ;
5058
5159 test ( 'value exists' , async ( ) => {
52- const getMock = jest . mocked ( browser . storage . local . get ) . mockResolvedValue ( { [ KEY ] : VALUE } ) ;
60+ // eslint-disable-next-line jest/unbound-method -- The Jest ESLint plugin doesn't support Vitest
61+ const getMock = vi . mocked ( browser . storage . local . get ) . mockResolvedValue ( { [ KEY ] : VALUE } ) ;
5362
5463 expect ( await BrowserStorageCache . get ( KEY ) ) . toBe ( VALUE ) ;
5564 expect ( getMock ) . toHaveBeenCalledTimes ( 1 ) ;
0 commit comments