55 lines
1.5 KiB
Python
55 lines
1.5 KiB
Python
from panda3d import core
|
|
import random
|
|
import pytest
|
|
|
|
try:
|
|
import hashlib
|
|
except ImportError:
|
|
hashlib = object()
|
|
hashlib.algorithms_available = ()
|
|
|
|
|
|
def test_hashval_hex():
|
|
hex = '%032x' % random.getrandbits(32 * 4)
|
|
val = core.HashVal()
|
|
val.input_hex(core.StringStream(hex.encode('ascii')))
|
|
assert str(val) == hex.lower()
|
|
|
|
|
|
def test_hashval_md5_known():
|
|
known_hashes = {
|
|
'd41d8cd98f00b204e9800998ecf8427e': b'',
|
|
'93b885adfe0da089cdf634904fd59f71': b'\000',
|
|
'3b5d3c7d207e37dceeedd301e35e2e58': b'\000' * 64,
|
|
'202cb962ac59075b964b07152d234b70': b'123',
|
|
'520620de89e220f9b5850cc97cbff46c': b'01234567' * 8,
|
|
'ad32d3ef227a5ebd800a40d4eeaff41f': b'01234567' * 8 + b'a',
|
|
}
|
|
|
|
for known, plain in known_hashes.items():
|
|
hv = core.HashVal()
|
|
hv.hash_bytes(plain)
|
|
assert hv.as_hex() == known
|
|
|
|
|
|
@pytest.mark.skipif('md5' not in hashlib.algorithms_available,
|
|
reason="MD5 algorithm not available in hashlib")
|
|
def test_hashval_md5_random():
|
|
data = bytearray()
|
|
|
|
for i in range(2500):
|
|
control = hashlib.md5(data).hexdigest()
|
|
|
|
# Test hash_bytes
|
|
hv = core.HashVal()
|
|
hv.hash_bytes(bytes(data))
|
|
assert hv.as_hex() == control
|
|
|
|
# Test hash_stream
|
|
hv = core.HashVal()
|
|
result = hv.hash_stream(core.StringStream(data))
|
|
assert result
|
|
assert hv.as_hex() == control
|
|
|
|
data.append(random.randint(0, 255))
|