mirror of
https://github.com/planetmint/planetmint.git
synced 2025-03-30 15:08:31 +00:00
67 lines
2.1 KiB
Python
67 lines
2.1 KiB
Python
# Copyright © 2020 Interplanetary Database Association e.V.,
|
|
# Planetmint and IPDB software contributors.
|
|
# SPDX-License-Identifier: (Apache-2.0 AND CC-BY-4.0)
|
|
# Code is Apache-2.0 and docs are CC-BY-4.0
|
|
|
|
import base64
|
|
import json
|
|
|
|
from pytest import mark
|
|
from hashlib import sha3_256
|
|
|
|
|
|
def test_encode_decode_transaction(b):
|
|
from planetmint.abci.utils import decode_transaction
|
|
from planetmint.abci.utils import encode_transaction
|
|
|
|
asset = {"value": "key"}
|
|
|
|
encode_tx = encode_transaction(asset)
|
|
new_encode_tx = base64.b64encode(json.dumps(asset).encode("utf8")).decode("utf8")
|
|
|
|
assert encode_tx == new_encode_tx
|
|
|
|
de64 = base64.b64decode(encode_tx)
|
|
assert asset == decode_transaction(de64)
|
|
|
|
|
|
def test_calculate_hash_no_key(b):
|
|
from planetmint.abci.utils import calculate_hash
|
|
|
|
# pass an empty list
|
|
assert calculate_hash([]) == ""
|
|
|
|
|
|
# TODO test for the case of an empty list of hashes, and possibly other cases.
|
|
def test_merkleroot():
|
|
from planetmint.abci.utils import merkleroot
|
|
|
|
hashes = [sha3_256(i.encode()).digest() for i in "abc"]
|
|
assert merkleroot(hashes) == ("78c7c394d3158c218916b7ae0ebdea502e0f4e85c08e3b371e3dfd824d389fa3")
|
|
|
|
|
|
SAMPLE_PUBLIC_KEY = {
|
|
"address": "53DC09497A6ED73B342C78AB1E916076A03A8B95",
|
|
"pub_key": {"type": "AC26791624DE60", "value": "7S+T/do70jvneAq0M1so2X3M1iWTSuwtuSAr3nVpfEw="},
|
|
}
|
|
|
|
|
|
@mark.skip(
|
|
reason="ripemd160, the core of pulbic_key64_to_address is no longer supported by hashlib (from python 3.9.13 on)"
|
|
)
|
|
def test_convert_base64_public_key_to_address():
|
|
from planetmint.abci.utils import public_key64_to_address
|
|
|
|
address = public_key64_to_address(SAMPLE_PUBLIC_KEY["pub_key"]["value"])
|
|
assert address == SAMPLE_PUBLIC_KEY["address"]
|
|
|
|
|
|
def test_public_key_encoding_decoding():
|
|
from planetmint.abci.utils import public_key_to_base64
|
|
from planetmint.abci.utils import public_key_from_base64
|
|
|
|
public_key = public_key_from_base64(SAMPLE_PUBLIC_KEY["pub_key"]["value"])
|
|
base64_public_key = public_key_to_base64(public_key)
|
|
|
|
assert base64_public_key == SAMPLE_PUBLIC_KEY["pub_key"]["value"]
|