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 | |||||