From aac9afa7c15dee062e2676217430e12dcf1e8e37 Mon Sep 17 00:00:00 2001
From: SomberNight <somber.night@protonmail.com>
Date: Mon, 20 Mar 2023 17:49:50 +0000
Subject: [PATCH] tests: add a failing testcase for
 daemon.update_password_for_directory

related https://github.com/spesmilo/electrum/issues/8259
---
 electrum/tests/__init__.py    |  1 +
 electrum/tests/test_daemon.py | 41 +++++++++++++++++++++++++++++++++++
 2 files changed, 42 insertions(+)
 create mode 100644 electrum/tests/test_daemon.py

diff --git a/electrum/tests/__init__.py b/electrum/tests/__init__.py
index 218a0db67..6ef060e7d 100644
--- a/electrum/tests/__init__.py
+++ b/electrum/tests/__init__.py
@@ -50,6 +50,7 @@ class ElectrumTestCase(unittest.IsolatedAsyncioTestCase):
         self.electrum_path = tempfile.mkdtemp()
 
     async def asyncSetUp(self):
+        await super().asyncSetUp()
         loop = util.get_asyncio_loop()
         # IsolatedAsyncioTestCase creates event loops with debug=True, which makes the tests take ~4x time
         if not (os.environ.get("PYTHONASYNCIODEBUG") or os.environ.get("PYTHONDEVMODE")):
diff --git a/electrum/tests/test_daemon.py b/electrum/tests/test_daemon.py
new file mode 100644
index 000000000..28a5e6198
--- /dev/null
+++ b/electrum/tests/test_daemon.py
@@ -0,0 +1,41 @@
+import os
+
+from electrum.daemon import Daemon
+from electrum.simple_config import SimpleConfig
+from electrum.wallet import restore_wallet_from_text, Abstract_Wallet, Standard_Wallet
+from electrum import util
+
+from . import ElectrumTestCase
+
+
+class TestUnifiedPassword(ElectrumTestCase):
+    config: 'SimpleConfig'
+
+    def setUp(self):
+        super().setUp()
+        self.config = SimpleConfig({'electrum_path': self.electrum_path})
+        self.config.set_key("single_password", 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)
+
+    async def asyncTearDown(self):
+        await self.daemon.stop()
+        await super().asyncTearDown()
+
+    async def test_update_password_for_directory(self):
+        wallet1: Standard_Wallet = restore_wallet_from_text(
+            "9dk", path=f"{self.wallet_dir}/w1", password=None, gap_limit=2, config=self.config)['wallet']
+        wallet2: Standard_Wallet = restore_wallet_from_text(
+            "x8", path=f"{self.wallet_dir}/w2", password="123456", gap_limit=2, config=self.config)['wallet']
+        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)
+
+
+