pabtc/test/test_wallet.py at master · libraries/pabtc · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
import itertools
import pabtc


def test_wallet_transfer():
    pabtc.config.current = pabtc.config.develop
    user_list = [
        pabtc.wallet.Wallet(pabtc.wallet.Signerp2pkh(pabtc.core.PriKey(1))),
        pabtc.wallet.Wallet(pabtc.wallet.Signerp2shp2ms(
            [pabtc.core.PriKey(e) for e in [1, 2]],
            [pabtc.core.PriKey(e).pubkey() for e in [1, 2]]),
        ),
        pabtc.wallet.Wallet(pabtc.wallet.Signerp2shp2wpkh(pabtc.core.PriKey(1))),
        pabtc.wallet.Wallet(pabtc.wallet.Signerp2wpkh(pabtc.core.PriKey(1))),
        pabtc.wallet.Wallet(pabtc.wallet.Signerp2wshp2ms(
            [pabtc.core.PriKey(e) for e in [1, 2]],
            [pabtc.core.PriKey(e).pubkey() for e in [1, 2]]),
        ),
        pabtc.wallet.Wallet(pabtc.wallet.Signerp2tr(pabtc.core.PriKey(1), bytearray())),
    ]
    mate_list = [
        pabtc.wallet.Wallet(pabtc.wallet.Signerp2pkh(pabtc.core.PriKey(2))),
        pabtc.wallet.Wallet(pabtc.wallet.Signerp2shp2ms(
            [pabtc.core.PriKey(e) for e in [2, 1]],
            [pabtc.core.PriKey(e).pubkey() for e in [2, 1]]),
        ),
        pabtc.wallet.Wallet(pabtc.wallet.Signerp2shp2wpkh(pabtc.core.PriKey(2))),
        pabtc.wallet.Wallet(pabtc.wallet.Signerp2wpkh(pabtc.core.PriKey(2))),
        pabtc.wallet.Wallet(pabtc.wallet.Signerp2wshp2ms(
            [pabtc.core.PriKey(e) for e in [2, 1]],
            [pabtc.core.PriKey(e).pubkey() for e in [2, 1]]),
        ),
        pabtc.wallet.Wallet(pabtc.wallet.Signerp2tr(pabtc.core.PriKey(2), bytearray())),
    ]
    for user, mate in itertools.product(user_list, mate_list):
        value = pabtc.denomination.bitcoin
        value_old = mate.balance()
        txid = user.transfer(mate.script, value)
        pabtc.rpc.wait(txid[::-1].hex())
        value_new = mate.balance()
        assert value_new - value_old == value
        value_old = value_new
        txid = user.transfer(mate.script, value)
        pabtc.rpc.wait(txid[::-1].hex())
        value_new = mate.balance()
        assert value_new - value_old == value
        pabtc.rpc.generate_to_address(6, user.addr)
        txid = mate.transfer_all(user.script)
        pabtc.rpc.wait(txid[::-1].hex())
        assert mate.balance() == 0