如何测试依赖于useContext钩子的react组件？

Z时代
2024-01-10
分类：问答

我有一个使用的组件，useContext然后它的输出取决于上下文中的值。一个简单的例子：

import React, { useContext } from 'react';
const MyComponent = () => {
  const name = useContext(NameContext);
  return <div>{name}</div>;
};

当使用来自react和jest快照的浅色渲染器测试此组件时。如何更改值NameContext？

回答：

通常，使用钩子应该不会改变测试策略。实际上，这里更大的问题不是问题，而是上下文的使用，这使事情变得有些复杂。

有很多方法可以使这项工作，但是我发现唯一可以使用的方法'react-test-renderer/shallow'是注入一个模拟钩子：

import ShallowRenderer from 'react-test-renderer/shallow';
let realUseContext;
let useContextMock;
// Setup mock
beforeEach(() => {
    realUseContext = React.useContext;
    useContextMock = React.useContext = jest.fn();
});
// Cleanup mock
afterEach(() => {
    React.useContext = realUseContext;
});
test("mock hook", () => {
    useContextMock.mockReturnValue("Test Value");
    const element = new ShallowRenderer().render(
        <MyComponent />
    );
    expect(element.props.children).toBe('Test Value');
});

但是，这有点脏，并且是特定于实现的，因此，如果您能够在使用浅渲染器时有所妥协，则可以使用其他一些选项：

非浅渲染

如果您不是浅层渲染，则可以将组件包装在上下文提供程序中以注入所需的值：

import TestRenderer from 'react-test-renderer';
test("non-shallow render", () => {
    const element = new TestRenderer.create(
        <NameContext.Provider value="Provided Value">
            <MyComponent />
        </NameContext.Provider>
    );
    expect(element.root.findByType("div").children).toEqual(['Provided Value']);
});

（免责声明：这应该可以，但是当我测试它时，我遇到了一个错误，我认为这是我的设置中的一个问题）

酶和潜水的浅渲染

正如@skyboyer所评论的那样，asen的浅渲染器支持.dive使您能够深渲染一个浅渲染组件的一部分：

import { shallow } from "./enzyme";
test("enzyme dive", () => {
    const TestComponent = () => (
        <NameContext.Provider value="Provided Value">
            <MyComponent />
        </NameContext.Provider>
    );
    const element = shallow(<TestComponent />);
    expect(element.find(MyComponent).dive().text()).toBe("Provided Value");
});

使用ReactDOM

最后，Hooks FAQ提供了一个使用来测试钩子的示例，该示例ReactDOM同样适用。自然地，使用ReactDOM手段这也是一个深渲染，而不是浅渲染。

let container;
beforeEach(() => {
    container = document.createElement('div');
    document.body.appendChild(container);
});
afterEach(() => {
    document.body.removeChild(container);
    container = null;
});
test("with ReactDOM", () => {
    act(() => {
        ReactDOM.render((
            <NameContext.Provider value="Provided Value">
                <MyComponent />
            </NameContext.Provider>
        ), container);
    });
    expect(container.textContent).toBe("Provided Value");
});