You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

155 lines
4.8 KiB

import shutil
import tempfile
import sys
import unittest
import os
10 years ago
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_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()
10 years ago
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)
10 years ago
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)
10 years ago
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.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.assertNotIn(self.import_key_address, self.wallet.addresses())
def test_update_password(self):
new_password = "secret2"
self.wallet.update_password(self.password, new_password)
10 years ago
self.wallet.get_seed(new_password)