# (c) Copyright 2020 by Coinkite Inc. This file is covered by license found in COPYING-CC.
#
# unit test for address decoding from various types of CTxOuts
from h import a2b_hex, b2a_hex
from serializations import CTxOut
from uio import BytesIO
from public_constants import AF_CLASSIC, AF_P2SH, AF_P2WPKH, AF_P2WSH


cases = [
    # TxOut, type, is_segwit, hash160/pubkey, 
    ( 'c4f33d0000000000160014ad46a001d55bd55d157e716bf17c02f8964b5a19',
        AF_P2WPKH, True,
        'ad46a001d55bd55d157e716bf17c02f8964b5a19' ),
    ( '202cb206000000001976a9148280b37df378db99f66f85c95a783a76ac7a6d5988ac',
        AF_CLASSIC, False,
        '8280b37df378db99f66f85c95a783a76ac7a6d59' ),

    # from legendary txid: 40eee3ae1760e3a8532263678cdf64569e6ad06abc133af64f735e52562bccc8
    ( '301b0f000000000017a914e9c3dd0c07aac76179ebc76a6c78d4d67c6c160a87',
        AF_P2SH, False,
        'e9c3dd0c07aac76179ebc76a6c78d4d67c6c160a'),

    # from testnet: a4c89e0ffb84d06a1e62f0f9f0f5974db250878caa1f71f9992a1f865b8ff2fa
    # via <https://github.com/bitcoinjs/bitcoinjs-lib/issues/856>
    ( 'b88201000000000017a914f0ca58dc8e539421a3cb4a9c22c059973075287c87',
        AF_P2SH, False,
        'f0ca58dc8e539421a3cb4a9c22c059973075287c'),

    # XXX missing: P2SH segwit, 1of1 and N of M
    ( 'd0f13d0000000000160014f2369bac6d24ed11313fa65adda1971d10e17bff',
        AF_P2WPKH, True,
        'f2369bac6d24ed11313fa65adda1971d10e17bff')
]

for i, (raw_txo, expect_type, expect_sw, expect_hash) in enumerate(cases):
    expect_hash = a2b_hex(expect_hash)

    out = CTxOut()
    out.deserialize(BytesIO(a2b_hex(raw_txo)))

    print("Case %d: %s... " % (i, raw_txo[0:30]))
    addr_type, addr_or_pubkey, is_segwit = out.get_address()

    assert is_segwit == expect_sw, 'wrong segwit'
    assert addr_or_pubkey == expect_hash, 'wrong pubkey/addr'
    assert addr_type == expect_type, addr_type