functional_tests: add tx_notes tests

This commit is contained in:
moneromooo-monero 2019-05-02 12:31:08 +00:00
parent 4c375c9442
commit 170e51f0d9
No known key found for this signature in database
GPG key ID: 686F07454D6CEFC3
2 changed files with 41 additions and 0 deletions

View file

@ -49,6 +49,7 @@ class TransferTest():
self.sweep_dust() self.sweep_dust()
self.sweep_single() self.sweep_single()
self.check_destinations() self.check_destinations()
self.check_tx_notes()
self.check_rescan() self.check_rescan()
def reset(self): def reset(self):
@ -719,6 +720,23 @@ class TransferTest():
daemon.generateblocks('42ey1afDFnn4886T7196doS9GPMzexD9gXpsZJDwVjeRVdFCSoHnv7KPbBeGpzJBzHRCAs9UxqeoyFQMYbqSWYTfJJQAWDm', 1) daemon.generateblocks('42ey1afDFnn4886T7196doS9GPMzexD9gXpsZJDwVjeRVdFCSoHnv7KPbBeGpzJBzHRCAs9UxqeoyFQMYbqSWYTfJJQAWDm', 1)
self.wallet[0].refresh() self.wallet[0].refresh()
def check_tx_notes(self):
daemon = Daemon()
print('Testing tx notes')
res = self.wallet[0].get_transfers()
assert len(res['in']) > 0
in_txid = res['in'][0].txid
assert len(res['out']) > 0
out_txid = res['out'][0].txid
res = self.wallet[0].get_tx_notes([in_txid, out_txid])
assert res.notes == ['', '']
res = self.wallet[0].set_tx_notes([in_txid, out_txid], ['in txid', 'out txid'])
res = self.wallet[0].get_tx_notes([in_txid, out_txid])
assert res.notes == ['in txid', 'out txid']
res = self.wallet[0].get_tx_notes([out_txid, in_txid])
assert res.notes == ['out txid', 'in txid']
def check_rescan(self): def check_rescan(self):
daemon = Daemon() daemon = Daemon()

View file

@ -911,6 +911,29 @@ class Wallet(object):
} }
return self.rpc.send_json_rpc_request(rescan_spent) return self.rpc.send_json_rpc_request(rescan_spent)
def set_tx_notes(self, txids = [], notes = []):
set_tx_notes = {
'method': 'set_tx_notes',
'jsonrpc': '2.0',
'params': {
'txids': txids,
'notes': notes,
},
'id': '0'
}
return self.rpc.send_json_rpc_request(set_tx_notes)
def get_tx_notes(self, txids = []):
get_tx_notes = {
'method': 'get_tx_notes',
'jsonrpc': '2.0',
'params': {
'txids': txids,
},
'id': '0'
}
return self.rpc.send_json_rpc_request(get_tx_notes)
def set_attribute(self, key, value): def set_attribute(self, key, value):
set_attribute = { set_attribute = {
'method': 'set_attribute', 'method': 'set_attribute',