a6caf84f55
- Add CurveSwapStructEncoder and tests - Add CurveSwapExecutorExposed and tests - Add needed interfaces #time 0m #time 0m #time 0m
149 lines
4.1 KiB
Python
149 lines
4.1 KiB
Python
from core.encoding.interface import EncodingContext
|
|
from core.models.evm.ethereum_token import EthereumToken
|
|
|
|
from propeller_swap_encoders.curve import CurveSwapStructEncoder
|
|
|
|
WETH = EthereumToken(
|
|
symbol="WETH",
|
|
address="0xc02aaa39b223fe8d0a0e5c4f27ead9083c756cc2",
|
|
decimals=18,
|
|
gas=0,
|
|
)
|
|
USDT = EthereumToken(
|
|
symbol="USDT", address="0xdAC17F958D2ee523a2206206994597C13D831ec7", decimals=6
|
|
)
|
|
WBTC = EthereumToken(
|
|
symbol="WBTC", address="0x2260fac5e5542a773aa44fbcfedf7c193bc2c599", decimals=8
|
|
)
|
|
|
|
|
|
def test_encode_curve_v2():
|
|
bob = "0x000000000000000000000000000000000000007B"
|
|
|
|
swap = {
|
|
"pool_id": "0xD51a44d3FaE010294C616388b506AcdA1bfAAE46",
|
|
"sell_token": USDT,
|
|
"buy_token": WETH,
|
|
"split": 0,
|
|
"sell_amount": 0,
|
|
"buy_amount": 100,
|
|
"token_approval_needed": False,
|
|
"pool_tokens": (USDT, WBTC, WETH),
|
|
"pool_type": "CurveV2PoolState",
|
|
"protocol_specific_attrs": {
|
|
"curve_v2_pool_type": "tricrypto2_non_factory",
|
|
"is_curve_tricrypto": None,
|
|
"quote": None,
|
|
"pool_fee": None,
|
|
},
|
|
}
|
|
|
|
curve_encoder = CurveSwapStructEncoder()
|
|
encoded = curve_encoder.encode_swap_struct(
|
|
swap, receiver=bob, encoding_context=EncodingContext()
|
|
)
|
|
assert (
|
|
encoded.hex()
|
|
==
|
|
# buy token
|
|
"c02aaa39b223fe8d0a0e5c4f27ead9083c756cc2"
|
|
# pool address
|
|
"d51a44d3fae010294c616388b506acda1bfaae46"
|
|
# receiver
|
|
"000000000000000000000000000000000000007b"
|
|
# pool type (tricrypto = 3)
|
|
"03"
|
|
# i (sell token index)
|
|
"00"
|
|
# j (buy token index)
|
|
"02"
|
|
# token_approval_needed
|
|
"00"
|
|
)
|
|
|
|
|
|
def test_encode_curve_v1():
|
|
bob = "0x000000000000000000000000000000000000007B"
|
|
swap = {
|
|
"pool_id": "bebc44782c7db0a1a60cb6fe97d0b483032ff1c7",
|
|
"sell_token": USDT,
|
|
"buy_token": WETH,
|
|
"split": 0,
|
|
"sell_amount": 0,
|
|
"buy_amount": 100,
|
|
"token_approval_needed": False,
|
|
"pool_tokens": (USDT, WBTC, WETH),
|
|
"pool_type": "CurveV1PoolState",
|
|
"protocol_specific_attrs": {
|
|
"curve_v2_pool_type": None,
|
|
"is_curve_tricrypto": None,
|
|
"quote": None,
|
|
"pool_fee": 1000000,
|
|
},
|
|
}
|
|
curve_encoder = CurveSwapStructEncoder()
|
|
encoded = curve_encoder.encode_swap_struct(
|
|
swap, receiver=bob, encoding_context=EncodingContext()
|
|
)
|
|
assert (
|
|
encoded.hex()
|
|
==
|
|
# buy token
|
|
"c02aaa39b223fe8d0a0e5c4f27ead9083c756cc2"
|
|
# pool address
|
|
"bebc44782c7db0a1a60cb6fe97d0b483032ff1c7"
|
|
# receiver
|
|
"000000000000000000000000000000000000007b"
|
|
# pool type (simple_no_amount = 1)
|
|
"01"
|
|
# i (sell token index)
|
|
"00"
|
|
# j (buy token index)
|
|
"02"
|
|
# token_approval_needed
|
|
"00"
|
|
)
|
|
|
|
|
|
def test_encode_curve_evm_crypto_pool():
|
|
bob = "0x000000000000000000000000000000000000007B"
|
|
swap = {
|
|
"pool_id": "bebc44782c7db0a1a60cb6fe97d0b483032ff1c7",
|
|
"sell_token": USDT,
|
|
"buy_token": WETH,
|
|
"split": 0,
|
|
"sell_amount": 0,
|
|
"buy_amount": 100,
|
|
"token_approval_needed": False,
|
|
"pool_tokens": (USDT, WBTC, WETH),
|
|
"pool_type": "CurveSimulatedPoolState",
|
|
"protocol_specific_attrs": {
|
|
"curve_v2_pool_type": None,
|
|
"is_curve_tricrypto": True,
|
|
"quote": None,
|
|
"pool_fee": None,
|
|
},
|
|
}
|
|
curve_encoder = CurveSwapStructEncoder()
|
|
encoded = curve_encoder.encode_swap_struct(
|
|
swap, receiver=bob, encoding_context=EncodingContext()
|
|
)
|
|
assert (
|
|
encoded.hex()
|
|
==
|
|
# buy token
|
|
"c02aaa39b223fe8d0a0e5c4f27ead9083c756cc2"
|
|
# pool address
|
|
"bebc44782c7db0a1a60cb6fe97d0b483032ff1c7"
|
|
# receiver
|
|
"000000000000000000000000000000000000007b"
|
|
# pool type (tricrypto = 3)
|
|
"03"
|
|
# i (sell token index)
|
|
"00"
|
|
# j (buy token index)
|
|
"02"
|
|
# token_approval_needed
|
|
"00"
|
|
)
|