如何验证store函数调用？

// store.js
export const store = {
  state: { count: 0 },
  updateCount(newCount) {
    this.state.count = newCount;
  }
};

// store.test.js
import { store } from './store';

describe('store', () => {
  it('should update count correctly', () => {
    // 调用函数
    store.updateCount(5);
    
    // 验证状态是否更新
    expect(store.state.count).toBe(5);
  });
});

// store.js
export const store = {
  state: { count: 0 },
  updateCount(newCount, callback) {
    this.state.count = newCount;
    if (typeof callback === 'function') {
      callback();
    }
  }
};

// store.test.js
import { store } from './store';
import { jest } from '@jest/globals';

describe('store', () => {
  it('should call callback after updating count', () => {
    const mockCallback = jest.fn();
    
    store.updateCount(10, mockCallback);
    
    expect(store.state.count).toBe(10);
    expect(mockCallback).toHaveBeenCalled();
  });
});