From ac776926f1aab0cfbd470d4b5e04a0f782f9dce2 Mon Sep 17 00:00:00 2001
From: cheatfate <eugene.kabanov@status.im>
Date: Fri, 23 Feb 2024 20:53:56 +0200
Subject: [PATCH] Adding test which is not working.

---
 tests/test_block_rewards.nim | 109 +++++++++++++++++++++++++++++++++++
 1 file changed, 109 insertions(+)
 create mode 100644 tests/test_block_rewards.nim

diff --git a/tests/test_block_rewards.nim b/tests/test_block_rewards.nim
new file mode 100644
index 000000000..c233e6d6e
--- /dev/null
+++ b/tests/test_block_rewards.nim
@@ -0,0 +1,109 @@
+# beacon_chain
+# Copyright (c) 2018-2024 Status Research & Development GmbH
+# Licensed and distributed under either of
+#   * MIT license (license terms in the root directory or at https://opensource.org/licenses/MIT).
+#   * Apache v2 license (license terms in the root directory or at https://www.apache.org/licenses/LICENSE-2.0).
+# at your option. This file may not be copied, modified, or distributed except according to those terms.
+
+{.used.}
+
+import
+  chronicles,
+  ../../beacon_chain/spec/forks,
+  ../../beacon_chain/spec/state_transition,
+  ../../beacon_chain/validators/rewards,
+  ./consensus_spec/os_ops,
+  ./testutil
+
+from std/sequtils import toSeq
+from std/strutils import toLowerAscii
+from ../../beacon_chain/spec/presets import
+  const_preset, defaultRuntimeConfig
+from ./consensus_spec/fixtures_utils import
+  SSZ, SszTestsDir, hash_tree_root, parseTest, readSszBytes, toSszType
+
+proc runTest(consensusFork: static ConsensusFork,
+             testDir: static[string], suiteName, path: string) =
+  test "Block rewards test -" & preset():
+    echo "testName = ", testName
+    echo "path = ", path
+
+    when consensusFork == ConsensusFork.Phase0:
+      skip()
+    else:
+      let
+        testPath = testDir / path
+        preState = newClone(parseTest(testPath / "pre.ssz_snappy",
+                     SSZ, consensusFork.BeaconState))
+        blockPath = testPath / "blocks_0.ssz_snappy"
+
+      if not(fileExists(blockPath)):
+        discard
+      else:
+        var
+          fhPreState = ForkedHashedBeaconState.new(preState[])
+          cache = StateCache()
+          info = ForkedEpochInfo()
+
+        let
+          blck = parseTest(testPath/"blocks_0.ssz_snappy",
+                           SSZ, consensusFork.SignedBeaconBlock)
+          forkedBlock = ForkedBeaconBlock.init(blck.message)
+          consensusBlockValue = collectBlockRewards(fhPreState[], forkedBlock)
+
+        let (proposerIndex, preStateBalance, blockValue) =
+          withStateAndBlck(fhPreState[], forkedBlock):
+            (forkyBlck.proposer_index,
+             forkyState.data.balances.item(forkyBlck.proposer_index),
+             consensusBlockValue.get())
+
+        let res = state_transition(
+          defaultRuntimeConfig, fhPreState[], blck, cache, info, flags = {},
+          noRollback)
+
+        if res.isErr():
+          # Ignore failed states.
+          return
+
+        echo "proposer_index = ", proposerIndex
+        echo "before balance = ", preStateBalance
+        echo "block value = ", blockValue
+
+        let balance =
+          withState(fhPreState[]):
+            forkyState.data.balances.item(proposerIndex)
+
+        echo "after balance = ", balance
+
+template runForkBlockTests(consensusFork: static ConsensusFork) =
+  const
+    forkHumanName = $consensusFork
+    forkDirName = forkHumanName.toLowerAscii()
+    FinalityDir =
+      SszTestsDir/const_preset/forkDirName/"finality"/"finality"/"pyspec_tests"
+    RandomDir =
+      SszTestsDir/const_preset/forkDirName/"random"/"random"/"pyspec_tests"
+    SanityBlocksDir =
+      SszTestsDir/const_preset/forkDirName/"sanity"/"blocks"/"pyspec_tests"
+
+  suite "Consensus block value calculation - " & forkHumanName &
+          " - Sanity -" & preset():
+    for kind, path in walkDir(SanityBlocksDir, relative = true,
+                              checkDir = true):
+      consensusFork.runTest(SanityBlocksDir, suiteName, path)
+
+  suite "Consensus block value calculation - " & forkHumanName &
+          " - Finality -" & preset():
+    for kind, path in walkDir(FinalityDir, relative = true,
+                              checkDir = true):
+      consensusFork.runTest(FinalityDir, suiteName, path)
+
+  suite "Consensus block value calculation - " & forkHumanName &
+          " - Random -" & preset():
+    for kind, path in walkDir(RandomDir, relative = true,
+                              checkDir = true):
+      consensusFork.runTest(RandomDir, suiteName, path)
+
+
+withAll(ConsensusFork):
+  runForkBlockTests(consensusFork)