import jmbitcoin as btc
import pytest


def test_is_bip21_uri():
    # invalid URIs
    assert(not btc.is_bip21_uri(''))
    assert(not btc.is_bip21_uri('nfdjksnfjkdsnfjkds'))
    assert(not btc.is_bip21_uri('175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W'))
    assert(not btc.is_bip21_uri('175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W?amount=20.3'))
    assert(not btc.is_bip21_uri('bitcoin:'))
    assert(not btc.is_bip21_uri('bitcoin:?amount=20.3'))
    # valid URIs
    assert(btc.is_bip21_uri('bitcoin:175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W'))
    assert(btc.is_bip21_uri('BITCOIN:175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W'))
    assert(btc.is_bip21_uri('BitCoin:175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W'))
    assert(btc.is_bip21_uri('bitcoin:175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W?label=Luke-Jr'))


def test_bip21_decode():

    # These should raise exception because of not being valid BIP21 URI's
    with pytest.raises(ValueError):
        btc.decode_bip21_uri('')
        btc.decode_bip21_uri('nfdjksnfjkdsnfjkds')
        btc.decode_bip21_uri('175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W')
        btc.decode_bip21_uri(
            '175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W?amount=20.3')
        btc.decode_bip21_uri('bitcoin:')
        btc.decode_bip21_uri('bitcoin:?amount=20.3')
        btc.decode_bip21_uri(
            'bitcoin:175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W?amount=')
        btc.decode_bip21_uri(
            'bitcoin:175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W?amount=XYZ')
        btc.decode_bip21_uri(
            'bitcoin:175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W?amount=100\'000')
        btc.decode_bip21_uri(
            'bitcoin:175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W?amount=100,000')
        btc.decode_bip21_uri(
            'bitcoin:175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W?amount=100000000')

    assert(btc.decode_bip21_uri('bitcoin:175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W'
        )['address'] == '175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W')
    assert(btc.decode_bip21_uri('BITCOIN:175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W'
        )['address'] == '175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W')
    assert(btc.decode_bip21_uri('BitCoin:175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W'
        )['address'] == '175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W')

    parsed = btc.decode_bip21_uri(
        'bitcoin:175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W?label=Luke-Jr')
    assert(parsed['address'] == '175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W')
    assert(parsed['label'] == 'Luke-Jr')

    parsed = btc.decode_bip21_uri(
        'bitcoin:175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W?amount=20.3&label=Luke-Jr')
    assert(parsed['address'] == '175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W')
    assert(parsed['amount'] == 2030000000)
    assert(parsed['label'] == 'Luke-Jr')

    parsed = btc.decode_bip21_uri(
        'bitcoin:175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W?amount=50&label=Luke-Jr&message=Donation%20for%20project%20xyz')
    assert(parsed['address'] == '175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W')
    assert(parsed['amount'] == 5000000000)
    assert(parsed['label'] == 'Luke-Jr')
    assert(parsed['message'] == 'Donation for project xyz')

    # This should raise exception because of unknown req-* parameters
    with pytest.raises(ValueError):
        btc.decode_bip21_uri(
            'bitcoin:175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W?req-somethingyoudontunderstand=50&req-somethingelseyoudontget=999')

    parsed = btc.decode_bip21_uri(
        'bitcoin:175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W?somethingyoudontunderstand=50&somethingelseyoudontget=999')
    assert(parsed['address'] == '175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W')
    assert(parsed['somethingyoudontunderstand'] == '50')
    assert(parsed['somethingelseyoudontget'] == '999')

    # Test multiple amount parameters, last value should win.
    parsed = btc.decode_bip21_uri(
        'bitcoin:175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W?amount=20.3&amount=50&label=Luke-Jr')
    assert(parsed['address'] == '175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W')
    assert(parsed['amount'] == 5000000000)
    assert(parsed['label'] == 'Luke-Jr')
    # Here are two amount parameters, first valid, second not valid, so URI is not valid.
    with pytest.raises(ValueError):
        btc.decode_bip21_uri(
            'bitcoin:175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W?amount=20.3&amount=100,000&label=Luke-Jr')


def test_bip21_encode():
    assert(
        btc.encode_bip21_uri('175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W', {}) ==
        'bitcoin:175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W'
    )
    assert(
        btc.encode_bip21_uri('175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W', {
            'label': 'Luke-Jr'
        }) ==
        'bitcoin:175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W?label=Luke-Jr'
    )
    # Both dictionary and list of tuples should work
    assert(
        btc.encode_bip21_uri('175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W', [
            ('label', 'Luke-Jr')
        ]) ==
        'bitcoin:175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W?label=Luke-Jr'
    )
    # Use list of tuples version for multiple parameter tests, as dicts don't
    # have guaranteed ordering.
    assert(
        btc.encode_bip21_uri('175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W', [
            ('amount', 20.3),
            ('label', 'Luke-Jr')
        ]) ==
        'bitcoin:175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W?amount=20.3&label=Luke-Jr'
    )
    assert(
        btc.encode_bip21_uri('175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W', [
            ('amount', 50),
            ('label', 'Luke-Jr'),
            ('message', 'Donation for project xyz')
        ]) ==
        'bitcoin:175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W?amount=50&label=Luke-Jr&message=Donation%20for%20project%20xyz'
    )
    assert(
        btc.encode_bip21_uri('175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W', [
            ('req-somethingyoudontunderstand', 50),
            ('req-somethingelseyoudontget', 999)
        ]) ==
        'bitcoin:175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W?req-somethingyoudontunderstand=50&req-somethingelseyoudontget=999'
    )
    assert(
        btc.encode_bip21_uri('175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W', [
            ('somethingyoudontunderstand', 50),
            ('somethingelseyoudontget', 999)
        ]) ==
        'bitcoin:175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W?somethingyoudontunderstand=50&somethingelseyoudontget=999'
    )
    # Invalid amounts must raise ValueError
    with pytest.raises(ValueError):
        # test dicts
        btc.encode_bip21_uri('175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W', {
            'amount': ''
        })
        btc.encode_bip21_uri('175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W', {
            'amount': 'XYZ'
        })
        btc.encode_bip21_uri('175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W', {
            'amount': '100\'000'
        })
        btc.encode_bip21_uri('175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W', {
            'amount': '100,000'
        })
        btc.encode_bip21_uri('175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W', {
            'amount': '100000000'
        })
        # test list of tuples
        btc.encode_bip21_uri('175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W', [
            ('amount', '')
        ])
        btc.encode_bip21_uri('175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W', [
            ('amount', 'XYZ')
        ])
        btc.encode_bip21_uri('175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W', [
            ('amount', '100\'000')
        ])
        btc.encode_bip21_uri('175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W', [
            ('amount', '100,000')
        ])
        btc.encode_bip21_uri('175tWpb8K1S7NmH4Zx6rewF9WQrcZv245W', [
            ('amount', '100000000')
        ])