From 57f678bbf322c412d09b7ea5530fd3f2801cd196 Mon Sep 17 00:00:00 2001
From: benbierens <thatbenbierens@gmail.com>
Date: Fri, 17 Nov 2023 16:35:29 +0100
Subject: [PATCH] Setting up testfixture for proof datasampler

---
 tests/codex/proof/testdatasampler.nim | 47 +++++++++++++++++++++++++++
 tests/codex/testproof.nim             |  3 ++
 tests/testCodex.nim                   |  1 +
 3 files changed, 51 insertions(+)
 create mode 100644 tests/codex/proof/testdatasampler.nim
 create mode 100644 tests/codex/testproof.nim

diff --git a/tests/codex/proof/testdatasampler.nim b/tests/codex/proof/testdatasampler.nim
new file mode 100644
index 00000000..f7cace76
--- /dev/null
+++ b/tests/codex/proof/testdatasampler.nim
@@ -0,0 +1,47 @@
+import std/os
+import std/strutils
+import std/sequtils
+
+import pkg/questionable
+import pkg/questionable/results
+
+import pkg/chronos
+import pkg/asynctest
+import pkg/stew/byteutils
+import pkg/stew/endians2
+import pkg/datastore
+
+import pkg/codex/rng
+import pkg/codex/stores/cachestore
+import pkg/codex/chunker
+import pkg/codex/stores
+import pkg/codex/blocktype as bt
+import pkg/codex/clock
+import pkg/codex/utils/asynciter
+
+import ../helpers
+import ../examples
+
+let
+  bytesPerBlock = 64 * 1024
+  numberOfSlotBlocks = 10
+
+asyncchecksuite "Test proof datasampler":
+  let chunker = RandomChunker.new(Rng.instance(),
+    size = bytesPerBlock * numberOfSlotBlocks,
+    chunkSize = bytesPerBlock)
+
+  var slotBlocks: seq[bt.Block]
+
+  proc createSlotBlocks(): Future[void] {.async.} =
+    while true:
+      let chunk = await chunker.getBytes()
+      if chunk.len <= 0:
+        break
+      slotBlocks.add(bt.Block.new(chunk).tryGet())
+
+  setup:
+    await createSlotBlocks()
+
+  test "Should pass":
+    check true
diff --git a/tests/codex/testproof.nim b/tests/codex/testproof.nim
new file mode 100644
index 00000000..e6c9b36a
--- /dev/null
+++ b/tests/codex/testproof.nim
@@ -0,0 +1,3 @@
+import ./proof/testdatasampler
+
+{.warning[UnusedImport]: off.}
diff --git a/tests/testCodex.nim b/tests/testCodex.nim
index 85f6b26c..d8c53289 100644
--- a/tests/testCodex.nim
+++ b/tests/testCodex.nim
@@ -6,6 +6,7 @@ import ./codex/testmanifest
 import ./codex/testnode
 import ./codex/teststorestream
 import ./codex/testpurchasing
+import ./codex/testproof
 import ./codex/testsales
 import ./codex/testerasure
 import ./codex/testutils