#3 Add testnet support

Author: oskyk

cashaddress/convert.py

@@ -10,23 +10,31 @@ class InvalidAddress(Exception):
 class Address:
     VERSION_MAP = {
         'legacy': [
-            ('P2SH', 5),
-            ('P2PKH', 0)
+            ('P2SH', 5, False),
+            ('P2PKH', 0, False),
+            ('P2SH-TESTNET', 196, True),
+            ('P2PKH-TESTNET', 111, True)
         ],
         'cash': [
-            ('P2SH', 8),
-            ('P2PKH', 0)
+            ('P2SH', 8, False),
+            ('P2PKH', 0, False),
+            ('P2SH-TESTNET', 8, True),
+            ('P2PKH-TESTNET', 0, True)
         ]
     }
-    DEFAULT_PREFIX = 'bitcoincash'
+    MAINNET_PREFIX = 'bitcoincash'
+    TESTNET_PREFIX = 'bchtest'
 
     def __init__(self, version, payload, prefix=None):
         self.version = version
         self.payload = payload
         if prefix:
             self.prefix = prefix
         else:
-            self.prefix = self.DEFAULT_PREFIX
+            if Address._address_type('cash', self.version)[2]:
+                self.prefix = self.TESTNET_PREFIX
+            else:
+                self.prefix = self.MAINNET_PREFIX
 
     def __str__(self):
         return 'version: {}\npayload: {}\nprefix: {}'.format(self.version, self.payload, self.prefix)
@@ -90,13 +98,15 @@ def _cash_string(address_string):
             raise InvalidAddress('Cash address contains uppercase and lowercase characters')
         address_string = address_string.lower()
         if ':' not in address_string:
-            address_string = Address.DEFAULT_PREFIX + ':' + address_string
+            address_string = Address.MAINNET_PREFIX + ':' + address_string
         prefix, base32string = address_string.split(':')
         decoded = b32decode(base32string)
         if not verify_checksum(prefix, decoded):
             raise InvalidAddress('Bad cash address checksum')
         converted = convertbits(decoded, 5, 8)
         version = Address._address_type('cash', converted[0])[0]
+        if prefix == Address.TESTNET_PREFIX:
+            version += '-TESTNET'
         payload = converted[1:-6]
         return Address(version, payload, prefix)
 

tests/test.py

@@ -27,6 +27,30 @@ def test_to_cash_p2pkh(self):
         self.assertEqual(convert.to_cash_address('bitcoincash:qqkv9wr69ry2p9l53lxp635va4h86wv435995w8p2h'),
                          'bitcoincash:qqkv9wr69ry2p9l53lxp635va4h86wv435995w8p2h')
 
+    def test_to_legacy_p2sh_testnet(self):
+        self.assertEqual(convert.to_legacy_address('2MwikwR6hoVijCmr1u8UgzFMHFP6rpQyRvP'),
+                         '2MwikwR6hoVijCmr1u8UgzFMHFP6rpQyRvP')
+        self.assertEqual(convert.to_legacy_address('bchtest:pqc3tyspqwn95retv5k3c5w4fdq0cxvv95u36gfk00'),
+                         '2MwikwR6hoVijCmr1u8UgzFMHFP6rpQyRvP')
+
+    def test_to_legacy_p2pkh_testnet(self):
+        self.assertEqual(convert.to_legacy_address('mqp7vM7eU7Vu9NPH1V7s7pPg5FFBMo6SWK'),
+                         'mqp7vM7eU7Vu9NPH1V7s7pPg5FFBMo6SWK')
+        self.assertEqual(convert.to_legacy_address('bchtest:qpc0qh2xc3tfzsljq79w37zx02kwvzm4gydm222qg8'),
+                         'mqp7vM7eU7Vu9NPH1V7s7pPg5FFBMo6SWK')
+
+    def test_to_cash_p2sh_testnet(self):
+        self.assertEqual(convert.to_cash_address('2MwikwR6hoVijCmr1u8UgzFMHFP6rpQyRvP'),
+                         'bchtest:pqc3tyspqwn95retv5k3c5w4fdq0cxvv95u36gfk00')
+        self.assertEqual(convert.to_cash_address('bchtest:pqc3tyspqwn95retv5k3c5w4fdq0cxvv95u36gfk00'),
+                         'bchtest:pqc3tyspqwn95retv5k3c5w4fdq0cxvv95u36gfk00')
+
+    def test_to_cash_p2pkh_testnet(self):
+        self.assertEqual(convert.to_cash_address('mqp7vM7eU7Vu9NPH1V7s7pPg5FFBMo6SWK'),
+                         'bchtest:qpc0qh2xc3tfzsljq79w37zx02kwvzm4gydm222qg8')
+        self.assertEqual(convert.to_cash_address('bchtest:qpc0qh2xc3tfzsljq79w37zx02kwvzm4gydm222qg8'),
+                         'bchtest:qpc0qh2xc3tfzsljq79w37zx02kwvzm4gydm222qg8')
+
 
 if __name__ == '__main__':
     unittest.main()