Skip to content

Commit 8bfef33

Browse files
Apply suggestions from code review: update enums
Co-Authored-By: hwwhww <[email protected]>
1 parent dfdde18 commit 8bfef33

File tree

3 files changed

+7
-7
lines changed

3 files changed

+7
-7
lines changed

eth/beacon/enums.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
class ValidatorStatusCode(IntEnum):
55
PENDING_ACTIVATION = 0
66
ACTIVE = 1
7-
PENDING_EXIT = 2
7+
ACTIVE_PENDING_EXIT = 2
88
EXITED_WITHOUT_PENALTY = 3
99
EXITED_WITH_PENALTY = 4
1010

@@ -18,4 +18,4 @@ class SignatureDomain(IntEnum):
1818
DOMAIN_DEPOSIT = 0
1919
DOMAIN_ATTESTATION = 1
2020
DOMAIN_PROPOSAL = 2
21-
DOMAIN_LOGOUT = 3
21+
DOMAIN_EXIT = 3

eth/beacon/helpers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -156,7 +156,7 @@ def get_active_validator_indices(validators: Sequence['ValidatorRecord']) -> Tup
156156
"""
157157
return tuple(
158158
i for i, v in enumerate(validators)
159-
if v.status in [ValidatorStatusCode.ACTIVE, ValidatorStatusCode.PENDING_EXIT]
159+
if v.status in [ValidatorStatusCode.ACTIVE, ValidatorStatusCode.ACTIVE_PENDING_EXIT]
160160
)
161161

162162

tests/beacon/test_helpers.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -503,14 +503,14 @@ def test_get_active_validator_indices(sample_validator_record_params):
503503
active_validator_indices = get_active_validator_indices(validators)
504504
assert len(active_validator_indices) == 3
505505

506-
# Make one validator becomes PENDING_EXIT.
506+
# Make one validator becomes ACTIVE_PENDING_EXIT.
507507
validators[0] = validators[0].copy(
508-
status=ValidatorStatusCode.PENDING_EXIT,
508+
status=ValidatorStatusCode.ACTIVE_PENDING_EXIT,
509509
)
510510
active_validator_indices = get_active_validator_indices(validators)
511511
assert len(active_validator_indices) == 3
512512

513-
# Make one validator becomes PENDING_EXIT.
513+
# Make one validator becomes EXITED_WITHOUT_PENALTY.
514514
validators[0] = validators[0].copy(
515515
status=ValidatorStatusCode.EXITED_WITHOUT_PENALTY,
516516
)
@@ -677,7 +677,7 @@ def test_get_new_validator_registry_delta_chain_tip(index,
677677
[
678678
(0, 0, 0, 0, 0),
679679
(0, 0, 0, 1, 0),
680-
(0, 0, 20, 10, 0),
680+
(0, 1, 20, 10, 0),
681681
(0, 1, 20, 20, 1),
682682
(0, 1, 10, 20, 1),
683683
]

0 commit comments

Comments
 (0)