edwin-sdk/tests/solana_wallet.test.ts

SDK for integrating AI agents with DeFi protocols

import { config } from 'dotenv';
config(); // Load test environment variables from .env file

import { describe, it, expect, beforeAll } from 'vitest';
import { SolanaWalletFactory } from '../src/core/wallets/solana_wallet/factory';
import { SolanaWalletService } from '../src/plugins/solana_wallet/solanaWalletService';
import { PublicKeyClient } from '../src/core/wallets/solana_wallet/clients/publickey';

// Check if Helius API key is available
const hasHeliusKey = Boolean(process.env.HELIUS_KEY);

// Skip entire test if no Helius API key
const describeIf = hasHeliusKey ? describe : describe.skip;

// Known Solana wallets with tokens for testing
const TEST_WALLET_ADDRESS = process.env.SOLANA_PUBLIC_KEY as string;

describeIf('Solana Wallet Token Balances Tests', () => {
    let wallet: PublicKeyClient;
    let solanaWalletService: SolanaWalletService;

    beforeAll(() => {
        // Create a public key wallet for testing
        wallet = SolanaWalletFactory.fromPublicKey(TEST_WALLET_ADDRESS as string);
        solanaWalletService = new SolanaWalletService(wallet);
        console.log('Running Solana wallet tests with Helius API key');
    });

    describe('Get Wallet Balances', () => {
        it('should get all token balances for a wallet', async () => {
            console.log(`Getting token balances for wallet: ${TEST_WALLET_ADDRESS}`);

            const result = await solanaWalletService.getSolanaWalletBalances({
                walletAddress: TEST_WALLET_ADDRESS,
            });

            expect(result).toBeDefined();
            expect(Array.isArray(result)).toBe(true);

            // Log the first few tokens for debugging
            if (result.length > 0) {
                console.log(`Found ${result.length} tokens in wallet`);
                console.log('First few tokens:');
                result.slice(0, 3).forEach(token => {
                    console.log(`- ${token.symbol}: ${token.balance}`);
                });

                // Validate token structure
                const firstToken = result[0];
                expect(firstToken.mint).toBeDefined();
                expect(firstToken.symbol).toBeDefined();
                expect(typeof firstToken.balance).toBe('number');
            } else {
                console.log('No tokens found in wallet');
            }
        }, 30000); // 30 second timeout
    });

    describe('Get Current Wallet Balances', () => {
        it('should get all token balances for the current wallet', async () => {
            console.log(`Getting token balances for current wallet: ${wallet.getAddress()}`);

            const result = await solanaWalletService.getCurrentSolanaWalletBalances();

            expect(result).toBeDefined();
            expect(Array.isArray(result)).toBe(true);

            // The current wallet is the same as TEST_WALLET_ADDRESS in this test
            // so we expect similar results as the previous test
            if (result.length > 0) {
                console.log(`Found ${result.length} tokens in current wallet`);

                // Validate token structure
                const firstToken = result[0];
                expect(firstToken.mint).toBeDefined();
                expect(firstToken.symbol).toBeDefined();
                expect(typeof firstToken.balance).toBe('number');
            } else {
                console.log('No tokens found in current wallet');
            }
        }, 30000); // 30 second timeout
    });

    describe('Error Handling', () => {
        it('should handle invalid wallet addresses gracefully', async () => {
            console.log('Testing with invalid wallet address');

            try {
                await solanaWalletService.getSolanaWalletBalances({
                    walletAddress: 'invalid-address',
                });
                // If it doesn't throw, the test fails
                expect(true).toBe(false);
            } catch (error) {
                // Expect an error for invalid addresses
                expect(error).toBeDefined();
                console.log('Error handling test passed');
            }
        }, 30000); // 30 second timeout
    });
});