diff --git a/jest/async-storage-mock.js b/jest/async-storage-mock.js index f743310b..b50ab37a 100644 --- a/jest/async-storage-mock.js +++ b/jest/async-storage-mock.js @@ -34,6 +34,14 @@ const asMock = { multiSet: jest.fn(_multiSet), multiRemove: jest.fn(_multiRemove), multiMerge: jest.fn(_multiMerge), + useAsyncStorage: jest.fn(key => { + return { + getItem: (...args) => asMock.getItem(key, ...args), + setItem: (...args) => asMock.setItem(key, ...args), + mergeItem: (...args) => asMock.mergeItem(key, ...args), + removeItem: (...args) => asMock.removeItem(key, ...args), + }; + }), }; async function _multiSet(keyValuePairs, callback) {