#!/usr/bin/env python3
# Copyright 2021 The Pigweed Authors
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may not
# use this file except in compliance with the License. You may obtain a copy of
# the License at
#
#     https://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations under
# the License.
"""Generate File List Tests."""

import pathlib
import unittest

from pw_stm32cube_build import find_files


class ParseProductStringTest(unittest.TestCase):
    """parse_product_str tests."""

    def test_start_with_stm32(self):
        with self.assertRaises(ValueError):
            find_files.parse_product_str('f439zit')

    def test_specify_chip(self):
        with self.assertRaises(ValueError):
            find_files.parse_product_str('stm32f43')

    def test_stm32f412zx(self):
        (family, defines, name) = find_files.parse_product_str('stm32f412zx')

        self.assertEqual(family, 'stm32f4xx')
        self.assertEqual(defines, {'STM32F412xx', 'STM32F412Zx'})
        self.assertEqual(name, 'stm32f412zx')

    def test_stm32f439xx(self):
        (family, defines, name) = find_files.parse_product_str('STM32F439xx')

        self.assertEqual(family, 'stm32f4xx')
        self.assertEqual(defines, {'STM32F439xx'})
        self.assertEqual(name, 'stm32f439xx')

    def test_stm32f439(self):
        (family, defines, name) = find_files.parse_product_str('STM32F439')

        self.assertEqual(family, 'stm32f4xx')
        self.assertEqual(defines, {'STM32F439xx'})
        self.assertEqual(name, 'stm32f439xx')

    def test_stm32f439xi(self):
        (family, defines, name) = find_files.parse_product_str('STM32F439xI')

        self.assertEqual(family, 'stm32f4xx')
        self.assertEqual(defines, {'STM32F439xx', 'STM32F439xI'})
        self.assertEqual(name, 'stm32f439xi')

    def test_stm32f439zit6u(self):
        (family, defines, name) = find_files.parse_product_str('stm32f439zit6u')

        self.assertEqual(family, 'stm32f4xx')
        self.assertEqual(
            defines,
            {'STM32F439xx', 'STM32F439Zx', 'STM32F439xI', 'STM32F439ZI'},
        )
        self.assertEqual(name, 'stm32f439zit6u')

    def test_stm32l552zet(self):
        (family, defines, name) = find_files.parse_product_str('stm32l552zet')

        self.assertEqual(family, 'stm32l5xx')
        self.assertEqual(
            defines,
            {'STM32L552xx', 'STM32L552Zx', 'STM32L552xE', 'STM32L552ZE'},
        )
        self.assertEqual(name, 'stm32l552zet')

    def test_stm32l552xc(self):
        (family, defines, name) = find_files.parse_product_str('stm32l552xc')

        self.assertEqual(family, 'stm32l5xx')
        self.assertEqual(defines, {'STM32L552xx', 'STM32L552xC'})
        self.assertEqual(name, 'stm32l552xc')

    def test_stm32wb5m(self):
        (family, defines, name) = find_files.parse_product_str('stm32wb5m')

        self.assertEqual(family, 'stm32wbxx')
        self.assertEqual(defines, {'STM32WB5Mxx'})
        self.assertEqual(name, 'stm32wb5mxx')


class SelectDefineTest(unittest.TestCase):
    """select_define tests."""

    def test_stm32f412zx_not_found(self):
        with self.assertRaises(ValueError):
            find_files.select_define({'STM32F412xx', 'STM32F412Zx'}, "")

    def test_stm32f412zx_found(self):
        define = find_files.select_define(
            {'STM32F412xx', 'STM32F412Zx'},
            "asdf\nfdas\n#if defined(STM32F412Zx)\n",
        )
        self.assertEqual(define, 'STM32F412Zx')

    def test_stm32f412zx_multiple_found(self):
        with self.assertRaises(ValueError):
            find_files.select_define(
                {'STM32F412xx', 'STM32F412Zx'},
                "asdf\n#if defined (STM32F412xx)\n#elif defined(STM32F412Zx)\n",
            )


class MatchFilenameTest(unittest.TestCase):
    """match_filename tests."""

    def test_stm32f412zx(self):
        # Match should fail if product name is not specific enough
        self.assertTrue(
            find_files.match_filename('stm32f412zx', 'stm32f412zx_flash.icf')
        )
        self.assertFalse(
            find_files.match_filename('stm32f412xx', 'stm32f412zx_flash.icf')
        )
        self.assertTrue(
            find_files.match_filename('stm32f412zx', 'startup_stm32f412zx.s')
        )
        self.assertFalse(
            find_files.match_filename('stm32f412xx', 'startup_stm32f429zx.s')
        )

    def test_stm32f439xx(self):
        self.assertTrue(
            find_files.match_filename('stm32f439xx', 'stm32f439xx_flash.icf')
        )
        self.assertFalse(
            find_files.match_filename('stm32f439xx', 'stm32f429xx_flash.icf')
        )
        self.assertTrue(
            find_files.match_filename('stm32f439xx', 'startup_stm32f439xx.s')
        )
        self.assertFalse(
            find_files.match_filename('stm32f439xx', 'startup_stm32f429xx.s')
        )

    def test_stm32f439xi(self):
        self.assertTrue(
            find_files.match_filename('stm32f439xi', 'stm32f439xx_flash.icf')
        )
        self.assertFalse(
            find_files.match_filename('stm32f439xi', 'stm32f429xx_flash.icf')
        )
        self.assertTrue(
            find_files.match_filename('stm32f439xi', 'startup_stm32f439xx.s')
        )
        self.assertFalse(
            find_files.match_filename('stm32f439xi', 'startup_stm32f429xx.s')
        )

    def test_stm32l552zet(self):
        self.assertTrue(
            find_files.match_filename('stm32l552zet', 'STM32L552xE_FLASH.ld')
        )
        self.assertTrue(
            find_files.match_filename('stm32l552zet', 'STM32L552xx_FLASH.ld')
        )
        self.assertFalse(
            find_files.match_filename('stm32l552zet', 'STM32L552xC_FLASH.ld')
        )
        self.assertTrue(
            find_files.match_filename('stm32l552zet', 'stm32l552xe_flash.icf')
        )
        self.assertFalse(
            find_files.match_filename('stm32l552zet', 'stm32l552xc_flash.icf')
        )
        self.assertTrue(
            find_files.match_filename('stm32l552zet', 'startup_stm32l552xx.s')
        )
        self.assertFalse(
            find_files.match_filename('stm32l552zet', 'startup_stm32l562xx.s')
        )


class FindLinkerFilesTest(unittest.TestCase):
    """find_linker_files tests."""

    TEST_PATH = pathlib.Path('/test/path')

    def test_stm32f439xx(self):
        files = [
            'path/to/stm32f439xx_flash.icf',
            'other/path/to/stm32f439xx_sram.icf',
        ]
        gcc_linker, iar_linker = find_files.find_linker_files(
            'stm32f439xx', files, self.TEST_PATH
        )

        self.assertEqual(gcc_linker, None)
        self.assertEqual(iar_linker, self.TEST_PATH / files[0])

    def test_stm32f439xx_find_ld(self):
        files = [
            'path/to/stm32f439xx_flash.icf',
            'other/path/to/stm32f439xx_sram.icf',
            'path/to/STM32F439xx_FLASH.ld',
        ]
        gcc_linker, iar_linker = find_files.find_linker_files(
            'stm32f439xx', files, self.TEST_PATH
        )

        self.assertEqual(gcc_linker, self.TEST_PATH / files[2])
        self.assertEqual(iar_linker, self.TEST_PATH / files[0])

    def test_stm32f439xc_error_multiple_matching_ld(self):
        files = [
            'path/to/stm32f439xx_flash.icf',
            'other/path/to/stm32f439xx_sram.icf',
            'other/path/to/STM32F439xI_FLASH.ld',
            'path/to/STM32F439xx_FLASH.ld',
        ]
        with self.assertRaises(ValueError):
            find_files.find_linker_files('stm32f439xi', files, self.TEST_PATH)

    def test_stm32f439xc_error_multiple_matching_icf(self):
        files = [
            'path/to/stm32f439xx_flash.icf',
            'other/path/to/stm32f439xc_flash.icf',
        ]
        with self.assertRaises(ValueError):
            find_files.find_linker_files('stm32f439xc', files, self.TEST_PATH)

    def test_stm32f439xc_error_none_found(self):
        files = [
            'path/to/stm32f439xc_flash.icf',
            'other/path/to/stm32f439xc_flash.icf',
        ]
        with self.assertRaises(ValueError):
            find_files.find_linker_files('stm32f439xx', files, self.TEST_PATH)

    # ignore secure and nonsecure variants for the M33 boards
    def test_stm32l552xe_ignore_s_ns(self):
        files = [
            'iar/linker/stm32l552xe_flash_ns.icf',
            'iar/linker/stm32l552xe_flash_s.icf',
            'iar/linker/stm32l552xe_flash.icf',
            'gcc/linker/STM32L552xE_FLASH_ns.ld',
            'gcc/linker/STM32L552xE_FLASH_s.ld',
            'gcc/linker/STM32L552xE_FLASH.ld',
        ]
        gcc_linker, iar_linker = find_files.find_linker_files(
            'stm32l552xe', files, self.TEST_PATH
        )

        self.assertEqual(gcc_linker, self.TEST_PATH / files[-1])
        self.assertEqual(iar_linker, self.TEST_PATH / files[2])


class FindStartupFileTest(unittest.TestCase):
    """find_startup_file tests."""

    TEST_PATH = pathlib.Path('/test/path')

    def test_stm32f439xx_none_found(self):
        files = [
            'path/to/stm32f439xx_flash.icf',
            'other/path/to/stm32f439xx_sram.icf',
            'path/iar/startup_stm32f439xx.s',
        ]
        with self.assertRaises(ValueError):
            find_files.find_startup_file('stm32f439xx', files, self.TEST_PATH)

    def test_stm32f439xx(self):
        files = [
            'path/to/stm32f439xx_flash.icf',
            'other/path/to/stm32f439xx_sram.icf',
            'path/iar/startup_stm32f439xx.s',
            'path/gcc/startup_stm32f439xx.s',
        ]
        startup_file = find_files.find_startup_file(
            'stm32f439xx', files, self.TEST_PATH
        )

        self.assertEqual(startup_file, self.TEST_PATH / files[3])

    def test_stm32f439xx_multiple_found(self):
        files = [
            'path/to/stm32f439xx_flash.icf',
            'other/path/to/stm32f439xx_sram.icf',
            'path/gcc/startup_stm32f439xc.s',
            'path/gcc/startup_stm32f439xx.s',
        ]
        with self.assertRaises(ValueError):
            find_files.find_startup_file('stm32f439xc', files, self.TEST_PATH)


class GetSourceAndHeadersTest(unittest.TestCase):
    """test_sources_and_headers tests."""

    def test_sources_and_headers(self):
        files = [
            'random/header.h',
            'random/source.c',
            'cmsis_core/Include/core_cm4.h',
            'cmsis_device/Include/stm32f4xx.h',
            'cmsis_device/Include/stm32f439xx.h',
            'hal_driver/Inc/stm32f4xx_hal_eth.h',
            'hal_driver/Src/stm32f4xx_hal_adc.c',
            'hal_driver/Inc/stm32f4xx_hal.h',
            'hal_driver/Src/stm32f4xx_hal_timebase_tim_template.c',
            'hal_driver/Src/stm32f4xx_hal_eth.c',
        ]
        path = pathlib.Path('/test/path/to/stm32cube')
        sources, headers = find_files.get_sources_and_headers(files, path)
        self.assertSetEqual(
            set(
                [
                    path / 'hal_driver/Src/stm32f4xx_hal_adc.c',
                    path / 'hal_driver/Src/stm32f4xx_hal_eth.c',
                ]
            ),
            set(sources),
        )
        self.assertSetEqual(
            set(
                [
                    path / 'cmsis_core/Include/core_cm4.h',
                    path / 'cmsis_device/Include/stm32f4xx.h',
                    path / 'cmsis_device/Include/stm32f439xx.h',
                    path / 'hal_driver/Inc/stm32f4xx_hal_eth.h',
                    path / 'hal_driver/Inc/stm32f4xx_hal.h',
                ]
            ),
            set(headers),
        )


if __name__ == '__main__':
    unittest.main()