Add wOETH donation fork tests

contracts/test/_fixture.js

@@ -82,6 +82,9 @@ const simpleOETHFixture = deployments.createFixture(async () => {
   );
   const oeth = await ethers.getContractAt("OETH", oethProxy.address);
 
+  const cWOETHProxy = await ethers.getContract("WOETHProxy");
+  const woeth = await ethers.getContractAt("WOETH", cWOETHProxy.address);
+
   const oethHarvesterProxy = await ethers.getContract("OETHHarvesterProxy");
   const oethHarvester = await ethers.getContractAt(
     "OETHHarvester",
@@ -182,6 +185,7 @@ const simpleOETHFixture = deployments.createFixture(async () => {
     // OETH
     oethVault,
     oeth,
+    woeth,
     nativeStakingSSVStrategy,
     oethDripper,
     oethHarvester,

contracts/test/token/woeth.mainnet.fork-test.js

@@ -0,0 +1,265 @@
+const { expect } = require("chai");
+
+const { simpleOETHFixture, createFixtureLoader } = require("./../_fixture");
+const { hardhatSetBalance } = require("../_fund");
+const { oethUnits } = require("../helpers");
+
+const oethWhaleFixture = async () => {
+  const fixture = await simpleOETHFixture();
+
+  const { weth, oeth, oethVault, woeth, domen } = fixture;
+
+  // Domen is a OETH whale
+  await oethVault
+    .connect(domen)
+    .mint(weth.address, oethUnits("20000"), oethUnits("19999"));
+
+  await oeth.connect(domen).approve(woeth.address, oethUnits("20000"));
+
+  return fixture;
+};
+
+const loadFixture = createFixtureLoader(oethWhaleFixture);
+
+describe("ForkTest: wOETH", function () {
+  this.timeout(0);
+
+  let fixture;
+  beforeEach(async () => {
+    fixture = await loadFixture();
+  });
+
+  it("Should prevent total asset manipulation by donations", async () => {
+    const { oeth, woeth, domen } = fixture;
+    const totalAssetsBefore = await woeth.totalAssets();
+    await oeth.connect(domen).transfer(woeth.address, oethUnits("100"));
+    const totalAssetsAfter = await woeth.totalAssets();
+
+    expect(totalAssetsBefore).to.be.equal(totalAssetsAfter);
+  });
+
+  it("Deposit should not be manipulated by donations", async () => {
+    const { oeth, woeth, domen } = fixture;
+
+    await expect(domen).to.have.approxBalanceOf("0", woeth);
+
+    // Wrap some OETH
+    await woeth.connect(domen).deposit(oethUnits("1000"), domen.address);
+
+    const sharePriceBeforeDonate = await woeth.convertToAssets(
+      oethUnits("1000")
+    );
+
+    // Donate some OETH
+    oeth.connect(domen).transfer(woeth.address, oethUnits("10000"));
+
+    // Ensure no change in share price
+    const sharePriceAfterDonate = await woeth.convertToAssets(
+      oethUnits("1000")
+    );
+    expect(sharePriceBeforeDonate).to.approxEqual(
+      sharePriceAfterDonate,
+      "Price manipulation"
+    );
+
+    // Wrap again
+    await woeth.connect(domen).deposit(oethUnits("1000"), domen.address);
+
+    // Ensure the balance is right
+    await expect(domen).to.have.approxBalanceOf(
+      // 2000 * 1000 / sharePrice(1000 OETH)
+      oethUnits("2000").mul(oethUnits("1000")).div(sharePriceAfterDonate),
+      woeth
+    );
+  });
+
+  it("Withdraw should not be manipulated by donations", async () => {
+    const { oeth, woeth, domen } = fixture;
+
+    await expect(domen).to.have.approxBalanceOf("0", woeth);
+    await expect(domen).to.have.approxBalanceOf("20000", oeth);
+
+    // Wrap some OETH
+    await woeth.connect(domen).deposit(oethUnits("3000"), domen.address);
+
+    const sharePriceBeforeDonate = await woeth.convertToAssets(
+      oethUnits("1000")
+    );
+
+    // Donate some OETH
+    oeth.connect(domen).transfer(woeth.address, oethUnits("10000"));
+
+    // Ensure no change in share price
+    const sharePriceAfterDonate = await woeth.convertToAssets(
+      oethUnits("1000")
+    );
+    expect(sharePriceBeforeDonate).to.approxEqual(
+      sharePriceAfterDonate,
+      "Price manipulation"
+    );
+
+    // Withdraw
+    await woeth
+      .connect(domen)
+      .withdraw(
+        await woeth.maxWithdraw(domen.address),
+        domen.address,
+        domen.address
+      );
+
+    // Ensure balance is right
+    await expect(domen).to.have.approxBalanceOf("10000", oeth);
+  });
+
+  describe("Funds in, Funds out", async () => {
+    it("should deposit at the correct ratio", async () => {
+      const { oeth, woeth, domen } = fixture;
+
+      const totalSupply = await woeth.totalSupply();
+      const balanceBefore = await oeth.balanceOf(domen.address);
+
+      // Wrap some OETH
+      const txResponse = await woeth
+        .connect(domen)
+        .deposit(oethUnits("50"), domen.address);
+      const txReceipt = await txResponse.wait();
+      const mintedShares = txReceipt.events[2].args.shares; // 0. transfer oeth, 1. transfer woeth, 2. deposit
+      const assetTransfered = txReceipt.events[2].args.assets; // 0. transfer oeth, 1. transfer woeth, 2. mint
+
+      await expect(assetTransfered).to.be.equal(oethUnits("50"));
+      await expect(
+        await woeth.convertToShares(assetTransfered)
+      ).to.be.approxEqual(mintedShares);
+      await expect(woeth).to.have.a.totalSupply(totalSupply.add(mintedShares));
+      await expect(await woeth.balanceOf(domen.address)).to.be.equal(
+        mintedShares
+      );
+      await expect(await oeth.balanceOf(domen.address)).to.be.equal(
+        balanceBefore.sub(assetTransfered)
+      );
+    });
+    it("should withdraw at the correct ratio", async () => {
+      const { oeth, woeth, domen } = fixture;
+      // First wrap some OETH
+      await woeth.connect(domen).deposit(oethUnits("50"), domen.address);
+
+      const totalSupply = await woeth.totalSupply();
+      const balanceBefore = await oeth.balanceOf(domen.address);
+
+      // Then unwrap some WOETH
+      const txResponse = await woeth
+        .connect(domen)
+        .withdraw(
+          await woeth.maxWithdraw(domen.address),
+          domen.address,
+          domen.address
+        );
+      const txReceipt = await txResponse.wait();
+      const burnedShares = txReceipt.events[2].args.shares; // 0. transfer oeth, 1. transfer woeth, 2. withdraw
+      const assetTransfered = txReceipt.events[2].args.assets; // 0. transfer oeth, 1. transfer woeth, 2. mint
+
+      await expect(assetTransfered).to.be.approxEqual(oethUnits("50"));
+      await expect(
+        await woeth.convertToShares(assetTransfered)
+      ).to.be.approxEqual(burnedShares);
+      await expect(woeth).to.have.a.totalSupply(totalSupply.sub(burnedShares));
+      await expect(await woeth.balanceOf(domen.address)).to.be.equal(0);
+      await expect(await oeth.balanceOf(domen.address)).to.be.approxEqual(
+        balanceBefore.add(assetTransfered)
+      );
+    });
+    it("should mint at the correct ratio", async () => {
+      const { oeth, woeth, domen } = fixture;
+
+      const totalSupply = await woeth.totalSupply();
+      const balanceBefore = await oeth.balanceOf(domen.address);
+
+      // Mint some WOETH
+      const txResponse = await woeth
+        .connect(domen)
+        .mint(oethUnits("25"), domen.address);
+      const txReceipt = await txResponse.wait();
+      const mintedShares = txReceipt.events[2].args.shares; // 0. transfer oeth, 1. transfer woeth, 2. mint
+      const assetTransfered = txReceipt.events[2].args.assets; // 0. transfer oeth, 1. transfer woeth, 2. mint
+
+      await expect(mintedShares).to.be.equal(oethUnits("25"));
+      await expect(await woeth.convertToAssets(mintedShares)).to.be.approxEqual(
+        assetTransfered
+      );
+      await expect(woeth).to.have.a.totalSupply(totalSupply.add(mintedShares));
+      await expect(await woeth.balanceOf(domen.address)).to.be.equal(
+        mintedShares
+      );
+      await expect(await oeth.balanceOf(domen.address)).to.be.equal(
+        balanceBefore.sub(assetTransfered)
+      );
+    });
+    it("should redeem at the correct ratio", async () => {
+      const { oeth, woeth, domen } = fixture;
+
+      // Mint some WOETH
+      await woeth.connect(domen).mint(oethUnits("25"), domen.address);
+
+      const totalSupply = await woeth.totalSupply();
+      const balanceBefore = await oeth.balanceOf(domen.address);
+
+      // Redeem some WOETH
+      const txResponse = await woeth
+        .connect(domen)
+        .redeem(
+          await woeth.maxRedeem(domen.address),
+          domen.address,
+          domen.address
+        );
+      const txReceipt = await txResponse.wait();
+      const burnedShares = txReceipt.events[2].args.shares; // 0. transfer oeth, 1. transfer woeth, 2. redeem
+      const assetTransfered = txReceipt.events[2].args.assets; // 0. transfer oeth, 1. transfer woeth, 2. redeem
+
+      await expect(burnedShares).to.be.equal(oethUnits("25"));
+      await expect(await woeth.convertToAssets(burnedShares)).to.be.approxEqual(
+        assetTransfered
+      );
+      await expect(woeth).to.have.a.totalSupply(totalSupply.sub(burnedShares));
+      await expect(await woeth.balanceOf(domen.address)).to.be.equal(0);
+      await expect(await oeth.balanceOf(domen.address)).to.be.approxEqual(
+        balanceBefore.add(assetTransfered)
+      );
+    });
+    it("should redeem at the correct ratio after rebase", async () => {
+      const { weth, oethVault, woeth, domen, josh } = fixture;
+
+      // Mint some WOETH
+      const initialDeposit = oethUnits("50");
+      await woeth.connect(domen).deposit(initialDeposit, domen.address);
+
+      const totalAssetsBefore = await woeth.totalAssets();
+      // Rebase
+      await hardhatSetBalance(josh.address, "250");
+      await weth.connect(josh).deposit({ value: oethUnits("200") });
+      await weth.connect(josh).transfer(oethVault.address, oethUnits("200"));
+      await oethVault.rebase();
+
+      const totalAssetsAfter = await woeth.totalAssets();
+      expect(totalAssetsAfter > totalAssetsBefore).to.be.true;
+
+      // Then unwrap some WOETH
+      const txResponse = await woeth
+        .connect(domen)
+        .redeem(
+          await woeth.maxRedeem(domen.address),
+          domen.address,
+          domen.address
+        );
+
+      const txReceipt = await txResponse.wait();
+      const burnedShares = txReceipt.events[2].args.shares; // 0. transfer oeth, 1. transfer woeth, 2. redeem
+      const assetTransfered = txReceipt.events[2].args.assets; // 0. transfer oeth, 1. transfer woeth, 2. redeem
+
+      await expect(assetTransfered > initialDeposit);
+      await expect(burnedShares).to.be.approxEqual(
+        await woeth.convertToShares(assetTransfered)
+      );
+      await expect(domen).to.have.a.balanceOf("0", woeth);
+    });
+  });
+});