23 lines
1.1 KiB
Python
23 lines
1.1 KiB
Python
import pytest
|
|
|
|
from frostfs_api.crypto.key_extension import KeyExtension
|
|
from tests.helpers.convert import convert_bytes_format_127_to_256
|
|
from tests.helpers.resources import WIF, PRIVATE_KEY
|
|
|
|
|
|
@pytest.mark.crypto
|
|
class TestKeyExtension:
|
|
def test_get_private_key_from_wif_success(self, client_key_extension: KeyExtension):
|
|
private_key = client_key_extension.get_private_key_from_wif(WIF)
|
|
|
|
assert len(private_key) == 32, f"Not correct len of private key, expected: 32 Actual: {len(private_key)} "
|
|
assert private_key == bytes(convert_bytes_format_127_to_256(PRIVATE_KEY)), \
|
|
f"Not match private key, Expected: {PRIVATE_KEY}, Actual: {private_key}"
|
|
|
|
def test_get_private_key_empty_wif(self, client_key_extension: KeyExtension):
|
|
with pytest.raises(ValueError, match="Empty WIF private"):
|
|
client_key_extension.get_private_key_from_wif("")
|
|
|
|
def test_get_private_key_incorrect_wif(self, client_key_extension: KeyExtension):
|
|
with pytest.raises(ValueError, match="Invalid checksum"):
|
|
client_key_extension.get_private_key_from_wif("AAAAAAABBBBBBBBBBBCCCCCCCDDDDDDDDDDDDDDDDD")
|