test: implements unit tests

Author: 0xItadaki

test/unit/L2Resolver.t.sol

@@ -313,22 +313,36 @@ contract UnitL2Resolver is Test {
     _l2Resolver.setRecord(_nodeNameHash, _key, _chainIdBytes);
   }
 
-  function test_GetRecordWhenTheRecordExists(bytes32 _node, string calldata _recordKey, bytes memory _value) external {
+  function test_GetRecordWhenTheRecordExists(bytes32 _node, string calldata _recordKey, bytes memory _data) external {
     vm.assume(_node != _reverseLookupNode);
 
-    _mockAndExpect(address(_ensRegistry), abi.encodeWithSelector(IENS.owner.selector, _node), abi.encode(_nodeOwner));
-    vm.prank(_nodeOwner);
-    _l2Resolver.setRecord(_node, _recordKey, _value);
+    _setRecord(_node, _recordKey, _data);
 
     // it returns the stored bytes
-    assertEq(_l2Resolver.getRecord(_node, _recordKey), _value);
+    assertEq(_l2Resolver.getRecord(_node, _recordKey), _data);
   }
 
   function test_GetRecordWhenTheRecordDoesNotExist(bytes32 _node, string calldata _recordKey) external view {
     // it returns empty bytes
     assertEq(_l2Resolver.getRecord(_node, _recordKey), '');
   }
 
+  function test_ChainIdWhenTheChainIdentifierIsSet(bytes calldata _data) external {
+    _setRecord(_nodeNameHash, _key, _data);
+
+    // it returns the stored EIP-7930 bytes
+    assertEq(_l2Resolver.chainId(_nodeNameHash), _data);
+  }
+
+  function test_ChainIdWhenTheChainIdentifierIsNotSet(string calldata _recordKey, bytes calldata _data) external {
+    // sets the _data for a different key
+    vm.assume(keccak256(bytes(_recordKey)) != keccak256(bytes(_key)));
+    _setRecord(_nodeNameHash, _recordKey, _data);
+
+    // it returns empty bytes
+    assertEq(_l2Resolver.chainId(_nodeNameHash), '');
+  }
+
   /**
    * @notice Sets up a mock and expects a call to it
    *
@@ -340,4 +354,17 @@ contract UnitL2Resolver is Test {
     vm.mockCall(_receiver, _calldata, _returned);
     vm.expectCall(_receiver, _calldata);
   }
+
+  /**
+   * @notice Sets a record for a given node
+   * @dev Mocks the _nodeOwner as the owner of the _node in the ENSRegistry
+   * @param _node The ENS node to set the record for
+   * @param _recordKey The key to set the record for
+   * @param _data The data to set the record for
+   */
+  function _setRecord(bytes32 _node, string memory _recordKey, bytes memory _data) internal {
+    _mockAndExpect(address(_ensRegistry), abi.encodeWithSelector(IENS.owner.selector, _node), abi.encode(_nodeOwner));
+    vm.prank(_nodeOwner);
+    _l2Resolver.setRecord(_node, _recordKey, _data);
+  }
 }

test/unit/L2Resolver.tree

@@ -44,3 +44,9 @@ UnitL2Resolver::getRecord
 │   └── it returns the stored bytes
 └── when the record does not exist
     └── it returns empty bytes
+
+UnitL2Resolver::chainId
+├── when the chain identifier is set
+│   └── it returns the stored EIP-7930 bytes
+└── when the chain identifier is not set
+    └── it returns empty bytes