git / moonbite-core / test/functional/rpc_dumptxoutset.py · raw

#!/usr/bin/env python3# Copyright (c) 2019-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 the generation of UTXO snapshots using `dumptxoutset`."""from test_framework.test_framework import BitcoinTestFrameworkfrom test_framework.util import assert_equal, assert_raises_rpc_errorimport hashlibfrom pathlib import Pathclass DumptxoutsetTest(BitcoinTestFramework):    def set_test_params(self):        self.setup_clean_chain = True        self.num_nodes = 1    def run_test(self):        """Test a trivial usage of the dumptxoutset RPC command."""        node = self.nodes[0]        mocktime = node.getblockheader(node.getblockhash(0))['time'] + 1        node.setmocktime(mocktime)        node.generate(100)        FILENAME = 'txoutset.dat'        out = node.dumptxoutset(FILENAME)        expected_path = Path(node.datadir) / self.chain / FILENAME        assert expected_path.is_file()        assert_equal(out['coins_written'], 100)        assert_equal(out['base_height'], 100)        assert_equal(out['path'], str(expected_path))        # Blockhash should be deterministic based on mocked time.        assert_equal(            out['base_hash'],            '3bc56dc7255f129e7ae88ebe791e131b0a813cc5254d629ee79c28c4191aefe7')        with open(str(expected_path), 'rb') as f:            digest = hashlib.sha256(f.read()).hexdigest()            # UTXO snapshot hash should be deterministic based on mocked time.            assert_equal(                digest, 'f1d6826ce1b9463355e2000d93797939940a9d5d83df3c53e504310dc7c13001')        # Specifying a path to an existing file will fail.        assert_raises_rpc_error(            -8, '{} already exists'.format(FILENAME),  node.dumptxoutset, FILENAME)if __name__ == '__main__':    DumptxoutsetTest().main()