mirror of
https://github.com/OALabs/hashdb
synced 2026-06-08 12:05:11 +00:00
herrcore
52 lines
2.3 KiB
Python
52 lines
2.3 KiB
Python
import pytest
|
|
|
|
|
|
def test_sanity():
|
|
import hashdb
|
|
|
|
|
|
def test_algorithm_attributes():
|
|
from hashdb import algorithms
|
|
for algorithm_name in list(algorithms.modules.keys()):
|
|
assert(algorithms.modules[algorithm_name].DESCRIPTION != None)
|
|
assert(algorithms.modules[algorithm_name].DESCRIPTION != '')
|
|
assert(algorithms.modules[algorithm_name].TYPE != None)
|
|
assert(algorithms.modules[algorithm_name].TYPE != '')
|
|
assert(algorithms.modules[algorithm_name].TEST_1 != None)
|
|
assert(algorithms.modules[algorithm_name].TEST_1 != '')
|
|
|
|
|
|
def test_algorithm_hash():
|
|
from hashdb import algorithms
|
|
for algorithm_name in list(algorithms.modules.keys()):
|
|
assert(algorithms.modules[algorithm_name].TEST_1 == algorithms.modules[algorithm_name].hash(b'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789'))
|
|
|
|
|
|
def test_algorithm_duplicates():
|
|
from hashdb import algorithms
|
|
hash_list = []
|
|
for algorithm_name in list(algorithms.modules.keys()):
|
|
# Skip custom API hashes
|
|
if 'TEST_API_1' not in dir(algorithms.modules[algorithm_name]):
|
|
hash_list.append(algorithms.modules[algorithm_name].TEST_1)
|
|
assert(len(hash_list) == len(set(hash_list)))
|
|
|
|
|
|
def test_custom_api_algorithms():
|
|
from hashdb import algorithms
|
|
hash_list = []
|
|
for algorithm_name in list(algorithms.modules.keys()):
|
|
# Custom API hashes must have a custom hash and data field
|
|
if 'TEST_API_1' in dir(algorithms.modules[algorithm_name]):
|
|
assert(algorithms.modules[algorithm_name].TEST_API_1 != None)
|
|
assert(algorithms.modules[algorithm_name].TEST_API_1 != '')
|
|
assert(algorithms.modules[algorithm_name].TEST_API_DATA_1 != None)
|
|
assert(algorithms.modules[algorithm_name].TEST_API_DATA_1 != '')
|
|
assert((type(algorithms.modules[algorithm_name].TEST_API_DATA_1) == bytes) or
|
|
(type(algorithms.modules[algorithm_name].TEST_API_DATA_1) == str))
|
|
if type(algorithms.modules[algorithm_name].TEST_API_DATA_1) == str:
|
|
data = algorithms.modules[algorithm_name].TEST_API_DATA_1.encode('utf-8')
|
|
else:
|
|
data = algorithms.modules[algorithm_name].TEST_API_DATA_1
|
|
assert(algorithms.modules[algorithm_name].TEST_API_1 == algorithms.modules[algorithm_name].hash(data))
|