能否为Enzyme JS添加自定义方法？以findByTestAttr为例（无需fork仓库）

当然可以给Enzyme添加自定义方法！除了fork仓库这种重操作，还有几个更轻量的方案，完全不用动源码，我给你详细说说：

这是最贴合你需求的方案——直接给Enzyme的ReactWrapper和ShallowWrapper原型添加方法，让所有wrapper实例都能像原生方法一样调用findByTestAttr。

你只需要在测试的全局入口文件（比如Create React App里的setupTests.js）中添加这段代码：

import { ReactWrapper, ShallowWrapper } from 'enzyme';

// 定义自定义方法
const findByTestAttr = function(selector) {
  return this.find(`[data-test="${selector}"]`);
};

// 挂载到Enzyme的原型上
ReactWrapper.prototype.findByTestAttr = findByTestAttr;
ShallowWrapper.prototype.findByTestAttr = findByTestAttr;

之后不管是用shallow还是mount渲染出来的wrapper，都能直接用wrapper.findByTestAttr('foo')替代冗长的属性选择器写法了，完美匹配你想要的语法。

如果团队里有人担心修改原型会污染全局环境，那可以把这个逻辑封装成工具函数，通过传入wrapper来调用，更模块化也更安全：

// 比如放在 utils/test-helpers.js 里
export const findByTestAttr = (wrapper, selector) => {
  return wrapper.find(`[data-test="${selector}"]`);
};

在测试文件里使用时：

import { shallow } from 'enzyme';
import { findByTestAttr } from '../utils/test-helpers';
import MyComponent from './MyComponent';

test('找到foo元素', () => {
  const wrapper = shallow(<MyComponent />);
  const fooElement = findByTestAttr(wrapper, 'foo');
  expect(fooElement.exists()).toBe(true);
});

这种方式不会修改Enzyme的原生代码，也方便团队统一维护测试工具方法。

如果你用Jest作为测试框架，还可以把这个逻辑做成自定义断言，让测试代码更语义化：

// 在setupTests.js里扩展Jest断言
import { expect } from '@jest/globals';
import { ReactWrapper, ShallowWrapper } from 'enzyme';

expect.extend({
  toContainElementWithTestAttr(wrapper, testAttrValue) {
    const targetElement = wrapper.find(`[data-test="${testAttrValue}"]`);
    const pass = targetElement.exists();
    
    if (pass) {
      return {
        pass: true,
        message: () => `✅ 成功找到data-test="${testAttrValue}"的元素`
      };
    } else {
      return {
        pass: false,
        message: () => `❌ 未找到data-test="${testAttrValue}"的元素`
      };
    }
  }
});

使用时直接用断言语法：

test('组件包含foo元素', () => {
  const wrapper = shallow(<MyComponent />);
  expect(wrapper).toContainElementWithTestAttr('foo');
});

这种方式更偏向断言验证，适合追求测试代码可读性的场景。

总结一下：原型扩展最贴合你想要的调用语法，工具函数更安全无副作用，自定义断言则让测试更语义化——完全不需要fork Enzyme仓库就能实现需求~

内容的提问来源于stack exchange，提问作者Ramin Mousavi