from common import *

if not utils.BITCOIN_ONLY:
    from trezor.messages import RipplePayment
    from trezor.messages import RippleSignTx
    from apps.ripple.serialize import serialize, serialize_amount
    from apps.ripple.sign_tx import get_network_prefix


@unittest.skipUnless(not utils.BITCOIN_ONLY, "altcoin")
class TestRippleSerializer(unittest.TestCase):
    def test_amount(self):
        # https://github.com/ripple/ripple-binary-codec/blob/4581f1b41e712f545ba08be15e188a557c731ecf/test/fixtures/data-driven-tests.json#L2494
        self.assertEqual(serialize_amount(0), unhexlify("4000000000000000"))
        self.assertEqual(serialize_amount(1), unhexlify("4000000000000001"))
        self.assertEqual(serialize_amount(93493429243), unhexlify("40000015c4a483fb"))
        with self.assertRaises(ValueError):
            serialize_amount(1000000000000000000)  # too large
        with self.assertRaises(ValueError):
            serialize_amount(-1)  # negative not supported
        with self.assertRaises(Exception):
            serialize_amount(1.1)  # float numbers not supported

    def test_transactions(self):
        # from https://github.com/miracle2k/ripple-python
        source_address = "r3P9vH81KBayazSTrQj6S25jW6kDb779Gi"
        payment = RipplePayment(
            amount=200000000, destination="r3kmLJN5D28dHuH8vZNUZpMC43pEHpaocV"
        )
        common = RippleSignTx(
            address_n=None,
            fee=10,
            flags=None,
            sequence=1,
            last_ledger_sequence=None,
            payment=payment,
        )
        self.assertEqual(
            serialize(common, source_address),
            unhexlify(
                "120000240000000161400000000bebc20068400000000000000a811450f97a072f1c4357f1ad84566a609479d927c9428314550fc62003e785dc231a1058a05e56e3f09cf4e6"
            ),
        )

        source_address = "r3kmLJN5D28dHuH8vZNUZpMC43pEHpaocV"
        payment = RipplePayment(
            amount=1, destination="r3P9vH81KBayazSTrQj6S25jW6kDb779Gi"
        )
        common = RippleSignTx(
            address_n=None,
            fee=99,
            flags=None,
            sequence=99,
            last_ledger_sequence=None,
            payment=payment,
        )
        self.assertEqual(
            serialize(common, source_address),
            unhexlify(
                "12000024000000636140000000000000016840000000000000638114550fc62003e785dc231a1058a05e56e3f09cf4e6831450f97a072f1c4357f1ad84566a609479d927c942"
            ),
        )

        # https://github.com/ripple/ripple-binary-codec/blob/4581f1b41e712f545ba08be15e188a557c731ecf/test/fixtures/data-driven-tests.json#L1579
        source_address = "r9TeThyi5xiuUUrFjtPKZiHcDxs7K9H6Rb"
        payment = RipplePayment(
            amount=25000000, destination="r4BPgS7DHebQiU31xWELvZawwSG2fSPJ7C"
        )
        common = RippleSignTx(
            address_n=None,
            fee=10,
            flags=0,
            sequence=2,
            last_ledger_sequence=None,
            payment=payment,
        )
        self.assertEqual(
            serialize(common, source_address),
            unhexlify(
                "120000220000000024000000026140000000017d784068400000000000000a81145ccb151f6e9d603f394ae778acf10d3bece874f68314e851bbbe79e328e43d68f43445368133df5fba5a"
            ),
        )

        # https://github.com/ripple/ripple-binary-codec/blob/4581f1b41e712f545ba08be15e188a557c731ecf/test/fixtures/data-driven-tests.json#L1651
        source_address = "rGWTUVmm1fB5QUjMYn8KfnyrFNgDiD9H9e"
        payment = RipplePayment(
            amount=200000, destination="rw71Qs1UYQrSQ9hSgRohqNNQcyjCCfffkQ"
        )
        common = RippleSignTx(
            address_n=None,
            fee=15,
            flags=0,
            sequence=144,
            last_ledger_sequence=None,
            payment=payment,
        )
        # 201b005ee9ba removed from the test vector because last ledger sequence is not supported
        self.assertEqual(
            serialize(common, source_address),
            unhexlify(
                "12000022000000002400000090614000000000030d4068400000000000000f8114aa1bd19d9e87be8069fdbf6843653c43837c03c6831467fe6ec28e0464dd24fb2d62a492aac697cfad02"
            ),
        )

        # https://github.com/ripple/ripple-binary-codec/blob/4581f1b41e712f545ba08be15e188a557c731ecf/test/fixtures/data-driven-tests.json#L1732
        source_address = "r4BPgS7DHebQiU31xWELvZawwSG2fSPJ7C"
        payment = RipplePayment(
            amount=25000000,
            destination="rBqSFEFg2B6GBMobtxnU1eLA1zbNC9NDGM",
            destination_tag=4146942154,
        )
        common = RippleSignTx(
            address_n=None,
            fee=12,
            flags=0,
            sequence=1,
            last_ledger_sequence=None,
            payment=payment,
        )
        self.assertEqual(
            serialize(common, source_address),
            unhexlify(
                "120000220000000024000000012ef72d50ca6140000000017d784068400000000000000c8114e851bbbe79e328e43d68f43445368133df5fba5a831476dac5e814cd4aa74142c3ab45e69a900e637aa2"
            ),
        )

    def test_transactions_for_signing(self):
        # https://github.com/ripple/ripple-binary-codec/blob/4581f1b41e712f545ba08be15e188a557c731ecf/test/signing-data-encoding-test.js
        source_address = "r9LqNeG6qHxjeUocjvVki2XR35weJ9mZgQ"
        payment = RipplePayment(
            amount=1000, destination="rHb9CJAWyB4rj91VRWn96DkukG4bwdtyTh"
        )
        common = RippleSignTx(
            address_n=None,
            fee=10,
            flags=2147483648,
            sequence=1,
            last_ledger_sequence=None,
            payment=payment,
        )

        tx = serialize(
            common,
            source_address,
            pubkey=unhexlify(
                "ed5f5ac8b98974a3ca843326d9b88cebd0560177b973ee0b149f782cfaa06dc66a"
            ),
        )
        tx = get_network_prefix() + tx

        self.assertEqual(tx[0:4], unhexlify("53545800"))  # signing prefix
        self.assertEqual(tx[4:7], unhexlify("120000"))  # transaction type
        self.assertEqual(tx[7:12], unhexlify("2280000000"))  # flags
        self.assertEqual(tx[12:17], unhexlify("2400000001"))  # sequence
        self.assertEqual(tx[17:26], unhexlify("6140000000000003e8"))  # amount
        self.assertEqual(tx[26:35], unhexlify("68400000000000000a"))  # fee
        self.assertEqual(
            tx[35:70],
            unhexlify(
                "7321ed5f5ac8b98974a3ca843326d9b88cebd0560177b973ee0b149f782cfaa06dc66a"
            ),
        )  # signing pub key
        self.assertEqual(
            tx[70:92], unhexlify("81145b812c9d57731e27a2da8b1830195f88ef32a3b6")
        )  # account
        self.assertEqual(
            tx[92:114], unhexlify("8314b5f762798a53d543a014caf8b297cff8f2f937e8")
        )  # destination
        self.assertEqual(len(tx[114:]), 0)  # that's it


if __name__ == "__main__":
    unittest.main()