![laanwj@gmail.com](/assets/img/avatar_default.png)
2 changed files with 97 additions and 1 deletions
@ -0,0 +1,94 @@ |
|||
#!/usr/bin/env python |
|||
# Copyright (c) 2014 The Bitcoin Core developers |
|||
# Distributed under the MIT/X11 software license, see the accompanying |
|||
# file COPYING or http://www.opensource.org/licenses/mit-license.php. |
|||
|
|||
# Exercise the listtransactions API |
|||
|
|||
from test_framework import BitcoinTestFramework |
|||
from bitcoinrpc.authproxy import AuthServiceProxy, JSONRPCException |
|||
from util import * |
|||
|
|||
|
|||
def check_array_result(object_array, to_match, expected): |
|||
""" |
|||
Pass in array of JSON objects, a dictionary with key/value pairs |
|||
to match against, and another dictionary with expected key/value |
|||
pairs. |
|||
""" |
|||
num_matched = 0 |
|||
for item in object_array: |
|||
all_match = True |
|||
for key,value in to_match.items(): |
|||
if item[key] != value: |
|||
all_match = False |
|||
if not all_match: |
|||
continue |
|||
for key,value in expected.items(): |
|||
if item[key] != value: |
|||
raise AssertionError("%s : expected %s=%s"%(str(item), str(key), str(value))) |
|||
num_matched = num_matched+1 |
|||
if num_matched == 0: |
|||
raise AssertionError("No objects matched %s"%(str(to_match))) |
|||
|
|||
import threading |
|||
|
|||
class LongpollThread(threading.Thread): |
|||
def __init__(self, node): |
|||
threading.Thread.__init__(self) |
|||
# query current longpollid |
|||
templat = node.getblocktemplate() |
|||
self.longpollid = templat['longpollid'] |
|||
# create a new connection to the node, we can't use the same |
|||
# connection from two threads |
|||
self.node = AuthServiceProxy(node.url, timeout=600) |
|||
|
|||
def run(self): |
|||
self.node.getblocktemplate({'longpollid':self.longpollid}) |
|||
|
|||
class GetBlockTemplateTest(BitcoinTestFramework): |
|||
''' |
|||
Test longpolling with getblocktemplate. |
|||
''' |
|||
|
|||
def run_test(self, nodes): |
|||
print "Warning: this test will take about 70 seconds in the best case. Be patient." |
|||
nodes[0].setgenerate(True, 10) |
|||
templat = nodes[0].getblocktemplate() |
|||
longpollid = templat['longpollid'] |
|||
# longpollid should not change between successive invocations if nothing else happens |
|||
templat2 = nodes[0].getblocktemplate() |
|||
assert(templat2['longpollid'] == longpollid) |
|||
|
|||
# Test 1: test that the longpolling wait if we do nothing |
|||
thr = LongpollThread(nodes[0]) |
|||
thr.start() |
|||
# check that thread still lives |
|||
thr.join(5) # wait 5 seconds or until thread exits |
|||
assert(thr.is_alive()) |
|||
|
|||
# Test 2: test that longpoll will terminate if another node generates a block |
|||
nodes[1].setgenerate(True, 1) # generate a block on another node |
|||
# check that thread will exit now that new transaction entered mempool |
|||
thr.join(5) # wait 5 seconds or until thread exits |
|||
assert(not thr.is_alive()) |
|||
|
|||
# Test 3: test that longpoll will terminate if we generate a block ourselves |
|||
thr = LongpollThread(nodes[0]) |
|||
thr.start() |
|||
nodes[0].setgenerate(True, 1) # generate a block on another node |
|||
thr.join(5) # wait 5 seconds or until thread exits |
|||
assert(not thr.is_alive()) |
|||
|
|||
# Test 4: test that introducing a new transaction into the mempool will terminate the longpoll |
|||
thr = LongpollThread(nodes[0]) |
|||
thr.start() |
|||
# generate a random transaction and submit it |
|||
(txid, txhex, fee) = random_transaction(nodes, Decimal("1.1"), Decimal("0.0"), Decimal("0.001"), 20) |
|||
# after one minute, every 10 seconds the mempool is probed, so in 80 seconds it should have returned |
|||
thr.join(60 + 20) |
|||
assert(not thr.is_alive()) |
|||
|
|||
if __name__ == '__main__': |
|||
GetBlockTemplateTest().main() |
|||
|
Loading…
Reference in new issue