Added basic testnet tests

11 years ago · 0d25c8974e
parent 22dd34baa1
commit 0d25c8974e
3 changed files with 200 additions and 5 deletions
--- a/tests/test.py
+++ b/tests/test.py
@ -14,10 +14,10 @@ TODO:
  * PassphraseRequest/PassphraseAck workflow (vyvolat napr. pomocou GetAddress)

 * rozsirit test_sign.tx o viac transakcii (zlozitejsich)
-    * fee over threshold
-    * not enough funds
-    * viac ako jeden vstup a jeden vystup
-    * iny cointype ako 0
+    x fee over threshold
+    x not enough funds
+    x viac ako jeden vstup a jeden vystup
+    x iny cointype ako 0

 - chceme v tomto release(?)
  * SignMessage workflow
--- a/tests/test_signtx.py
+++ b/tests/test_signtx.py
@ -1,9 +1,40 @@
+# tx 4a7b7e0403ae5607e473949cfa03f09f2cd8b0f404bf99ce10b7303d86280bf7
+# 100 UTXO for spending for unittests
+
 import unittest
 import common
 import binascii

 import trezorlib.messages_pb2 as proto
 import trezorlib.types_pb2 as proto_types
+from trezorlib.client import CallException
+
+class FakeTestnetBlockchain(object):
+    def get_tx(self, txhash):
+        if txhash != '6f90f3c7cbec2258b0971056ef3fe34128dbde30daa9c0639a898f9977299d54':
+            raise Exception("Unexpected hash")
+
+        t = proto_types.TransactionType()
+
+        i = t.inputs.add()
+        i.prev_hash = binascii.unhexlify('ee336e79153d51f4f3e45278f1f77ab29fd5bb135dce467282e2aff22cb9c570')
+        i.prev_index = 1
+        i.script_sig = binascii.unhexlify('483045022066c418874dbe5628296700382d727ce1734928796068c26271472df09dccf1a20221009dec59d19f9d73db381fcd35c0fff757ad73e54ef59157b0d7c57e6739a092f00121033fef08c603943dc7d25f4ce65771762143b1cd8678343d660a1a76b9d1d3ced7')
+
+        i = t.inputs.add()
+        i.prev_hash = binascii.unhexlify('2fe4d8af2b44faccc10dd5a6578c923491d2d21269a1dfe8c83f492a30fb8f9f')
+        i.prev_index = 1
+        i.script_sig = binascii.unhexlify('47304402206fbb8e14be706b8557a2280d2a2a75c0a65c4f7936d90d510f0971c93f41f74402201b79c8c4e4ac4c944913611633c230193558296e70a36269b7fc3a80efa27d120121030cb5be79bdc36a4ff4443dbac43068cc43d638ea06ff2fa1b8dab389e39aefc7')
+
+        o = t.outputs.add()
+        o.amount = 403850989
+        o.script_pubkey = binascii.unhexlify('76a914f5a05c2664b40d3116b1c5086c9ba38ed15b742e88ac')
+
+        o = t.outputs.add()
+        o.amount = 1000000000
+        o.script_pubkey = binascii.unhexlify('76a91424a56db43cf6f2b02e838ea493f95d8d6047423188ac')
+
+        return t

 class TestSignTx(common.TrezorTest):

@ -29,6 +60,60 @@ class TestSignTx(common.TrezorTest):
        # Accepted by network: tx fd79435246dee76b2f159d2db08032d666c95adc544de64c8c49f474df4a7fee
        self.assertEqual(binascii.hexlify(tx.serialized_tx), '010000000182488650ef25a58fef6788bd71b8212038d7f2bbe4750bc7bcb44701e85ef6d5000000006b4830450221009a0b7be0d4ed3146ee262b42202841834698bb3ee39c24e7437df208b8b7077102202b79ab1e7736219387dffe8d615bbdba87e11477104b867ef47afed1a5ede7810121023230848585885f63803a0a8aecdd6538792d5c539215c91698e315bf0253b43dffffffff0160cc0500000000001976a914de9b2a8da088824e8fe51debea566617d851537888ac00000000')

+    def test_testnet_one_two_fee(self):
+        # tx: 6f90f3c7cbec2258b0971056ef3fe34128dbde30daa9c0639a898f9977299d54
+        # input 1: 10.00000000 BTC
+        inp1 = proto_types.TxInputType(address_n=[0],  # mirio8q3gtv7fhdnmb3TpZ4EuafdzSs7zL
+                             # amount=1000000000,
+                             prev_hash=binascii.unhexlify('6f90f3c7cbec2258b0971056ef3fe34128dbde30daa9c0639a898f9977299d54'),
+                             prev_index=1,
+                             )
+
+        out1 = proto_types.TxOutputType(address='mfiGQVPcRcaEvQPYDErR34DcCovtxYvUUV',
+                              amount=1000000000 - 500000000 - 10000000,
+                              script_type=proto_types.PAYTOADDRESS,
+                              )
+
+        out2 = proto_types.TxOutputType(address_n=[2],
+                              amount=500000000,
+                              script_type=proto_types.PAYTOADDRESS,
+                              )
+
+        self.client.blockchain = FakeTestnetBlockchain()
+        msg = self.client._prepare_simple_sign_tx('Testnet', [inp1, ], [out1, out2])
+        expected_btn = [proto_types.ButtonRequest_ConfirmOutput] * 2
+        tx = self.client.call(msg, expected_buttonrequests=expected_btn)
+
+        self.assertEqual(binascii.hexlify(tx.serialized_tx), '0100000001549d2977998f899a63c0a9da30dedb2841e33fef561097b05822eccbc7f3906f010000006b4830450221009c2d30385519fdb13dce13d5ac038be07d7b2dad0b0f7b2c1c339d7255bcf553022056a2f5bceab3cd0ffed4d388387e631f419d67ff9ce7798e3d7dfe6a6d6ec4bd0121023230848585885f63803a0a8aecdd6538792d5c539215c91698e315bf0253b43dffffffff0280ce341d000000001976a9140223b1a09138753c9cb0baf95a0a62c82711567a88ac0065cd1d000000001976a9142db345c36563122e2fd0f5485fb7ea9bbf7cb5a288ac00000000')
+
+    def test_testnet_fee_too_high(self):
+        # tx: 6f90f3c7cbec2258b0971056ef3fe34128dbde30daa9c0639a898f9977299d54
+        # input 1: 10.00000000 BTC
+        inp1 = proto_types.TxInputType(address_n=[0],  # mirio8q3gtv7fhdnmb3TpZ4EuafdzSs7zL
+                             # amount=1000000000,
+                             prev_hash=binascii.unhexlify('6f90f3c7cbec2258b0971056ef3fe34128dbde30daa9c0639a898f9977299d54'),
+                             prev_index=1,
+                             )
+
+        out1 = proto_types.TxOutputType(address='mfiGQVPcRcaEvQPYDErR34DcCovtxYvUUV',
+                              amount=1000000000 - 500000000 - 20000000,
+                              script_type=proto_types.PAYTOADDRESS,
+                              )
+
+        out2 = proto_types.TxOutputType(address_n=[2],
+                              amount=500000000,
+                              script_type=proto_types.PAYTOADDRESS,
+                              )
+
+        self.client.blockchain = FakeTestnetBlockchain()
+        msg = self.client._prepare_simple_sign_tx('Testnet', [inp1, ], [out1, out2])
+        expected_btn = [proto_types.ButtonRequest_ConfirmOutput,
+                        proto_types.ButtonRequest_ConfirmOutput,
+                        proto_types.ButtonRequest_FeeOverThreshold]
+        tx = self.client.call(msg, expected_buttonrequests=expected_btn)
+
+        self.assertEqual(binascii.hexlify(tx.serialized_tx), '0100000001549d2977998f899a63c0a9da30dedb2841e33fef561097b05822eccbc7f3906f010000006b483045022100d74e9fa5c7ff5966d52bce8d1d772c1e3ef1376395fb85a0bbf910e723bd606d02204cb8df6debd7c4c076632011bb1e180495bcf3630d2471c354bed56c2e45a2180121023230848585885f63803a0a8aecdd6538792d5c539215c91698e315bf0253b43dffffffff0200389c1c000000001976a9140223b1a09138753c9cb0baf95a0a62c82711567a88ac0065cd1d000000001976a9142db345c36563122e2fd0f5485fb7ea9bbf7cb5a288ac00000000')
+
    def test_one_two_fee(self):
        # tx: d5f65ee80147b4bcc70b75e4bbf2d7382021b871bd8867ef8fa525ef50864882
        # input 0: 0.0039 BTC
@ -122,6 +207,8 @@ class TestSignTx(common.TrezorTest):
        self.assertEqual(binascii.hexlify(tx.serialized_tx), '01000000021c032e5715d1da8115a2fe4f57699e15742fe113b0d2d1ca3b594649d322bec6010000006c493046022100f773c403b2f85a5c1d6c9c4ad69c43de66930fff4b1bc818eb257af98305546a022100bbc421b41bc60d89593186c99b4b9ca6ac8dce1fe62342d38a8e14e0cbf279dc01210338d78612e990f2eea0c426b5e48a8db70b9d7ed66282b3b26511e0b1c75515a6ffffffff6ea42cd8d9c8e5441c4c5f85bfe50311078730d2881494f11f4d2257777a4958010000006c49304602210090cff1c1911e771605358a8cddd5ae94c7b60cc96e50275908d9bf9d6367c79f022100d4058d1efd9f5eb9542a62fad91ecca30db32a326f311f51dc2b1a71cbb578670121038caebd6f753bbbd2bb1f3346a43cd32140648583673a31d62f2dfb56ad0ab9e3ffffffff02a0860100000000001976a9142f4490d5263906e4887ca2996b9e207af3e7824088aca0860100000000001976a914812c13d97f9159e54e326b481b8f88a73df8507a88ac00000000')

    def test_255_outputs(self):
+        # Tests if device implements serialization of len(outputs) correctly
+
        # tx: c63e24ed820c5851b60c54613fbc4bcb37df6cd49b4c96143e99580a472f79fb
        # index 1: 0.0010 BTC
        # tx: 39a29e954977662ab3879c66fb251ef753e0912223a83d1dcb009111d28265e5
@ -175,6 +262,30 @@ class TestSignTx(common.TrezorTest):

        self.assertEqual(binascii.hexlify(tx.serialized_tx), '010000000182488650ef25a58fef6788bd71b8212038d7f2bbe4750bc7bcb44701e85ef6d5000000006b4830450220361b8268718533055682f4532d30c553ce778f70b217457a9aa1956e457c5aac022100ac751d7a44cbfb7b44f65989ca2d6e6c19607b0e071044b0b6c7cba8e6b825400121023230848585885f63803a0a8aecdd6538792d5c539215c91698e315bf0253b43dffffffff0120300500000000001976a914de9b2a8da088824e8fe51debea566617d851537888ac00000000')

+    def test_not_enough_funds(self):
+        # tx: d5f65ee80147b4bcc70b75e4bbf2d7382021b871bd8867ef8fa525ef50864882
+        # input 0: 0.0039 BTC
+
+        inp1 = proto_types.TxInputType(address_n=[0],  # 14LmW5k4ssUrtbAB4255zdqv3b4w1TuX9e
+                             # amount=390000,
+                             prev_hash=binascii.unhexlify('d5f65ee80147b4bcc70b75e4bbf2d7382021b871bd8867ef8fa525ef50864882'),
+                             prev_index=0,
+                             )
+
+        out1 = proto_types.TxOutputType(address='1MJ2tj2ThBE62zXbBYA5ZaN3fdve5CPAz1',
+                              amount=400000,
+                              script_type=proto_types.PAYTOADDRESS,
+                              )
+
+        msg = self.client._prepare_simple_sign_tx('Bitcoin', [inp1, ], [out1, ])
+        expected_btn = [proto_types.ButtonRequest_ConfirmOutput]
+        try:
+            self.client.call(msg, expected_buttonrequests=expected_btn)
+        except CallException as e:
+            self.assertEqual(e.args[0], proto_types.Failure_NotEnoughFunds)
+        else:
+            self.assert_(False, "types.Failure_NotEnoughFunds expected")
+
    def test_estimate_size(self):
        inp1 = proto_types.TxInputType(address_n=[0],  # 14LmW5k4ssUrtbAB4255zdqv3b4w1TuX9e
                             # amount=390000,
--- a/tests/test_zerosig.py
+++ b/tests/test_zerosig.py
@ -0,0 +1,84 @@
+import unittest
+import common
+import binascii
+
+import trezorlib.messages_pb2 as proto
+import trezorlib.types_pb2 as proto_types
+
+# address_n = [177] < 68
+# address_n = [16518] < 66
+class TestZeroSig(common.TrezorTest):
+
+    '''
+    def test_mine_zero_signature(self):
+        # tx: d5f65ee80147b4bcc70b75e4bbf2d7382021b871bd8867ef8fa525ef50864882
+        # input 0: 0.0039 BTC
+
+        inp1 = proto_types.TxInputType(address_n=[0],  # 14LmW5k4ssUrtbAB4255zdqv3b4w1TuX9e
+                             # amount=390000,
+                             prev_hash=binascii.unhexlify('d5f65ee80147b4bcc70b75e4bbf2d7382021b871bd8867ef8fa525ef50864882'),
+                             prev_index=0,
+                             )
+
+        msg = self.client._prepare_simple_sign_tx('Bitcoin', [inp1, ], [])
+
+        for n in range(3500, 200000):
+            out1 = proto_types.TxOutputType(address_n=[n],
+                                  amount=390000 - 10000,
+                                  script_type=proto_types.PAYTOADDRESS,
+                                  )
+            msg.ClearField('outputs')
+            msg.outputs.extend([out1, ])
+
+            tx = self.client.call(msg)
+    
+            siglen = ord(tx.serialized_tx[44])
+            print siglen
+            if siglen < 67:
+                print "!!!!", n
+                print binascii.hexlify(tx.serialized_tx)
+                return
+    '''
+
+    def test_one_zero_signature(self):
+        inp1 = proto_types.TxInputType(address_n=[0],  # 14LmW5k4ssUrtbAB4255zdqv3b4w1TuX9e
+                             # amount=390000,
+                             prev_hash=binascii.unhexlify('d5f65ee80147b4bcc70b75e4bbf2d7382021b871bd8867ef8fa525ef50864882'),
+                             prev_index=0,
+                             )
+
+        # Following address_n has been mined by 'test_mine_zero_signature'
+        out1 = proto_types.TxOutputType(address_n=[177],
+                              amount=390000 - 10000,
+                              script_type=proto_types.PAYTOADDRESS,
+                              )
+
+        msg = self.client._prepare_simple_sign_tx('Bitcoin', [inp1], [out1])
+        tx = self.client.call(msg)   
+        siglen = ord(tx.serialized_tx[44])
+
+        # Trezor must strip leading zero from signature
+        self.assertEqual(siglen, 67)
+
+    def test_two_zero_signature(self):
+        inp1 = proto_types.TxInputType(address_n=[0],  # 14LmW5k4ssUrtbAB4255zdqv3b4w1TuX9e
+                             # amount=390000,
+                             prev_hash=binascii.unhexlify('d5f65ee80147b4bcc70b75e4bbf2d7382021b871bd8867ef8fa525ef50864882'),
+                             prev_index=0,
+                             )
+
+        # Following address_n has been mined by 'test_mine_zero_signature'
+        out1 = proto_types.TxOutputType(address_n=[16518],
+                              amount=390000 - 10000,
+                              script_type=proto_types.PAYTOADDRESS,
+                              )
+
+        msg = self.client._prepare_simple_sign_tx('Bitcoin', [inp1], [out1])
+        tx = self.client.call(msg)
+        siglen = ord(tx.serialized_tx[44])
+
+        # Trezor must strip leading zero from signature
+        self.assertEqual(siglen, 66)
+        
+if __name__ == '__main__':
+    unittest.main()