From 8d5d7387f959e6393da5b29684d52ae04e64722f Mon Sep 17 00:00:00 2001
From: protolambda <proto@protolambda.com>
Date: Thu, 12 Dec 2019 02:36:20 +0100
Subject: [PATCH] Two tests for proposer indices being off because of active
 validators status, fixes #1515

---
 .../eth2spec/test/sanity/test_blocks.py       | 52 ++++++++++++++++++-
 1 file changed, 51 insertions(+), 1 deletion(-)

diff --git a/test_libs/pyspec/eth2spec/test/sanity/test_blocks.py b/test_libs/pyspec/eth2spec/test/sanity/test_blocks.py
index 41316e92d..3c42446cb 100644
--- a/test_libs/pyspec/eth2spec/test/sanity/test_blocks.py
+++ b/test_libs/pyspec/eth2spec/test/sanity/test_blocks.py
@@ -3,7 +3,7 @@ from copy import deepcopy
 from eth2spec.utils.ssz.ssz_impl import hash_tree_root
 from eth2spec.utils.bls import bls_sign
 
-from eth2spec.test.helpers.state import get_balance, state_transition_and_sign_block
+from eth2spec.test.helpers.state import get_balance, state_transition_and_sign_block, next_slot
 from eth2spec.test.helpers.block import build_empty_block_for_next_slot, build_empty_block, sign_block, \
     transition_unsigned_block
 from eth2spec.test.helpers.keys import privkeys, pubkeys
@@ -253,6 +253,56 @@ def test_attester_slashing(spec, state):
     )
 
 
+@with_all_phases
+@spec_state_test
+def test_proposer_after_inactive_index(spec, state):
+    # disable some low validator index to check after for
+    inactive_index = 10
+    state.validators[inactive_index].exit_epoch = spec.get_current_epoch(state)
+
+    # skip forward, get brand new proposers
+    state.slot = spec.SLOTS_PER_EPOCH * 2
+
+    while True:
+        proposer_index = spec.get_beacon_proposer_index(state)
+        if proposer_index > inactive_index:
+            # found a proposer that has a higher index than a disabled validator
+            yield 'pre', state
+            # test if the proposer can be recognized correctly after the inactive validator
+            signed_block = sign_block(spec, state, build_empty_block(spec, state), proposer_index=proposer_index)
+            yield 'blocks', [signed_block]
+            yield 'post', state
+            break
+        else:
+            next_slot(spec, state)
+
+
+@with_all_phases
+@spec_state_test
+def test_high_proposer_index(spec, state):
+    # disable a good amount of validators to make the active count lower, for a faster test
+    current_epoch = spec.get_current_epoch(state)
+    for i in range(len(state.validators) // 3):
+        state.validators[i].exit_epoch = current_epoch
+
+    # skip forward, get brand new proposers
+    state.slot = spec.SLOTS_PER_EPOCH * 2
+
+    active_count = len(spec.get_active_validator_indices(state, current_epoch))
+    while True:
+        proposer_index = spec.get_beacon_proposer_index(state)
+        if proposer_index >= active_count:
+            # found a proposer that has a higher index than the active validator count
+            yield 'pre', state
+            # test if the proposer can be recognized correctly, even while it has a high index.
+            signed_block = sign_block(spec, state, build_empty_block(spec, state), proposer_index=proposer_index)
+            yield 'blocks', [signed_block]
+            yield 'post', state
+            break
+        else:
+            next_slot(spec, state)
+
+
 @with_all_phases
 @spec_state_test
 def test_expected_deposit_in_block(spec, state):