Skip to content
Snippets Groups Projects
Commit c6721586 authored by inso's avatar inso
Browse files

Mock libnacl in a different way

parent 8b1ace26
No related branches found
No related tags found
No related merge requests found
...@@ -31,7 +31,7 @@ class Mock(MagicMock): ...@@ -31,7 +31,7 @@ class Mock(MagicMock):
def __getattr__(cls, name): def __getattr__(cls, name):
return Mock() return Mock()
MOCK_MODULES = ['libnacl.sign'] MOCK_MODULES = ['libnacl']
sys.modules.update((mod_name, Mock()) for mod_name in MOCK_MODULES) sys.modules.update((mod_name, Mock()) for mod_name in MOCK_MODULES)
# If your documentation needs a minimal Sphinx version, state it here. # If your documentation needs a minimal Sphinx version, state it here.
......
...@@ -5,9 +5,8 @@ Ucoin public and private keys ...@@ -5,9 +5,8 @@ Ucoin public and private keys
""" """
import base58 import base58
import base64 import libnacl
from pylibscrypt import scrypt from pylibscrypt import scrypt
from libnacl.sign import Signer as NaclSigningKey
SEED_LENGTH = 32 # Length of the key SEED_LENGTH = 32 # Length of the key
...@@ -23,7 +22,8 @@ def _ensure_bytes(data): ...@@ -23,7 +22,8 @@ def _ensure_bytes(data):
return data return data
class SigningKey(NaclSigningKey):
class SigningKey(libnacl.sign.Signer):
def __init__(self, salt, password): def __init__(self, salt, password):
salt = _ensure_bytes(salt) salt = _ensure_bytes(salt)
password = _ensure_bytes(password) password = _ensure_bytes(password)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment