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

Fix tests

Browse Source
This commit is contained in:
Louise Poole
2024-04-15 16:56:19 +01:00
parent 98857be305
commit 306293764c
2 changed files with 12 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
evm/src/balancer-v2/BalancerV2SwapAdapter.sol
View File

@@ -187,7 +187,7 @@ contract BalancerV2SwapAdapter is ISwapAdapter {
override override
returns (Capability[] memory capabilities) returns (Capability[] memory capabilities)
{ {
capabilities = new Capability[](2); capabilities = new Capability[](3);
capabilities[0] = Capability.SellOrder; capabilities[0] = Capability.SellOrder;
capabilities[1] = Capability.BuyOrder; capabilities[1] = Capability.BuyOrder;
capabilities[2] = Capability.PriceFunction; capabilities[2] = Capability.PriceFunction;

20
evm/test/BalancerV2SwapAdapter.t.sol
View File

@@ -40,14 +40,15 @@ contract BalancerV2SwapAdapterTest is Test, ISwapAdapterTypes {
function testPrice() public { function testPrice() public {
uint256[] memory amounts = new uint256[](2); uint256[] memory amounts = new uint256[](2);
amounts[0] = 100; amounts[0] = amount0;
amounts[1] = 200; amounts[1] = amount1;
vm.expectRevert(
abi.encodeWithSelector( Fraction[] memory prices = adapter.price(B_80BAL_20WETH_POOL_ID, BAL, WETH, amounts);
NotImplemented.selector, "BalancerV2SwapAdapter.price"
) for (uint256 i = 0; i < prices.length; i++) {
); assertGt(prices[i].numerator, 0);
adapter.price(B_80BAL_20WETH_POOL_ID, BAL, WETH, amounts); assertGt(prices[i].denominator, 0);
}
} }
function testPriceSingleFuzz() public { function testPriceSingleFuzz() public {
@@ -206,9 +207,10 @@ contract BalancerV2SwapAdapterTest is Test, ISwapAdapterTypes {
{ {
Capability[] memory res = adapter.getCapabilities(pool, t0, t1); Capability[] memory res = adapter.getCapabilities(pool, t0, t1);
assertEq(res.length, 2); assertEq(res.length, 3);
assertEq(uint256(res[0]), uint256(Capability.SellOrder)); assertEq(uint256(res[0]), uint256(Capability.SellOrder));
assertEq(uint256(res[1]), uint256(Capability.BuyOrder)); assertEq(uint256(res[1]), uint256(Capability.BuyOrder));
assertEq(uint256(res[2]), uint256(Capability.PriceFunction));
} }
function testGetTokens() public { function testGetTokens() public {