mirror of
https://github.com/JoinMarket-Org/joinmarket-clientserver.git
synced 2026-08-13 12:33:40 +02:00
125 lines
4.1 KiB
Python
125 lines
4.1 KiB
Python
#! /usr/bin/env python
|
|
'''Wallet functionality tests.'''
|
|
|
|
import os
|
|
import time
|
|
import binascii
|
|
from commontest import create_wallet_for_sync, make_sign_and_push
|
|
import json
|
|
|
|
import pytest
|
|
from jmbase import get_log, hextobin
|
|
from jmclient import (
|
|
load_test_config, jm_single,
|
|
estimate_tx_fee, BitcoinCoreInterface, Mnemonic)
|
|
from taker_test_data import t_raw_signed_tx
|
|
|
|
pytestmark = pytest.mark.usefixtures("setup_regtest_bitcoind")
|
|
|
|
testdir = os.path.dirname(os.path.realpath(__file__))
|
|
log = get_log()
|
|
|
|
|
|
def do_tx(wallet_service, amount):
|
|
ins_full = wallet_service.select_utxos(0, amount)
|
|
cj_addr = wallet_service.get_internal_addr(1)
|
|
change_addr = wallet_service.get_internal_addr(0)
|
|
wallet_service.save_wallet()
|
|
txid = make_sign_and_push(ins_full,
|
|
wallet_service,
|
|
amount,
|
|
output_addr=cj_addr,
|
|
change_addr=change_addr,
|
|
estimate_fee=True)
|
|
assert txid
|
|
time.sleep(2) #blocks
|
|
wallet_service.sync_unspent()
|
|
return txid
|
|
|
|
|
|
def test_query_utxo_set(setup_wallets):
|
|
load_test_config()
|
|
jm_single().bc_interface.tick_forward_chain_interval = 1
|
|
wallet_service = create_wallet_for_sync([2, 3, 0, 0, 0],
|
|
["wallet4utxo.json", "4utxo", [2, 3]])
|
|
wallet_service.sync_wallet(fast=True)
|
|
txid = do_tx(wallet_service, 90000000)
|
|
txid2 = do_tx(wallet_service, 20000000)
|
|
print("Got txs: ", txid, txid2)
|
|
res1 = jm_single().bc_interface.query_utxo_set(
|
|
(txid, 0), include_mempool=True)
|
|
res2 = jm_single().bc_interface.query_utxo_set(
|
|
[(txid, 0), (txid2, 1)],
|
|
includeconfs=True, include_mempool=True)
|
|
assert len(res1) == 1
|
|
assert len(res2) == 2
|
|
assert all([x in res1[0] for x in ['script', 'value']])
|
|
assert not 'confirms' in res1[0]
|
|
assert 'confirms' in res2[0]
|
|
assert 'confirms' in res2[1]
|
|
res3 = jm_single().bc_interface.query_utxo_set((b"\xee" * 32, 25))
|
|
assert res3 == [None]
|
|
|
|
|
|
"""Purely blockchaininterface related error condition tests"""
|
|
|
|
|
|
def test_wrong_network_bci(setup_wallets):
|
|
rpc = jm_single().bc_interface.jsonRpc
|
|
with pytest.raises(Exception) as e_info:
|
|
x = BitcoinCoreInterface(rpc, 'mainnet')
|
|
|
|
|
|
def test_pushtx_errors(setup_wallets):
|
|
"""Ensure pushtx fails return False
|
|
"""
|
|
badtx = b"\xaa\xaa"
|
|
assert not jm_single().bc_interface.pushtx(badtx)
|
|
#Break the authenticated jsonrpc and try again
|
|
jm_single().bc_interface.jsonRpc.port = 18333
|
|
assert not jm_single().bc_interface.pushtx(hextobin(t_raw_signed_tx))
|
|
#rebuild a valid jsonrpc inside the bci
|
|
load_test_config()
|
|
|
|
|
|
"""Tests mainly for wallet.py"""
|
|
|
|
|
|
def test_absurd_fee(setup_wallets):
|
|
jm_single().config.set("POLICY", "absurd_fee_per_kb", "1000")
|
|
with pytest.raises(ValueError) as e_info:
|
|
estimate_tx_fee(10, 2)
|
|
load_test_config()
|
|
|
|
|
|
def check_bip39_case(vectors, language="english"):
|
|
mnemo = Mnemonic(language)
|
|
for v in vectors:
|
|
code = mnemo.to_mnemonic(binascii.unhexlify(v[0]))
|
|
seed = binascii.hexlify(Mnemonic.to_seed(code, passphrase=v[4])).decode('ascii')
|
|
print('checking this phrase: ' + v[1])
|
|
assert mnemo.check(v[1])
|
|
assert v[1] == code
|
|
assert v[2] == seed
|
|
|
|
"""
|
|
Sanity check of basic bip39 functionality for 12 words seed, copied from
|
|
https://github.com/trezor/python-mnemonic/blob/master/tests/test_mnemonic.py
|
|
"""
|
|
def test_bip39_vectors(setup_wallets):
|
|
with open(os.path.join(testdir, 'bip39vectors.json'), 'r') as f:
|
|
vectors_full = json.load(f)
|
|
vectors = vectors_full['english']
|
|
#default from-file cases use passphrase 'TREZOR'; TODO add other
|
|
#extensions, but note there is coverage of that in the below test
|
|
for v in vectors:
|
|
v.append("TREZOR")
|
|
#12 word seeds only
|
|
vectors = filter(lambda x: len(x[1].split())==12, vectors)
|
|
check_bip39_case(vectors)
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def setup_wallets():
|
|
load_test_config()
|
|
jm_single().bc_interface.tick_forward_chain_interval = 2
|