132 lines
3.7 KiB
Go
132 lines
3.7 KiB
Go
package testutils
|
|
|
|
import (
|
|
"context"
|
|
"math/big"
|
|
"time"
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/core/types"
|
|
"github.com/fraktal/mev-beta/internal/config"
|
|
"github.com/fraktal/mev-beta/internal/logger"
|
|
"github.com/fraktal/mev-beta/pkg/events"
|
|
"github.com/fraktal/mev-beta/pkg/market"
|
|
"github.com/fraktal/mev-beta/pkg/scanner"
|
|
"github.com/holiman/uint256"
|
|
)
|
|
|
|
// CreateTestConfig creates a test configuration
|
|
func CreateTestConfig() *config.Config {
|
|
return &config.Config{
|
|
Arbitrum: config.ArbitrumConfig{
|
|
RPCEndpoint: "https://arb1.arbitrum.io/rpc",
|
|
ChainID: 42161,
|
|
RateLimit: config.RateLimitConfig{
|
|
RequestsPerSecond: 10,
|
|
MaxConcurrent: 5,
|
|
Burst: 20,
|
|
},
|
|
},
|
|
Bot: config.BotConfig{
|
|
Enabled: true,
|
|
PollingInterval: 1,
|
|
MinProfitThreshold: 10.0,
|
|
GasPriceMultiplier: 1.2,
|
|
MaxWorkers: 10,
|
|
ChannelBufferSize: 100,
|
|
RPCTimeout: 30,
|
|
},
|
|
Uniswap: config.UniswapConfig{
|
|
FactoryAddress: "0x1F98431c8aD98523631AE4a59f267346ea31F984",
|
|
PositionManagerAddress: "0xC36442b4a4522E871399CD717aBDD847Ab11FE88",
|
|
FeeTiers: []int64{500, 3000, 10000},
|
|
Cache: config.CacheConfig{
|
|
Enabled: true,
|
|
Expiration: 300,
|
|
MaxSize: 10000,
|
|
},
|
|
},
|
|
Log: config.LogConfig{
|
|
Level: "info",
|
|
Format: "text",
|
|
File: "",
|
|
},
|
|
Database: config.DatabaseConfig{
|
|
File: "mev-bot.db",
|
|
MaxOpenConnections: 10,
|
|
MaxIdleConnections: 5,
|
|
},
|
|
}
|
|
}
|
|
|
|
// CreateTestLogger creates a test logger
|
|
func CreateTestLogger() *logger.Logger {
|
|
return logger.New("info", "text", "")
|
|
}
|
|
|
|
// CreateTestEvent creates a test event
|
|
func CreateTestEvent() *events.Event {
|
|
return &events.Event{
|
|
Type: events.Swap,
|
|
Protocol: "UniswapV3",
|
|
PoolAddress: common.HexToAddress("0x88e6A0c2dDD26FEEb64F039a2c41296FcB3f5640"),
|
|
Token0: common.HexToAddress("0xA0b86991c6218b36c1d19D4a2e9Eb0cE3606eB48"),
|
|
Token1: common.HexToAddress("0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2"),
|
|
Amount0: big.NewInt(1000000000),
|
|
Amount1: big.NewInt(500000000000000000),
|
|
SqrtPriceX96: uint256.NewInt(2505414483750470000),
|
|
Liquidity: uint256.NewInt(1000000000000000000),
|
|
Tick: 200000,
|
|
Timestamp: uint64(time.Now().Unix()),
|
|
TransactionHash: common.HexToHash("0x1234567890abcdef1234567890abcdef1234567890abcdef1234567890abcdef"),
|
|
BlockNumber: 12345,
|
|
}
|
|
}
|
|
|
|
// CreateTestTransaction creates a test transaction
|
|
func CreateTestTransaction() *types.Transaction {
|
|
to := common.HexToAddress("0x88e6A0c2dDD26FEEb64F039a2c41296FcB3f5640")
|
|
return types.NewTransaction(0, to, big.NewInt(0), 0, big.NewInt(0), nil)
|
|
}
|
|
|
|
// CreateTestMarketManager creates a test market manager
|
|
func CreateTestMarketManager() *market.MarketManager {
|
|
cfg := &config.UniswapConfig{
|
|
Cache: config.CacheConfig{
|
|
Expiration: 300,
|
|
MaxSize: 10000,
|
|
},
|
|
}
|
|
logger := CreateTestLogger()
|
|
return market.NewMarketManager(cfg, logger)
|
|
}
|
|
|
|
// CreateTestScanner creates a test market scanner
|
|
func CreateTestScanner() *scanner.MarketScanner {
|
|
cfg := &config.BotConfig{
|
|
MaxWorkers: 5,
|
|
ChannelBufferSize: 10,
|
|
RPCTimeout: 30,
|
|
MinProfitThreshold: 10.0,
|
|
}
|
|
logger := CreateTestLogger()
|
|
return scanner.NewMarketScanner(cfg, logger)
|
|
}
|
|
|
|
// CreateTestPipeline creates a test pipeline
|
|
func CreateTestPipeline() *market.Pipeline {
|
|
cfg := &config.BotConfig{
|
|
MaxWorkers: 5,
|
|
ChannelBufferSize: 10,
|
|
}
|
|
logger := CreateTestLogger()
|
|
marketMgr := CreateTestMarketManager()
|
|
scanner := CreateTestScanner()
|
|
return market.NewPipeline(cfg, logger, marketMgr, scanner)
|
|
}
|
|
|
|
// CreateTestContext creates a test context
|
|
func CreateTestContext() context.Context {
|
|
return context.Background()
|
|
}
|