Skip to content

Commit

Permalink
test: use quoter for AMO rebalancing.
Browse files Browse the repository at this point in the history
  • Loading branch information
@clement-ux
clement-ux committed Aug 29, 2024
1 parent d29dc6f commit 4e9d161
Showing 1 changed file with 135 additions and 35 deletions.
170 changes: 135 additions & 35 deletions contracts/test/strategies/aerodrome-amo.base.fork-test.js
View file
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@ const { expect } = require("chai");
const { oethUnits } = require("../helpers");
const ethers = require("ethers");
const { impersonateAndFund } = require("../../utils/signers");
const { deployWithConfirmation } = require("../../utils/deploy");
//const { formatUnits } = ethers.utils;
const { BigNumber } = ethers;

Expand Down Expand Up @@ -124,7 +125,8 @@ describe("ForkTest: Aerodrome AMO Strategy (Base)", function () {
strategist,
rafael,
aeroSwapRouter,
aeroNftManager;
aeroNftManager,
quoter;

beforeEach(async () => {
fixture = await baseFixture();
Expand All @@ -141,6 +143,9 @@ describe("ForkTest: Aerodrome AMO Strategy (Base)", function () {
oethbVaultSigner = await impersonateAndFund(oethbVault.address);
gauge = fixture.aeroClGauge;

await deployWithConfirmation("AerodromeAMOQuoter");
quoter = await hre.ethers.getContract("AerodromeAMOQuoter");

await setup();
await weth
.connect(rafael)
Expand Down Expand Up @@ -275,7 +280,7 @@ describe("ForkTest: Aerodrome AMO Strategy (Base)", function () {
aeroBalanceBefore
);

expect(aeroBalancediff).to.equal(oethUnits("1337"));
expect(aeroBalancediff).to.gte(oethUnits("1337")); // Gte to take into account rewards already accumulated.
await assetLpStakedInGauge();
});
});
Expand Down Expand Up @@ -322,9 +327,9 @@ describe("ForkTest: Aerodrome AMO Strategy (Base)", function () {
balanceBefore.add(oethUnits("1"))
);

// Little to no weth should be left on the strategy contract - 10 wei is really small
// Little to no weth should be left on the strategy contract - 100 wei is really small
expect(await weth.balanceOf(aerodromeAmoStrategy.address)).to.lte(
BigNumber.from("10")
BigNumber.from("100")
);

await assetLpStakedInGauge();
Expand Down Expand Up @@ -412,9 +417,9 @@ describe("ForkTest: Aerodrome AMO Strategy (Base)", function () {
balanceBefore.add(oethUnits("1"))
);

// Little to no weth should be left on the strategy contract - 10 wei is really small
// Little to no weth should be left on the strategy contract - 100 wei is really small
expect(await weth.balanceOf(aerodromeAmoStrategy.address)).to.lte(
BigNumber.from("10")
BigNumber.from("100")
);

await assetLpStakedInGauge();
Expand Down Expand Up @@ -497,9 +502,9 @@ describe("ForkTest: Aerodrome AMO Strategy (Base)", function () {
balanceBefore.add(oethUnits("1"))
);

// Little to no weth should be left on the strategy contract - 10 wei is really small
// Little to no weth should be left on the strategy contract - 100 wei is really small
expect(await weth.balanceOf(aerodromeAmoStrategy.address)).to.lte(
BigNumber.from("10")
BigNumber.from("100")
);

await assetLpStakedInGauge();
Expand Down Expand Up @@ -564,11 +569,21 @@ describe("ForkTest: Aerodrome AMO Strategy (Base)", function () {
it("Should be able to deposit to the pool & rebalance", async () => {
await mintAndDepositToStrategy({ amount: oethUnits("5") });

// prettier-ignore
const tx = await rebalance(
const currentPrice = await aerodromeAmoStrategy.getPoolX96Price();
const priceTick0 = await aerodromeAmoStrategy.sqrtRatioX96TickLower();
const priceTick1 = await aerodromeAmoStrategy.sqrtRatioX96TickHigher();
const targetPrice = priceTick0 * 0.2 + priceTick1 * 0.8;

let direction = currentPrice > targetPrice ? true : false;
const value = await quoteAmountToSwapBeforeRebalance(
direction,
oethUnits("0.00001"),
true, // _swapWETHs
oethUnits("0.000009")
oethUnits("1")
);
const tx = await rebalance(
value,
direction,
0
);

await expect(tx).to.emit(aerodromeAmoStrategy, "PoolRebalanced");
Expand All @@ -578,11 +593,21 @@ describe("ForkTest: Aerodrome AMO Strategy (Base)", function () {
it("Should be able to deposit to the pool & rebalance multiple times", async () => {
await mintAndDepositToStrategy({ amount: oethUnits("5") });

// prettier-ignore
const tx = await rebalance(
const currentPrice = await aerodromeAmoStrategy.getPoolX96Price();
const priceTick0 = await aerodromeAmoStrategy.sqrtRatioX96TickLower();
const priceTick1 = await aerodromeAmoStrategy.sqrtRatioX96TickHigher();
const targetPrice = priceTick0 * 0.2 + priceTick1 * 0.8;

let direction = currentPrice > targetPrice ? true : false;
const value = await quoteAmountToSwapBeforeRebalance(
direction,
oethUnits("0.00001"),
true, // _swapWETHs
oethUnits("0.000009")
oethUnits("1")
);
const tx = await rebalance(
value,
direction,
0
);

await expect(tx).to.emit(aerodromeAmoStrategy, "PoolRebalanced");
Expand Down Expand Up @@ -640,19 +665,31 @@ describe("ForkTest: Aerodrome AMO Strategy (Base)", function () {

await expect(
rebalance(
oethUnits("0.01"),
oethUnits("1000000000"),
true, // _swapWETH
oethUnits("0.009")
)
).to.be.revertedWith("NotEnoughWethForSwap");
});

it("Should revert when pool rebalance is off target", async () => {
const currentPrice = await aerodromeAmoStrategy.getPoolX96Price();
const priceTick0 = await aerodromeAmoStrategy.sqrtRatioX96TickLower();
const priceTick1 = await aerodromeAmoStrategy.sqrtRatioX96TickHigher();
const targetPrice = priceTick0 * 0.2 + priceTick1 * 0.8;

let direction = currentPrice > targetPrice ? true : false;
const value = await quoteAmountToSwapBeforeRebalance(
direction,
oethUnits("0.00000001"),
oethUnits("1000")
);

await expect(
rebalance(
oethUnits("0.001"),
true, // _swapWETH
oethUnits("0.0009")
value.mul("200").div("100"),
direction,
0
)
).to.be.revertedWith("PoolRebalanceOutOfBounds");
});
Expand All @@ -666,10 +703,21 @@ describe("ForkTest: Aerodrome AMO Strategy (Base)", function () {
// supply some WETH for the rebalance
await mintAndDepositToStrategy({ amount: oethUnits("1") });

const currentPrice = await aerodromeAmoStrategy.getPoolX96Price();
const priceTick0 = await aerodromeAmoStrategy.sqrtRatioX96TickLower();
const priceTick1 = await aerodromeAmoStrategy.sqrtRatioX96TickHigher();
const targetPrice = priceTick0 * 0.2 + priceTick1 * 0.8;

let direction = currentPrice > targetPrice ? true : false;
const value = await quoteAmountToSwapBeforeRebalance(
direction,
oethUnits("0.00000001"),
oethUnits("1000")
);
await rebalance(
oethUnits("0.0083"),
true, // _swapWETH
oethUnits("0.0082")
value,
direction,
0
);

await assetLpStakedInGauge();
Expand All @@ -681,29 +729,39 @@ describe("ForkTest: Aerodrome AMO Strategy (Base)", function () {
swapWeth: true,
});

const currentPrice = await aerodromeAmoStrategy.getPoolX96Price();
const priceTick0 = await aerodromeAmoStrategy.sqrtRatioX96TickLower();
const priceTick1 = await aerodromeAmoStrategy.sqrtRatioX96TickHigher();
const targetPrice = priceTick0 * 0.2 + priceTick1 * 0.8;

let direction = currentPrice > targetPrice ? true : false;
const value = await quoteAmountToSwapBeforeRebalance(
direction,
oethUnits("0.00000001"),
oethUnits("1000")
);
await rebalance(
oethUnits("0.0079"),
false, // _swapWETH
oethUnits("0.0036")
value,
direction,
0
);

await assetLpStakedInGauge();
});

it("Should have the correct balance within some tolerance", async () => {
await expect(
await aerodromeAmoStrategy.checkBalance(weth.address)
).to.approxEqualTolerance(oethUnits("23.2"));
const balance = await aerodromeAmoStrategy.checkBalance(weth.address);
await mintAndDepositToStrategy({ amount: oethUnits("6") });
await expect(
await aerodromeAmoStrategy.checkBalance(weth.address)
).to.approxEqualTolerance(oethUnits("29.35"));
).to.equal(balance.add(oethUnits("6")));

// just add liquidity don't move the active trading position
await rebalance(BigNumber.from("0"), true, BigNumber.from("0"));

await expect(
await aerodromeAmoStrategy.checkBalance(weth.address)
).to.approxEqualTolerance(oethUnits("51.09"));
).to.approxEqualTolerance(balance.add(oethUnits("6").mul("4")), 1.5);

await assetLpStakedInGauge();
});
Expand All @@ -723,7 +781,11 @@ describe("ForkTest: Aerodrome AMO Strategy (Base)", function () {
});

await expect(
rebalance(oethUnits("4.99"), true, oethUnits("4"))
rebalance(
(await weth.balanceOf(await aerodromeAmoStrategy.clPool())).mul("2"),
true,
oethUnits("4")
)
).to.be.revertedWith("NotEnoughWethForSwap");

await assetLpStakedInGauge();
Expand Down Expand Up @@ -833,11 +895,24 @@ describe("ForkTest: Aerodrome AMO Strategy (Base)", function () {
const setup = async () => {
await mintAndDepositToStrategy({ amount: oethUnits("5") });

const currentPrice = await aerodromeAmoStrategy.getPoolX96Price();
const priceTick0 = await aerodromeAmoStrategy.sqrtRatioX96TickLower();
const priceTick1 = await aerodromeAmoStrategy.sqrtRatioX96TickHigher();
const targetPrice = priceTick0 * 0.2 + priceTick1 * 0.8;

let direction = currentPrice > targetPrice ? true : false;

const value = await quoteAmountToSwapBeforeRebalance(
direction,
oethUnits("0.00000001"),
oethUnits("1000")
);

// move the price to pre-configured 20% value
await rebalance(
oethUnits("0.00875"),
true, // _swapWETH
oethUnits("0.0086")
value,
direction, // _swapWETH
value.mul(90).div(100)
);
};

Expand Down Expand Up @@ -889,6 +964,31 @@ describe("ForkTest: Aerodrome AMO Strategy (Base)", function () {
});
};

const quoteAmountToSwapBeforeRebalance = async (
swapWETH,
minAmount,
maxAmount
) => {
await oethbVault
.connect(await impersonateAndFund(addresses.base.governor))
.setStrategistAddr(await quoter.quoterHelper());
const txResponse = await quoter.quoteAmountToSwapBeforeRebalance(
aerodromeAmoStrategy.address,
swapWETH,
minAmount,
maxAmount,
50
);
const txReceipt = await txResponse.wait();
const [transferEvent] = txReceipt.events;
const value = transferEvent.args.value;
await oethbVault
.connect(await impersonateAndFund(addresses.base.governor))
.setStrategistAddr(addresses.base.strategist);
//console.log("Value to swap", value.toString());
return value;
};

const rebalance = async (amountToSwap, swapWETH, minTokenReceived) => {
return await aerodromeAmoStrategy
.connect(strategist)
Expand Down

0 comments on commit 4e9d161

Please sign in to comment.