basicswap_miserver/tests/basicswap/extended/test_wallet_init.py

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

# Copyright (c) 2020-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_init
mkdir -p ${TEST_PATH}/bin
cp -r ~/tmp/basicswap_bin/* ${TEST_PATH}/bin
export PYTHONPATH=$(pwd)
python tests/basicswap/extended/test_wallet_init.py


"""

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

from tests.basicswap.mnemonics import mnemonics
from tests.basicswap.common import (
    read_json_api,
    waitForServer,
)
from tests.basicswap.common_xmr import (
    run_prepare,
)
import bin.basicswap_run as runSystem

TEST_PATH = os.path.expanduser(os.getenv('TEST_PATH', '~/test_basicswap1'))

stop_test = False

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


class Test(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        super(Test, cls).setUpClass()

    # Load both wallets from the same mnemonic
    bins_path = os.path.join(TEST_PATH, 'bin')
    for i in range(2):
        logging.info('Preparing node: %d.', i)
        client_path = os.path.join(TEST_PATH, 'client{}'.format(i))
        try:
            shutil.rmtree(client_path)
        except Exception as ex:
            logging.warning('setUpClass %s', str(ex))

        run_prepare(i, client_path, bins_path, 'monero,bitcoin', mnemonics[0])

    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 test_wallet(self):
        global stop_test
        update_thread = None
        processes = []

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

        try:
            waitForServer(12700)

            wallets_0 = read_json_api(12700, 'wallets')
            assert(wallets_0['1']['expected_seed'] is True)
            assert(wallets_0['6']['expected_seed'] is True)

            waitForServer(12701)
            wallets_1 = read_json_api(12701, 'wallets')

            assert(wallets_0['1']['expected_seed'] is True)
            assert(wallets_1['6']['expected_seed'] is True)

            # TODO: Check other coins

            assert(wallets_0['1']['deposit_address'] == wallets_1['1']['deposit_address'])
            assert(wallets_0['6']['deposit_address'] == wallets_1['6']['deposit_address'])
        except Exception:
            traceback.print_exc()

        stop_test = True
        if update_thread:
            update_thread.join()
        for p in processes:
            p.terminate()
        for p in processes:
            p.join()


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