import shutil import tempfile import sys import unittest import os import json from StringIO import StringIO from lib.wallet import WalletStorage, NewWallet class FakeConfig(object): """A stub config file to be used in tests""" def __init__(self, path): self.path = path self.store = {} def set(self, key, value): self.store[key] = value def get(self, key, default=None): return self.store.get(key, default) class FakeSynchronizer(object): def __init__(self): self.store = [] def add(self, address): self.store.append(address) class WalletTestCase(unittest.TestCase): def setUp(self): super(WalletTestCase, self).setUp() self.user_dir = tempfile.mkdtemp() self.fake_config = FakeConfig(self.user_dir) self._saved_stdout = sys.stdout self._stdout_buffer = StringIO() sys.stdout = self._stdout_buffer def tearDown(self): super(WalletTestCase, self).tearDown() shutil.rmtree(self.user_dir) # Restore the "real" stdout sys.stdout = self._saved_stdout class TestWalletStorage(WalletTestCase): def test_init_wallet_default_path(self): storage = WalletStorage(self.fake_config) expected = os.path.join(self.user_dir, "wallets", "default_wallet") self.assertEqual(expected, storage.path) def test_init_wallet_explicit_path(self): path = os.path.join(self.user_dir, "somewallet") self.fake_config.set("wallet_path", path) storage = WalletStorage(self.fake_config) self.assertEqual(path, storage.path) def test_init_wallet_default_wallet_path(self): path = os.path.join(self.user_dir, "somewallet") self.fake_config.set("default_wallet_path", path) storage = WalletStorage(self.fake_config) self.assertEqual(path, storage.path) def test_read_dictionnary_from_file(self): path = os.path.join(self.user_dir, "somewallet") self.fake_config.set("wallet_path", path) some_dict = {"a":"b", "c":"d"} contents = repr(some_dict) with open(path, "w") as f: contents = f.write(contents) storage = WalletStorage(self.fake_config) self.assertEqual("b", storage.get("a")) self.assertEqual("d", storage.get("c")) def test_write_dictionnary_to_file(self): path = os.path.join(self.user_dir, "somewallet") self.fake_config.set("wallet_path", path) storage = WalletStorage(self.fake_config) some_dict = {"a":"b", "c":"d"} storage.data = some_dict storage.write() contents = "" with open(path, "r") as f: contents = f.read() self.assertEqual(some_dict, json.loads(contents)) class TestNewWallet(WalletTestCase): seed_text = "travel nowhere air position hill peace suffer parent beautiful rise blood power home crumble teach" password = "secret" first_account_name = "account1" import_private_key = "L52XzL2cMkHxqxBXRyEpnPQZGUs3uKiL3R11XbAdHigRzDozKZeW" import_key_address = "15mKKb2eos1hWa6tisdPwwDC1a5J1y9nma" def setUp(self): super(TestNewWallet, self).setUp() self.storage = WalletStorage(self.fake_config) self.wallet = NewWallet(self.storage) # This cannot be constructed by electrum at random, it should be safe # from eventual collisions. self.wallet.add_seed(self.seed_text, self.password) self.wallet.create_master_keys(self.password) self.wallet.create_main_account(self.password) def test_wallet_with_seed_is_not_watching_only(self): self.assertFalse(self.wallet.is_watching_only()) def test_wallet_without_seed_is_watching_only(self): # We need a new storage , since the default storage was already seeded # in setUp() new_dir = tempfile.mkdtemp() config = FakeConfig(new_dir) storage = WalletStorage(config) wallet = NewWallet(storage) self.assertTrue(wallet.is_watching_only()) shutil.rmtree(new_dir) # Don't leave useless stuff in /tmp def test_new_wallet_is_deterministic(self): self.assertTrue(self.wallet.is_deterministic()) def test_get_seed_returns_correct_seed(self): self.assertEqual(self.wallet.get_seed(self.password), self.seed_text) def test_key_import(self): # Wallets have no imported keys by default. self.assertFalse(self.wallet.has_imported_keys()) # Importing a key works. self.wallet.import_key(self.import_private_key, "") self.assertEqual(2, len(self.wallet.addresses())) self.assertIn(self.import_key_address, self.wallet.addresses()) self.assertTrue(self.wallet.has_imported_keys()) # Deleting the key works. self.wallet.delete_imported_key(self.import_key_address) self.assertFalse(self.wallet.has_imported_keys()) self.assertEqual(1, len(self.wallet.addresses())) self.assertNotIn(self.import_key_address, self.wallet.addresses()) def test_update_password(self): new_password = "secret2" self.wallet.update_password(self.password, new_password) self.wallet.get_seed(new_password)