#!/usr/bin/env python3
# Copyright (c) 2020 The Bitcoin Core developers
# Distributed under the MIT software license, see the accompanying
# file COPYING or http://www.opensource.org/licenses/mit-license.php.
"""Test deprecation of getpeerinfo RPC fields."""

from test_framework.test_framework import BitcoinTestFramework


class GetpeerinfoDeprecationTest(BitcoinTestFramework):
    def set_test_params(self):
        self.num_nodes = 2
        self.extra_args = [[], ["-deprecatedrpc=banscore"]]

    def run_test(self):
        self.test_banscore_deprecation()
        self.test_addnode_deprecation()

    def test_banscore_deprecation(self):
        self.log.info("Test getpeerinfo by default no longer returns a banscore field")
        assert "banscore" not in self.nodes[0].getpeerinfo()[0].keys()

        self.log.info("Test getpeerinfo returns banscore with -deprecatedrpc=banscore")
        assert "banscore" in self.nodes[1].getpeerinfo()[0].keys()

    def test_addnode_deprecation(self):
        self.restart_node(1, ["-deprecatedrpc=getpeerinfo_addnode"])
        self.connect_nodes(0, 1)

        self.log.info("Test getpeerinfo by default no longer returns an addnode field")
        assert "addnode" not in self.nodes[0].getpeerinfo()[0].keys()

        self.log.info("Test getpeerinfo returns addnode with -deprecatedrpc=addnode")
        assert "addnode" in self.nodes[1].getpeerinfo()[0].keys()


if __name__ == "__main__":
    GetpeerinfoDeprecationTest().main()