Pedro Branco
9 years ago
7 changed files with 698 additions and 0 deletions
@ -0,0 +1,293 @@
@@ -0,0 +1,293 @@
|
||||
#!/usr/bin/env python3 |
||||
# Copyright (c) 2014-2016 The Bitcoin Core 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 * |
||||
|
||||
class ImportMultiTest (BitcoinTestFramework): |
||||
def __init__(self): |
||||
super().__init__() |
||||
self.num_nodes = 2 |
||||
self.setup_clean_chain = True |
||||
|
||||
def setup_network(self, split=False): |
||||
self.nodes = start_nodes(2, self.options.tmpdir) |
||||
self.is_network_split=False |
||||
|
||||
def run_test (self): |
||||
print ("Mining blocks...") |
||||
self.nodes[0].generate(1) |
||||
self.nodes[1].generate(1) |
||||
|
||||
# keyword definition |
||||
PRIV_KEY = 'privkey' |
||||
PUB_KEY = 'pubkey' |
||||
ADDRESS_KEY = 'address' |
||||
SCRIPT_KEY = 'script' |
||||
|
||||
|
||||
node0_address1 = self.nodes[0].validateaddress(self.nodes[0].getnewaddress()) |
||||
node0_address2 = self.nodes[0].validateaddress(self.nodes[0].getnewaddress()) |
||||
node0_address3 = self.nodes[0].validateaddress(self.nodes[0].getnewaddress()) |
||||
|
||||
#Check only one address |
||||
assert_equal(node0_address1['ismine'], True) |
||||
|
||||
#Node 1 sync test |
||||
assert_equal(self.nodes[1].getblockcount(),1) |
||||
|
||||
#Address Test - before import |
||||
address_info = self.nodes[1].validateaddress(node0_address1['address']) |
||||
assert_equal(address_info['iswatchonly'], False) |
||||
assert_equal(address_info['ismine'], False) |
||||
|
||||
|
||||
# RPC importmulti ----------------------------------------------- |
||||
|
||||
# Bitcoin Address |
||||
print("Should import an address") |
||||
address = self.nodes[0].validateaddress(self.nodes[0].getnewaddress()) |
||||
result = self.nodes[1].importmulti([{ |
||||
"scriptPubKey": { |
||||
"address": address['address'] |
||||
} |
||||
}]) |
||||
assert_equal(result[0]['success'], True) |
||||
address_assert = self.nodes[1].validateaddress(address['address']) |
||||
assert_equal(address_assert['iswatchonly'], True) |
||||
assert_equal(address_assert['ismine'], False) |
||||
|
||||
|
||||
# ScriptPubKey + internal |
||||
print("Should import a scriptPubKey with internal flag") |
||||
address = self.nodes[0].validateaddress(self.nodes[0].getnewaddress()) |
||||
result = self.nodes[1].importmulti([{ |
||||
"scriptPubKey": address['scriptPubKey'], |
||||
"internal": True |
||||
}]) |
||||
assert_equal(result[0]['success'], True) |
||||
address_assert = self.nodes[1].validateaddress(address['address']) |
||||
assert_equal(address_assert['iswatchonly'], True) |
||||
assert_equal(address_assert['ismine'], False) |
||||
|
||||
# ScriptPubKey + !internal |
||||
print("Should not import a scriptPubKey without internal flag") |
||||
address = self.nodes[0].validateaddress(self.nodes[0].getnewaddress()) |
||||
result = self.nodes[1].importmulti([{ |
||||
"scriptPubKey": address['scriptPubKey'] |
||||
}]) |
||||
assert_equal(result[0]['success'], False) |
||||
assert_equal(result[0]['error']['code'], -8) |
||||
assert_equal(result[0]['error']['message'], 'Internal must be set for hex scriptPubKey') |
||||
address_assert = self.nodes[1].validateaddress(address['address']) |
||||
assert_equal(address_assert['iswatchonly'], False) |
||||
assert_equal(address_assert['ismine'], False) |
||||
|
||||
|
||||
# Address + Public key + !Internal |
||||
print("Should import an address with public key") |
||||
address = self.nodes[0].validateaddress(self.nodes[0].getnewaddress()) |
||||
result = self.nodes[1].importmulti([{ |
||||
"scriptPubKey": { |
||||
"address": address['address'] |
||||
}, |
||||
"pubkeys": [ address['pubkey'] ] |
||||
}]) |
||||
assert_equal(result[0]['success'], True) |
||||
address_assert = self.nodes[1].validateaddress(address['address']) |
||||
assert_equal(address_assert['iswatchonly'], True) |
||||
assert_equal(address_assert['ismine'], False) |
||||
|
||||
|
||||
# ScriptPubKey + Public key + internal |
||||
print("Should import a scriptPubKey with internal and with public key") |
||||
address = self.nodes[0].validateaddress(self.nodes[0].getnewaddress()) |
||||
request = [{ |
||||
"scriptPubKey": address['scriptPubKey'], |
||||
"pubkeys": [ address['pubkey'] ], |
||||
"internal": True |
||||
}]; |
||||
result = self.nodes[1].importmulti(request) |
||||
assert_equal(result[0]['success'], True) |
||||
address_assert = self.nodes[1].validateaddress(address['address']) |
||||
assert_equal(address_assert['iswatchonly'], True) |
||||
assert_equal(address_assert['ismine'], False) |
||||
|
||||
# ScriptPubKey + Public key + !internal |
||||
print("Should not import a scriptPubKey without internal and with public key") |
||||
address = self.nodes[0].validateaddress(self.nodes[0].getnewaddress()) |
||||
request = [{ |
||||
"scriptPubKey": address['scriptPubKey'], |
||||
"pubkeys": [ address['pubkey'] ] |
||||
}]; |
||||
result = self.nodes[1].importmulti(request) |
||||
assert_equal(result[0]['success'], False) |
||||
assert_equal(result[0]['error']['code'], -8) |
||||
assert_equal(result[0]['error']['message'], 'Internal must be set for hex scriptPubKey') |
||||
address_assert = self.nodes[1].validateaddress(address['address']) |
||||
assert_equal(address_assert['iswatchonly'], False) |
||||
assert_equal(address_assert['ismine'], False) |
||||
|
||||
# Address + Private key + !watchonly |
||||
print("Should import an address with private key") |
||||
address = self.nodes[0].validateaddress(self.nodes[0].getnewaddress()) |
||||
result = self.nodes[1].importmulti([{ |
||||
"scriptPubKey": { |
||||
"address": address['address'] |
||||
}, |
||||
"keys": [ self.nodes[0].dumpprivkey(address['address']) ] |
||||
}]) |
||||
assert_equal(result[0]['success'], True) |
||||
address_assert = self.nodes[1].validateaddress(address['address']) |
||||
assert_equal(address_assert['iswatchonly'], False) |
||||
assert_equal(address_assert['ismine'], True) |
||||
|
||||
# Address + Private key + watchonly |
||||
print("Should not import an address with private key and with watchonly") |
||||
address = self.nodes[0].validateaddress(self.nodes[0].getnewaddress()) |
||||
result = self.nodes[1].importmulti([{ |
||||
"scriptPubKey": { |
||||
"address": address['address'] |
||||
}, |
||||
"keys": [ self.nodes[0].dumpprivkey(address['address']) ], |
||||
"watchonly": True |
||||
}]) |
||||
assert_equal(result[0]['success'], False) |
||||
assert_equal(result[0]['error']['code'], -8) |
||||
assert_equal(result[0]['error']['message'], 'Incompatibility found between watchonly and keys') |
||||
address_assert = self.nodes[1].validateaddress(address['address']) |
||||
assert_equal(address_assert['iswatchonly'], False) |
||||
assert_equal(address_assert['ismine'], False) |
||||
|
||||
# ScriptPubKey + Private key + internal |
||||
print("Should import a scriptPubKey with internal and with private key") |
||||
address = self.nodes[0].validateaddress(self.nodes[0].getnewaddress()) |
||||
result = self.nodes[1].importmulti([{ |
||||
"scriptPubKey": address['scriptPubKey'], |
||||
"keys": [ self.nodes[0].dumpprivkey(address['address']) ], |
||||
"internal": True |
||||
}]) |
||||
assert_equal(result[0]['success'], True) |
||||
address_assert = self.nodes[1].validateaddress(address['address']) |
||||
assert_equal(address_assert['iswatchonly'], False) |
||||
assert_equal(address_assert['ismine'], True) |
||||
|
||||
# ScriptPubKey + Private key + !internal |
||||
print("Should not import a scriptPubKey without internal and with private key") |
||||
address = self.nodes[0].validateaddress(self.nodes[0].getnewaddress()) |
||||
result = self.nodes[1].importmulti([{ |
||||
"scriptPubKey": address['scriptPubKey'], |
||||
"keys": [ self.nodes[0].dumpprivkey(address['address']) ] |
||||
}]) |
||||
assert_equal(result[0]['success'], False) |
||||
assert_equal(result[0]['error']['code'], -8) |
||||
assert_equal(result[0]['error']['message'], 'Internal must be set for hex scriptPubKey') |
||||
address_assert = self.nodes[1].validateaddress(address['address']) |
||||
assert_equal(address_assert['iswatchonly'], False) |
||||
assert_equal(address_assert['ismine'], False) |
||||
|
||||
|
||||
# P2SH address |
||||
sig_address_1 = self.nodes[0].validateaddress(self.nodes[0].getnewaddress()) |
||||
sig_address_2 = self.nodes[0].validateaddress(self.nodes[0].getnewaddress()) |
||||
sig_address_3 = self.nodes[0].validateaddress(self.nodes[0].getnewaddress()) |
||||
multi_sig_script = self.nodes[0].createmultisig(2, [sig_address_1['address'], sig_address_2['address'], sig_address_3['pubkey']]) |
||||
self.nodes[1].generate(100) |
||||
transactionid = self.nodes[1].sendtoaddress(multi_sig_script['address'], 10.00) |
||||
self.nodes[1].generate(1) |
||||
transaction = self.nodes[1].gettransaction(transactionid); |
||||
|
||||
print("Should import a p2sh") |
||||
result = self.nodes[1].importmulti([{ |
||||
"scriptPubKey": { |
||||
"address": multi_sig_script['address'] |
||||
} |
||||
}]) |
||||
assert_equal(result[0]['success'], True) |
||||
address_assert = self.nodes[1].validateaddress(multi_sig_script['address']) |
||||
assert_equal(address_assert['isscript'], True) |
||||
assert_equal(address_assert['iswatchonly'], True) |
||||
p2shunspent = self.nodes[1].listunspent(0,999999, [multi_sig_script['address']])[0] |
||||
assert_equal(p2shunspent['spendable'], False) |
||||
assert_equal(p2shunspent['solvable'], False) |
||||
|
||||
|
||||
# P2SH + Redeem script |
||||
sig_address_1 = self.nodes[0].validateaddress(self.nodes[0].getnewaddress()) |
||||
sig_address_2 = self.nodes[0].validateaddress(self.nodes[0].getnewaddress()) |
||||
sig_address_3 = self.nodes[0].validateaddress(self.nodes[0].getnewaddress()) |
||||
multi_sig_script = self.nodes[0].createmultisig(2, [sig_address_1['address'], sig_address_2['address'], sig_address_3['pubkey']]) |
||||
self.nodes[1].generate(100) |
||||
transactionid = self.nodes[1].sendtoaddress(multi_sig_script['address'], 10.00) |
||||
self.nodes[1].generate(1) |
||||
transaction = self.nodes[1].gettransaction(transactionid); |
||||
|
||||
print("Should import a p2sh with respective redeem script") |
||||
result = self.nodes[1].importmulti([{ |
||||
"scriptPubKey": { |
||||
"address": multi_sig_script['address'] |
||||
}, |
||||
"redeemscript": multi_sig_script['redeemScript'] |
||||
}]) |
||||
assert_equal(result[0]['success'], True) |
||||
|
||||
p2shunspent = self.nodes[1].listunspent(0,999999, [multi_sig_script['address']])[0] |
||||
assert_equal(p2shunspent['spendable'], False) |
||||
assert_equal(p2shunspent['solvable'], True) |
||||
|
||||
|
||||
# P2SH + Redeem script + Private Keys + !Watchonly |
||||
sig_address_1 = self.nodes[0].validateaddress(self.nodes[0].getnewaddress()) |
||||
sig_address_2 = self.nodes[0].validateaddress(self.nodes[0].getnewaddress()) |
||||
sig_address_3 = self.nodes[0].validateaddress(self.nodes[0].getnewaddress()) |
||||
multi_sig_script = self.nodes[0].createmultisig(2, [sig_address_1['address'], sig_address_2['address'], sig_address_3['pubkey']]) |
||||
self.nodes[1].generate(100) |
||||
transactionid = self.nodes[1].sendtoaddress(multi_sig_script['address'], 10.00) |
||||
self.nodes[1].generate(1) |
||||
transaction = self.nodes[1].gettransaction(transactionid); |
||||
|
||||
print("Should import a p2sh with respective redeem script and private keys") |
||||
result = self.nodes[1].importmulti([{ |
||||
"scriptPubKey": { |
||||
"address": multi_sig_script['address'] |
||||
}, |
||||
"redeemscript": multi_sig_script['redeemScript'], |
||||
"keys": [ self.nodes[0].dumpprivkey(sig_address_1['address']), self.nodes[0].dumpprivkey(sig_address_2['address'])] |
||||
}]) |
||||
assert_equal(result[0]['success'], True) |
||||
|
||||
p2shunspent = self.nodes[1].listunspent(0,999999, [multi_sig_script['address']])[0] |
||||
assert_equal(p2shunspent['spendable'], False) |
||||
assert_equal(p2shunspent['solvable'], True) |
||||
|
||||
# P2SH + Redeem script + Private Keys + Watchonly |
||||
sig_address_1 = self.nodes[0].validateaddress(self.nodes[0].getnewaddress()) |
||||
sig_address_2 = self.nodes[0].validateaddress(self.nodes[0].getnewaddress()) |
||||
sig_address_3 = self.nodes[0].validateaddress(self.nodes[0].getnewaddress()) |
||||
multi_sig_script = self.nodes[0].createmultisig(2, [sig_address_1['address'], sig_address_2['address'], sig_address_3['pubkey']]) |
||||
self.nodes[1].generate(100) |
||||
transactionid = self.nodes[1].sendtoaddress(multi_sig_script['address'], 10.00) |
||||
self.nodes[1].generate(1) |
||||
transaction = self.nodes[1].gettransaction(transactionid); |
||||
|
||||
print("Should import a p2sh with respective redeem script and private keys") |
||||
result = self.nodes[1].importmulti([{ |
||||
"scriptPubKey": { |
||||
"address": multi_sig_script['address'] |
||||
}, |
||||
"redeemscript": multi_sig_script['redeemScript'], |
||||
"keys": [ self.nodes[0].dumpprivkey(sig_address_1['address']), self.nodes[0].dumpprivkey(sig_address_2['address'])], |
||||
"watchonly": True |
||||
}]) |
||||
assert_equal(result[0]['success'], False) |
||||
assert_equal(result[0]['error']['code'], -8) |
||||
assert_equal(result[0]['error']['message'], 'Incompatibility found between watchonly and keys') |
||||
|
||||
# TODO Consistency tests? |
||||
|
||||
|
||||
|
||||
if __name__ == '__main__': |
||||
ImportMultiTest ().main () |
Loading…
Reference in new issue