Update module name to github.com/fraktal/mev-beta and fix channel closing issues in pipeline stages
This commit is contained in:
131
test/testutils/testutils.go
Normal file
131
test/testutils/testutils.go
Normal file
@@ -0,0 +1,131 @@
|
||||
package testutils
|
||||
|
||||
import (
|
||||
"context"
|
||||
"math/big"
|
||||
"time"
|
||||
|
||||
"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/ethereum/go-ethereum/common"
|
||||
"github.com/ethereum/go-ethereum/core/types"
|
||||
"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()
|
||||
}
|
||||
Reference in New Issue
Block a user