commands: add convert_xkey for converting between {x,y,z}|{pub,prv}

2019-01-20 15:49:42 +01:00
parent 9cff42328d
commit 5fc715cdee
2 changed files with 56 additions and 0 deletions
--- a/electrum/tests/test_commands.py
+++ b/electrum/tests/test_commands.py
@@ -3,6 +3,8 @@ from decimal import Decimal

 from electrum.commands import Commands, eval_bool

+from . import TestCaseForTestnet
+

 class TestCommands(unittest.TestCase):

@@ -39,3 +41,46 @@ class TestCommands(unittest.TestCase):
        self.assertTrue(eval_bool("True"))
        self.assertTrue(eval_bool("true"))
        self.assertTrue(eval_bool("1"))
+
+    def test_convert_xkey(self):
+        cmds = Commands(config=None, wallet=None, network=None)
+        xpubs = {
+            ("xpub6CCWFbvCbqF92kGwm9nV7t7RvVoQUKaq5USMdyVP6jvv1NgN52KAX6NNYCeE8Ca7JQC4K5tZcnQrubQcjJ6iixfPs4pwAQJAQgTt6hBjg11", "standard"),
+            ("ypub6X2mZGb7kWnct3U4bWa7KyCw6TwrQwaKzaxaRNPGUkJo4UVbKgUj9A2WZQbp87E2i3Js4ZV85SmQnt2BSzWjXCLzjQXMkK7egQXXVHT4eKn", "p2wpkh-p2sh"),
+            ("zpub6qs2rwG2uCL6jLfBRsMjY4JSGS6JMZZpuhUoCmH9rkgg7aJpaLeHmDgeacZQ81sx7gRfp35gY77xgAdkAgvkKS2bbkDnLDw8x8bAsuKBrvP", "p2wpkh"),
+        }
+        for xkey1, xtype1 in xpubs:
+            for xkey2, xtype2 in xpubs:
+                self.assertEqual(xkey2, cmds.convert_xkey(xkey1, xtype2))
+
+        xprvs = {
+            ("xprv9yD9r6PJmTgqpGCUf8FUkkAhNTxv4rryiFWkqb5mYQPw8aMDXUzuyJ3tgv5vUqYkdK1E6Q5jKxPss4HkMBYV4q8AfG8t7rxgyS4xQX4ndAm", "standard"),
+            ("yprvAJ3R9m4Dv9EKfZPbVV36xqGCYS7N1UrUdN2ycyyevQmpBgASn9AUbMi2i83WUkCg2x82qsgHnckRkLuK4sxVs4omXbqJhmnBFA8bo8ssinK", "p2wpkh-p2sh"),
+            ("zprvAcsgTRj94pmoWraiKqpjAvMhiQFox6qyYUZCQNsYJR9hEmyg2oL3DRNAjL16UerbSbEqbMGrFH6yddWsnaNWfJVNPwXjHgbfWtCFBgDxFkX", "p2wpkh"),
+        }
+        for xkey1, xtype1 in xprvs:
+            for xkey2, xtype2 in xprvs:
+                self.assertEqual(xkey2, cmds.convert_xkey(xkey1, xtype2))
+
+
+class TestCommandsTestnet(TestCaseForTestnet):
+
+    def test_convert_xkey(self):
+        cmds = Commands(config=None, wallet=None, network=None)
+        xpubs = {
+            ("tpubD8p5qNfjczgTGbh9qgNxsbFgyhv8GgfVkmp3L88qtRm5ibUYiDVCrn6WYfnGey5XVVw6Bc5QNQUZW5B4jFQsHjmaenvkFUgWtKtgj5AdPm9", "standard"),
+            ("upub59wfQ8qJTg6ZSuvwtR313Qdp8gP8TSBwTof5dPQ3QVsYp1N9t29Rr9TGF1pj8kAXUg3mKbmrTKasA2qmBJKb1bGUzB6ApDZpVC7LoHhyvBo", "p2wpkh-p2sh"),
+            ("vpub5UmvhoWDcMe3JD84impdFVjKJeXaQ4BSNvBJQnHvnWFRs7BP8gJzUD7QGDnK8epStKAa55NQuywR3KTKtzjbopx5rWnbQ8PJkvAzBtgaGBc", "p2wpkh"),
+        }
+        for xkey1, xtype1 in xpubs:
+            for xkey2, xtype2 in xpubs:
+                self.assertEqual(xkey2, cmds.convert_xkey(xkey1, xtype2))
+
+        xprvs = {
+            ("tprv8c83gxdVUcznP8fMx2iNUBbaQgQC7MUbBUDG3c6YU9xgt7Dn5pfcgHUeNZTAvuYmNgVHjyTzYzGWwJr7GvKCm2FkPaaJipyipbfJeB3tdPW", "standard"),
+            ("uprv8vxJzdJQdJYGERrUnPVzgGh5aeYe3yU66ajUpzzRrALZwD31LUqBJM8nPmQkvpCgnKc6VT4Z1ed4pbTfzcjDZFwMFvGjJjoD6Kix2pCwVe7", "p2wpkh-p2sh"),
+            ("vprv9FnaJHyKmz5k5j3bckHctMnakch5zbTb1hFhcPtKEAiSzJrEb8zjvQnvQyNLvircBxiuEvf7UJycht5EiK9EMVcx8Fy9techN3nbRQRFhEv", "p2wpkh"),
+        }
+        for xkey1, xtype1 in xprvs:
+            for xkey2, xtype2 in xprvs:
+                self.assertEqual(xkey2, cmds.convert_xkey(xkey1, xtype2))