在 React 组件的单元测试中，我们经常需要根据特定的类名来查找元素，以便进行断言或进一步的操作。React Test Renderer 提供了一种轻量级的方式来渲染 React 组件，并允许我们访问组件的底层结构。虽然它没有直接提供按类名查找元素的方法，但我们可以通过自定义选择器函数来实现这一目标。
自定义选择器函数
以下是一个自定义的选择器函数，它接受一个选择器字符串作为参数，并返回一个函数，该函数用于检查 ReactTestInstance 是否匹配该选择器。
import { ReactTestInstance } from 'react-test-renderer';
function testSelector(selector = ''): (instance: ReactTestInstance) => boolean {
const [type, ...classNames] = selector.split('.');
return instance => {
if (type && instance.type !== type) {
return false;
}
const {className = ''} = instance.props;
const instanceClassNames = (className as string).split(' ');
return classNames.every(className => instanceClassNames.includes(className));
};
}代码解释：
- selector.split('.'): 将选择器字符串按 . 分割成数组。第一个元素是元素类型（例如 'span'，'div'），后面的元素是类名。
- instance.type !== type: 如果指定了元素类型，则检查 ReactTestInstance 的类型是否匹配。如果不匹配，则返回 false。
- instance.props.className: 获取 ReactTestInstance 的 className 属性。
- (className as string).split(' '): 将 className 字符串按空格分割成数组，得到所有类名。
- classNames.every(className => instanceClassNames.includes(className)): 检查选择器中指定的所有类名是否都包含在 ReactTestInstance 的类名中。如果所有类名都匹配，则返回 true，否则返回 false。
使用 findAll 和自定义选择器
有了这个自定义选择器函数，我们就可以在 findAll 方法中使用它来查找具有特定类名的元素。
import { create } from 'react-test-renderer';
import Footer from './Footer'; // 假设 Footer 组件在 Footer.js 文件中
const component = create();
// 假设 Footer 组件包含两个类名为 "footer_link" 的 span 元素
expect(component.root.findAll(testSelector('span.footer_link')).length).toBe(2);代码解释：
- create(): 使用 create 函数渲染 Footer 组件。
- component.root.findAll(testSelector('span.footer_link')): 使用 findAll 方法和 testSelector 函数查找所有类型为 span 且类名包含 footer_link 的元素。
- .length: 获取找到的元素的数量。
- expect(...).toBe(2): 断言找到的元素数量是否为 2。
注意事项
- 确保选择器字符串的格式正确。例如，'span.footer_link' 表示查找类型为 span 且类名包含 footer_link 的元素。
- 自定义选择器函数可以根据需要进行扩展，以支持更复杂的选择器。例如，可以添加对属性选择器的支持。
- 这种方法适用于使用 React Test Renderer 进行单元测试的场景。如果需要进行更复杂的集成测试或端到端测试，建议使用其他的测试框架，例如 Jest 和 React Testing Library。
总结
通过自定义选择器函数，我们可以方便地使用 React Test Renderer 的 findAll 方法按类名查找元素。这种方法可以帮助我们编写更简洁、更易于维护的单元测试。记住，根据项目需求选择合适的测试工具和方法非常重要。
