Testing direct return of depositBalance plus any donations

thesis · May 7, 2024 · c5e767e · c5e767e
1 parent bb42bce
commit c5e767e
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/solidity/test/MezoAllocator.test.ts b/solidity/test/MezoAllocator.test.ts
@@ -357,6 +357,8 @@ describe("MezoAllocator", () => {
     })
 
     context("when there is a deposit", () => {
+      beforeAfterSnapshotWrapper()
+
       before(async () => {
         await tbtc.mint(await stbtc.getAddress(), to1e18(5))
         await mezoAllocator.connect(maintainer).allocate()
@@ -372,6 +374,22 @@ describe("MezoAllocator", () => {
         expect(await mezoAllocator.totalAssets()).to.equal(depositBalance)
       })
     })
+
+    context("when there is a deposit plus 'donation' made", () => {
+      beforeAfterSnapshotWrapper()
+
+      before(async () => {
+        await tbtc.mint(await stbtc.getAddress(), to1e18(5))
+        await mezoAllocator.connect(maintainer).allocate()
+        // donation
+        await tbtc.mint(await mezoAllocator.getAddress(), to1e18(1))
+      })
+
+      it("should return the total assets value", async () => {
+        const totalAssets = await mezoAllocator.totalAssets()
+        expect(totalAssets).to.equal(to1e18(6))
+      })
+    })
   })
 
   describe("addMaintainer", () => {