# This file is part of the Trezor project.
#
# Copyright (C) 2012-2018 SatoshiLabs and contributors
#
# This library is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License version 3
# as published by the Free Software Foundation.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the License along with this library.
# If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>.

import pytest

from trezorlib import messages, ontology
from trezorlib.tools import parse_path

from ..common import MNEMONIC12

ONTOLOGY_TEST_MNEMONIC = "immune annual decorate major humble surprise dismiss trend edit suit alert uncover release transfer suit torch small timber lock mind tomorrow north lend diet"


@pytest.mark.altcoin
@pytest.mark.ontology
@pytest.mark.skip_t1
@pytest.mark.setup_client(mnemonic=MNEMONIC12)
def test_ontology_sign_transfer_ont(client):
    transaction = messages.OntologyTransaction(
        version=0x00,
        nonce=0x7F7F1CEB,
        gas_price=500,
        gas_limit=30000,
        payer="AGn8JFPGM5S4jkWhTC89Xtz1Y76sPz29Rc",
        tx_attributes=[],
    )

    transfer = messages.OntologyTransfer(
        asset=1,
        amount=100,
        from_address="AGn8JFPGM5S4jkWhTC89Xtz1Y76sPz29Rc",
        to_address="AcyLq3tokVpkMBMLALVMWRdVJ83TTgBUwU",
    )

    signature = ontology.sign(
        client, parse_path("m/44'/1024'/0'/0/0"), transaction, transfer
    )
    assert (
        signature.hex()
        == "0102f9b0c43b2ed35aa89b0927a60e692cb8a74280c2da819a909150c8b3fd2b0b401806c97797fcc4b93d34f210ad01740cfd13b720a389a80f384c1f94fb749e"
    )


# original test didn't work, had to replace with data from
# https://github.com/ontio/ontology-ts-sdk/blob/master/test/transfer.sign.test.ts
@pytest.mark.altcoin
@pytest.mark.ontology
@pytest.mark.skip_t1
@pytest.mark.setup_client(mnemonic=ONTOLOGY_TEST_MNEMONIC)
def test_ontology_sign_transfer_ong(client):
    transaction = messages.OntologyTransaction(
        version=0x00,
        nonce=0x7F7F1CEB,
        gas_price=500,
        gas_limit=30000,
        payer="AGn8JFPGM5S4jkWhTC89Xtz1Y76sPz29Rc",
        tx_attributes=[],
    )

    transfer = messages.OntologyTransfer(
        asset=2,
        amount=12000000,
        from_address="AGn8JFPGM5S4jkWhTC89Xtz1Y76sPz29Rc",
        to_address="AcyLq3tokVpkMBMLALVMWRdVJ83TTgBUwU",
    )

    signature = ontology.sign(
        client, parse_path("m/44'/888'/0'/0/0"), transaction, transfer
    )
    assert (
        signature.hex()
        == "017da1b8268e1272d7471eef58fa0884108073c09d5efdae0143da5d281019682e5a1562f1d76484eb0379e3febe7025a958bb14855107b9ad26daec2fee0119f4"
    )