mirror of
https://github.com/ElementsProject/elements.git
synced 2026-08-14 12:43:40 +02:00
re-add pegging python test
This commit is contained in:
Gregory Sanders
parent
3ba146d872
commit
be735cf397
1 changed files with 213 additions and 0 deletions
213
qa/rpc-tests/pegging.py
Executable file
213
qa/rpc-tests/pegging.py
Executable file
|
|
@ -0,0 +1,213 @@
|
|||
#!/usr/bin/env python2
|
||||
|
||||
from test_framework.authproxy import AuthServiceProxy, JSONRPCException
|
||||
import os
|
||||
import random
|
||||
import sys
|
||||
import time
|
||||
import subprocess
|
||||
import shutil
|
||||
|
||||
if len(sys.argv) != 3:
|
||||
print("paths to bitcoind and sidechain daemon must be included as arguments")
|
||||
sys.exit(0)
|
||||
print(sys.argv[1])
|
||||
print(sys.argv[2])
|
||||
|
||||
def sync_all(sidechain, sidechain2):
|
||||
timeout = 20
|
||||
while len(sidechain.getrawmempool()) != len(sidechain2.getrawmempool()):
|
||||
time.sleep(1)
|
||||
timeout -= 1
|
||||
if timeout == 0:
|
||||
raise Exception("Peg-in has failed to propagate.")
|
||||
sidechain2.generate(1)
|
||||
while sidechain.getblockcount() != sidechain2.getblockcount():
|
||||
time.sleep(1)
|
||||
timeout -= 1
|
||||
if timeout == 0:
|
||||
raise Exception("Blocks are not propagating.")
|
||||
|
||||
fedpeg_key="cPxqWyf1HDGpGFH1dnfjz8HbiWxvwG8WXyetbuAiw4thKXUdXLpR"
|
||||
fedpeg_pubkey="512103dff4923d778550cc13ce0d887d737553b4b58f4e8e886507fc39f5e447b2186451ae"
|
||||
|
||||
bitcoin_datadir="/tmp/"+''.join(random.choice('0123456789ABCDEF') for i in range(5))
|
||||
bitcoin_pass=''.join(random.choice('0123456789ABCDEF') for i in range(10))
|
||||
sidechain_datadir="/tmp/"+''.join(random.choice('0123456789ABCDEF') for i in range(5))
|
||||
sidechain_pass=''.join(random.choice('0123456789ABCDEF') for i in range(10))
|
||||
sidechain2_datadir="/tmp/"+''.join(random.choice('0123456789ABCDEF') for i in range(5))
|
||||
sidechain2_pass=''.join(random.choice('0123456789ABCDEF') for i in range(10))
|
||||
|
||||
bitcoin_port = 8000 + os.getpid()%999
|
||||
sidechain_port = bitcoin_port + 1
|
||||
sidechain2_port = bitcoin_port + 2
|
||||
sidechain1_p2p_port = bitcoin_port + 3
|
||||
sidechain2_p2p_port = bitcoin_port + 4
|
||||
|
||||
os.makedirs(bitcoin_datadir)
|
||||
os.makedirs(sidechain_datadir)
|
||||
os.makedirs(sidechain2_datadir)
|
||||
|
||||
with open(os.path.join(bitcoin_datadir, "bitcoin.conf"), 'w') as f:
|
||||
f.write("regtest=1\n")
|
||||
f.write("rpcuser=bitcoinrpc\n")
|
||||
f.write("rpcpassword="+bitcoin_pass+"\n")
|
||||
f.write("rpcport="+str(bitcoin_port)+"\n")
|
||||
f.write("discover=0\n")
|
||||
f.write("listen=0\n")
|
||||
f.write("testnet=0\n")
|
||||
f.write("txindex=1\n")
|
||||
f.write("daemon=1\n")
|
||||
f.write("listen=0\n")
|
||||
|
||||
with open(os.path.join(sidechain_datadir, "elements.conf"), 'w') as f:
|
||||
f.write("regtest=1\n")
|
||||
f.write("rpcuser=sidechainrpc\n")
|
||||
f.write("rpcpassword="+sidechain_pass+"\n")
|
||||
f.write("rpcport="+str(sidechain_port)+"\n")
|
||||
f.write("discover=0\n")
|
||||
f.write("testnet=0\n")
|
||||
f.write("txindex=1\n")
|
||||
f.write("fedpegscript="+fedpeg_pubkey+"\n")
|
||||
f.write("daemon=1\n")
|
||||
f.write("mainchainrpchost=127.0.0.1\n")
|
||||
f.write("mainchainrpcport="+str(bitcoin_port)+"\n")
|
||||
f.write("mainchainrpcuser=bitcoinrpc\n")
|
||||
f.write("mainchainrpcpassword="+bitcoin_pass+"\n")
|
||||
f.write("validatepegin=1\n")
|
||||
f.write("validatepegout=0\n")
|
||||
f.write("port="+str(sidechain1_p2p_port)+"\n")
|
||||
f.write("connect=localhost:"+str(sidechain2_p2p_port)+"\n")
|
||||
f.write("listen=1\n")
|
||||
|
||||
with open(os.path.join(sidechain2_datadir, "elements.conf"), 'w') as f:
|
||||
f.write("regtest=1\n")
|
||||
f.write("rpcuser=sidechainrpc2\n")
|
||||
f.write("rpcpassword="+sidechain2_pass+"\n")
|
||||
f.write("rpcport="+str(sidechain2_port)+"\n")
|
||||
f.write("discover=0\n")
|
||||
f.write("testnet=0\n")
|
||||
f.write("txindex=1\n")
|
||||
f.write("fedpegscript="+fedpeg_pubkey+"\n")
|
||||
f.write("daemon=1\n")
|
||||
f.write("mainchainrpchost=127.0.0.1\n")
|
||||
f.write("mainchainrpcport="+str(bitcoin_port)+"\n")
|
||||
f.write("mainchainrpcuser=bitcoinrpc\n")
|
||||
f.write("mainchainrpcpassword="+bitcoin_pass+"\n")
|
||||
f.write("validatepegin=1\n")
|
||||
f.write("validatepegout=0\n")
|
||||
f.write("port="+str(sidechain2_p2p_port)+"\n")
|
||||
f.write("connect=localhost:"+str(sidechain1_p2p_port)+"\n")
|
||||
f.write("listen=1\n")
|
||||
|
||||
try:
|
||||
|
||||
# Default is 8, meaning 8+2 confirms for wallet acceptance normally
|
||||
# this will require 10+2.
|
||||
sidechain_args = " -peginconfirmationdepth=10 "
|
||||
|
||||
# Start daemons
|
||||
print("Starting daemons at "+bitcoin_datadir+", "+sidechain_datadir+" and "+sidechain2_datadir)
|
||||
bitcoindstart = sys.argv[1]+"/bitcoind -datadir="+bitcoin_datadir
|
||||
subprocess.Popen(bitcoindstart.split(), stdout=subprocess.PIPE)
|
||||
|
||||
sidechainstart = sys.argv[2]+"/elementsd -datadir="+sidechain_datadir + sidechain_args
|
||||
subprocess.Popen(sidechainstart.split(), stdout=subprocess.PIPE)
|
||||
|
||||
sidechain2start = sys.argv[2]+"/elementsd -datadir="+sidechain2_datadir + sidechain_args
|
||||
subprocess.Popen(sidechain2start.split(), stdout=subprocess.PIPE)
|
||||
|
||||
print("Daemons started")
|
||||
time.sleep(3)
|
||||
|
||||
bitcoin = AuthServiceProxy("http://bitcoinrpc:"+bitcoin_pass+"@127.0.0.1:"+str(bitcoin_port))
|
||||
sidechain = AuthServiceProxy("http://sidechainrpc:"+sidechain_pass+"@127.0.0.1:"+str(sidechain_port))
|
||||
sidechain2 = AuthServiceProxy("http://sidechainrpc2:"+sidechain2_pass+"@127.0.0.1:"+str(sidechain2_port))
|
||||
print("Daemons started, making blocks to get funds")
|
||||
|
||||
bitcoin.generate(101)
|
||||
sidechain.generate(101)
|
||||
|
||||
addr = bitcoin.getnewaddress()
|
||||
|
||||
addrs = sidechain.getpeginaddress()
|
||||
txid1 = bitcoin.sendtoaddress(addrs["mainchain_address"], 24)
|
||||
# 10+2 confirms required to get into mempool and confirm
|
||||
bitcoin.generate(11)
|
||||
time.sleep(2)
|
||||
proof = bitcoin.gettxoutproof([txid1])
|
||||
raw = bitcoin.getrawtransaction(txid1)
|
||||
|
||||
print("Attempting peg-in")
|
||||
try:
|
||||
pegtxid = sidechain.claimpegin(raw, proof)
|
||||
raise Exception("Peg-in should not mature enough yet, need another block.")
|
||||
except JSONRPCException as e:
|
||||
assert("Peg-in Bitcoin transaction needs more confirmations to be sent." in e.error["message"])
|
||||
pass
|
||||
|
||||
# Should fail due to non-matching wallet address
|
||||
try:
|
||||
pegtxid = sidechain.claimpegin(raw, proof, sidechain.getnewaddress())
|
||||
raise Exception("Peg-in with non-matching witness_program should fail.")
|
||||
except JSONRPCException as e:
|
||||
assert("Given witness_program is not a valid v0 witness program" in e.error["message"])
|
||||
pass
|
||||
|
||||
# 12 confirms allows in mempool
|
||||
bitcoin.generate(1)
|
||||
|
||||
# Should succeed via wallet lookup for address match, and when given
|
||||
pegtxid1 = sidechain.claimpegin(raw, proof)
|
||||
|
||||
sync_all(sidechain, sidechain2)
|
||||
|
||||
tx1 = sidechain.gettransaction(pegtxid1)
|
||||
|
||||
if "confirmations" in tx1 and tx1["confirmations"] > 0:
|
||||
print("Peg-in is confirmed: Success!")
|
||||
else:
|
||||
raise Exception("Peg-in confirmation has failed.")
|
||||
|
||||
# Do many claims in mempool
|
||||
n_claims = 100
|
||||
|
||||
print("Flooding mempool with many small claims")
|
||||
pegtxs = []
|
||||
sidechain.generate(101)
|
||||
|
||||
for i in range(n_claims):
|
||||
addrs = sidechain.getpeginaddress()
|
||||
txid = bitcoin.sendtoaddress(addrs["mainchain_address"], 1)
|
||||
bitcoin.generate(12)
|
||||
proof = bitcoin.gettxoutproof([txid])
|
||||
raw = bitcoin.getrawtransaction(txid)
|
||||
pegtxs += [sidechain.claimpegin(raw, proof)]
|
||||
|
||||
sync_all(sidechain, sidechain2)
|
||||
|
||||
sidechain2.generate(1)
|
||||
for pegtxid in pegtxs:
|
||||
tx = sidechain.gettransaction(pegtxid)
|
||||
if "confirmations" not in tx or tx["confirmations"] == 0:
|
||||
raise Exception("Peg-in confirmation has failed.")
|
||||
|
||||
# TODO test reorgs, conflicting peg-ins
|
||||
|
||||
print("Success!")
|
||||
|
||||
except JSONRPCException as e:
|
||||
print("Pegging testing failed, aborting:")
|
||||
print(e.error)
|
||||
except Exception as e:
|
||||
print("Pegging testing failed, aborting:")
|
||||
print(e)
|
||||
|
||||
print("Stopping daemons and cleaning up")
|
||||
bitcoin.stop()
|
||||
sidechain.stop()
|
||||
|
||||
time.sleep(5)
|
||||
|
||||
shutil.rmtree(sidechain_datadir)
|
||||
shutil.rmtree(bitcoin_datadir)
|
||||
Loading…
Add table
Add a link
Reference in a new issue