From c26d09540df8ccdeb66d37af9bf87435f4dbd6de Mon Sep 17 00:00:00 2001
From: Danny Ryan <dannyjryan@gmail.com>
Date: Wed, 20 Mar 2019 08:51:40 -0600
Subject: [PATCH] port header tests

---
 .../test_process_block_header.py              | 68 +++++++++++++++++--
 1 file changed, 63 insertions(+), 5 deletions(-)

diff --git a/tests/phase0/block_processing/test_process_block_header.py b/tests/phase0/block_processing/test_process_block_header.py
index 4ec7e336f..8998a6e8f 100644
--- a/tests/phase0/block_processing/test_process_block_header.py
+++ b/tests/phase0/block_processing/test_process_block_header.py
@@ -3,7 +3,8 @@ import pytest
 
 
 from build.phase0.spec import (
-    get_beacon_proposer_index,
+    cache_state,
+    advance_slot,
     process_block_header,
 )
 from tests.phase0.helpers import (
@@ -14,13 +15,70 @@ from tests.phase0.helpers import (
 pytestmark = pytest.mark.header
 
 
-def test_proposer_slashed(state):
+def test_sucess(state):
+    pre_state = deepcopy(state)
+    block = build_empty_block_for_next_slot(pre_state)
+
+    #
+    # setup pre_state to be ready for block transition
+    #
+    cache_state(pre_state)
+    advance_slot(pre_state)
+
+    post_state = deepcopy(pre_state)
+
+    #
+    # test block header
+    #
+    process_block_header(post_state, block)
+
+    return state, [block], post_state
+
+
+def test_invalid_slot(state):
     pre_state = deepcopy(state)
 
+    # mess up previous block root
     block = build_empty_block_for_next_slot(pre_state)
-    proposer_index = get_beacon_proposer_index(pre_state, block.slot)
-    pre_state.validator_registry[proposer_index].slashed = True
+    block.previous_block_root = b'\12'*32
+
+    #
+    # setup pre_state advancing two slots to induce error
+    #
+    cache_state(pre_state)
+    advance_slot(pre_state)
+    advance_slot(pre_state)
+
+    post_state = deepcopy(pre_state)
+
+    #
+    # test block header
+    #
     with pytest.raises(AssertionError):
-        process_block_header(pre_state, block)
+        process_block_header(post_state, block)
+
+    return state, [block], None
+
+
+def test_invalid_previous_block_root(state):
+    pre_state = deepcopy(state)
+
+    # mess up previous block root
+    block = build_empty_block_for_next_slot(pre_state)
+    block.previous_block_root = b'\12'*32
+
+    #
+    # setup pre_state to be ready for block transition
+    #
+    cache_state(pre_state)
+    advance_slot(pre_state)
+
+    post_state = deepcopy(pre_state)
+
+    #
+    # test block header
+    #
+    with pytest.raises(AssertionError):
+        process_block_header(post_state, block)
 
     return state, [block], None