Differential D3949 Diff 11048 test/functional/wallet_labels.py

Changeset View

Standalone View

test/functional/wallet_labels.py

#!/usr/bin/env python3		#!/usr/bin/env python3
# Copyright (c) 2016-2017 The Bitcoin Core developers		# Copyright (c) 2016-2017 The Bitcoin Core developers
# Distributed under the MIT software license, see the accompanying		# Distributed under the MIT software license, see the accompanying
# file COPYING or http://www.opensource.org/licenses/mit-license.php.		# file COPYING or http://www.opensource.org/licenses/mit-license.php.
"""Test label RPCs.		"""Test label RPCs.

RPCs tested are:		RPCs tested are:
- getlabeladdress		- getlabeladdress
- getaddressesbyaccount		- getaddressesbyaccount
- listaddressgroupings		- listaddressgroupings
- setlabel		- setlabel
- sendfrom (with account arguments)		- sendfrom (with account arguments)
- move (with account arguments)		- move (with account arguments)
"""		"""
		from collections import defaultdict

from test_framework.test_framework import BitcoinTestFramework		from test_framework.test_framework import BitcoinTestFramework
from test_framework.util import assert_equal		from test_framework.util import assert_equal


class WalletLabelsTest(BitcoinTestFramework):		class WalletLabelsTest(BitcoinTestFramework):
def set_test_params(self):		def set_test_params(self):
self.setup_clean_chain = True		self.setup_clean_chain = True
▲ Show 20 Lines • Show All 47 Lines • ▼ Show 20 Lines	def run_test(self):
node.generate(1)		node.generate(1)

amount_to_send = 1.0		amount_to_send = 1.0

# Create labels and make sure subsequent label API calls		# Create labels and make sure subsequent label API calls
# recognize the label/address associations.		# recognize the label/address associations.
labels = [Label(name) for name in ("a", "b", "c", "d", "e")]		labels = [Label(name) for name in ("a", "b", "c", "d", "e")]
for label in labels:		for label in labels:
label.add_receive_address(node.getlabeladdress(label.name))		label.add_receive_address(
		node.getlabeladdress(label=label.name, force=True))
label.verify(node)		label.verify(node)

		# Check all labels are returned by listlabels.
		assert_equal(node.listlabels(), [label.name for label in labels])

# Send a transaction to each label, and make sure this forces		# Send a transaction to each label, and make sure this forces
# getlabeladdress to generate a new receiving address.		# getlabeladdress to generate a new receiving address.
for label in labels:		for label in labels:
node.sendtoaddress(label.receive_address, amount_to_send)		node.sendtoaddress(label.receive_address, amount_to_send)
label.add_receive_address(node.getlabeladdress(label.name))		label.add_receive_address(node.getlabeladdress(label.name))
label.verify(node)		label.verify(node)

# Check the amounts received.		# Check the amounts received.
Show All 18 Lines	def run_test(self):
expected_account_balances = {"": 5200 + fee}		expected_account_balances = {"": 5200 + fee}
for label in labels:		for label in labels:
expected_account_balances[label.name] = 0		expected_account_balances[label.name] = 0
assert_equal(node.listaccounts(), expected_account_balances)		assert_equal(node.listaccounts(), expected_account_balances)
assert_equal(node.getbalance(""), 5200 + fee)		assert_equal(node.getbalance(""), 5200 + fee)

# Check that setlabel can assign a label to a new unused address.		# Check that setlabel can assign a label to a new unused address.
for label in labels:		for label in labels:
address = node.getlabeladdress("")		address = node.getlabeladdress(label="", force=True)
node.setlabel(address, label.name)		node.setlabel(address, label.name)
label.add_address(address)		label.add_address(address)
label.verify(node)		label.verify(node)
assert(address not in node.getaddressesbyaccount(""))		assert(address not in node.getaddressesbyaccount(""))

# Check that addmultisigaddress can assign labels.		# Check that addmultisigaddress can assign labels.
for label in labels:		for label in labels:
addresses = []		addresses = []
for x in range(10):		for x in range(10):
addresses.append(node.getnewaddress())		addresses.append(node.getnewaddress())
multisig_address = node.addmultisigaddress(		multisig_address = node.addmultisigaddress(
5, addresses, label.name)['address']		5, addresses, label.name)['address']
label.add_address(multisig_address)		label.add_address(multisig_address)
		label.purpose[multisig_address] = "send"
label.verify(node)		label.verify(node)
node.sendfrom("", multisig_address, 50)		node.sendfrom("", multisig_address, 50)
node.generate(101)		node.generate(101)
for label in labels:		for label in labels:
assert_equal(node.getbalance(label.name), 50)		assert_equal(node.getbalance(label.name), 50)

# Check that setlabel can change the label of an address from a		# Check that setlabel can change the label of an address from a
# different label.		# different label.
change_label(node, labels[0].addresses[0], labels[0], labels[1])		change_label(node, labels[0].addresses[0], labels[0], labels[1])

# Check that setlabel can change the label of an address which		# Check that setlabel can change the label of an address which
# is the receiving address of a different label.		# is the receiving address of a different label.
change_label(node, labels[0].receive_address, labels[0], labels[1])		change_label(node, labels[0].receive_address, labels[0], labels[1])

# Check that setlabel can set the label of an address already		# Check that setlabel can set the label of an address already
# in the label. This is a no-op.		# in the label. This is a no-op.
change_label(node, labels[2].addresses[0], labels[2], labels[2])		change_label(node, labels[2].addresses[0], labels[2], labels[2])

# Check that setlabel can set the label of an address which is		# Check that setlabel can set the label of an address which is
# already the receiving address of the label. It would probably make		# already the receiving address of the label. This is a no-op.
# sense for this to be a no-op, but right now it resets the receiving
# address, causing getlabeladdress to return a brand new address.
change_label(node, labels[2].receive_address, labels[2], labels[2])		change_label(node, labels[2].receive_address, labels[2], labels[2])


class Label:		class Label:
def __init__(self, name):		def __init__(self, name):
# Label name		# Label name
self.name = name		self.name = name
# Current receiving address associated with this label.		# Current receiving address associated with this label.
self.receive_address = None		self.receive_address = None
# List of all addresses assigned with this label		# List of all addresses assigned with this label
self.addresses = []		self.addresses = []
		# Map of address to address purpose
		self.purpose = defaultdict(lambda: "receive")

def add_address(self, address):		def add_address(self, address):
assert_equal(address not in self.addresses, True)		assert_equal(address not in self.addresses, True)
self.addresses.append(address)		self.addresses.append(address)

def add_receive_address(self, address):		def add_receive_address(self, address):
self.add_address(address)		self.add_address(address)
self.receive_address = address		self.receive_address = address

def verify(self, node):		def verify(self, node):
if self.receive_address is not None:		if self.receive_address is not None:
assert self.receive_address in self.addresses		assert self.receive_address in self.addresses
assert_equal(node.getlabeladdress(self.name), self.receive_address)		assert_equal(node.getlabeladdress(self.name), self.receive_address)

for address in self.addresses:		for address in self.addresses:
		assert_equal(
		node.getaddressinfo(address)['labels'][0],
		{"name": self.name,
		"purpose": self.purpose[address]})
assert_equal(node.getaccount(address), self.name)		assert_equal(node.getaccount(address), self.name)

assert_equal(		assert_equal(
		node.getaddressesbylabel(self.name),
		{address: {"purpose": self.purpose[address]} for address in self.addresses})
		assert_equal(
set(node.getaddressesbyaccount(self.name)), set(self.addresses))		set(node.getaddressesbyaccount(self.name)), set(self.addresses))


def change_label(node, address, old_label, new_label):		def change_label(node, address, old_label, new_label):
assert_equal(address in old_label.addresses, True)		assert_equal(address in old_label.addresses, True)
node.setlabel(address, new_label.name)		node.setlabel(address, new_label.name)

old_label.addresses.remove(address)		old_label.addresses.remove(address)
new_label.add_address(address)		new_label.add_address(address)

# Calling setlabel on an address which was previously the receiving		# Calling setlabel on an address which was previously the receiving
# address of a different label should reset the receiving address of		# address of a different label should reset the receiving address of
# the old label, causing getlabeladdress to return a brand new		# the old label, causing getlabeladdress to return a brand new
# address.		# address.
if address == old_label.receive_address:		if old_label.name != new_label.name and address == old_label.receive_address:
new_address = node.getlabeladdress(old_label.name)		new_address = node.getlabeladdress(old_label.name)
assert_equal(new_address not in old_label.addresses, True)		assert_equal(new_address not in old_label.addresses, True)
assert_equal(new_address not in new_label.addresses, True)		assert_equal(new_address not in new_label.addresses, True)
old_label.add_receive_address(new_address)		old_label.add_receive_address(new_address)

old_label.verify(node)		old_label.verify(node)
new_label.verify(node)		new_label.verify(node)


if __name__ == '__main__':		if __name__ == '__main__':
WalletLabelsTest().main()		WalletLabelsTest().main()