dash/test/functional/rpc_mnauth.py
Wladimir J. van der Laan c02337e1b0
Merge #13517: qa: Remove need to handle the network thread in tests
fa87da2f172ae2e6dc15e9ed156a3564a8ecfbdd qa: Avoid start/stop of the network thread mid-test (MarcoFalke)

Pull request description:

  This simplifies test writing by removing the need to handle the network thread in tests. E.g. start thread, join thread, restart thread mid-test, adding p2p connections at the "right" time, ...

Tree-SHA512: 533642f12fef5496f1933855edcdab1a7ed901d088d34911749cd0f9e044c8a6cb1f89985ac3a7f41a512943663e4e270a61978f6f072143ae050cd102d4eab8
2021-07-02 12:59:26 +03:00

57 lines
2.5 KiB
Python
Executable File

#!/usr/bin/env python3
# Copyright (c) 2021 The Dash Core developers
# Distributed under the MIT software license, see the accompanying
# file COPYING or http://www.opensource.org/licenses/mit-license.php.
from test_framework.mininode import *
from test_framework.test_framework import DashTestFramework
from test_framework.util import assert_equal, assert_raises_rpc_error
'''
rpc_mnauth.py
Tests mnauth RPC command
'''
class FakeMNAUTHTest(DashTestFramework):
def set_test_params(self):
self.set_dash_test_params(2, 1, fast_dip3_enforcement=True)
def run_test(self):
masternode = self.mninfo[0]
p2p_masternode = masternode.node.add_p2p_connection(P2PInterface())
p2p_masternode.wait_for_verack()
protx_hash = masternode.proTxHash
public_key = masternode.pubKeyOperator
# The peerinfo should not yet contain verified_proregtx_hash/verified_pubkey_hash
assert("verified_proregtx_hash" not in masternode.node.getpeerinfo()[-1])
assert("verified_pubkey_hash" not in masternode.node.getpeerinfo()[-1])
# Fake-Authenticate the P2P connection to the masternode
node_id = masternode.node.getpeerinfo()[-1]["id"]
assert(masternode.node.mnauth(node_id, protx_hash, public_key))
# The peerinfo should now contain verified_proregtx_hash and verified_pubkey_hash
peerinfo = masternode.node.getpeerinfo()[-1]
assert("verified_proregtx_hash" in peerinfo)
assert("verified_pubkey_hash" in peerinfo)
assert_equal(peerinfo["verified_proregtx_hash"], protx_hash)
assert_equal(peerinfo["verified_pubkey_hash"], bytes_to_hex_str(hash256(hex_str_to_bytes(public_key))[::-1]))
# Test some error cases
null_hash = "0000000000000000000000000000000000000000000000000000000000000000"
assert_raises_rpc_error(-8, "proTxHash invalid", masternode.node.mnauth,
node_id,
null_hash,
public_key)
assert_raises_rpc_error(-8, "publicKey invalid", masternode.node.mnauth,
node_id,
protx_hash,
null_hash)
assert(not masternode.node.mnauth(-1, protx_hash, public_key))
if __name__ == '__main__':
FakeMNAUTHTest().main()