web3agent-aof/dist/test/swap.test.js

AI Framework for Onchain Operations and DeFi Interactions

import { jest } from '@jest/globals';
import { EVMAgent } from '../src/EVMAgent';
describe('EVMAgent', () => {
    let agent;
    let mockProvider;
    let mockWallet;
    beforeEach(() => {
        mockProvider = {
            getNetwork: jest.fn().mockResolvedValue({ chainId: 1 }),
        };
        mockWallet = {
            provider: mockProvider,
            getAddress: jest.fn().mockResolvedValue('0x123'),
            connect: jest.fn().mockReturnThis(),
        };
        agent = new EVMAgent(mockProvider, mockWallet);
    });
    afterEach(() => {
        jest.clearAllMocks();
    });
    describe('price monitoring', () => {
        it('should monitor price changes', async () => {
            const callback = jest.fn();
            agent.monitorPriceChange('ETH', callback);
            // Wait for initial price check
            await new Promise(resolve => setTimeout(resolve, 100));
            expect(callback).toHaveBeenCalled();
        });
        it('should stop monitoring when requested', async () => {
            const callback = jest.fn();
            agent.monitorPriceChange('ETH', callback);
            // Wait for initial price check
            await new Promise(resolve => setTimeout(resolve, 100));
            agent.stopPriceMonitoring('ETH');
            // Wait to ensure no more callbacks
            await new Promise(resolve => setTimeout(resolve, 100));
            const callCount = callback.mock.calls.length;
            await new Promise(resolve => setTimeout(resolve, 100));
            expect(callback.mock.calls.length).toBe(callCount);
        });
    });
});
//# sourceMappingURL=swap.test.js.map