Skip to content

Commit

Permalink
qa: Remove mocktime unless required
Browse files Browse the repository at this point in the history
  • Loading branch information
MarcoFalke committed Feb 19, 2019
1 parent 3e4fd40 commit fab0d85
Show file tree
Hide file tree
Showing 4 changed files with 24 additions and 22 deletions.
12 changes: 12 additions & 0 deletions test/functional/rpc_blockchain.py
Expand Up @@ -35,6 +35,7 @@
from test_framework.blocktools import (
create_block,
create_coinbase,
TIME_GENESIS_BLOCK,
)
from test_framework.messages import (
msg_block,
Expand All @@ -46,9 +47,11 @@

class BlockchainTest(BitcoinTestFramework):
def set_test_params(self):
self.setup_clean_chain = True
self.num_nodes = 1

def run_test(self):
self.mine_chain()
self.restart_node(0, extra_args=['-stopatheight=207', '-prune=1']) # Set extra args with pruning after rescan is complete

self._test_getblockchaininfo()
Expand All @@ -61,6 +64,15 @@ def run_test(self):
self._test_waitforblockheight()
assert self.nodes[0].verifychain(4, 0)

def mine_chain(self):
self.log.info('Create some old blocks')
address = self.nodes[0].get_deterministic_priv_key().address
for t in range(TIME_GENESIS_BLOCK, TIME_GENESIS_BLOCK + 200 * 600, 600):
# ten-minute steps from genesis block time
self.nodes[0].setmocktime(t)
self.nodes[0].generatetoaddress(1, address)
assert_equal(self.nodes[0].getblockchaininfo()['blocks'], 200)

def _test_getblockchaininfo(self):
self.log.info("Test getblockchaininfo")

Expand Down
19 changes: 1 addition & 18 deletions test/functional/test_framework/test_framework.py
Expand Up @@ -29,11 +29,11 @@
get_datadir_path,
initialize_datadir,
p2p_port,
set_node_times,
sync_blocks,
sync_mempools,
)


class TestStatus(Enum):
PASSED = 1
FAILED = 2
Expand Down Expand Up @@ -94,7 +94,6 @@ def __init__(self):
self.setup_clean_chain = False
self.nodes = []
self.network_thread = None
self.mocktime = 0
self.rpc_timeout = 60 # Wait for up to 60 seconds for the RPC server to respond
self.supports_cli = False
self.bind_to_localhost_only = True
Expand Down Expand Up @@ -316,7 +315,6 @@ def add_nodes(self, num_nodes, extra_args=None, *, rpchost=None, binary=None):
timewait=self.rpc_timeout,
bitcoind=binary[i],
bitcoin_cli=self.options.bitcoincli,
mocktime=self.mocktime,
coverage_dir=self.options.coveragedir,
cwd=self.options.tmpdir,
extra_conf=extra_confs[i],
Expand Down Expand Up @@ -468,7 +466,6 @@ def _initialize_chain(self):
timewait=self.rpc_timeout,
bitcoind=self.options.bitcoind,
bitcoin_cli=self.options.bitcoincli,
mocktime=self.mocktime,
coverage_dir=None,
cwd=self.options.tmpdir,
))
Expand All @@ -479,32 +476,18 @@ def _initialize_chain(self):
for node in self.nodes:
node.wait_for_rpc_connection()

# For backward compatibility of the python scripts with previous
# versions of the cache, set mocktime to Jan 1,
# 2014 + (201 * 10 * 60)"""
self.mocktime = 1388534400 + (201 * 10 * 60)

# Create a 200-block-long chain; each of the 4 first nodes
# gets 25 mature blocks and 25 immature.
# Note: To preserve compatibility with older versions of
# initialize_chain, only 4 nodes will generate coins.
#
# blocks are created with timestamps 10 minutes apart
# starting from 2010 minutes in the past
block_time = self.mocktime - (201 * 10 * 60)
for i in range(2):
for peer in range(4):
for j in range(25):
set_node_times(self.nodes, block_time)
self.nodes[peer].generatetoaddress(1, self.nodes[peer].get_deterministic_priv_key().address)
block_time += 10 * 60
# Must sync before next peer starts generating blocks
sync_blocks(self.nodes)

# Shut them down, and clean up cache directories:
self.stop_nodes()
self.nodes = []
self.mocktime = 0

def cache_path(n, *paths):
return os.path.join(get_datadir_path(self.options.cachedir, n), "regtest", *paths)
Expand Down
5 changes: 2 additions & 3 deletions test/functional/test_framework/test_node.py
Expand Up @@ -61,7 +61,7 @@ class TestNode():
To make things easier for the test writer, any unrecognised messages will
be dispatched to the RPC connection."""

def __init__(self, i, datadir, *, rpchost, timewait, bitcoind, bitcoin_cli, mocktime, coverage_dir, cwd, extra_conf=None, extra_args=None, use_cli=False, start_perf=False):
def __init__(self, i, datadir, *, rpchost, timewait, bitcoind, bitcoin_cli, coverage_dir, cwd, extra_conf=None, extra_args=None, use_cli=False, start_perf=False):
"""
Kwargs:
start_perf (bool): If True, begin profiling the node with `perf` as soon as
Expand Down Expand Up @@ -90,8 +90,7 @@ def __init__(self, i, datadir, *, rpchost, timewait, bitcoind, bitcoin_cli, mock
"-debug",
"-debugexclude=libevent",
"-debugexclude=leveldb",
"-mocktime=" + str(mocktime),
"-uacomment=testnode%d" % i
"-uacomment=testnode%d" % i,
]

self.cli = TestNodeCLI(bitcoin_cli, self.datadir)
Expand Down
10 changes: 9 additions & 1 deletion test/functional/wallet_create_tx.py
Expand Up @@ -7,17 +7,25 @@
from test_framework.util import (
assert_equal,
)
from test_framework.blocktools import (
TIME_GENESIS_BLOCK,
)


class CreateTxWalletTest(BitcoinTestFramework):
def set_test_params(self):
self.setup_clean_chain = False
self.setup_clean_chain = True
self.num_nodes = 1

def skip_test_if_missing_module(self):
self.skip_if_no_wallet()

def run_test(self):
self.log.info('Create some old blocks')
self.nodes[0].setmocktime(TIME_GENESIS_BLOCK)
self.nodes[0].generate(200)
self.nodes[0].setmocktime(0)

self.log.info('Check that we have some (old) blocks and that anti-fee-sniping is disabled')
assert_equal(self.nodes[0].getblockchaininfo()['blocks'], 200)
txid = self.nodes[0].sendtoaddress(self.nodes[0].getnewaddress(), 1)
Expand Down

0 comments on commit fab0d85

Please sign in to comment.