mirror of
https://github.com/logos-blockchain/logos-blockchain-specs.git
synced 2026-01-02 13:13:06 +00:00
3f3427ee9f
* Modify and rename hash_commitment method. Now we just hash the column commitment itself. * Fix calls * Remove certificate from verifier * Update verifier * Fix tests * Fix verifier imports * Fix more imports * Fix dispersal * Fix more imports * Fix missing parameter in dispersal * Fix tests * Full flow renaming * Disperse encoded data in full flow test * Make da verification indempotent (#118) --------- Co-authored-by: Gusto <bacvinka@gmail.com> Co-authored-by: gusto <bacv@users.noreply.github.com>
73 lines
2.7 KiB
Python
73 lines
2.7 KiB
Python
from unittest import TestCase
|
|
|
|
from da.common import Column
|
|
from da.encoder import DAEncoder
|
|
from da.kzg_rs import kzg
|
|
from da.kzg_rs.common import GLOBAL_PARAMETERS, ROOTS_OF_UNITY
|
|
from da.test_encoder import TestEncoder
|
|
from da.verifier import DAVerifier, DABlob
|
|
|
|
|
|
class TestVerifier(TestCase):
|
|
|
|
def setUp(self):
|
|
self.verifier = DAVerifier()
|
|
|
|
def test_verify_column(self):
|
|
column = Column(int.to_bytes(i, length=32) for i in range(8))
|
|
_, column_commitment = kzg.bytes_to_commitment(column.as_bytes(), GLOBAL_PARAMETERS)
|
|
aggregated_poly, aggregated_column_commitment = kzg.bytes_to_commitment(
|
|
DAEncoder.hash_commitment_blake2b31(column_commitment), GLOBAL_PARAMETERS
|
|
)
|
|
aggregated_proof = kzg.generate_element_proof(0, aggregated_poly, GLOBAL_PARAMETERS, ROOTS_OF_UNITY)
|
|
self.assertTrue(
|
|
self.verifier._verify_column(
|
|
column, 0, column_commitment, aggregated_column_commitment, aggregated_proof,
|
|
)
|
|
)
|
|
|
|
def test_verify(self):
|
|
_ = TestEncoder()
|
|
_.setUp()
|
|
encoded_data = _.encoder.encode(_.data)
|
|
for i, column in enumerate(encoded_data.chunked_data.columns):
|
|
verifier = DAVerifier()
|
|
da_blob = DABlob(
|
|
Column(column),
|
|
i,
|
|
encoded_data.column_commitments[i],
|
|
encoded_data.aggregated_column_commitment,
|
|
encoded_data.aggregated_column_proofs[i],
|
|
encoded_data.row_commitments,
|
|
[row[i] for row in encoded_data.row_proofs],
|
|
)
|
|
self.assertIsNotNone(verifier.verify(da_blob))
|
|
|
|
def test_verify_duplicated_blob(self):
|
|
_ = TestEncoder()
|
|
_.setUp()
|
|
encoded_data = _.encoder.encode(_.data)
|
|
columns = enumerate(encoded_data.chunked_data.columns)
|
|
i, column = next(columns)
|
|
da_blob = DABlob(
|
|
Column(column),
|
|
i,
|
|
encoded_data.column_commitments[i],
|
|
encoded_data.aggregated_column_commitment,
|
|
encoded_data.aggregated_column_proofs[i],
|
|
encoded_data.row_commitments,
|
|
[row[i] for row in encoded_data.row_proofs],
|
|
)
|
|
self.assertIsNotNone(self.verifier.verify(da_blob))
|
|
for i, column in columns:
|
|
da_blob = DABlob(
|
|
Column(column),
|
|
i,
|
|
encoded_data.column_commitments[i],
|
|
encoded_data.aggregated_column_commitment,
|
|
encoded_data.aggregated_column_proofs[i],
|
|
encoded_data.row_commitments,
|
|
[row[i] for row in encoded_data.row_proofs],
|
|
)
|
|
self.assertTrue(self.verifier.verify(da_blob))
|