在reactjs中测试计数器

import React, { useState } from 'react';

const Counter = () => {
  const [count, setCount] = useState(0);

  const increment = () => {
    setCount(count + 1);
  };

  const decrement = () => {
    setCount(count - 1);
  };

  return (
    <div>
      <button onClick={increment}>+</button>
      <span>{count}</span>
      <button onClick={decrement}>-</button>
    </div>
  );
};

export default Counter;

import React from 'react';
import { shallow } from 'enzyme';
import Counter from './Counter';

describe('Counter', () => {
  it('should render initial count', () => {
    const wrapper = shallow(<Counter />);
    const countText = wrapper.find('span').text();
    expect(countText).toBe('0');
  });

  it('should increment count when + button is clicked', () => {
    const wrapper = shallow(<Counter />);
    const incrementButton = wrapper.find('button').at(0);
    incrementButton.simulate('click');
    const countText = wrapper.find('span').text();
    expect(countText).toBe('1');
  });

  it('should decrement count when - button is clicked', () => {
    const wrapper = shallow(<Counter />);
    const decrementButton = wrapper.find('button').at(1);
    decrementButton.simulate('click');
    const countText = wrapper.find('span').text();
    expect(countText).toBe('-1');
  });
});