Add denies

centrifuge · Jul 30, 2024 · 1c2bbbb · 1c2bbbb
1 parent f0e4945
commit 1c2bbbb
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/src/spell/MigrationSpell-mainnet.sol b/src/spell/MigrationSpell-mainnet.sol
@@ -81,8 +81,9 @@ contract MigrationSpell {
         IRoot rootOld = IRoot(address(ROOT_OLD));
         IRoot rootNew = IRoot(address(ROOT_NEW));
         IPoolManager poolManager = IPoolManager(address(POOLMANAGER));
-        IInvestmentManager investmentManagerOld = IInvestmentManager(address(INVESTMENTMANAGER_OLD));
         ITranche trancheTokenOld = ITranche(TRANCHE_TOKEN_OLD);
+        IInvestmentManager investmentManagerOld = IInvestmentManager(address(INVESTMENTMANAGER_OLD));
+        rootOld.relyContract(address(investmentManagerOld), self);
 
         // deploy new tranche token
         rootNew.relyContract(address(POOLMANAGER), self);
@@ -126,6 +127,11 @@ contract MigrationSpell {
         trancheTokenOld.file("symbol", SYMBOL_OLD);
 
         // denies
+        rootNew.denyContract(address(POOLMANAGER), self);
+        rootNew.denyContract(address(trancheTokenNew), self);
+        rootOld.denyContract(address(investmentManagerOld), self);
+        rootOld.deny(self);
+        rootNew.deny(self);
     }
 
     function getNumberOfMigratedMembers() public view returns (uint256) {

diff --git a/src/spell/MigrationSpell.t.sol b/src/spell/MigrationSpell.t.sol
@@ -73,6 +73,13 @@ contract MigrationSpellTest is Test {
         // assert renaming of old trancheToken worked
         assertEq(trancheTokenOld.name(), spell.NAME_OLD());
         assertEq(trancheTokenOld.symbol(), spell.SYMBOL_OLD());
+
+        // assert denies
+        assertEq(spell.POOLMANAGER().wards(address(spell)), 0);
+        assertEq(spell.trancheTokenNew().wards(address(spell)), 0);
+        assertEq(spell.INVESTMENTMANAGER_OLD().wards(address(spell)), 0);
+        assertEq(spell.ROOT_OLD().wards(address(spell)), 0);
+        assertEq(spell.ROOT_NEW().wards(address(spell)), 0);
     }
 
     function _loadDeployment(string memory folder, string memory name) internal {