74 lines
1.7 KiB
Python
74 lines
1.7 KiB
Python
"""Cryptography primitives test module."""
|
|
|
|
import pytest
|
|
from merkle_tree_stream import MerkleTreeNode
|
|
from pysodium import crypto_sign_PUBLICKEYBYTES, crypto_sign_SECRETKEYBYTES
|
|
|
|
from hypercore_crypto import (
|
|
data,
|
|
discovery_key,
|
|
key_pair,
|
|
parent,
|
|
random_bytes,
|
|
sign,
|
|
verify,
|
|
)
|
|
|
|
|
|
def test_key_pair_seed_length():
|
|
with pytest.raises(ValueError) as exception:
|
|
key_pair(b"world hello")
|
|
assert "must be of length" in str(exception)
|
|
|
|
|
|
def test_key_pair_length():
|
|
public_key, secret_key = key_pair()
|
|
assert len(public_key) == crypto_sign_PUBLICKEYBYTES
|
|
assert len(secret_key) == crypto_sign_SECRETKEYBYTES
|
|
|
|
|
|
def test_sign():
|
|
message = b"hello world"
|
|
_, secret_key = key_pair()
|
|
signature = sign(message, secret_key)
|
|
assert message not in signature
|
|
assert len(signature) == 64
|
|
|
|
|
|
def test_verify():
|
|
message = b"hello world"
|
|
public_key, secret_key = key_pair()
|
|
signature = sign(message, secret_key)
|
|
assert verify(message, signature, public_key)
|
|
|
|
|
|
def test_data_digest():
|
|
assert (
|
|
data(b"hello world").hex()
|
|
== "ccfa4259ee7c41e411e5770973a49c5ceffb5272d6a37f2c6f2dac2190f7e2b7"
|
|
)
|
|
|
|
|
|
def test_random_bytes():
|
|
assert len(random_bytes(32)) == 32
|
|
|
|
|
|
def test_parent_digest():
|
|
_data = b"hello world"
|
|
_parent = parent(
|
|
MerkleTreeNode(
|
|
index=0, size=11, hash=data(_data), parent=None, data=None
|
|
),
|
|
MerkleTreeNode(
|
|
index=2, size=11, hash=data(_data), parent=None, data=None
|
|
),
|
|
)
|
|
assert (
|
|
_parent.hex()
|
|
== "43563406adba8b34b133fdca32d0a458c5be769615e01df30e6535ccd3c075f0"
|
|
)
|
|
|
|
|
|
def test_discovery_key_generated():
|
|
assert discovery_key(random_bytes(32)) is not None
|