basicswap/tests/test_prepare.py

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

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

import os
import sys
import unittest
from unittest.mock import patch
from io import StringIO
import logging
import shutil

import bin.basicswap_prepare as prepareSystem
test_path = os.path.expanduser('~/test_basicswap')

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


class Test(unittest.TestCase):
    @classmethod
    def tearDownClass(self):
        try:
            shutil.rmtree(test_path)
        except Exception as e:
            logger.warning('tearDownClass %s', str(e))

    def test_no_overwrite(self):
        testargs = ['basicswap-prepare', '-datadir=' + test_path]
        with patch.object(sys, 'argv', testargs):
            prepareSystem.main()

        self.assertTrue(os.path.exists(os.path.join(test_path, 'basicswap.json')))

        testargs = ['basicswap-prepare', '-datadir=' + test_path]
        with patch('sys.stderr', new=StringIO()) as fake_stderr:
            with patch.object(sys, 'argv', testargs):
                with self.assertRaises(SystemExit) as cm:
                    prepareSystem.main()

        self.assertEqual(cm.exception.code, 1)
        logger.info('fake_stderr.getvalue() %s', fake_stderr.getvalue())
        self.assertTrue('exists, exiting' in fake_stderr.getvalue())


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