Add liquidtyPoolFactory file and add migration test

centrifuge · Nov 14, 2023 · c21366e · c21366e
1 parent 149494f
commit c21366e
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 1 deletion.
diff --git a/src/PoolManager.sol b/src/PoolManager.sol
@@ -85,7 +85,7 @@ contract PoolManager is Auth {
     uint8 internal constant MAX_DECIMALS = 18;
 
     EscrowLike public immutable escrow;
-    LiquidityPoolFactoryLike public immutable liquidityPoolFactory;
+    LiquidityPoolFactoryLike public liquidityPoolFactory;
     TrancheTokenFactoryLike public immutable trancheTokenFactory;
 
     GatewayLike public gateway;
@@ -151,6 +151,7 @@ contract PoolManager is Auth {
         if (what == "gateway") gateway = GatewayLike(data);
         else if (what == "investmentManager") investmentManager = InvestmentManagerLike(data);
         else if (what == "restrictionManagerFactory") restrictionManagerFactory = RestrictionManagerFactoryLike(data);
+        else if (what == "liquidityPoolFactory") liquidityPoolFactory = LiquidityPoolFactoryLike(data);
         else revert("PoolManager/file-unrecognized-param");
         emit File(what, data);
     }

diff --git a/test/PoolManager.t.sol b/test/PoolManager.t.sol
@@ -35,6 +35,10 @@ contract PoolManagerTest is TestSetup {
         poolManager.file("restrictionManagerFactory", newRestrictionManagerFactory);
         assertEq(address(poolManager.restrictionManagerFactory()), newRestrictionManagerFactory);
 
+        address newLiquidityPoolFactory = makeAddr("newLiquidityPoolFactory");
+        poolManager.file("liquidityPoolFactory", newLiquidityPoolFactory);
+        assertEq(address(poolManager.liquidityPoolFactory()), newLiquidityPoolFactory);
+
         address newEscrow = makeAddr("newEscrow");
         vm.expectRevert("PoolManager/file-unrecognized-param");
         poolManager.file("escrow", newEscrow);
@@ -549,6 +553,30 @@ contract PoolManagerTest is TestSetup {
         poolManager.removeLiquidityPool(poolId, trancheId, address(erc20));
     }
 
+    function testLiquidityPoolMigration() public {
+        address oldLiquidityPool_ = deploySimplePool();
+
+        LiquidityPool oldLiquidityPool =  LiquidityPool(oldLiquidityPool_);
+        uint64 poolId = oldLiquidityPool.poolId();
+        bytes16 trancheId = oldLiquidityPool.trancheId();
+        address currency = address(oldLiquidityPool.asset());
+
+        LiquidityPoolFactory newLiquidityPoolFactory = new LiquidityPoolFactory(address(root));
+
+        // rewire factory contracts
+        newLiquidityPoolFactory.rely(address(poolManager));
+        poolManager.file("liquidityPoolFactory", address(newLiquidityPoolFactory));
+
+        // Remove old liquidity pool
+        root.relyContract(address(poolManager), address(this));
+        poolManager.removeLiquidityPool(poolId, trancheId, currency);
+        assertEq(poolManager.getLiquidityPool(poolId, trancheId, currency), address(0));
+
+        // Deploy new liquidity pool
+        address newLiquidityPool = poolManager.deployLiquidityPool(poolId, trancheId, currency);
+        assertEq(poolManager.getLiquidityPool(poolId, trancheId, currency), newLiquidityPool);
+    }
+
     // helpers
     function hasDuplicates(bytes16[4] calldata array) internal pure returns (bool) {
         uint256 length = array.length;