tests: move /electrum/tests to /tests
This commit is contained in:
Sander van Grieken
238
tests/test_daemon.py
Normal file
238
tests/test_daemon.py
Normal file
@@ -0,0 +1,238 @@
|
||||
import os
|
||||
from typing import Optional, Iterable
|
||||
|
||||
from electrum.commands import Commands
|
||||
from electrum.daemon import Daemon
|
||||
from electrum.simple_config import SimpleConfig
|
||||
from electrum.wallet import restore_wallet_from_text, Abstract_Wallet
|
||||
from electrum import util
|
||||
|
||||
from . import ElectrumTestCase, as_testnet
|
||||
|
||||
|
||||
class DaemonTestCase(ElectrumTestCase):
|
||||
config: 'SimpleConfig'
|
||||
|
||||
def setUp(self):
|
||||
super().setUp()
|
||||
self.config = SimpleConfig({'electrum_path': self.electrum_path})
|
||||
self.config.NETWORK_OFFLINE = True
|
||||
|
||||
self.wallet_dir = os.path.dirname(self.config.get_wallet_path())
|
||||
assert "wallets" == os.path.basename(self.wallet_dir)
|
||||
|
||||
async def asyncSetUp(self):
|
||||
await super().asyncSetUp()
|
||||
self.daemon = Daemon(config=self.config, listen_jsonrpc=False)
|
||||
assert self.daemon.network is None
|
||||
|
||||
async def asyncTearDown(self):
|
||||
await self.daemon.stop()
|
||||
await super().asyncTearDown()
|
||||
|
||||
def _restore_wallet_from_text(self, text, *, password: Optional[str], encrypt_file: bool = None) -> str:
|
||||
"""Returns path for created wallet."""
|
||||
basename = util.get_new_wallet_name(self.wallet_dir)
|
||||
path = os.path.join(self.wallet_dir, basename)
|
||||
wallet_dict = restore_wallet_from_text(
|
||||
text,
|
||||
path=path,
|
||||
password=password,
|
||||
encrypt_file=encrypt_file,
|
||||
gap_limit=2,
|
||||
config=self.config,
|
||||
)
|
||||
# We return the path instead of the wallet object, as extreme
|
||||
# care would be needed to use the wallet object directly:
|
||||
# Unless the daemon knows about it, daemon._load_wallet might create a conflicting wallet object
|
||||
# for the same fs path, and there would be two wallet objects contending for the same file.
|
||||
return path
|
||||
|
||||
|
||||
class TestUnifiedPassword(DaemonTestCase):
|
||||
|
||||
def setUp(self):
|
||||
super().setUp()
|
||||
self.config.WALLET_USE_SINGLE_PASSWORD = True
|
||||
|
||||
def _run_post_unif_sanity_checks(self, paths: Iterable[str], *, password: str):
|
||||
for path in paths:
|
||||
w = self.daemon.load_wallet(path, password)
|
||||
self.assertIsNotNone(w)
|
||||
w.check_password(password)
|
||||
self.assertTrue(w.has_storage_encryption())
|
||||
if w.can_have_keystore_encryption():
|
||||
self.assertTrue(w.has_keystore_encryption())
|
||||
if w.has_seed():
|
||||
self.assertIsInstance(w.get_seed(password), str)
|
||||
can_be_unified, is_unified = self.daemon._check_password_for_directory(old_password=password, wallet_dir=self.wallet_dir)
|
||||
self.assertEqual((True, True), (can_be_unified, is_unified))
|
||||
|
||||
# "cannot unify pw" tests --->
|
||||
|
||||
async def test_cannot_unify_two_std_wallets_both_have_ks_and_sto_enc(self):
|
||||
path1 = self._restore_wallet_from_text("9dk", password="123456", encrypt_file=True)
|
||||
path2 = self._restore_wallet_from_text("x8", password="asdasd", encrypt_file=True)
|
||||
with open(path1, "rb") as f:
|
||||
raw1_before = f.read()
|
||||
with open(path2, "rb") as f:
|
||||
raw2_before = f.read()
|
||||
|
||||
can_be_unified, is_unified = self.daemon._check_password_for_directory(old_password="123456", wallet_dir=self.wallet_dir)
|
||||
self.assertEqual((False, False), (can_be_unified, is_unified))
|
||||
is_unified = self.daemon.update_password_for_directory(old_password="123456", new_password="123456")
|
||||
self.assertFalse(is_unified)
|
||||
# verify that files on disk haven't changed:
|
||||
with open(path1, "rb") as f:
|
||||
raw1_after = f.read()
|
||||
with open(path2, "rb") as f:
|
||||
raw2_after = f.read()
|
||||
self.assertEqual(raw1_before, raw1_after)
|
||||
self.assertEqual(raw2_before, raw2_after)
|
||||
|
||||
async def test_cannot_unify_mixed_wallets(self):
|
||||
path1 = self._restore_wallet_from_text("9dk", password="123456", encrypt_file=True)
|
||||
path2 = self._restore_wallet_from_text("9dk", password="asdasd", encrypt_file=False)
|
||||
path3 = self._restore_wallet_from_text("9dk", password=None)
|
||||
with open(path1, "rb") as f:
|
||||
raw1_before = f.read()
|
||||
with open(path2, "rb") as f:
|
||||
raw2_before = f.read()
|
||||
with open(path3, "rb") as f:
|
||||
raw3_before = f.read()
|
||||
|
||||
can_be_unified, is_unified = self.daemon._check_password_for_directory(old_password="123456", wallet_dir=self.wallet_dir)
|
||||
self.assertEqual((False, False), (can_be_unified, is_unified))
|
||||
is_unified = self.daemon.update_password_for_directory(old_password="123456", new_password="123456")
|
||||
self.assertFalse(is_unified)
|
||||
# verify that files on disk haven't changed:
|
||||
with open(path1, "rb") as f:
|
||||
raw1_after = f.read()
|
||||
with open(path2, "rb") as f:
|
||||
raw2_after = f.read()
|
||||
with open(path3, "rb") as f:
|
||||
raw3_after = f.read()
|
||||
self.assertEqual(raw1_before, raw1_after)
|
||||
self.assertEqual(raw2_before, raw2_after)
|
||||
self.assertEqual(raw3_before, raw3_after)
|
||||
|
||||
# "can unify pw" tests --->
|
||||
|
||||
async def test_can_unify_two_std_wallets_both_have_ks_and_sto_enc(self):
|
||||
path1 = self._restore_wallet_from_text("9dk", password="123456", encrypt_file=True)
|
||||
path2 = self._restore_wallet_from_text("x8", password="123456", encrypt_file=True)
|
||||
can_be_unified, is_unified = self.daemon._check_password_for_directory(old_password="123456", wallet_dir=self.wallet_dir)
|
||||
self.assertEqual((True, True), (can_be_unified, is_unified))
|
||||
is_unified = self.daemon.update_password_for_directory(old_password="123456", new_password="123456")
|
||||
self.assertTrue(is_unified)
|
||||
self._run_post_unif_sanity_checks([path1, path2], password="123456")
|
||||
|
||||
async def test_can_unify_two_std_wallets_one_has_ks_enc_other_has_both_enc(self):
|
||||
path1 = self._restore_wallet_from_text("9dk", password="123456", encrypt_file=True)
|
||||
path2 = self._restore_wallet_from_text("x8", password="123456", encrypt_file=False)
|
||||
with open(path2, "rb") as f:
|
||||
raw2_before = f.read()
|
||||
|
||||
can_be_unified, is_unified = self.daemon._check_password_for_directory(old_password="123456", wallet_dir=self.wallet_dir)
|
||||
self.assertEqual((True, False), (can_be_unified, is_unified))
|
||||
is_unified = self.daemon.update_password_for_directory(old_password="123456", new_password="123456")
|
||||
self.assertTrue(is_unified)
|
||||
self._run_post_unif_sanity_checks([path1, path2], password="123456")
|
||||
# verify that file at path2 changed:
|
||||
with open(path2, "rb") as f:
|
||||
raw2_after = f.read()
|
||||
self.assertNotEqual(raw2_before, raw2_after)
|
||||
|
||||
async def test_can_unify_two_std_wallets_one_without_password(self):
|
||||
path1 = self._restore_wallet_from_text("9dk", password=None)
|
||||
path2 = self._restore_wallet_from_text("x8", password="123456", encrypt_file=True)
|
||||
can_be_unified, is_unified = self.daemon._check_password_for_directory(old_password="123456", wallet_dir=self.wallet_dir)
|
||||
self.assertEqual((True, False), (can_be_unified, is_unified))
|
||||
is_unified = self.daemon.update_password_for_directory(old_password="123456", new_password="123456")
|
||||
self.assertTrue(is_unified)
|
||||
self._run_post_unif_sanity_checks([path1, path2], password="123456")
|
||||
|
||||
@as_testnet
|
||||
async def test_can_unify_large_folder_yet_to_be_unified(self):
|
||||
paths = []
|
||||
# seed
|
||||
paths.append(self._restore_wallet_from_text("9dk", password="123456", encrypt_file=True))
|
||||
paths.append(self._restore_wallet_from_text("9dk", password="123456", encrypt_file=False))
|
||||
paths.append(self._restore_wallet_from_text("9dk", password=None))
|
||||
# xpub
|
||||
xpub = "vpub5UqWay427dCjkpE3gPKLnkBUqDRoBed1328uNrLDoTyKo6HFSs9agfDMy1VXbVtcuBVRiAZQsPPsPdu1Ge8m8qvNZPyzJ4ecPsf6U1ieW4x"
|
||||
paths.append(self._restore_wallet_from_text(xpub, password="123456", encrypt_file=True))
|
||||
paths.append(self._restore_wallet_from_text(xpub, password="123456", encrypt_file=False))
|
||||
paths.append(self._restore_wallet_from_text(xpub, password=None))
|
||||
# xprv
|
||||
xprv = "vprv9FrABTX8HFeSYL9aaMnLRcEkHBbJnBu9foDJaTvcF8SLvHx6uKqL8rtt7kTd66V4QPLfWPaCJMVZa3h9zuzLr7YFZd1uoEevqqyxp66oSbN"
|
||||
paths.append(self._restore_wallet_from_text(xprv, password="123456", encrypt_file=True))
|
||||
paths.append(self._restore_wallet_from_text(xprv, password="123456", encrypt_file=False))
|
||||
paths.append(self._restore_wallet_from_text(xprv, password=None))
|
||||
# WIFs
|
||||
wifs= "p2wpkh:cRyfp9nJ8soK1bBUJAcWbMrsJZxKJpe7HBSxz5uXVbwydvUxz9zT p2wpkh:cV6J6T2AG4oXAXdYHAV6dbzR41QnGumDSVvWrmj2yYpos81RtyBK"
|
||||
paths.append(self._restore_wallet_from_text(wifs, password="123456", encrypt_file=True))
|
||||
paths.append(self._restore_wallet_from_text(wifs, password="123456", encrypt_file=False))
|
||||
paths.append(self._restore_wallet_from_text(wifs, password=None))
|
||||
# addrs
|
||||
addrs = "tb1qq2tmmcngng78nllq2pvrkchcdukemtj5s6l0zu tb1qm7ckcjsed98zhvhv3dr56a22w3fehlkxyh4wgd"
|
||||
paths.append(self._restore_wallet_from_text(addrs, password="123456", encrypt_file=True))
|
||||
paths.append(self._restore_wallet_from_text(addrs, password="123456", encrypt_file=False))
|
||||
paths.append(self._restore_wallet_from_text(addrs, password=None))
|
||||
# do unification
|
||||
can_be_unified, is_unified = self.daemon._check_password_for_directory(old_password="123456", wallet_dir=self.wallet_dir)
|
||||
self.assertEqual((True, False), (can_be_unified, is_unified))
|
||||
is_unified = self.daemon.update_password_for_directory(old_password="123456", new_password="123456")
|
||||
self.assertTrue(is_unified)
|
||||
self._run_post_unif_sanity_checks(paths, password="123456")
|
||||
|
||||
|
||||
class TestCommandsWithDaemon(DaemonTestCase):
|
||||
TESTNET = True
|
||||
|
||||
async def test_wp_command_with_inmemory_wallet_has_password(self):
|
||||
cmds = Commands(config=self.config, daemon=self.daemon)
|
||||
wallet = restore_wallet_from_text('bitter grass shiver impose acquire brush forget axis eager alone wine silver',
|
||||
gap_limit=2,
|
||||
path=None,
|
||||
password="123456",
|
||||
config=self.config)['wallet']
|
||||
self.assertEqual("bitter grass shiver impose acquire brush forget axis eager alone wine silver",
|
||||
await cmds.getseed(wallet=wallet, password="123456"))
|
||||
|
||||
async def test_wp_command_with_inmemory_wallet_no_password(self):
|
||||
cmds = Commands(config=self.config, daemon=self.daemon)
|
||||
wallet = restore_wallet_from_text('bitter grass shiver impose acquire brush forget axis eager alone wine silver',
|
||||
gap_limit=2,
|
||||
path=None,
|
||||
config=self.config)['wallet']
|
||||
self.assertEqual("bitter grass shiver impose acquire brush forget axis eager alone wine silver",
|
||||
await cmds.getseed(wallet=wallet))
|
||||
|
||||
async def test_wp_command_with_diskfile_wallet_has_password(self):
|
||||
cmds = Commands(config=self.config, daemon=self.daemon)
|
||||
wpath = self._restore_wallet_from_text("bitter grass shiver impose acquire brush forget axis eager alone wine silver", password="123456", encrypt_file=True)
|
||||
await cmds.load_wallet(wallet_path=wpath, password="123456")
|
||||
wallet = self.daemon.get_wallet(wpath)
|
||||
self.assertIsInstance(wallet, Abstract_Wallet)
|
||||
|
||||
# when using the CLI/RPC to run commands, the "wallet" param is a path:
|
||||
self.assertEqual("bitter grass shiver impose acquire brush forget axis eager alone wine silver",
|
||||
await cmds.getseed(wallet=wpath, password="123456"))
|
||||
# in unit tests or custom code, the "wallet" param is often an Abstract_Wallet:
|
||||
self.assertEqual("bitter grass shiver impose acquire brush forget axis eager alone wine silver",
|
||||
await cmds.getseed(wallet=wallet, password="123456"))
|
||||
|
||||
async def test_wp_command_with_diskfile_wallet_no_password(self):
|
||||
cmds = Commands(config=self.config, daemon=self.daemon)
|
||||
wpath = self._restore_wallet_from_text("bitter grass shiver impose acquire brush forget axis eager alone wine silver", password=None)
|
||||
await cmds.load_wallet(wallet_path=wpath, password=None)
|
||||
wallet = self.daemon.get_wallet(wpath)
|
||||
self.assertIsInstance(wallet, Abstract_Wallet)
|
||||
|
||||
# when using the CLI/RPC to run commands, the "wallet" param is a path:
|
||||
self.assertEqual("bitter grass shiver impose acquire brush forget axis eager alone wine silver",
|
||||
await cmds.getseed(wallet=wpath))
|
||||
# in unit tests or custom code, the "wallet" param is often an Abstract_Wallet:
|
||||
self.assertEqual("bitter grass shiver impose acquire brush forget axis eager alone wine silver",
|
||||
await cmds.getseed(wallet=wallet))
|
||||
Reference in New Issue
Block a user