pr review fix

alchemyplatform · Dec 21, 2023 · ec21a90 · ec21a90
1 parent 88fc999
commit ec21a90
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/test/factory/MultiOwnerMSCAFactoryTest.t.sol b/test/factory/MultiOwnerMSCAFactoryTest.t.sol
@@ -130,6 +130,15 @@ contract MultiOwnerMSCAFactoryTest is Test {
         assertEq(address(factory).balance, 0);
     }
 
+    function test_2StepOwnershipTransfer() public {
+        assertEq(factory.owner(), address(this));
+        factory.transferOwnership(owner1);
+        assertEq(factory.owner(), address(this));
+        vm.prank(owner1);
+        factory.acceptOwnership();
+        assertEq(factory.owner(), owner1);
+    }
+
     // to receive funds from withdraw
     receive() external payable {}
 }
diff --git a/test/factory/MultiOwnerTokenReceiverFactoryTest.t.sol b/test/factory/MultiOwnerTokenReceiverFactoryTest.t.sol
@@ -186,6 +186,15 @@ contract MultiOwnerTokenReceiverMSCAFactoryTest is Test {
         assertEq(address(factory).balance, 0);
     }
 
+    function test_2StepOwnershipTransfer() public {
+        assertEq(factory.owner(), address(this));
+        factory.transferOwnership(owner1);
+        assertEq(factory.owner(), address(this));
+        vm.prank(owner1);
+        factory.acceptOwnership();
+        assertEq(factory.owner(), owner1);
+    }
+
     // to receive funds from withdraw
     receive() external payable {}
 }