Liquidity.Party/tycho-protocol-sdk
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: Add execution for curve

Browse Source 
- Add CurveSwapStructEncoder and tests
- Add CurveSwapExecutorExposed and tests
  - Add needed interfaces

#time 0m


#time 0m

#time 0m
This commit is contained in:
Diana Carvalho
2024-09-05 13:13:09 +01:00
parent f181688090
commit a6caf84f55
11 changed files with 1282 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

148
propeller-swap-encoders/propeller_swap_encoders/tests/test_curve.py Normal file
View File

@@ -0,0 +1,148 @@
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"
)