You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
26 lines
721 B
Python
26 lines
721 B
Python
import hashlib
|
|
|
|
from kademlia.utils import digest, shared_prefix
|
|
|
|
|
|
class TestUtils:
|
|
def test_digest(self): # pylint: disable=no-self-use
|
|
dig = hashlib.sha1(b'1').digest()
|
|
assert dig == digest(1)
|
|
|
|
dig = hashlib.sha1(b'another').digest()
|
|
assert dig == digest('another')
|
|
|
|
def test_shared_prefix(self): # pylint: disable=no-self-use
|
|
args = ['prefix', 'prefixasdf', 'prefix', 'prefixxxx']
|
|
assert shared_prefix(args) == 'prefix'
|
|
|
|
args = ['p', 'prefixasdf', 'prefix', 'prefixxxx']
|
|
assert shared_prefix(args) == 'p'
|
|
|
|
args = ['one', 'two']
|
|
assert shared_prefix(args) == ''
|
|
|
|
args = ['hi']
|
|
assert shared_prefix(args) == 'hi'
|