# This file is part of the Trezor project.
#
# Copyright (C) 2012-2019 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 monero
from trezorlib.debuglink import SessionDebugWrapper as Session
from trezorlib.tools import parse_path

from ...common import MNEMONIC12
from ...input_flows import InputFlowShowAddressQRCode

TEST_VECTORS = [
    (
        "m/44h/128h/0h",
        b"4Ahp23WfMrMFK3wYL2hLWQFGt87ZTeRkufS6JoQZu6MEFDokAQeGWmu9MA3GFq1yVLSJQbKJqVAn9F9DLYGpRzRAEXqAXKM",
    ),
    (
        "m/44h/128h/1h",
        b"44iAazhoAkv5a5RqLNVyh82a1n3ceNggmN4Ho7bUBJ14WkEVR8uFTe9f7v5rNnJ2kEbVXxfXiRzsD5Jtc6NvBi4D6WNHPie",
    ),
    (
        "m/44h/128h/2h",
        b"47ejhmbZ4wHUhXaqA4b7PN667oPMkokf4ZkNdWrMSPy9TNaLVr7vLqVUQHh2MnmaAEiyrvLsX8xUf99q3j1iAeMV8YvSFcH",
    ),
]

pytestmark = [
    pytest.mark.altcoin,
    pytest.mark.monero,
    pytest.mark.models("core"),
    pytest.mark.setup_client(mnemonic=MNEMONIC12),
]


@pytest.mark.parametrize("path, expected_address", TEST_VECTORS)
def test_monero_getaddress(session: Session, path: str, expected_address: bytes):
    address = monero.get_address(session, parse_path(path), show_display=True)
    assert address == expected_address


@pytest.mark.parametrize("path, expected_address", TEST_VECTORS)
def test_monero_getaddress_chunkify_details(
    session: Session, path: str, expected_address: bytes
):
    with session.client as client:
        IF = InputFlowShowAddressQRCode(client)
        client.set_input_flow(IF.get())
        address = monero.get_address(
            session, parse_path(path), show_display=True, chunkify=True
        )
        assert address == expected_address