#
# Tests of lib/hash.py
#
from functools import partial

import pytest

import electrumx.lib.hash as lib_hash


def test_sha256():
    assert lib_hash.sha256(b'sha256') == b'][\t\xf6\xdc\xb2\xd5:_\xff\xc6\x0cJ\xc0\xd5_\xab\xdfU`i\xd6c\x15E\xf4*\xa6\xe3P\x0f.'
    with pytest.raises(TypeError):
        lib_hash.sha256('sha256')

def ripemd160(x):
    assert lib_hash.ripemd160(b'ripemd160') == b'\x903\x91\xa1\xc0I\x9e\xc8\xdf\xb5\x1aSK\xa5VW\xf9|W\xd5'
    with pytest.raises(TypeError):
        lib_hash.ripemd160('ripemd160')

def test_double_sha256():
    assert lib_hash.double_sha256(b'double_sha256') == b'ksn\x8e\xb7\xb9\x0f\xf6\xd9\xad\x88\xd9#\xa1\xbcU(j1Bx\xce\xd5;s\xectL\xe7\xc5\xb4\x00'

def test_hmac_sha512():
    assert lib_hash.hmac_sha512(b'key', b'message') == b"\xe4w8M|\xa2)\xdd\x14&\xe6Kc\xeb\xf2\xd3n\xbdm~f\x9ag5BNr\xeal\x01\xd3\xf8\xb5n\xb3\x9c6\xd8#/T'\x99\x9b\x8d\x1a?\x9c\xd1\x12\x8f\xc6\x9fMu\xb44!h\x10\xfa6~\x98"

def test_hash160():
    assert lib_hash.hash160(b'hash_160') == b'\xb3\x96\x94\xfc\x978R\xa7)XqY\xbb\xdc\xeb\xac\xa7%\xb8$'

def test_hash_to_hex_str():
    assert lib_hash.hash_to_hex_str(b'hash_to_str') == '7274735f6f745f68736168'

def test_hex_str_to_hash():
    assert lib_hash.hex_str_to_hash('7274735f6f745f68736168') == b'hash_to_str'

def test_Base58_char_value():
    chars = '123456789ABCDEFGHJKLMNPQRSTUVWXYZabcdefghijkmnopqrstuvwxyz'
    for value, c in enumerate(chars):
        assert lib_hash.Base58.char_value(c) == value
    for c in (' ', 'I', '0', 'l', 'O'):
        with pytest.raises(lib_hash.Base58Error):
            lib_hash.Base58.char_value(c)

def test_Base58_decode():
    with pytest.raises(TypeError):
        lib_hash.Base58.decode(b'foo')
    with pytest.raises(lib_hash.Base58Error):
        lib_hash.Base58.decode('')
    assert lib_hash.Base58.decode('123456789ABCDEFGHJKLMNPQRSTUVWXYZabcdefghijkmnopqrstuvwxyz') == b'\x00\x01\x11\xd3\x8e_\xc9\x07\x1f\xfc\xd2\x0bJv<\xc9\xaeO%+\xb4\xe4\x8f\xd6j\x83^%*\xda\x93\xffH\rm\xd4=\xc6*d\x11U\xa5'
    assert lib_hash.Base58.decode('3i37NcgooY8f1S') == b'0123456789'

def test_Base58_encode():
    with pytest.raises(TypeError):
        lib_hash.Base58.encode('foo')
    assert lib_hash.Base58.encode(b'') == ''
    assert lib_hash.Base58.encode(b'\0') == '1'
    assert lib_hash.Base58.encode(b'0123456789') == '3i37NcgooY8f1S'

def test_Base58_decode_check():
    with pytest.raises(TypeError):
        lib_hash.Base58.decode_check(b'foo')
    assert lib_hash.Base58.decode_check('4t9WKfuAB8') == b'foo'
    with pytest.raises(lib_hash.Base58Error):
        lib_hash.Base58.decode_check('4t9WKfuAB9')

def test_Base58_encode_check():
    with pytest.raises(TypeError):
        lib_hash.Base58.encode_check('foo')
    assert lib_hash.Base58.encode_check(b'foo') == '4t9WKfuAB8'

def test_Base58_decode_check_custom():
    decode_check_sha256 = partial(lib_hash.Base58.decode_check,
                                  hash_fn=lib_hash.sha256)
    with pytest.raises(TypeError):
        decode_check_sha256(b'foo')
    assert decode_check_sha256('4t9WFhKfWr') == b'foo'
    with pytest.raises(lib_hash.Base58Error):
        decode_check_sha256('4t9WFhKfWp')

def test_Base58_encode_check_custom():
    encode_check_sha256 = partial(lib_hash.Base58.encode_check,
                                  hash_fn=lib_hash.sha256)
    with pytest.raises(TypeError):
        encode_check_sha256('foo')
    assert encode_check_sha256(b'foo') == '4t9WFhKfWr'