.

taikoxyz · May 9, 2023 · ef1c54c · ef1c54c
1 parent fc15d8c
commit ef1c54c
Showing 1 changed file with 30 additions and 1 deletion.
diff --git a/packages/protocol/test/genesis/GenerateGenesis.g.sol b/packages/protocol/test/genesis/GenerateGenesis.g.sol
@@ -32,6 +32,8 @@ contract TestGenerateGenesis is Test, AddressResolver {
             string.concat(vm.projectRoot(), "/deployments/genesis_alloc.json")
         );
     address private owner = configJSON.readAddress(".contractOwner");
+    address private admin = configJSON.readAddress(".contractOwner");
+
     uint64 public constant BLOCK_GAS_LIMIT = 30000000;
 
     function testContractDeployment() public {
@@ -46,12 +48,22 @@ contract TestGenerateGenesis is Test, AddressResolver {
         checkDeployedCode("ProxiedSignalService");
 
         // check proxies
-        checkDeployedCode("TaikoL1Proxy");
+        checkDeployedCode("TaikoL2Proxy");
         checkDeployedCode("TokenVaultProxy");
         checkDeployedCode("EtherVaultProxy");
         checkDeployedCode("BridgeProxy");
         checkDeployedCode("AddressManagerProxy");
         checkDeployedCode("SignalServiceProxy");
+
+        checkProxyImplementation("TaikoL2Proxy", "ProxiedTaikoL2");
+        checkProxyImplementation("TokenVaultProxy", "ProxiedTokenVault");
+        checkProxyImplementation("EtherVaultProxy", "ProxiedEtherVault");
+        checkProxyImplementation("BridgeProxy", "ProxiedBridge");
+        checkProxyImplementation(
+            "AddressManagerProxy",
+            "ProxiedAddressManager"
+        );
+        checkProxyImplementation("SignalServiceProxy", "ProxiedSignalService");
     }
 
     function testAddressManager() public {
@@ -210,6 +222,23 @@ contract TestGenerateGenesis is Test, AddressResolver {
         assertEq(address(contractAddress).code, vm.parseBytes(deployedCode));
     }
 
+    function checkProxyImplementation(
+        string memory proxyName,
+        string memory contractName
+    ) private {
+        address contractAddress = getPredeployedContractAddress(contractName);
+        address proxyAddress = getPredeployedContractAddress(proxyName);
+        assertEq(
+            TransparentUpgradeableProxy(payable(proxyAddress)).implementation(),
+            address(contractAddress)
+        );
+
+        assertEq(
+            TransparentUpgradeableProxy(payable(proxyAddress)).admin(),
+            owner
+        );
+    }
+
     function checkSavedAddress(
         AddressManager addressManager,
         string memory contractName,