#!/usr/bin/env python3
# Copyright (c) 2020 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.test_framework import DashTestFramework
from test_framework.util import assert_equal

'''
rpc_masternode.py

Test "masternode" rpc subcommands
'''

class RPCMasternodeTest(DashTestFramework):
    def set_test_params(self):
        self.set_dash_test_params(4, 3, fast_dip3_enforcement=True)

    def run_test(self):
        self.log.info("test that results from `winners` and `payments` RPCs match")
        bi = self.nodes[0].getblockchaininfo()
        height = bi["blocks"]
        blockhash = bi["bestblockhash"]
        winners_payee = self.nodes[0].masternode("winners")[str(height)]
        payments = self.nodes[0].masternode("payments", blockhash)
        assert_equal(len(payments), 1)
        payments_block = payments[0]
        payments_payee = payments_block["masternodes"][0]["payees"][0]["address"]
        assert_equal(payments_block["height"], height)
        assert_equal(payments_block["blockhash"], blockhash)
        assert_equal(winners_payee, payments_payee)

        self.log.info("test various `payments` RPC options")
        payments1 = self.nodes[0].masternode("payments", blockhash, -1)
        assert_equal(payments, payments1)
        payments2_1 = self.nodes[0].masternode("payments", blockhash, 2)
        # using chaintip as a start block should return 1 block only
        assert_equal(len(payments2_1), 1)
        assert_equal(payments[0], payments2_1[0])
        payments2_2 = self.nodes[0].masternode("payments", blockhash, -2)
        # using chaintip as a start block should return 2 blocks now, with the tip being the last one
        assert_equal(len(payments2_2), 2)
        assert_equal(payments[0], payments2_2[-1])

        self.log.info("test that `masternode payments` results at chaintip match `getblocktemplate` results for that block")
        gbt_masternode = self.nodes[0].getblocktemplate()["masternode"]
        self.nodes[0].generate(1)
        payments_masternode = self.nodes[0].masternode("payments")[0]["masternodes"][0]
        assert_equal(gbt_masternode[0]["payee"], payments_masternode["payees"][0]["address"])
        assert_equal(gbt_masternode[0]["script"], payments_masternode["payees"][0]["script"])
        assert_equal(gbt_masternode[0]["amount"], payments_masternode["payees"][0]["amount"])


if __name__ == '__main__':
    RPCMasternodeTest().main()