from unittest import TestCase
from unittest.mock import patch, Mock

from parameterized import parameterized

from samcli.local.docker.lambda_debug_settings import DebuggingNotSupported, LambdaDebugSettings, Runtime

_DEBUG_RUNTIMES = [
    Runtime.java8,
    Runtime.java8al2,
    Runtime.java11,
    Runtime.java17,
    Runtime.dotnet6,
    Runtime.go1x,
    Runtime.nodejs12x,
    Runtime.nodejs14x,
    Runtime.nodejs16x,
    Runtime.nodejs18x,
    Runtime.python37,
    Runtime.python38,
    Runtime.python39,
    Runtime.python310,
    Runtime.python311,
]


class TestLambdaDebugSettings(TestCase):
    @parameterized.expand(
        [
            (["-delveAPI=2"], 2),
            (["-delveAPI=1"], 1),
            (["-delveAPI", "2"], 2),
            (["-delveAPI", "1"], 1),
            # default should be 1
            ([], 1),
        ]
    )
    def test_delve_api_version_parsing(self, debug_arg_list, expected_api_version):
        self.assertEqual(LambdaDebugSettings.parse_go_delve_api_version(debug_arg_list), expected_api_version)

    @parameterized.expand(
        [
            (["-delveApi=2"],),
            (["-delveApi", "2"],),
        ]
    )
    def test_unrecognized_delve_api_version_parsing(self, debug_arg_list):
        with patch("samcli.local.docker.lambda_debug_settings.LOG.warning") as warning_mock:
            self.assertEqual(LambdaDebugSettings.parse_go_delve_api_version(debug_arg_list), 1)
            warning_mock.assert_called_once_with(
                'Ignoring unrecognized arguments: %s. Only "-delveAPI" is supported.', debug_arg_list
            )

    @parameterized.expand([(runtime,) for runtime in _DEBUG_RUNTIMES])
    @patch("samcli.local.docker.lambda_debug_settings.DebugSettings")
    def test_only_one_debug_setting_is_created(self, runtime, debug_settings_mock):
        LambdaDebugSettings.get_debug_settings(1234, [], {}, runtime.value, {})
        debug_settings_mock.assert_called_once()

    @parameterized.expand([(runtime,) for runtime in Runtime if runtime not in _DEBUG_RUNTIMES])
    @patch("samcli.local.docker.lambda_debug_settings.DebugSettings")
    def test_debugging_not_supported_raised(self, runtime, debug_settings_mock):
        with self.assertRaises(DebuggingNotSupported):
            LambdaDebugSettings.get_debug_settings(1234, [], {}, runtime.value, {})
        debug_settings_mock.assert_not_called()

    @patch("samcli.local.docker.lambda_debug_settings.LambdaDebugSettings.parse_go_delve_api_version")
    def test_parse_go_delve_api_version_called_for_go_runtimes(self, parse_go_delve_api_version_mock):
        debug_args_list = Mock()
        LambdaDebugSettings.get_debug_settings(1234, debug_args_list, {}, Runtime.go1x.value, {})
        parse_go_delve_api_version_mock.assert_called_once_with(debug_args_list)

    @parameterized.expand([(runtime,) for runtime in _DEBUG_RUNTIMES if runtime != Runtime.go1x])
    @patch("samcli.local.docker.lambda_debug_settings.LambdaDebugSettings.parse_go_delve_api_version")
    def test_parse_go_delve_api_version_not_called_for_other_runtimes(self, runtime, parse_go_delve_api_version_mock):
        LambdaDebugSettings.get_debug_settings(1234, [], {}, runtime.value, {})
        parse_go_delve_api_version_mock.assert_not_called()