test: make functional test wallet_mnemonicbits.py works for descriptor wallets

Author: knst

test/functional/wallet_mnemonicbits.py

@@ -24,7 +24,42 @@ def run_test(self):
         self.stop_node(0)
         self.nodes[0].assert_start_raises_init_error(['-mnemonicbits=123'], "Error: Invalid '-mnemonicbits'. Allowed values: 128, 160, 192, 224, 256.")
         self.start_node(0)
-        assert_equal(len(self.nodes[0].dumphdinfo()["mnemonic"].split()), 12)  # 12 words by default
+
+        mnemonic_pre = self.nodes[0].listdescriptors(True)['descriptors'][1]["mnemonic"] if self.options.descriptors else self.nodes[0].dumphdinfo()["mnemonic"]
+
+        self.nodes[0].encryptwallet('pass')
+        self.nodes[0].walletpassphrase('pass', 100)
+        if self.options.descriptors:
+            assert "mnemonic" not in self.nodes[0].listdescriptors()['descriptors'][0]
+            assert "mnemonic" in self.nodes[0].listdescriptors(True)['descriptors'][0]
+
+            descriptors = self.nodes[0].listdescriptors(True)['descriptors']
+            assert_equal(descriptors[0]['mnemonic'], descriptors[1]['mnemonic'])
+
+            mnemonic_count = 0
+            found_in_encrypted = 0
+            for desc in descriptors:
+                if 'mnemonic' not in desc:
+                    # skip imported coinbase private key
+                    continue
+                assert_equal(len(desc['mnemonic'].split()), 12)
+                mnemonic_count += 1
+                if desc['mnemonic'] == mnemonic_pre:
+                    found_in_encrypted += 1
+                    assert not desc['active']
+                else:
+                    assert desc['active']
+            # there should 5 descriptors in total
+            # one of them imported private key for coinbase without mnemonic
+            # encryption of descriptor wallet creates new private keys,
+            # it should be 2 active and 2 inactive mnemonics
+            assert_equal(found_in_encrypted, 2)
+            assert_equal(mnemonic_count, 4)
+            assert_equal(len(descriptors), 5)
+        else:
+            assert_equal(len(self.nodes[0].dumphdinfo()["mnemonic"].split()), 12)  # 12 words by default
+            # legacy HD wallets could have only one chain
+            assert_equal(mnemonic_pre, self.nodes[0].dumphdinfo()["mnemonic"])
 
         self.log.info("Can have multiple wallets with different mnemonic length loaded at the same time")
         self.restart_node(0, extra_args=["-mnemonicbits=160"])
@@ -34,16 +69,27 @@ def run_test(self):
         self.restart_node(0, extra_args=["-mnemonicbits=224"])
         self.nodes[0].createwallet("wallet_224")
         self.restart_node(0, extra_args=["-mnemonicbits=256"])
+        self.nodes[0].get_wallet_rpc(self.default_wallet_name).walletpassphrase('pass', 100)
         self.nodes[0].loadwallet("wallet_160")
         self.nodes[0].loadwallet("wallet_192")
         self.nodes[0].loadwallet("wallet_224")
-        self.nodes[0].createwallet("wallet_256", False, True)  # blank
-        self.nodes[0].get_wallet_rpc("wallet_256").upgradetohd()
-        assert_equal(len(self.nodes[0].get_wallet_rpc(self.default_wallet_name).dumphdinfo()["mnemonic"].split()), 12)  # 12 words by default
-        assert_equal(len(self.nodes[0].get_wallet_rpc("wallet_160").dumphdinfo()["mnemonic"].split()), 15)              # 15 words
-        assert_equal(len(self.nodes[0].get_wallet_rpc("wallet_192").dumphdinfo()["mnemonic"].split()), 18)              # 18 words
-        assert_equal(len(self.nodes[0].get_wallet_rpc("wallet_224").dumphdinfo()["mnemonic"].split()), 21)              # 21 words
-        assert_equal(len(self.nodes[0].get_wallet_rpc("wallet_256").dumphdinfo()["mnemonic"].split()), 24)              # 24 words
+        if self.options.descriptors:
+            self.nodes[0].createwallet("wallet_256", False, True, "", False, True)  # blank Descriptors
+            self.nodes[0].get_wallet_rpc("wallet_256").upgradetohd()
+            # first descriptor is private key with no mnemonic for CbTx (see node.importprivkey), we use number#1 here instead
+            assert_equal(len(self.nodes[0].get_wallet_rpc(self.default_wallet_name).listdescriptors(True)["descriptors"][1]["mnemonic"].split()), 12)  # 12 words by default
+            assert_equal(len(self.nodes[0].get_wallet_rpc("wallet_160").listdescriptors(True)["descriptors"][0]["mnemonic"].split()), 15)              # 15 words
+            assert_equal(len(self.nodes[0].get_wallet_rpc("wallet_192").listdescriptors(True)["descriptors"][0]["mnemonic"].split()), 18)              # 18 words
+            assert_equal(len(self.nodes[0].get_wallet_rpc("wallet_224").listdescriptors(True)["descriptors"][0]["mnemonic"].split()), 21)              # 21 words
+            assert_equal(len(self.nodes[0].get_wallet_rpc("wallet_256").listdescriptors(True)["descriptors"][0]["mnemonic"].split()), 24)              # 24 words
+        else:
+            self.nodes[0].createwallet("wallet_256", False, True)  # blank HD legacy
+            self.nodes[0].get_wallet_rpc("wallet_256").upgradetohd()
+            assert_equal(len(self.nodes[0].get_wallet_rpc(self.default_wallet_name).dumphdinfo()["mnemonic"].split()), 12)  # 12 words by default
+            assert_equal(len(self.nodes[0].get_wallet_rpc("wallet_160").dumphdinfo()["mnemonic"].split()), 15)              # 15 words
+            assert_equal(len(self.nodes[0].get_wallet_rpc("wallet_192").dumphdinfo()["mnemonic"].split()), 18)              # 18 words
+            assert_equal(len(self.nodes[0].get_wallet_rpc("wallet_224").dumphdinfo()["mnemonic"].split()), 21)              # 21 words
+            assert_equal(len(self.nodes[0].get_wallet_rpc("wallet_256").dumphdinfo()["mnemonic"].split()), 24)              # 24 words
 
 
 if __name__ == '__main__':