diff --git a/test/allocators/RariFuseAllocator.test.ts b/test/allocators/RariFuseAllocator.test.ts index e83e520a3..f4c80ba24 100644 --- a/test/allocators/RariFuseAllocator.test.ts +++ b/test/allocators/RariFuseAllocator.test.ts @@ -501,13 +501,26 @@ describe(ALLOCATORN, () => { } // migrate + + let fallocator: RariFuseAllocator; async function beforeEachMigrateTest(): Promise { + for (let i = 1; i < 4; i++) { + await allocator.connect(guardian).update(i); + } + + await helpers.tmine(24 * 3600 * 43); + + for (let i = 1; i < 4; i++) { + await allocator.connect(guardian).update(i); + } + await allocator.connect(guardian).prepareMigration(); - let fallocator = await factory.connect(owner).deploy(FAID); + fallocator = (await factory.connect(owner).deploy(FAID)) as RariFuseAllocator; await fallocator.connect(guardian).fusePoolAdd(troller.address); + await fallocator.connect(guardian).fDataAdd(FDA[0]); await extender.connect(guardian).registerDeposit(fallocator.address); @@ -518,12 +531,25 @@ describe(ALLOCATORN, () => { async function migrateRevertingTests(): Promise { it("revert: should fail with wrong access ", async () => { await expect(allocator.connect(owner).migrate()).to.be.reverted; + for (let i = 1; i < 4; i++) { + await expect(allocator.connect(guardian).update(i)).to.be.reverted; + } }); } async function migratePassingTests(): Promise { it("passing: should migrate", async () => { - await allocator.connect(guardian).migrate(); + let balances: BigNumber[] = []; + + for ( let i = 0; i < 3; i++) { + balances.push(await utility[i].balanceOf(fallocator.address)); + } + + await allocator.connect(guardian).migrate() + + for ( let i = 0; i < 3; i++) { + expect(await utility[i].balanceOf(fallocator.address)).to.be.above(balances[i]); + } }); }