litex

#
# This file is part of LiteX.
#
# SPDX-License-Identifier: BSD-2-Clause

from unittest.mock import patch

from litex.build.altera.programmer import USBBlaster
from litex.build.xilinx.programmer import Adept, XC3SProg


def test_usbblaster_programmer_command():
    programmer = USBBlaster(cable_name="USB-Blaster", device_id=2)
    with patch.object(programmer, "call") as call:
        programmer.load_bitstream("/tmp/top.sof", cable_suffix="-1")

    assert call.call_args_list[0].args[0] == [
        "quartus_pgm", "-m", "jtag", "-c", "USB-Blaster-1",
        "-o", "p;/tmp/top.sof@2",
    ]


def test_xc3sprog_programmer_load_command():
    programmer = XC3SProg(cable="xpc", position=1)
    with patch.object(programmer, "call") as call:
        programmer.load_bitstream("/tmp/top.bit")

    assert call.call_args_list[0].args[0] == [
        "xc3sprog", "-v", "-c", "xpc", "-p", "1", "/tmp/top.bit"
    ]


def test_adept_programmer_load_command():
    programmer = Adept(board="arty", index=0)
    with patch.object(programmer, "call") as call:
        programmer.load_bitstream("/tmp/top.bit")

    assert call.call_args_list[0].args[0] == [
        "djtgcfg", "--verbose", "prog", "-d", "arty", "-i", "0", "-f", "/tmp/top.bit"
    ]