Changeset View
Changeset View
Standalone View
Standalone View
test/functional/wallet_address_types.py
Show All 33 Lines | |||||
from decimal import Decimal | from decimal import Decimal | ||||
import itertools | import itertools | ||||
from test_framework.test_framework import BitcoinTestFramework | from test_framework.test_framework import BitcoinTestFramework | ||||
from test_framework.util import ( | from test_framework.util import ( | ||||
assert_equal, | assert_equal, | ||||
assert_greater_than, | assert_greater_than, | ||||
connect_nodes_bi, | connect_nodes, | ||||
) | ) | ||||
class AddressTypeTest(BitcoinTestFramework): | class AddressTypeTest(BitcoinTestFramework): | ||||
def set_test_params(self): | def set_test_params(self): | ||||
self.num_nodes = 5 | self.num_nodes = 5 | ||||
# whitelist all peers to speed up tx relay / mempool sync | # whitelist all peers to speed up tx relay / mempool sync | ||||
self.extra_args = [["-whitelist=127.0.0.1"]] * self.num_nodes | self.extra_args = [["-whitelist=127.0.0.1"]] * self.num_nodes | ||||
def skip_test_if_missing_module(self): | def skip_test_if_missing_module(self): | ||||
self.skip_if_no_wallet() | self.skip_if_no_wallet() | ||||
def setup_network(self): | def setup_network(self): | ||||
self.setup_nodes() | self.setup_nodes() | ||||
# Fully mesh-connect nodes for faster mempool sync | # Fully mesh-connect nodes for faster mempool sync | ||||
for i, j in itertools.product(range(self.num_nodes), repeat=2): | for i, j in itertools.product(range(self.num_nodes), repeat=2): | ||||
if i > j: | if i > j: | ||||
connect_nodes_bi(self.nodes[i], self.nodes[j]) | connect_nodes(self.nodes[i], self.nodes[j]) | ||||
self.sync_all() | self.sync_all() | ||||
def get_balances(self, confirmed=True): | def get_balances(self, confirmed=True): | ||||
"""Return a list of confirmed or unconfirmed balances.""" | """Return a list of confirmed or unconfirmed balances.""" | ||||
if confirmed: | if confirmed: | ||||
return [self.nodes[i].getbalance() for i in range(4)] | return [self.nodes[i].getbalance() for i in range(4)] | ||||
else: | else: | ||||
return [self.nodes[i].getunconfirmedbalance() for i in range(4)] | return [self.nodes[i].getunconfirmedbalance() for i in range(4)] | ||||
▲ Show 20 Lines • Show All 215 Lines • Show Last 20 Lines |