Initial commit: Mantle AI Trading Bot

Features:
- AI-powered signal generation with multi-factor analysis
- Fundamental news aggregation from multiple sources
- Technical analysis with 6+ indicators
- VectorDB integration for semantic search
- Backtesting engine with performance metrics
- Demo/paper trading mode
- Real-time WebSocket updates
- Comprehensive dashboard UI

Built for Mantle Turing Test Hackathon
- AI Trading track
- AI Alpha & Data track
This commit is contained in:
Mantle AI Trader
2026-06-06 06:02:07 +00:00
Unverified
parent 6664758a6d
commit b1da4ee01d
100 changed files with 16113 additions and 0 deletions

View File

@@ -0,0 +1,184 @@
/**
* Unit Tests for Demo Trader
*/
import { describe, test, expect, beforeEach } from 'bun:test';
import { DemoTrader } from '../../src/lib/trading/demo/demo-trader';
import { TradeAction, OrderType } from '../../src/lib/trading/core/types';
describe('DemoTrader', () => {
let demoTrader: DemoTrader;
beforeEach(() => {
demoTrader = new DemoTrader(10000);
});
describe('initialization', () => {
test('should initialize with correct portfolio', () => {
const portfolio = demoTrader.getPortfolio();
expect(portfolio.totalValue).toBe(10000);
expect(portfolio.cashBalance).toBe(10000);
expect(portfolio.realizedPnL).toBe(0);
expect(portfolio.unrealizedPnL).toBe(0);
});
test('should have no positions initially', () => {
const positions = demoTrader.getPositions();
expect(positions.length).toBe(0);
});
});
describe('placeOrder', () => {
test('should place a market buy order', () => {
// Set price first
demoTrader.updatePrice('BTCUSDT', 45000);
const order = demoTrader.placeOrder({
symbol: 'BTCUSDT',
side: TradeAction.BUY,
type: OrderType.MARKET,
quantity: 0.1
});
expect(order).toBeDefined();
expect(order.status).toBe('FILLED');
expect(order.symbol).toBe('BTCUSDT');
expect(order.side).toBe(TradeAction.BUY);
});
test('should update cash balance after buy order', () => {
demoTrader.updatePrice('ETHUSDT', 2500);
demoTrader.placeOrder({
symbol: 'ETHUSDT',
side: TradeAction.BUY,
type: OrderType.MARKET,
quantity: 2
});
const portfolio = demoTrader.getPortfolio();
expect(portfolio.cashBalance).toBe(10000 - (2500 * 2));
});
test('should create a position after buy order', () => {
demoTrader.updatePrice('SOLUSDT', 100);
demoTrader.placeOrder({
symbol: 'SOLUSDT',
side: TradeAction.BUY,
type: OrderType.MARKET,
quantity: 10
});
const positions = demoTrader.getPositions();
expect(positions.length).toBe(1);
expect(positions[0].symbol).toBe('SOLUSDT');
expect(positions[0].side).toBe('LONG');
expect(positions[0].quantity).toBe(10);
});
test('should throw error for insufficient capital', () => {
demoTrader.updatePrice('BTCUSDT', 45000);
expect(() => {
demoTrader.placeOrder({
symbol: 'BTCUSDT',
side: TradeAction.BUY,
type: OrderType.MARKET,
quantity: 1 // Would cost 45000, but we only have 10000
});
}).toThrow('Insufficient capital');
});
});
describe('closePosition', () => {
test('should close an existing position', () => {
demoTrader.updatePrice('BTCUSDT', 45000);
demoTrader.placeOrder({
symbol: 'BTCUSDT',
side: TradeAction.BUY,
type: OrderType.MARKET,
quantity: 0.1
});
const closeOrder = demoTrader.closePosition('BTCUSDT');
expect(closeOrder).toBeDefined();
expect(closeOrder?.side).toBe(TradeAction.SELL);
const positions = demoTrader.getPositions();
expect(positions.length).toBe(0);
});
test('should return null when closing non-existent position', () => {
const result = demoTrader.closePosition('NONEXISTENT');
expect(result).toBeNull();
});
});
describe('updatePrice', () => {
test('should update position PnL when price changes', () => {
demoTrader.updatePrice('ETHUSDT', 2500);
demoTrader.placeOrder({
symbol: 'ETHUSDT',
side: TradeAction.BUY,
type: OrderType.MARKET,
quantity: 2
});
// Price goes up
demoTrader.updatePrice('ETHUSDT', 2600);
const positions = demoTrader.getPositions();
expect(positions[0].unrealizedPnL).toBeGreaterThan(0);
expect(positions[0].currentPrice).toBe(2600);
});
});
describe('reset', () => {
test('should reset portfolio to initial state', () => {
demoTrader.updatePrice('BTCUSDT', 45000);
demoTrader.placeOrder({
symbol: 'BTCUSDT',
side: TradeAction.BUY,
type: OrderType.MARKET,
quantity: 0.1
});
demoTrader.reset(15000);
const portfolio = demoTrader.getPortfolio();
expect(portfolio.totalValue).toBe(15000);
expect(portfolio.cashBalance).toBe(15000);
const positions = demoTrader.getPositions();
expect(positions.length).toBe(0);
});
});
describe('getStatistics', () => {
test('should return correct trade statistics', () => {
demoTrader.updatePrice('BTCUSDT', 45000);
// Open and close a position
demoTrader.placeOrder({
symbol: 'BTCUSDT',
side: TradeAction.BUY,
type: OrderType.MARKET,
quantity: 0.1
});
demoTrader.updatePrice('BTCUSDT', 46000); // Price up
demoTrader.closePosition('BTCUSDT');
const stats = demoTrader.getStatistics();
expect(stats.totalTrades).toBe(2); // Open and close
expect(stats.winningTrades + stats.losingTrades).toBe(stats.totalTrades);
});
});
});