Merge pull request #1634 from Bhargavasomu/validator_registry_delta_block

Validator registry delta block

Author: hwwhww

eth/beacon/types/validator_registry_delta_block.py

@@ -0,0 +1,52 @@
+from eth_typing import (
+    Hash32,
+)
+import rlp
+
+from eth.beacon.utils.hash import (
+    hash_eth2,
+)
+from eth.rlp.sedes import (
+    hash32,
+    uint24,
+    uint64,
+    uint384,
+)
+
+
+class ValidatorRegistryDeltaBlock(rlp.Serializable):
+    """
+    Note: using RLP until we have standardized serialization format.
+    """
+    fields = [
+        ('latest_registry_delta_root', hash32),
+        ('validator_index', uint24),
+        ('pubkey', uint384),
+        ('flag', uint64)
+    ]
+
+    def __init__(self,
+                 latest_registry_delta_root: Hash32,
+                 validator_index: int,
+                 pubkey: int,
+                 flag: int) -> None:
+        super().__init__(
+            latest_registry_delta_root=latest_registry_delta_root,
+            validator_index=validator_index,
+            pubkey=pubkey,
+            flag=flag,
+        )
+
+    _hash = None
+
+    @property
+    def hash(self) -> Hash32:
+        if self._hash is None:
+            self._hash = hash_eth2(rlp.encode(self))
+        return self._hash
+
+    @property
+    def root(self) -> Hash32:
+        # Alias of `hash`.
+        # Using flat hash, might change to SSZ tree hash.
+        return self.hash

eth/rlp/sedes.py

@@ -15,4 +15,5 @@
 uint32 = BigEndianInt(32)
 uint64 = BigEndianInt(64)
 uint256 = BigEndianInt(256)
+uint384 = BigEndianInt(384)
 trie_root = Binary.fixed_length(32, allow_empty=True)

tests/beacon/conftest.py

@@ -294,6 +294,16 @@ def sample_validator_record_params():
     }
 
 
+@pytest.fixture
+def sample_validator_registry_delta_block_params():
+    return {
+        'latest_registry_delta_root': b'\x01' * 32,
+        'validator_index': 1,
+        'pubkey': 123,
+        'flag': 1,
+    }
+
+
 #
 # Temporary default values
 #

tests/beacon/types/test_validator_registry_delta_block.py

@@ -0,0 +1,13 @@
+from eth.beacon.types.validator_registry_delta_block import (
+    ValidatorRegistryDeltaBlock,
+)
+
+
+def test_defaults(sample_validator_registry_delta_block_params):
+    validator_registry_delta_block = ValidatorRegistryDeltaBlock(
+        **sample_validator_registry_delta_block_params
+    )
+    assert validator_registry_delta_block.latest_registry_delta_root == sample_validator_registry_delta_block_params['latest_registry_delta_root']  # noqa: E501
+    assert validator_registry_delta_block.validator_index == sample_validator_registry_delta_block_params['validator_index']  # noqa: E501
+    assert validator_registry_delta_block.pubkey == sample_validator_registry_delta_block_params['pubkey']  # noqa: E501
+    assert validator_registry_delta_block.flag == sample_validator_registry_delta_block_params['flag']  # noqa: E501