Make tranche token factory fileable (#224)

* Make tranche token factory fileable * IERC20Metadata => IERC20 * Revert erc20 interfaces * Undo interface change again
centrifuge · Dec 19, 2023 · 333cc86 · 333cc86
1 parent 2f07d31
commit 333cc86
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/src/PoolManager.sol b/src/PoolManager.sol
@@ -87,10 +87,10 @@ contract PoolManager is Auth {
     uint8 internal constant MAX_DECIMALS = 18;
 
     EscrowLike public immutable escrow;
-    TrancheTokenFactoryLike public immutable trancheTokenFactory;
 
     GatewayLike public gateway;
     InvestmentManagerLike public investmentManager;
+    TrancheTokenFactoryLike public trancheTokenFactory;
     LiquidityPoolFactoryLike public liquidityPoolFactory;
     RestrictionManagerFactoryLike public restrictionManagerFactory;
 
@@ -155,8 +155,9 @@ contract PoolManager is Auth {
     function file(bytes32 what, address data) external auth {
         if (what == "gateway") gateway = GatewayLike(data);
         else if (what == "investmentManager") investmentManager = InvestmentManagerLike(data);
-        else if (what == "restrictionManagerFactory") restrictionManagerFactory = RestrictionManagerFactoryLike(data);
+        else if (what == "trancheTokenFactory") trancheTokenFactory = TrancheTokenFactoryLike(data);
         else if (what == "liquidityPoolFactory") liquidityPoolFactory = LiquidityPoolFactoryLike(data);
+        else if (what == "restrictionManagerFactory") restrictionManagerFactory = RestrictionManagerFactoryLike(data);
         else revert("PoolManager/file-unrecognized-param");
         emit File(what, data);
     }

diff --git a/test/PoolManager.t.sol b/test/PoolManager.t.sol
@@ -34,6 +34,10 @@ contract PoolManagerTest is TestSetup {
         poolManager.file("investmentManager", newInvestmentManager);
         assertEq(address(poolManager.investmentManager()), newInvestmentManager);
 
+        address newTrancheTokenFactory = makeAddr("newTrancheTokenFactory");
+        poolManager.file("trancheTokenFactory", newTrancheTokenFactory);
+        assertEq(address(poolManager.trancheTokenFactory()), newTrancheTokenFactory);
+
         address newRestrictionManagerFactory = makeAddr("newRestrictionManagerFactory");
         poolManager.file("restrictionManagerFactory", newRestrictionManagerFactory);
         assertEq(address(poolManager.restrictionManagerFactory()), newRestrictionManagerFactory);