[test] Test abortrescan command.
This commit is contained in:
parent
a86255b116
commit
ed60970c83
2 changed files with 67 additions and 0 deletions
66
test/functional/import-abort-rescan.py
Executable file
66
test/functional/import-abort-rescan.py
Executable file
|
@ -0,0 +1,66 @@
|
||||||
|
#!/usr/bin/env python3
|
||||||
|
# Copyright (c) 2017 The Bitcoin Core developers
|
||||||
|
# Distributed under the MIT software license, see the accompanying
|
||||||
|
# file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
||||||
|
"""Test wallet import RPCs.
|
||||||
|
|
||||||
|
Test rescan behavior of importprivkey when aborted. The test ensures that:
|
||||||
|
1. The abortrescan command indeed stops the rescan process.
|
||||||
|
2. Subsequent rescan catches the aborted address UTXO
|
||||||
|
"""
|
||||||
|
|
||||||
|
from test_framework.test_framework import BitcoinTestFramework
|
||||||
|
from test_framework.util import (assert_equal, get_rpc_proxy)
|
||||||
|
from decimal import Decimal
|
||||||
|
import threading # for bg importprivkey
|
||||||
|
import time # for sleep
|
||||||
|
|
||||||
|
class ImportAbortRescanTest(BitcoinTestFramework):
|
||||||
|
def __init__(self):
|
||||||
|
super().__init__()
|
||||||
|
self.setup_clean_chain = True
|
||||||
|
|
||||||
|
def run_test(self):
|
||||||
|
# Generate for BTC
|
||||||
|
assert_equal(self.nodes[0].getbalance(), 0)
|
||||||
|
assert_equal(self.nodes[1].getbalance(), 0)
|
||||||
|
self.nodes[0].generate(300)
|
||||||
|
assert_equal(self.nodes[1].getbalance(), 0)
|
||||||
|
# Make blocks with spam to cause rescan delay
|
||||||
|
for i in range(5):
|
||||||
|
for j in range(5):
|
||||||
|
self.nodes[0].sendtoaddress(self.nodes[0].getnewaddress(), 0.1)
|
||||||
|
self.nodes[0].generate(10)
|
||||||
|
addr = self.nodes[0].getnewaddress()
|
||||||
|
privkey = self.nodes[0].dumpprivkey(addr)
|
||||||
|
self.nodes[0].sendtoaddress(addr, 0.123)
|
||||||
|
self.nodes[0].generate(10) # mature tx
|
||||||
|
self.sync_all()
|
||||||
|
|
||||||
|
# Import this address in the background ...
|
||||||
|
node1ref = get_rpc_proxy(self.nodes[1].url, 1, timeout=600)
|
||||||
|
importthread = threading.Thread(target=node1ref.importprivkey, args=[privkey])
|
||||||
|
importthread.start()
|
||||||
|
# ... then abort rescan; try a bunch until abortres becomes true,
|
||||||
|
# because we will start checking before above thread starts processing
|
||||||
|
for i in range(2000):
|
||||||
|
time.sleep(0.001)
|
||||||
|
abortres = self.nodes[1].abortrescan()
|
||||||
|
if abortres: break
|
||||||
|
assert abortres # if false, we failed to abort
|
||||||
|
# import should die soon
|
||||||
|
for i in range(10):
|
||||||
|
time.sleep(0.1)
|
||||||
|
deadres = not importthread.isAlive()
|
||||||
|
if deadres: break
|
||||||
|
|
||||||
|
assert deadres # if false, importthread did not die soon enough
|
||||||
|
assert_equal(self.nodes[1].getbalance(), 0.0)
|
||||||
|
|
||||||
|
# Import a different address and let it run
|
||||||
|
self.nodes[1].importprivkey(self.nodes[0].dumpprivkey(self.nodes[0].getnewaddress()))
|
||||||
|
# Expect original privkey to now also be discovered and added to balance
|
||||||
|
assert_equal(self.nodes[1].getbalance(), Decimal("0.123"))
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
ImportAbortRescanTest().main()
|
|
@ -108,6 +108,7 @@ BASE_SCRIPTS= [
|
||||||
'rpcnamedargs.py',
|
'rpcnamedargs.py',
|
||||||
'listsinceblock.py',
|
'listsinceblock.py',
|
||||||
'p2p-leaktests.py',
|
'p2p-leaktests.py',
|
||||||
|
'import-abort-rescan.py',
|
||||||
]
|
]
|
||||||
|
|
||||||
ZMQ_SCRIPTS = [
|
ZMQ_SCRIPTS = [
|
||||||
|
|
Loading…
Reference in a new issue