zephyr/scripts/west_commands/tests/test_rfp.py

# Copyright (c) 2025 Pete Johanson
#
# SPDX-License-Identifier: Apache-2.0

import argparse
import os
from unittest.mock import call, patch

from conftest import RC_KERNEL_HEX
from runners.rfp import RfpBinaryRunner

TEST_RFP_PORT = 'test-rfp-serial'
TEST_RFP_PORT_SPEED = '115200'
TEST_RFP_DEVICE = 'RA'
TEST_RFP_USR_LOCAL_RFP_CLI = '/usr/local/bin/rfp-cli'

EXPECTED_COMMANDS = [
    [
        'rfp-cli',
        '-port',
        TEST_RFP_PORT,
        '-device',
        TEST_RFP_DEVICE,
        '-run',
        '-noerase',
        '-p',
        '-file',
        RC_KERNEL_HEX,
    ],
]

EXPECTED_COMMANDS_WITH_SPEED = [
    [
        'rfp-cli',
        '-port',
        TEST_RFP_PORT,
        '-s',
        TEST_RFP_PORT_SPEED,
        '-device',
        TEST_RFP_DEVICE,
        '-run',
        '-noerase',
        '-p',
        '-file',
        RC_KERNEL_HEX,
    ],
]


EXPECTED_COMMANDS_WITH_ERASE = [
    [
        'rfp-cli',
        '-port',
        TEST_RFP_PORT,
        '-device',
        TEST_RFP_DEVICE,
        '-run',
        '-erase',
        '-p',
        '-file',
        RC_KERNEL_HEX,
    ],
]

EXPECTED_COMMANDS_WITH_VERIFY = [
    [
        'rfp-cli',
        '-port',
        TEST_RFP_PORT,
        '-device',
        TEST_RFP_DEVICE,
        '-run',
        '-noerase',
        '-v',
        '-p',
        '-file',
        RC_KERNEL_HEX,
    ],
]

EXPECTED_COMMANDS_WITH_RFP_CLI = [
    [
        TEST_RFP_USR_LOCAL_RFP_CLI,
        '-port',
        TEST_RFP_PORT,
        '-device',
        TEST_RFP_DEVICE,
        '-run',
        '-noerase',
        '-p',
        '-file',
        RC_KERNEL_HEX,
    ],
]


def require_patch(program):
    assert program in ['rfp']


os_path_isfile = os.path.isfile


def os_path_isfile_patch(filename):
    if filename == RC_KERNEL_HEX:
        return True
    return os_path_isfile(filename)


@patch('runners.core.ZephyrBinaryRunner.require', side_effect=require_patch)
@patch('runners.core.ZephyrBinaryRunner.check_call')
def test_rfp_init(cc, req, runner_config, tmpdir):
    """
    Test commands using a runner created by constructor.

    Input:
        port=A
    device=B

    Output:
        -port A
        -device B
    """
    runner = RfpBinaryRunner(runner_config, port=TEST_RFP_PORT, device=TEST_RFP_DEVICE)
    with patch('os.path.isfile', side_effect=os_path_isfile_patch):
        runner.run('flash')
    assert cc.call_args_list == [call(x) for x in EXPECTED_COMMANDS]


@patch('runners.core.ZephyrBinaryRunner.require', side_effect=require_patch)
@patch('runners.core.ZephyrBinaryRunner.check_call')
def test_rfp_create(cc, req, runner_config, tmpdir):
    """
    Test commands using a runner created from command line parameters.


    Input:
        ---port A
    --device B

    Output:
        -port A
        -device B
    """
    args = ['--port', str(TEST_RFP_PORT), "--device", str(TEST_RFP_DEVICE)]
    parser = argparse.ArgumentParser(allow_abbrev=False)
    RfpBinaryRunner.add_parser(parser)
    arg_namespace = parser.parse_args(args)
    runner = RfpBinaryRunner.create(runner_config, arg_namespace)
    with patch('os.path.isfile', side_effect=os_path_isfile_patch):
        runner.run('flash')
    assert cc.call_args_list == [call(x) for x in EXPECTED_COMMANDS]


@patch('runners.core.ZephyrBinaryRunner.require', side_effect=require_patch)
@patch('runners.core.ZephyrBinaryRunner.check_call')
def test_rfp_create_with_speed(cc, req, runner_config, tmpdir):
    """
    Test commands using a runner created from command line parameters.

    Input:
        --port
        --speed
    --speed

    Output:
        -s SPEED
    """
    args = [
        '--device',
        str(TEST_RFP_DEVICE),
        '--port',
        str(TEST_RFP_PORT),
        '--speed',
        str(TEST_RFP_PORT_SPEED),
    ]
    parser = argparse.ArgumentParser(allow_abbrev=False)
    RfpBinaryRunner.add_parser(parser)
    arg_namespace = parser.parse_args(args)
    runner = RfpBinaryRunner.create(runner_config, arg_namespace)
    with patch('os.path.isfile', side_effect=os_path_isfile_patch):
        runner.run('flash')
    assert cc.call_args_list == [call(x) for x in EXPECTED_COMMANDS_WITH_SPEED]


@patch('runners.core.ZephyrBinaryRunner.require', side_effect=require_patch)
@patch('runners.core.ZephyrBinaryRunner.check_call')
def test_rfp_create_with_erase(cc, req, runner_config, tmpdir):
    """
    Test commands using a runner created from command line parameters.

    Input:
        --erase

    Output:
        -erase
    """
    args = ['--device', str(TEST_RFP_DEVICE), '--port', str(TEST_RFP_PORT), '--erase']
    parser = argparse.ArgumentParser(allow_abbrev=False)
    RfpBinaryRunner.add_parser(parser)
    arg_namespace = parser.parse_args(args)
    runner = RfpBinaryRunner.create(runner_config, arg_namespace)
    with patch('os.path.isfile', side_effect=os_path_isfile_patch):
        runner.run('flash')
    assert cc.call_args_list == [call(x) for x in EXPECTED_COMMANDS_WITH_ERASE]


@patch('runners.core.ZephyrBinaryRunner.require', side_effect=require_patch)
@patch('runners.core.ZephyrBinaryRunner.check_call')
def test_rfp_create_with_verify(cc, req, runner_config, tmpdir):
    """
    Test commands using a runner created from command line parameters.

    Input:
        --verify

    Output:
        -v
    """
    args = ['--device', str(TEST_RFP_DEVICE), '--port', str(TEST_RFP_PORT), '--verify']
    parser = argparse.ArgumentParser(allow_abbrev=False)
    RfpBinaryRunner.add_parser(parser)
    arg_namespace = parser.parse_args(args)
    runner = RfpBinaryRunner.create(runner_config, arg_namespace)
    with patch('os.path.isfile', side_effect=os_path_isfile_patch):
        runner.run('flash')
    assert cc.call_args_list == [call(x) for x in EXPECTED_COMMANDS_WITH_VERIFY]


@patch('runners.core.ZephyrBinaryRunner.require', side_effect=require_patch)
@patch('runners.core.ZephyrBinaryRunner.check_call')
def test_rfp_create_with_rfp_cli(cc, req, runner_config, tmpdir):
    """
    Test commands using a runner created from command line parameters.

    Input:
        --rfp-cli /usr/local/bin/rfp-cli

    Output:
        /usr/local/bin/rfp-cli
    """
    args = [
        '--device',
        str(TEST_RFP_DEVICE),
        '--port',
        str(TEST_RFP_PORT),
        '--rfp-cli',
        str(TEST_RFP_USR_LOCAL_RFP_CLI),
    ]
    parser = argparse.ArgumentParser(allow_abbrev=False)
    RfpBinaryRunner.add_parser(parser)
    arg_namespace = parser.parse_args(args)
    runner = RfpBinaryRunner.create(runner_config, arg_namespace)
    with patch('os.path.isfile', side_effect=os_path_isfile_patch):
        runner.run('flash')
    assert cc.call_args_list == [call(x) for x in EXPECTED_COMMANDS_WITH_RFP_CLI]