From 26155149096c90ad84eef13211b072c0633aacc4 Mon Sep 17 00:00:00 2001
From: Agnish Ghosh <iamagnix@gmail.com>
Date: Thu, 13 Jun 2024 16:19:15 +0530
Subject: [PATCH] add: testing init for peerdas eip

---
 beacon_chain/spec/datatypes/eip7594.nim |  2 +-
 beacon_chain/spec/eip7594_helpers.nim   | 24 ++++------
 tests/all_tests.nim                     |  1 +
 tests/test_eip7594_helpers.nim          | 59 +++++++++++++++++++++++++
 4 files changed, 70 insertions(+), 16 deletions(-)
 create mode 100644 tests/test_eip7594_helpers.nim

diff --git a/beacon_chain/spec/datatypes/eip7594.nim b/beacon_chain/spec/datatypes/eip7594.nim
index cfdafefd7..b39a5190c 100644
--- a/beacon_chain/spec/datatypes/eip7594.nim
+++ b/beacon_chain/spec/datatypes/eip7594.nim
@@ -9,7 +9,7 @@
 
 import "."/[base, deneb], kzg4844
 
-export base, kzg4844
+export base
 
 const
   FIELD_ELEMENTS_PER_EXT_BLOB* = 2 * kzg_abi.FIELD_ELEMENTS_PER_BLOB
diff --git a/beacon_chain/spec/eip7594_helpers.nim b/beacon_chain/spec/eip7594_helpers.nim
index bc09fdebf..524ea6d3c 100644
--- a/beacon_chain/spec/eip7594_helpers.nim
+++ b/beacon_chain/spec/eip7594_helpers.nim
@@ -14,6 +14,7 @@ import
   ssz_serialization/proofs,
   chronicles,
   ./[beacon_time, crypto],
+  kzg4844/kzg_ex,
   eth/p2p/discoveryv5/[node],
   ./helpers,
   ./datatypes/[eip7594, deneb]
@@ -67,14 +68,15 @@ proc get_custody_columns*(node_id: NodeId, custody_subnet_count: uint64): Result
 proc compute_extended_matrix* (blobs: seq[KzgBlob]): Result[ExtendedMatrix, cstring] =
   # This helper demonstrates the relationship between blobs and `ExtendedMatrix`
   var extended_matrix: ExtendedMatrix
-  for blob in blobs:
-    let res = computeCells(blob)
-
+  for i in 0..<blobs.len:
+    debugEcho "Checkpoint 1"
+    let res = computeCells(blobs[i])
+    debugEcho "Checkpoint 2"
     if res.isErr:
         return err("Error computing kzg cells and kzg proofs")
-
+    debugEcho "Checkpoint 3"
     discard extended_matrix.add(res.get())
-
+    debugEcho "Checkpoint 4"
   ok(extended_matrix)
 
 # https://github.com/ethereum/consensus-specs/blob/5f48840f4d768bf0e0a8156a3ed06ec333589007/specs/_features/eip7594/das-core.md#recover_matrix    
@@ -117,7 +119,7 @@ proc get_data_column_sidecars*(signed_block: deneb.SignedBeaconBlock, blobs: seq
   var signed_block_header: deneb.SignedBeaconBlockHeader
   var blck = signed_block.message
 
-  var cellsAndProofs: seq[KzgCellsAndKzgProofs] = @[]
+  var cellsAndProofs: seq[KzgCellsAndKzgProofs]
 
   for blob in blobs:
     let
@@ -137,7 +139,7 @@ proc get_data_column_sidecars*(signed_block: deneb.SignedBeaconBlock, blobs: seq
     cells[i].add(cellsAndProofs[i].cells[0])
     proofs[i].add(cellsAndProofs[i].proofs[1])
 
-  var sidecars: seq[DataColumnSidecar] = @[]
+  var sidecars: seq[DataColumnSidecar]
 
   for columnIndex in 0..<NUMBER_OF_COLUMNS:
     var column: DataColumn
@@ -206,14 +208,6 @@ proc verify_data_column_sidecar_kzg_proofs*(sidecar: DataColumnSidecar): Result[
 
   let res = verifyCellKzgProofBatch(kzgCommits, rowIndices, colIndices, sidecarCol, kzgProofs)
 
-  # # KZG batch verifies that the cells match the corresponding commitments and KZG proofs
-  # let res = validate_data_column_sidecar(
-  #   kzgCommits,
-  #   rowIndices,
-  #   colIndices,
-  #   sidecarCol,
-  #   kzgProofs)
-
   if res.isErr():
     return err("DataColumnSidecar: validation failed")
 
diff --git a/tests/all_tests.nim b/tests/all_tests.nim
index 18f698da6..7acb870d6 100644
--- a/tests/all_tests.nim
+++ b/tests/all_tests.nim
@@ -26,6 +26,7 @@ import # Unit test
   ./test_datatypes,
   ./test_deposit_snapshots,
   ./test_discovery,
+  ./test_eip7594_helpers,
   ./test_engine_authentication,
   ./test_el_manager,
   ./test_el_conf,
diff --git a/tests/test_eip7594_helpers.nim b/tests/test_eip7594_helpers.nim
new file mode 100644
index 000000000..f1d2245c8
--- /dev/null
+++ b/tests/test_eip7594_helpers.nim
@@ -0,0 +1,59 @@
+# 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.
+
+{.push raises: [].}
+{.used.}
+
+# Uncategorized helper functions from the spec
+import
+  unittest2,
+  random,
+  std/[algorithm, macros, tables, sysrand],
+  stew/results,
+  stint,
+  kzg4844/[kzg_abi, kzg_ex],
+  ssz_serialization/proofs,
+  chronicles,
+  ../beacon_chain/spec/beacon_time,
+  eth/p2p/discoveryv5/[node],
+  ./consensus_spec/[os_ops, fixtures_utils],
+  ../beacon_chain/spec/[helpers, eip7594_helpers],
+  ../beacon_chain/spec/datatypes/[eip7594, deneb]
+
+from std/sequtils import anyIt, mapIt, toSeq
+from std/strutils import rsplit
+
+block:
+  template sourceDir: string = currentSourcePath.rsplit(DirSep, 1)[0]
+  doAssert Kzg.loadTrustedSetup(
+    sourceDir &
+      "/../vendor/nim-kzg4844/kzg4844/csources/src/trusted_setup.txt").isOk
+
+const MAX_TOP_BYTE = 114
+
+proc createSampleKzgBlobs(n: int): Result[seq[KzgBlob], cstring] =
+  var blob: KzgBlob
+  var blobs: seq[KzgBlob]
+  for i in 0..<n:
+    discard urandom(blob)
+    for i in 0..<blob.len:
+      if blob[i] > MAX_TOP_BYTE and i %% kzg_abi.BYTES_PER_FIELD_ELEMENT == 0:
+        blob[i] = MAX_TOP_BYTE
+    blobs.add(blob)
+
+  ok(blobs)
+
+suite "EIP-7594 Unit Tests":
+  test "EIP-7594: Compute Extended Matrix":
+    proc testComputeExtendedMatrix() =
+      let blob_count = 2
+      let input_blobs = createSampleKzgBlobs(blob_count)
+      let extended_matrix = compute_extended_matrix(input_blobs.get)
+
+      doAssert extended_matrix.get.len == kzg_abi.CELLS_PER_EXT_BLOB * blob_count
+
+    testComputeExtendedMatrix()
\ No newline at end of file