kid / test_trading.py
nastaseshot's picture
Create test_trading.py
ed7aef8 verified
# test_trading.py
import pytest
from trading.liquidity_checker import check_token_liquidity
from trading.trade_executor import execute_trade
def test_check_token_liquidity():
result = check_token_liquidity("FakeTokenAddress")
assert isinstance(result, bool), "Liquidity check must return a boolean"
def test_execute_trade(monkeypatch):
class DummyWallet:
pass
monkeypatch.setattr('utils.solana_sdk_helper.send_transaction', lambda w, t, a: "DummyTxHash")
txn = execute_trade(DummyWallet(), "FakeTokenAddress", 0.1)
assert txn == "DummyTxHash", "Trade should return dummy transaction hash"