mirror of
https://github.com/monero-project/monero.git
synced 2024-11-17 08:17:37 +00:00
functional_tests: add proofs tests (tx key, in/out tx, reserve)
This commit is contained in:
parent
a3144bd7c8
commit
cf6d775964
3 changed files with 350 additions and 1 deletions
|
@ -9,7 +9,7 @@ import socket
|
|||
import string
|
||||
|
||||
USAGE = 'usage: functional_tests_rpc.py <python> <srcdir> <builddir> [<tests-to-run> | all]'
|
||||
DEFAULT_TESTS = ['daemon_info', 'blockchain', 'wallet_address', 'integrated_address', 'mining', 'transfer', 'txpool', 'multisig', 'cold_signing', 'sign_message']
|
||||
DEFAULT_TESTS = ['daemon_info', 'blockchain', 'wallet_address', 'integrated_address', 'mining', 'transfer', 'txpool', 'multisig', 'cold_signing', 'sign_message', 'proofs']
|
||||
try:
|
||||
python = sys.argv[1]
|
||||
srcdir = sys.argv[2]
|
||||
|
|
282
tests/functional_tests/proofs.py
Executable file
282
tests/functional_tests/proofs.py
Executable file
|
@ -0,0 +1,282 @@
|
|||
#!/usr/bin/env python3
|
||||
|
||||
# Copyright (c) 2019 The Monero Project
|
||||
#
|
||||
# All rights reserved.
|
||||
#
|
||||
# Redistribution and use in source and binary forms, with or without modification, are
|
||||
# permitted provided that the following conditions are met:
|
||||
#
|
||||
# 1. Redistributions of source code must retain the above copyright notice, this list of
|
||||
# conditions and the following disclaimer.
|
||||
#
|
||||
# 2. Redistributions in binary form must reproduce the above copyright notice, this list
|
||||
# of conditions and the following disclaimer in the documentation and/or other
|
||||
# materials provided with the distribution.
|
||||
#
|
||||
# 3. Neither the name of the copyright holder nor the names of its contributors may be
|
||||
# used to endorse or promote products derived from this software without specific
|
||||
# prior written permission.
|
||||
#
|
||||
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY
|
||||
# EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
|
||||
# MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL
|
||||
# THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||||
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
# PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
|
||||
# STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF
|
||||
# THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
import time
|
||||
|
||||
"""Test misc proofs (tx key, send, receive, reserve)
|
||||
"""
|
||||
|
||||
from test_framework.daemon import Daemon
|
||||
from test_framework.wallet import Wallet
|
||||
|
||||
class ProofsTest():
|
||||
def run_test(self):
|
||||
self.mine('42ey1afDFnn4886T7196doS9GPMzexD9gXpsZJDwVjeRVdFCSoHnv7KPbBeGpzJBzHRCAs9UxqeoyFQMYbqSWYTfJJQAWDm', 80)
|
||||
self.create_wallets()
|
||||
txid, tx_key, amount = self.transfer()
|
||||
self.check_tx_key(txid, tx_key, amount)
|
||||
self.check_tx_proof(txid, amount)
|
||||
self.check_reserve_proof()
|
||||
|
||||
def mine(self, address, blocks):
|
||||
print("Mining some blocks")
|
||||
daemon = Daemon()
|
||||
daemon.generateblocks(address, blocks)
|
||||
|
||||
def transfer(self):
|
||||
print('Creating transaction')
|
||||
self.wallet[0].refresh()
|
||||
dst = {'address': '44Kbx4sJ7JDRDV5aAhLJzQCjDz2ViLRduE3ijDZu3osWKBjMGkV1XPk4pfDUMqt1Aiezvephdqm6YD19GKFD9ZcXVUTp6BW', 'amount':123456789000}
|
||||
res = self.wallet[0].transfer([dst], get_tx_key = True)
|
||||
assert len(res.tx_hash) == 64
|
||||
assert len(res.tx_key) == 64
|
||||
daemon = Daemon()
|
||||
daemon.generateblocks('42ey1afDFnn4886T7196doS9GPMzexD9gXpsZJDwVjeRVdFCSoHnv7KPbBeGpzJBzHRCAs9UxqeoyFQMYbqSWYTfJJQAWDm', 1)
|
||||
return (res.tx_hash, res.tx_key, 123456789000)
|
||||
|
||||
def create_wallets(self):
|
||||
print('Creating wallets')
|
||||
seeds = [
|
||||
'velvet lymph giddy number token physics poetry unquoted nibs useful sabotage limits benches lifestyle eden nitrogen anvil fewest avoid batch vials washing fences goat unquoted',
|
||||
'peeled mixture ionic radar utopia puddle buying illness nuns gadget river spout cavernous bounced paradise drunk looking cottage jump tequila melting went winter adjust spout',
|
||||
]
|
||||
self.wallet = [None, None]
|
||||
for i in range(2):
|
||||
self.wallet[i] = Wallet(idx = i)
|
||||
try: self.wallet[i].close_wallet()
|
||||
except: pass
|
||||
res = self.wallet[i].restore_deterministic_wallet(seed = seeds[i])
|
||||
|
||||
def check_tx_key(self, txid, tx_key, amount):
|
||||
daemon = Daemon()
|
||||
|
||||
print('Checking tx key')
|
||||
|
||||
self.wallet[0].refresh()
|
||||
self.wallet[1].refresh()
|
||||
|
||||
sending_address = '42ey1afDFnn4886T7196doS9GPMzexD9gXpsZJDwVjeRVdFCSoHnv7KPbBeGpzJBzHRCAs9UxqeoyFQMYbqSWYTfJJQAWDm'
|
||||
receiving_address = '44Kbx4sJ7JDRDV5aAhLJzQCjDz2ViLRduE3ijDZu3osWKBjMGkV1XPk4pfDUMqt1Aiezvephdqm6YD19GKFD9ZcXVUTp6BW'
|
||||
res = self.wallet[0].get_tx_key(txid)
|
||||
assert res.tx_key == tx_key
|
||||
res = self.wallet[0].check_tx_key(txid = txid, tx_key = tx_key, address = receiving_address)
|
||||
assert res.received == amount
|
||||
assert not res.in_pool
|
||||
assert res.confirmations == 1
|
||||
res = self.wallet[1].check_tx_key(txid = txid, tx_key = tx_key, address = receiving_address)
|
||||
assert res.received == amount
|
||||
assert not res.in_pool
|
||||
assert res.confirmations == 1
|
||||
|
||||
self.wallet[1].check_tx_key(txid = txid, tx_key = tx_key, address = sending_address)
|
||||
assert res.received >= 0 # might be change
|
||||
assert not res.in_pool
|
||||
assert res.confirmations == 1
|
||||
|
||||
ok = False
|
||||
try: self.wallet[1].check_tx_key(txid = '0' * 64, tx_key = tx_key, address = receiving_address)
|
||||
except: ok = True
|
||||
assert ok
|
||||
|
||||
res = self.wallet[1].check_tx_key(txid = txid, tx_key = '0' * 64, address = receiving_address)
|
||||
assert res.received == 0
|
||||
assert not res.in_pool
|
||||
assert res.confirmations == 1
|
||||
|
||||
def check_tx_proof(self, txid, amount):
|
||||
daemon = Daemon()
|
||||
|
||||
print('Checking tx proof')
|
||||
|
||||
self.wallet[0].refresh()
|
||||
self.wallet[1].refresh()
|
||||
|
||||
sending_address = '42ey1afDFnn4886T7196doS9GPMzexD9gXpsZJDwVjeRVdFCSoHnv7KPbBeGpzJBzHRCAs9UxqeoyFQMYbqSWYTfJJQAWDm'
|
||||
receiving_address = '44Kbx4sJ7JDRDV5aAhLJzQCjDz2ViLRduE3ijDZu3osWKBjMGkV1XPk4pfDUMqt1Aiezvephdqm6YD19GKFD9ZcXVUTp6BW'
|
||||
res = self.wallet[0].get_tx_proof(txid, sending_address, 'foo');
|
||||
assert res.signature.startswith('InProof');
|
||||
signature0i = res.signature
|
||||
res = self.wallet[0].get_tx_proof(txid, receiving_address, 'bar');
|
||||
assert res.signature.startswith('OutProof');
|
||||
signature0o = res.signature
|
||||
res = self.wallet[1].get_tx_proof(txid, receiving_address, 'baz');
|
||||
assert res.signature.startswith('InProof');
|
||||
signature1 = res.signature
|
||||
|
||||
res = self.wallet[0].check_tx_proof(txid, sending_address, 'foo', signature0i);
|
||||
assert res.good
|
||||
assert res.received > 0 # likely change
|
||||
assert not res.in_pool
|
||||
assert res.confirmations == 1
|
||||
|
||||
ok = False
|
||||
try: res = self.wallet[0].check_tx_proof('0' * 64, sending_address, 'foo', signature0i);
|
||||
except: ok = True
|
||||
assert ok or not res.good
|
||||
|
||||
ok = False
|
||||
try: res = self.wallet[0].check_tx_proof(txid, receiving_address, 'foo', signature0i);
|
||||
except: ok = True
|
||||
assert ok or not res.good
|
||||
|
||||
ok = False
|
||||
try: res = self.wallet[0].check_tx_proof(txid, sending_address, '', signature0i);
|
||||
except: ok = True
|
||||
assert ok or not res.good
|
||||
|
||||
ok = False
|
||||
try: res = self.wallet[0].check_tx_proof(txid, sending_address, 'foo', signature1);
|
||||
except: ok = True
|
||||
assert ok or not res.good
|
||||
|
||||
|
||||
res = self.wallet[0].check_tx_proof(txid, receiving_address, 'bar', signature0o);
|
||||
assert res.good
|
||||
assert res.received == amount
|
||||
assert not res.in_pool
|
||||
assert res.confirmations == 1
|
||||
|
||||
ok = False
|
||||
try: res = self.wallet[0].check_tx_proof('0' * 64, receiving_address, 'bar', signature0o);
|
||||
except: ok = True
|
||||
assert ok or not res.good
|
||||
|
||||
ok = False
|
||||
try: res = self.wallet[0].check_tx_proof(txid, sending_address, 'bar', signature0o);
|
||||
except: ok = True
|
||||
assert ok or not res.good
|
||||
|
||||
ok = False
|
||||
try: res = self.wallet[0].check_tx_proof(txid, receiving_address, '', signature0o);
|
||||
except: ok = True
|
||||
assert ok or not res.good
|
||||
|
||||
ok = False
|
||||
try: res = self.wallet[0].check_tx_proof(txid, receiving_address, 'bar', signature0i);
|
||||
except: ok = True
|
||||
assert ok or not res.good
|
||||
|
||||
|
||||
res = self.wallet[1].check_tx_proof(txid, receiving_address, 'baz', signature1);
|
||||
assert res.good
|
||||
assert res.received == amount
|
||||
assert not res.in_pool
|
||||
assert res.confirmations == 1
|
||||
|
||||
ok = False
|
||||
try: res = self.wallet[1].check_tx_proof('0' * 64, receiving_address, 'baz', signature1);
|
||||
except: ok = True
|
||||
assert ok or not res.good
|
||||
|
||||
ok = False
|
||||
try: res = self.wallet[1].check_tx_proof(txid, sending_address, 'baz', signature1);
|
||||
except: ok = True
|
||||
assert ok or not res.good
|
||||
|
||||
ok = False
|
||||
try: res = self.wallet[1].check_tx_proof(txid, receiving_address, '', signature1);
|
||||
except: ok = True
|
||||
assert ok or not res.good
|
||||
|
||||
ok = False
|
||||
try: res = self.wallet[1].check_tx_proof(txid, receiving_address, 'baz', signature0o);
|
||||
except: ok = True
|
||||
assert ok or not res.good
|
||||
|
||||
|
||||
def check_reserve_proof(self):
|
||||
daemon = Daemon()
|
||||
|
||||
print('Checking reserve proof')
|
||||
|
||||
address0 = '42ey1afDFnn4886T7196doS9GPMzexD9gXpsZJDwVjeRVdFCSoHnv7KPbBeGpzJBzHRCAs9UxqeoyFQMYbqSWYTfJJQAWDm'
|
||||
address1 = '44Kbx4sJ7JDRDV5aAhLJzQCjDz2ViLRduE3ijDZu3osWKBjMGkV1XPk4pfDUMqt1Aiezvephdqm6YD19GKFD9ZcXVUTp6BW'
|
||||
|
||||
self.wallet[0].refresh()
|
||||
res = self.wallet[0].get_balance()
|
||||
balance0 = res.balance
|
||||
self.wallet[1].refresh()
|
||||
res = self.wallet[1].get_balance()
|
||||
balance1 = res.balance
|
||||
|
||||
res = self.wallet[0].get_reserve_proof(all_ = True, message = 'foo')
|
||||
assert res.signature.startswith('ReserveProof')
|
||||
signature = res.signature
|
||||
for i in range(2):
|
||||
res = self.wallet[i].check_reserve_proof(address = address0, message = 'foo', signature = signature)
|
||||
assert res.good
|
||||
assert res.total == balance0
|
||||
|
||||
ok = False
|
||||
try: res = self.wallet[i].check_reserve_proof(address = address0, message = 'bar', signature = signature)
|
||||
except: ok = True
|
||||
assert ok or not res.good
|
||||
|
||||
ok = False
|
||||
try: res = self.wallet[i].check_reserve_proof(address = address1, message = 'foo', signature = signature)
|
||||
except: ok = True
|
||||
assert ok or not res.good
|
||||
|
||||
amount = int(balance0 / 10)
|
||||
res = self.wallet[0].get_reserve_proof(all_ = False, amount = amount, message = 'foo')
|
||||
assert res.signature.startswith('ReserveProof')
|
||||
signature = res.signature
|
||||
for i in range(2):
|
||||
res = self.wallet[i].check_reserve_proof(address = address0, message = 'foo', signature = signature)
|
||||
assert res.good
|
||||
assert res.total >= amount and res.total <= balance0
|
||||
|
||||
ok = False
|
||||
try: res = self.wallet[i].check_reserve_proof(address = address0, message = 'bar', signature = signature)
|
||||
except: ok = True
|
||||
assert ok or not res.good
|
||||
|
||||
ok = False
|
||||
try: res = self.wallet[i].check_reserve_proof(address = address1, message = 'foo', signature = signature)
|
||||
except: ok = True
|
||||
assert ok or not res.good
|
||||
|
||||
ok = False
|
||||
try: self.wallet[0].get_reserve_proof(all_ = False, amount = balance0 + 1, message = 'foo')
|
||||
except: ok = True
|
||||
assert ok
|
||||
|
||||
|
||||
class Guard:
|
||||
def __enter__(self):
|
||||
for i in range(4):
|
||||
Wallet(idx = i).auto_refresh(False)
|
||||
def __exit__(self, exc_type, exc_value, traceback):
|
||||
for i in range(4):
|
||||
Wallet(idx = i).auto_refresh(True)
|
||||
|
||||
if __name__ == '__main__':
|
||||
with Guard() as guard:
|
||||
ProofsTest().run_test()
|
|
@ -495,6 +495,73 @@ class Wallet(object):
|
|||
}
|
||||
return self.rpc.send_json_rpc_request(get_tx_key)
|
||||
|
||||
def check_tx_key(self, txid = '', tx_key = '', address = ''):
|
||||
check_tx_key = {
|
||||
'method': 'check_tx_key',
|
||||
'params' : {
|
||||
'txid': txid,
|
||||
'tx_key': tx_key,
|
||||
'address': address,
|
||||
},
|
||||
'jsonrpc': '2.0',
|
||||
'id': '0'
|
||||
}
|
||||
return self.rpc.send_json_rpc_request(check_tx_key)
|
||||
|
||||
def get_tx_proof(self, txid = '', address = '', message = ''):
|
||||
get_tx_proof = {
|
||||
'method': 'get_tx_proof',
|
||||
'params' : {
|
||||
'txid': txid,
|
||||
'address': address,
|
||||
'message': message,
|
||||
},
|
||||
'jsonrpc': '2.0',
|
||||
'id': '0'
|
||||
}
|
||||
return self.rpc.send_json_rpc_request(get_tx_proof)
|
||||
|
||||
def check_tx_proof(self, txid = '', address = '', message = '', signature = ''):
|
||||
check_tx_proof = {
|
||||
'method': 'check_tx_proof',
|
||||
'params' : {
|
||||
'txid': txid,
|
||||
'address': address,
|
||||
'message': message,
|
||||
'signature': signature,
|
||||
},
|
||||
'jsonrpc': '2.0',
|
||||
'id': '0'
|
||||
}
|
||||
return self.rpc.send_json_rpc_request(check_tx_proof)
|
||||
|
||||
def get_reserve_proof(self, all_ = True, account_index = 0, amount = 0, message = ''):
|
||||
get_reserve_proof = {
|
||||
'method': 'get_reserve_proof',
|
||||
'params' : {
|
||||
'all': all_,
|
||||
'account_index': account_index,
|
||||
'amount': amount,
|
||||
'message': message,
|
||||
},
|
||||
'jsonrpc': '2.0',
|
||||
'id': '0'
|
||||
}
|
||||
return self.rpc.send_json_rpc_request(get_reserve_proof)
|
||||
|
||||
def check_reserve_proof(self, address = '', message = '', signature = ''):
|
||||
check_reserve_proof = {
|
||||
'method': 'check_reserve_proof',
|
||||
'params' : {
|
||||
'address': address,
|
||||
'message': message,
|
||||
'signature': signature,
|
||||
},
|
||||
'jsonrpc': '2.0',
|
||||
'id': '0'
|
||||
}
|
||||
return self.rpc.send_json_rpc_request(check_reserve_proof)
|
||||
|
||||
def sign(self, data):
|
||||
sign = {
|
||||
'method': 'sign',
|
||||
|
|
Loading…
Reference in a new issue