在玩笑中模拟 window.sessionStorage 的最佳方法是什么

IT技术 javascript reactjs unit-testing testing jestjs
2021-05-12 09:46:17

下面是一个非常简单的 Jest 单元测试，运行它时，你会得到类似的错误

无法窥探原始值；给定的未定义

类型错误：无法读取未定义的属性“getItem”

但是根据这篇文章的最后两条评论，localStorage 和 sessionStorage 已经添加到最新的 JSDOM 和 jest 中。如果使用jest-localstorage-mock并将其添加到我的 jest setupFiles 中，那么您将看到类似的奇怪错误

TypeError: object[methodName].mockImplementation 不是函数

所以我的问题是在玩笑中模拟 localStorage/sessionStorage 的最佳方法是什么。谢谢

describe('window.sessionStorage', () => {
    let mockSessionStorage;
    beforeEach(() => {
        mockSessionStorage = {};
        jest.spyOn(window.sessionStorage, "getItem").mockImplementation(key => {
            return mockSessionStorage[key];
        });
    });

    describe('getItem-', () => {
        beforeEach(() => {
            mockSessionStorage = {
                foo: 'bar',
            }
        });

        it('gets string item', () => {
            const ret = window.sessionStorage.getItem('foo');
            expect(ret).toBe('bar');
        });
    });
});

下面是我的笑话配置

module.exports = {
    verbose: true,
    //setupFiles: ["jest-localstorage-mock"],
    testURL: "http://localhost/"
};
2个回答

这是仅使用jestjsand的解决方案，仅此typescript而已。

index.ts：

export function getUserInfo() {
  const userInfo = window.sessionStorage.getItem('userInfo');
  if (userInfo) {
    return JSON.parse(userInfo);
  }
  return {};
}

index.spec.ts：

import { getUserInfo } from './';

const localStorageMock = (() => {
  let store = {};

  return {
    getItem(key) {
      return store[key] || null;
    },
    setItem(key, value) {
      store[key] = value.toString();
    },
    removeItem(key) {
      delete store[key];
    },
    clear() {
      store = {};
    }
  };
})();

Object.defineProperty(window, 'sessionStorage', {
  value: localStorageMock
});

describe('getUserInfo', () => {
  beforeEach(() => {
    window.sessionStorage.clear();
    jest.restoreAllMocks();
  });
  it('should get user info from session storage', () => {
    const getItemSpy = jest.spyOn(window.sessionStorage, 'getItem');
    window.sessionStorage.setItem('userInfo', JSON.stringify({ userId: 1, userEmail: 'example@gmail.com' }));
    const actualValue = getUserInfo();
    expect(actualValue).toEqual({ userId: 1, userEmail: 'example@gmail.com' });
    expect(getItemSpy).toBeCalledWith('userInfo');
  });

  it('should get empty object if no user info in session storage', () => {
    const getItemSpy = jest.spyOn(window.sessionStorage, 'getItem');
    const actualValue = getUserInfo();
    expect(actualValue).toEqual({});
    expect(window.sessionStorage.getItem).toBeCalledWith('userInfo');
    expect(getItemSpy).toBeCalledWith('userInfo');
  });
});

带有 100% 覆盖率报告的单元测试结果：

 PASS  src/stackoverflow/51566816/index.spec.ts
  getUserInfo
    ✓ should get user info from session storage (6ms)
    ✓ should get empty object if no user info in session storage (1ms)

----------|----------|----------|----------|----------|-------------------|
File      |  % Stmts | % Branch |  % Funcs |  % Lines | Uncovered Line #s |
----------|----------|----------|----------|----------|-------------------|
All files |      100 |      100 |      100 |      100 |                   |
 index.ts |      100 |      100 |      100 |      100 |                   |
----------|----------|----------|----------|----------|-------------------|
Test Suites: 1 passed, 1 total
Tests:       2 passed, 2 total
Snapshots:   0 total
Time:        4.548s, estimated 6s

这是完整的演示：https : //github.com/mrdulin/jest-codelab/tree/master/src/stackoverflow/51566816

您可能甚至不需要模拟。只需window.sessionStorage像往常一样使用并根据window.sessionStorage.getItem(...)结果而不是 spy编写您的条件window.sessionStorage.setItem。根本就没有忘记调用window.sessionStorage.clear()中beforeEach所展示的。

来自Eric Burel 的评论

其它你可能感兴趣的问题
上一篇用酶模拟“更改”后，复选框未“选中” 下一篇如何在 Create React App 中不显示警告