#!/usr/bin/env python2
# Copyright (c) 2018 SuperNET 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 BitcoinTestFramework
from test_framework.authproxy import JSONRPCException
from test_framework.util import assert_equal, assert_greater_than, \
    initialize_chain_clean, start_nodes, start_node, connect_nodes_bi, \
    stop_nodes, sync_blocks, sync_mempools, wait_bitcoinds, rpc_port

import time
from decimal import Decimal

class CryptoConditionsTest (BitcoinTestFramework):

    def setup_chain(self):
        print("Initializing CC test directory "+self.options.tmpdir)
        self.num_nodes = 1
        initialize_chain_clean(self.options.tmpdir, self.num_nodes)

    def setup_network(self, split = False):
        print("Setting up network...")
        self.pubkey  = "RWPg8B91kfK5UtUN7z6s6TeV9cHSGtVY8D"
        self.privkey = "UqMgxk7ySPNQ4r9nKAFPjkXy6r5t898yhuNCjSZJLg3RAM4WW1m9"
        self.nodes   = start_nodes(self.num_nodes, self.options.tmpdir,
                    extra_args=[[
                    '-conf='+self.options.tmpdir+'/node0/REGTEST.conf',
                    # TODO: AC.conf instead of komodo.conf
                    #'-conf='+self.options.tmpdir+'/node0/komodo.conf',
                    '-regtest',
                    '-port=64367',
                    '-rpcport=64368',
                    '-ac_name=REGTEST',
                    '-addressindex=1',
                    '-spentindex=1',
                    '-ac_supply=5555555',
                    '-ac_reward=10000000',
                    '-pubkey=02676d00110c2cd14ae24f95969e8598f7ccfaa675498b82654a5b5bd57fc1d8cf',
                    '-ac_cc=1',
                    '-whitelist=127.0.0.1',
                    '-debug',
                    '-daemon',
                    '-rpcuser=rt',
                    '-rpcpassword=rt'
                    ]]
        )
        self.is_network_split = split
        self.sync_all()
        print("Done setting up network")

    def run_test (self):
        print("Mining blocks...")
        rpc     = self.nodes[0]
        rpc.generate(4)
        self.sync_all()
        # this corresponds to -pubkey above
        print("Importing privkey")
        rpc.importprivkey(self.privkey)
        validate = rpc.validateaddress(self.pubkey)

        # Begin actual CC tests
        faucet  = rpc.faucetaddress()
        assert_equal(faucet['result'], 'success')
        assert_equal(faucet['myaddress'][0], 'R')

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