John Newbery
8 years ago
2 changed files with 0 additions and 67 deletions
@ -1,66 +0,0 @@ |
|||||||
#!/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() |
|
Loading…
Reference in new issue