78 lines
2.5 KiB
Python
Executable File
78 lines
2.5 KiB
Python
Executable File
#!/usr/bin/env python2
|
|
# Copyright (c) 2018 The Zcash developers
|
|
# Distributed under the MIT software license, see the accompanying
|
|
# file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
|
|
from test_framework.test_framework import BitcoinTestFramework
|
|
from test_framework.util import assert_equal, start_nodes
|
|
|
|
# Test wallet address behaviour across network upgradesa\
|
|
class WalletAddressesTest(BitcoinTestFramework):
|
|
|
|
def setup_nodes(self):
|
|
return start_nodes(4, self.options.tmpdir, [[
|
|
'-nuparams=5ba81b19:202', # Overwinter
|
|
'-nuparams=76b809bb:204', # Sapling
|
|
]] * 4)
|
|
|
|
def run_test(self):
|
|
def addr_checks(default_type):
|
|
# Check default type, as well as explicit types
|
|
types_and_addresses = [
|
|
(default_type, self.nodes[0].z_getnewaddress()),
|
|
('sprout', self.nodes[0].z_getnewaddress('sprout')),
|
|
('sapling', self.nodes[0].z_getnewaddress('sapling')),
|
|
]
|
|
|
|
all_addresses = self.nodes[0].z_listaddresses()
|
|
|
|
for addr_type, addr in types_and_addresses:
|
|
res = self.nodes[0].z_validateaddress(addr)
|
|
assert(res['isvalid'])
|
|
assert(res['ismine'])
|
|
assert_equal(res['type'], addr_type)
|
|
assert(addr in all_addresses)
|
|
|
|
# Sanity-check the test harness
|
|
assert_equal(self.nodes[0].getblockcount(), 200)
|
|
|
|
# Current height = 200 -> Sprout
|
|
# Default address type is Sapling
|
|
print "Testing height 200 (Sprout)"
|
|
addr_checks('sapling')
|
|
|
|
self.nodes[0].generate(1)
|
|
self.sync_all()
|
|
|
|
# Current height = 201 -> Sprout
|
|
# Default address type is Sapling
|
|
print "Testing height 201 (Sprout)"
|
|
addr_checks('sapling')
|
|
|
|
self.nodes[0].generate(1)
|
|
self.sync_all()
|
|
|
|
# Current height = 202 -> Overwinter
|
|
# Default address type is Sapling
|
|
print "Testing height 202 (Overwinter)"
|
|
addr_checks('sapling')
|
|
|
|
self.nodes[0].generate(1)
|
|
self.sync_all()
|
|
|
|
# Current height = 203 -> Overwinter
|
|
# Default address type is Sapling
|
|
print "Testing height 203 (Overwinter)"
|
|
addr_checks('sapling')
|
|
|
|
self.nodes[0].generate(1)
|
|
self.sync_all()
|
|
|
|
# Current height = 204 -> Sapling
|
|
# Default address type is Sapling
|
|
print "Testing height 204 (Sapling)"
|
|
addr_checks('sapling')
|
|
|
|
if __name__ == '__main__':
|
|
WalletAddressesTest().main()
|