added fork_at for transition forks, WIP

Author: felix314159

src/ethereum_test_types/blob_types.py

@@ -212,6 +212,10 @@ def get_cells(fork: Fork, data: Bytes) -> List[Bytes] | None:
         # loaded trusted setup if it is not already loaded
         Blob.trusted_setup()
 
+        # handle transition forks (not needed when default timestamp of 0 is used)
+        if timestamp > 0:
+            fork = fork.fork_at(timestamp=timestamp)
+
         # get data for blob parameters
         data: Bytes = generate_blob_data(seed)
         commitment: Bytes = get_commitment(data)
@@ -433,15 +437,6 @@ def corrupt_byte(b: bytes) -> Bytes:
             self.proof = Bytes(b"".join(corrupt_byte(bytes([byte])) for byte in self.proof))
 
 
-# TODO: what does he mean with fork.fork_at()? This function also seems to require a 'block number' which is not available here. And to which file is the logic of that function hidden?  # noqa: E501
-# mytestfork: Fork = CancunToPragueAtTime15k
-# mytestfork.fork_at(timestamp=14999)  # returns cancun
-# mytestfork.fork_at(timestamp=15000)  # returns prague
-# TODO: call this once only at blob creation (not when reading blob from file) + remove timestamp from json  # noqa: E501
-
-# TODO: fix tests/osaka/eip7594_peerdas/test_get_blobs.py
-# TODO: test all functions again
-
 # --------- generate static blobs ------------
 # fork_list = [Cancun, Prague, Osaka]
 # amount_static_blobs_per_fork = 10
@@ -509,6 +504,22 @@ def corrupt_byte(b: bytes) -> Bytes:
 # # myparis: Fork = Paris
 # # parisapriaparip: Blob = Blob.from_fork(myparis, myseed, 123)
 
+# test fork_at for when timestamp > 0
+# transitionfork: Fork = Cancun
+# new_timestamp = 15000  # is supposed to return prague, but still returns cancun.. TODO: ?
+# transitionforkblob: Blob = Blob.from_fork(transitionfork, timestamp=new_timestamp)
+# transitionforkblob.write_to_file()
+# restored_transitionforkblob: Blob = Blob.from_file("blob_0_cell_proofs_0")
+# assert transitionforkblob.data == restored_transitionforkblob.data
+# assert transitionforkblob.commitment == restored_transitionforkblob.commitment
+# assert transitionforkblob.proof == restored_transitionforkblob.proof
+# assert transitionforkblob.cells == restored_transitionforkblob.cells
+# assert transitionforkblob.versioned_hash == restored_transitionforkblob.versioned_hash
+# assert transitionforkblob.name == restored_transitionforkblob.name
+# assert transitionforkblob.fork == restored_transitionforkblob.fork
+# assert transitionforkblob.seed == restored_transitionforkblob.seed
+# assert transitionforkblob.timestamp == restored_transitionforkblob.timestamp
+
 # # test proof corruption
 # #  osaka
 # testseed = 55