# Copyright (c) 2018 Foundries.io
# Copyright (c) 2019 Nordic Semiconductor ASA.
#
# SPDX-License-Identifier: Apache-2.0

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

import pytest

from runners.dediprog import DediProgBinaryRunner
from conftest import RC_KERNEL_BIN

DPCMD_EXE = 'dpcmd.exe' if platform.system() == 'Windows' else 'dpcmd'

EXPECTED_COMMAND = {
    (RC_KERNEL_BIN, None):
    [DPCMD_EXE,
     '--auto', RC_KERNEL_BIN,
     '-x', 'ff',
     '--silent', '--verify'],


    (RC_KERNEL_BIN, '0'):
    [DPCMD_EXE,
     '--auto', RC_KERNEL_BIN, '--vcc', '0',
     '-x', 'ff',
     '--silent', '--verify'],

    (RC_KERNEL_BIN, '1'):
    [DPCMD_EXE,
     '--auto', RC_KERNEL_BIN, '--vcc', '1',
     '-x', 'ff',
     '--silent', '--verify'],
}

def require_patch(program):
    assert program in [DPCMD_EXE]

def id_fn(tc):
    return 'spi_image={},vcc={}'.format(*tc)

@pytest.mark.parametrize('tc', [
    (RC_KERNEL_BIN, None),
    (RC_KERNEL_BIN, '0'),
    (RC_KERNEL_BIN, '1'),
], ids=id_fn)
@patch('runners.core.ZephyrBinaryRunner.require', side_effect=require_patch)
@patch('runners.core.ZephyrBinaryRunner.check_call')
def test_dediprog_init(cc, req, tc, runner_config):
    '''Test commands using a runner created by constructor.'''
    spi_image, vcc = tc
    runner = DediProgBinaryRunner(runner_config, spi_image=spi_image,
                                  vcc=vcc, retries=0)
    runner.run('flash')
    assert cc.call_args_list == [call(EXPECTED_COMMAND[tc])]

@pytest.mark.parametrize('tc', [
    (RC_KERNEL_BIN, None),
    (RC_KERNEL_BIN, '0'),
    (RC_KERNEL_BIN, '1'),
], ids=id_fn)
@patch('runners.core.ZephyrBinaryRunner.require', side_effect=require_patch)
@patch('runners.core.ZephyrBinaryRunner.check_call')
def test_dediprog_create(cc, req, tc, runner_config):
    '''Test commands using a runner created from command line parameters.'''
    spi_image, vcc = tc
    args = ['--spi-image', spi_image, '--retries', '0']
    if vcc:
        args.extend(['--vcc', vcc])
    parser = argparse.ArgumentParser()
    DediProgBinaryRunner.add_parser(parser)
    arg_namespace = parser.parse_args(args)
    runner = DediProgBinaryRunner.create(runner_config, arg_namespace)
    runner.run('flash')
    assert cc.call_args_list == [call(EXPECTED_COMMAND[tc])]