GlcoinCore/elements
1
0
Fork 0
mirror of https://github.com/ElementsProject/elements.git synced 2026-08-20 13:37:28 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

pset, test: Test a PSET coinjoin workflow

Browse source
This commit is contained in:
Andrew Chow 2021-09-08 23:06:00 -04:00
parent ac580db4ec
commit a7ef9f325d
1 changed files with 60 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

60
test/functional/rpc_psbt.py
View file

@ -12,6 +12,7 @@ from test_framework.util import (
assert_greater_than,
assert_raises_rpc_error,
find_output,
find_vout_for_address,
)
from decimal import Decimal
@ -637,6 +638,65 @@ class PSBTTest(BitcoinTestFramework):
self.nodes[0].generate(1)
self.sync_all()
# Try a multiparty blinded tx
# Prepare wallets and UTXOs for inputs
self.nodes[2].createwallet("w1")
w1 = self.nodes[2].get_wallet_rpc("w1")
self.nodes[2].createwallet("w2")
w2 = self.nodes[2].get_wallet_rpc("w2")
self.nodes[2].createwallet("w3")
w3 = self.nodes[2].get_wallet_rpc("w3")
w1_addr = w1.getaddressinfo(w1.getnewaddress())["confidential"]
w2_addr = w2.getaddressinfo(w2.getnewaddress())["confidential"]
w3_addr = w3.getaddressinfo(w3.getnewaddress())["confidential"]
txid1 = self.nodes[0].sendtoaddress(w1_addr, 10)
txid2 = self.nodes[0].sendtoaddress(w2_addr, 10)
txid3 = self.nodes[0].sendtoaddress(w3_addr, 10)
self.sync_all()
vout1 = find_vout_for_address(self.nodes[2], txid1, w1_addr)
vout2 = find_vout_for_address(self.nodes[2], txid2, w2_addr)
vout3 = find_vout_for_address(self.nodes[2], txid3, w3_addr)
self.nodes[0].generate(1)
self.sync_all()
# Make the PSBT
created_psbt = self.nodes[0].createpsbt(
[
{"txid": txid1, "vout": vout1},
{"txid": txid2, "vout": vout2},
{"txid": txid3, "vout": vout3},
],
[
{self.get_address(True, 0): Decimal("9.999"), "blinder_index": 0},
{self.get_address(True, 0): Decimal("9.999"), "blinder_index": 1},
{self.get_address(True, 0): Decimal("9.999"), "blinder_index": 2},
{"fee": Decimal("0.003")}
]
)
# Update all but don't blind
up_psbt1 = w1.walletprocesspsbt(psbt=created_psbt, sign=False)["psbt"]
up_psbt2 = w2.walletprocesspsbt(psbt=created_psbt, sign=False)["psbt"]
up_psbt3 = w3.walletprocesspsbt(psbt=created_psbt, sign=False)["psbt"]
# Combine updated
comb_psbt1 = self.nodes[0].combinepsbt([up_psbt1, up_psbt2, up_psbt3])
# 1 and 2 blind
blind_psbt1 = w1.walletprocesspsbt(psbt=comb_psbt1, sign=False)["psbt"]
blind_psbt2 = w2.walletprocesspsbt(psbt=comb_psbt1, sign=False)["psbt"]
# Combine 1 and 2 blinded
comb_psbt2 = self.nodes[0].combinepsbt([blind_psbt1, blind_psbt2])
# 3 Updates and blinds combined
blind_psbt = w3.walletprocesspsbt(psbt=comb_psbt2, sign=False)["psbt"]
# All sign
sign_psbt1 = w1.walletprocesspsbt(psbt=blind_psbt)["psbt"]
sign_psbt2 = w2.walletprocesspsbt(psbt=blind_psbt)["psbt"]
sign_psbt3 = w3.walletprocesspsbt(psbt=blind_psbt)["psbt"]
# Combine sigs
comb_psbt2 = self.nodes[0].combinepsbt([sign_psbt1, sign_psbt2, sign_psbt3])
# Finalize and send
tx = self.nodes[0].finalizepsbt(comb_psbt2)["hex"]
self.nodes[0].sendrawtransaction(tx)
self.nodes[0].generate(1)
self.sync_all()
def run_test(self):
self.nodes[0].generate(200)
self.sync_all()