|
|
|
|
#!/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()
|
