basicswap_miserver/tests/basicswap/extended/test_wallet_restore.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

# Copyright (c) 2022 tecnovert
# Distributed under the MIT software license, see the accompanying
# file LICENSE or http://www.opensource.org/licenses/mit-license.php.

"""
export TEST_PATH=/tmp/test_basicswap_wallet_restore
mkdir -p ${TEST_PATH}/bin
cp -r ~/tmp/basicswap_bin/* ${TEST_PATH}/bin
export PGP_KEYS_DIR_PATH=$(pwd)/pgp/keys
export PYTHONPATH=$(pwd)
python tests/basicswap/extended/test_wallet_restore.py


"""

import os
import sys
import shutil
import logging
import unittest
import threading
import traceback
import multiprocessing
from unittest.mock import patch

from tests.basicswap.util import (
    read_json_api,
    post_json_api,
    waitForServer,
)
from tests.basicswap.common import (
    waitForNumOffers,
    waitForNumBids,
)
from tests.basicswap.common_xmr import (
    TestBase,
    run_prepare,
    waitForBidState,
)
from basicswap.rpc import (
    callrpc,
)
from tests.basicswap.mnemonics import mnemonics
import bin.basicswap_run as runSystem
from tests.basicswap.common import (
    BTC_BASE_RPC_PORT,
    LTC_BASE_RPC_PORT,
)

PORT_OFS = int(os.getenv('PORT_OFS', 1))
TEST_PATH = os.path.expanduser(os.getenv('TEST_PATH', '~/test_basicswap1'))
LITECOIN_RPC_PORT_BASE = int(os.getenv('LITECOIN_RPC_PORT_BASE', LTC_BASE_RPC_PORT))
BITCOIN_RPC_PORT_BASE = int(os.getenv('BITCOIN_RPC_PORT_BASE', BTC_BASE_RPC_PORT))

logger = logging.getLogger()
logger.level = logging.DEBUG
if not len(logger.handlers):
    logger.addHandler(logging.StreamHandler(sys.stdout))


def callbtcnoderpc(node_id, method, params=[], wallet=None, base_rpc_port=BITCOIN_RPC_PORT_BASE):
    auth = 'test_btc_{0}:test_btc_pwd_{0}'.format(node_id)
    return callrpc(base_rpc_port + node_id, auth, method, params, wallet)


def callltcnoderpc(node_id, method, params=[], wallet=None, base_rpc_port=LITECOIN_RPC_PORT_BASE):
    auth = 'test_ltc_{0}:test_ltc_pwd_{0}'.format(node_id)
    return callrpc(base_rpc_port + node_id, auth, method, params, wallet)


def updateThread(self):
    while not self.delay_event.is_set():
        callbtcnoderpc(2, 'generatetoaddress', [1, self.btc_addr])
        callltcnoderpc(1, 'generatetoaddress', [1, self.ltc_addr])
        self.delay_event.wait(2)


def prepare_node(node_id, mnemonic):
    logging.info('Preparing node: %d.', node_id)
    bins_path = os.path.join(TEST_PATH, 'bin')
    client_path = os.path.join(TEST_PATH, 'client{}'.format(node_id))
    try:
        shutil.rmtree(client_path)
    except Exception as ex:
        logging.warning('setUpClass %s', str(ex))
    return run_prepare(node_id, client_path, bins_path, 'monero,bitcoin,litecoin', mnemonic, 3, use_rpcauth=True)


class Test(TestBase):
    @classmethod
    def setUpClass(cls):
        super(Test, cls).setUpClass(cls)
        cls.update_thread = None
        cls.used_mnemonics = []
        # Load wallets from random mnemonics, except node0 which needs to import PART from the genesis block
        for i in range(3):
            cls.used_mnemonics.append(prepare_node(i, mnemonics[0] if i == 0 else None))

    def run_thread(self, client_id):
        client_path = os.path.join(TEST_PATH, 'client{}'.format(client_id))
        testargs = ['basicswap-run', '-datadir=' + client_path, '-regtest']
        with patch.object(sys, 'argv', testargs):
            runSystem.main()

    def finalise(self, processes):
        self.delay_event.set()
        if self.update_thread:
            self.update_thread.join()
        for p in processes:
            p.terminate()
        for p in processes:
            p.join()

    def test_wallet(self):
        processes = []

        for i in range(3):
            processes.append(multiprocessing.Process(target=self.run_thread, args=(i,)))
            processes[-1].start()

        try:
            waitForServer(self.delay_event, 12700)
            waitForServer(self.delay_event, 12701)
            waitForServer(self.delay_event, 12702)

            num_blocks = 500  # Mine enough to activate segwit
            self.btc_addr = callbtcnoderpc(2, 'getnewaddress', ['mining_addr', 'bech32'])
            logging.info('Mining %d Bitcoin blocks to %s', num_blocks, self.btc_addr)
            callbtcnoderpc(2, 'generatetoaddress', [num_blocks, self.btc_addr])

            num_blocks = 431
            self.ltc_addr = callltcnoderpc(1, 'getnewaddress', ['mining_addr', 'bech32'], wallet='wallet.dat')
            logging.info('Mining %d Litecoin blocks to %s', num_blocks, self.ltc_addr)
            callltcnoderpc(1, 'generatetoaddress', [num_blocks, self.ltc_addr])

            mweb_addr = callltcnoderpc(1, 'getnewaddress', ['mweb_addr', 'mweb'], wallet='mweb')
            callltcnoderpc(1, 'sendtoaddress', [mweb_addr, 1], wallet='wallet.dat')
            num_blocks = 69
            callltcnoderpc(1, 'generatetoaddress', [num_blocks, self.ltc_addr])

            self.update_thread = threading.Thread(target=updateThread, args=(self,))
            self.update_thread.start()

            self.wait_for_particl_height(12701, num_blocks=3)

            data = {
                'addr_from': '-1',
                'coin_from': 'part',
                'coin_to': 'ltc',
                'amt_from': '1',
                'amt_to': '1',
                'lockhrs': '24',
                'automation_strat_id': 1}

            offer_id = post_json_api(12700, 'offers/new', data)['offer_id']
            summary = read_json_api(12700)
            assert (summary['num_sent_offers'] == 1)

            logger.info('Waiting for offer')
            waitForNumOffers(self.delay_event, 12701, 1)

            offers = read_json_api(12701, 'offers')
            offer = offers[0]

            data = {
                'offer_id': offer['offer_id'],
                'amount_from': offer['amount_from']}

            bid_id = post_json_api(12701, 'bids/new', data)['bid_id']

            waitForNumBids(self.delay_event, 12700, 1)

            waitForBidState(self.delay_event, 12700, bid_id, 'Completed', wait_for=120)
            waitForBidState(self.delay_event, 12701, bid_id, 'Completed', wait_for=120)

            logging.info('Starting a new node on the same mnemonic as the first')
            prepare_node(3, self.used_mnemonics[0])
            processes.append(multiprocessing.Process(target=self.run_thread, args=(3,)))
            processes[-1].start()
            waitForServer(self.delay_event, 12703)

            self.wait_seconds(5)

            # TODO: Attempt to detect past swaps

            for coin in ('part', 'btc', 'ltc'):
                logging.info(f'Checking {coin} balance')
                original = read_json_api(12700, f'wallets/{coin}')
                restored = read_json_api(12703, f'wallets/{coin}')
                assert (float(original['balance']) + float(original['unconfirmed']) == float(restored['balance']) + float(restored['unconfirmed']))

            wallets_original = read_json_api(12700, 'wallets')
            # TODO: After restoring a new deposit address should be generated, should be automated
            #       Swaps should use a new key path, not the external path
            next_addr = read_json_api(12700, 'wallets/part/nextdepositaddr')
            next_addr = read_json_api(12703, 'wallets/part/nextdepositaddr')
            wallets_restored = read_json_api(12703, 'wallets')
            for k, w in wallets_original.items():
                assert (w['deposit_address'] == wallets_restored[k]['deposit_address'])

        except Exception as e:
            traceback.print_exc()
            self.finalise(processes)
            logging.info('Test failed.')
            raise e

        self.finalise(processes)
        logging.info('Test passed.')


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