LBRYCommunity/lbrycrd
1
0
Fork 0
Code Issues 29 Pull requests 1 Projects Releases 32 Packages Wiki Activity

[qa] Extend import-rescan.py to test specific key timestamps

Browse source
This commit is contained in:
Russell Yanofsky 2017-02-14 17:39:26 -05:00
parent 8be0866883
commit c28583d062
1 changed files with 15 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
qa/rpc-tests/import-rescan.py
View file

@ -20,7 +20,7 @@ happened previously.
""" """
from test_framework.test_framework import BitcoinTestFramework from test_framework.test_framework import BitcoinTestFramework
from test_framework.util import (start_nodes, connect_nodes, sync_blocks, assert_equal) from test_framework.util import (start_nodes, connect_nodes, sync_blocks, assert_equal, set_node_times)
from decimal import Decimal from decimal import Decimal
import collections import collections
@ -29,13 +29,13 @@ import itertools
Call = enum.Enum("Call", "single multi") Call = enum.Enum("Call", "single multi")
Data = enum.Enum("Data", "address pub priv") Data = enum.Enum("Data", "address pub priv")
Rescan = enum.Enum("Rescan", "no yes") Rescan = enum.Enum("Rescan", "no yes late_timestamp")
class Variant(collections.namedtuple("Variant", "call data rescan")): class Variant(collections.namedtuple("Variant", "call data rescan")):
"""Helper for importing one key and verifying scanned transactions.""" """Helper for importing one key and verifying scanned transactions."""
def do_import(self): def do_import(self, timestamp):
"""Call one key import RPC.""" """Call one key import RPC."""
if self.call == Call.single: if self.call == Call.single:
@ -51,15 +51,15 @@ class Variant(collections.namedtuple("Variant", "call data rescan")):
"scriptPubKey": { "scriptPubKey": {
"address": self.address["address"] "address": self.address["address"]
}, },
"timestamp": "now", "timestamp": timestamp + (1 if self.rescan == Rescan.late_timestamp else 0),
"pubkeys": [self.address["pubkey"]] if self.data == Data.pub else [], "pubkeys": [self.address["pubkey"]] if self.data == Data.pub else [],
"keys": [self.key] if self.data == Data.priv else [], "keys": [self.key] if self.data == Data.priv else [],
"label": self.label, "label": self.label,
"watchonly": self.data != Data.priv "watchonly": self.data != Data.priv
}], {"rescan": self.rescan == Rescan.yes}) }], {"rescan": self.rescan in (Rescan.yes, Rescan.late_timestamp)})
assert_equal(response, [{"success": True}]) assert_equal(response, [{"success": True}])
def check(self, txid=None, amount=None): def check(self, txid=None, amount=None, confirmations=None):
"""Verify that getbalance/listtransactions return expected values.""" """Verify that getbalance/listtransactions return expected values."""
balance = self.node.getbalance(self.label, 0, True) balance = self.node.getbalance(self.label, 0, True)
@ -76,7 +76,7 @@ class Variant(collections.namedtuple("Variant", "call data rescan")):
assert_equal(tx["category"], "receive") assert_equal(tx["category"], "receive")
assert_equal(tx["label"], self.label) assert_equal(tx["label"], self.label)
assert_equal(tx["txid"], txid) assert_equal(tx["txid"], txid)
assert_equal(tx["confirmations"], 1) assert_equal(tx["confirmations"], confirmations)
assert_equal("trusted" not in tx, True) assert_equal("trusted" not in tx, True)
if self.data != Data.priv: if self.data != Data.priv:
assert_equal(tx["involvesWatchonly"], True) assert_equal(tx["involvesWatchonly"], True)
@ -109,9 +109,13 @@ class ImportRescanTest(BitcoinTestFramework):
variant.initial_amount = 25 - (i + 1) / 4.0 variant.initial_amount = 25 - (i + 1) / 4.0
variant.initial_txid = self.nodes[0].sendtoaddress(variant.address["address"], variant.initial_amount) variant.initial_txid = self.nodes[0].sendtoaddress(variant.address["address"], variant.initial_amount)
# Generate a block containing the initial transactions. # Generate a block containing the initial transactions, then another
# block further in the future (past the rescan window).
self.nodes[0].generate(1) self.nodes[0].generate(1)
assert_equal(self.nodes[0].getrawmempool(), []) assert_equal(self.nodes[0].getrawmempool(), [])
timestamp = self.nodes[0].getblockheader(self.nodes[0].getbestblockhash())["time"]
set_node_times(self.nodes, timestamp + 1)
self.nodes[0].generate(1)
sync_blocks(self.nodes) sync_blocks(self.nodes)
# For each variation of wallet key import, invoke the import RPC and # For each variation of wallet key import, invoke the import RPC and
@ -124,11 +128,11 @@ class ImportRescanTest(BitcoinTestFramework):
# balances in the second part of the test.) # balances in the second part of the test.)
for variant in IMPORT_VARIANTS: for variant in IMPORT_VARIANTS:
variant.node = self.nodes[1 if variant.rescan == Rescan.yes else 2] variant.node = self.nodes[1 if variant.rescan == Rescan.yes else 2]
variant.do_import() variant.do_import(timestamp)
if variant.rescan == Rescan.yes: if variant.rescan == Rescan.yes:
variant.expected_balance = variant.initial_amount variant.expected_balance = variant.initial_amount
variant.expected_txs = 1 variant.expected_txs = 1
variant.check(variant.initial_txid, variant.initial_amount) variant.check(variant.initial_txid, variant.initial_amount, 2)
else: else:
variant.expected_balance = 0 variant.expected_balance = 0
variant.expected_txs = 0 variant.expected_txs = 0
@ -149,7 +153,7 @@ class ImportRescanTest(BitcoinTestFramework):
for variant in IMPORT_VARIANTS: for variant in IMPORT_VARIANTS:
variant.expected_balance += variant.sent_amount variant.expected_balance += variant.sent_amount
variant.expected_txs += 1 variant.expected_txs += 1
variant.check(variant.sent_txid, variant.sent_amount) variant.check(variant.sent_txid, variant.sent_amount, 1)
if __name__ == "__main__": if __name__ == "__main__":