#
# Copyright (c) 2021-2023 NVIDIA CORPORATION & AFFILIATES.
# Apache-2.0
#
# 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
#
# http://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.
#

import os
import sys
if sys.version_info.major == 3:
    from unittest import mock
else:
    import mock

test_path = os.path.dirname(os.path.abspath(__file__))
modules_path = os.path.dirname(test_path)
sys.path.insert(0, modules_path)

from sonic_platform import utils
from sonic_platform.chassis import Chassis
from sonic_platform.fan import Fan
from sonic_platform.fan_drawer import RealDrawer, VirtualDrawer
from sonic_platform.led import Led
from sonic_platform.psu import FixedPsu, Psu

class TestLed:
    @mock.patch('sonic_platform.led.Led._wait_files_ready', mock.MagicMock(return_value=True))
    @mock.patch('sonic_platform.chassis.extract_RJ45_ports_index', mock.MagicMock(return_value=True))
    def test_chassis_led(self):
        chassis = Chassis()
        assert chassis._led is None
        assert chassis.set_status_led('red') is False
        physical_led = chassis._led
        assert physical_led is not None
        self._verify_non_shared_led(physical_led, chassis)

    def test_uid_led(self):
        chassis = Chassis()
        assert chassis.set_uid_led('blue') is False
        physical_led = chassis._led_uid
        assert physical_led is not None
        self._verify_uid_led(physical_led, chassis)

    def _verify_non_shared_led(self, physical_led, obj):
        mock_file_content = self._mock_led_file_content(physical_led)

        def mock_read_str_from_file(file_path, **kwargs):
            return mock_file_content[file_path]

        def mock_write_file(file_path, content, **kwargs):
            mock_file_content[file_path] = content

        utils.read_str_from_file = mock_read_str_from_file
        utils.write_file = mock_write_file

        assert obj.get_status_led() == Led.STATUS_LED_COLOR_GREEN
        mock_file_content[physical_led.get_led_path('green')] = Led.LED_OFF
        assert obj.set_status_led(Led.STATUS_LED_COLOR_RED) is True
        assert obj.get_status_led() == Led.STATUS_LED_COLOR_RED
        mock_file_content[physical_led.get_led_path('red')] = Led.LED_OFF
        assert obj.set_status_led(Led.STATUS_LED_COLOR_GREEN) is True
        assert obj.get_status_led() == Led.STATUS_LED_COLOR_GREEN
        mock_file_content[physical_led.get_led_path('green')] = Led.LED_OFF
        assert obj.set_status_led(Led.STATUS_LED_COLOR_ORANGE) is True
        assert obj.get_status_led() == Led.STATUS_LED_COLOR_RED
        mock_file_content[physical_led.get_led_path('orange')] = Led.LED_OFF

        assert obj.set_status_led(Led.STATUS_LED_COLOR_RED_BLINK)
        assert obj.get_status_led() == Led.STATUS_LED_COLOR_RED_BLINK

        mock_file_content[physical_led.get_led_delay_off_path('red')] = Led.LED_OFF
        mock_file_content[physical_led.get_led_delay_on_path('red')] = Led.LED_OFF

        assert obj.set_status_led(Led.STATUS_LED_COLOR_GREEN_BLINK)
        assert obj.get_status_led() == Led.STATUS_LED_COLOR_GREEN_BLINK
        mock_file_content[physical_led.get_led_delay_off_path('green')] = Led.LED_OFF
        mock_file_content[physical_led.get_led_delay_on_path('green')] = Led.LED_OFF

        assert obj.set_status_led(Led.STATUS_LED_COLOR_ORANGE_BLINK)
        assert obj.get_status_led() == Led.STATUS_LED_COLOR_RED_BLINK
        mock_file_content[physical_led.get_led_delay_off_path('green')] = Led.LED_OFF
        mock_file_content[physical_led.get_led_delay_on_path('green')] = Led.LED_OFF

    def _verify_uid_led(self, physical_led, obj):
        mock_file_content = self._mock_led_file_content(physical_led)

        def mock_read_str_from_file(file_path, **kwargs):
            return mock_file_content[file_path]

        def mock_write_file(file_path, content, **kwargs):
            mock_file_content[file_path] = content

        utils.read_str_from_file = mock_read_str_from_file
        utils.write_file = mock_write_file

        assert obj.get_uid_led() == Led.STATUS_LED_COLOR_GREEN
        mock_file_content[physical_led.get_led_path('green')] = Led.LED_OFF
        assert obj.set_uid_led(Led.STATUS_LED_COLOR_BLUE) is True
        assert obj.get_uid_led() == Led.STATUS_LED_COLOR_BLUE
        mock_file_content[physical_led.get_led_path('blue')] = Led.LED_OFF

    def _mock_led_file_content(self, led):
        return {
            led.get_led_path('green'): Led.LED_ON,
            led.get_led_path('red'): Led.LED_OFF,
            led.get_led_path('orange'): Led.LED_OFF,
            led.get_led_path('blue'): Led.LED_OFF,
            led.get_led_cap_path(): 'none green green_blink red red_blink orange blue',
            led.get_led_delay_off_path('green'): Led.LED_OFF,
            led.get_led_delay_on_path('green'): Led.LED_OFF,
            led.get_led_delay_off_path('red'): Led.LED_OFF,
            led.get_led_delay_on_path('red'): Led.LED_OFF,
            led.get_led_delay_off_path('orange'): Led.LED_OFF,
            led.get_led_delay_on_path('orange'): Led.LED_OFF,
            led.get_led_delay_off_path('blue'): Led.LED_OFF,
            led.get_led_delay_on_path('blue'): Led.LED_OFF,
        }

    @mock.patch('sonic_platform.led.Led._wait_files_ready', mock.MagicMock(return_value=True))
    def test_fan_led(self):
        fan_drawer = RealDrawer(0)
        self._verify_fan_led(fan_drawer)
        fan_drawer = VirtualDrawer(0)
        self._verify_fan_led(fan_drawer)

    def _verify_fan_led(self, fan_drawer):
        fan1 = Fan(0, fan_drawer, 1)
        fan2 = Fan(1, fan_drawer, 2)
        physical_led = fan_drawer.get_led()._led
        self._verify_shared_led(physical_led, fan1, fan2)

    def _verify_shared_led(self, physical_led, obj1, obj2):
        mock_file_content = self._mock_led_file_content(physical_led)

        def mock_read_str_from_file(file_path, **kwargs):
            return mock_file_content[file_path]

        def mock_write_file(file_path, content, **kwargs):
            mock_file_content[file_path] = content

        utils.read_str_from_file = mock_read_str_from_file
        utils.write_file = mock_write_file
        assert obj1.set_status_led(Led.STATUS_LED_COLOR_GREEN)
        assert obj2.get_status_led() == Led.STATUS_LED_COLOR_GREEN
        mock_file_content[physical_led.get_led_path('green')] = Led.LED_OFF
        assert obj2.set_status_led(Led.STATUS_LED_COLOR_RED)
        assert obj2.get_status_led() == Led.STATUS_LED_COLOR_RED
        assert obj1.set_status_led(Led.STATUS_LED_COLOR_RED)
        assert obj2.get_status_led() == Led.STATUS_LED_COLOR_RED

        mock_file_content[physical_led.get_led_path('red')] = Led.LED_OFF
        assert obj1.set_status_led(Led.STATUS_LED_COLOR_GREEN)
        assert obj1.get_status_led() == Led.STATUS_LED_COLOR_RED
        assert obj2.get_status_led() == Led.STATUS_LED_COLOR_RED
        mock_file_content[physical_led.get_led_path('red')] = Led.LED_OFF
        assert obj2.set_status_led(Led.STATUS_LED_COLOR_GREEN)
        assert obj1.get_status_led() == Led.STATUS_LED_COLOR_GREEN
        assert obj2.get_status_led() == Led.STATUS_LED_COLOR_GREEN

    @mock.patch('sonic_platform.led.Led._wait_files_ready', mock.MagicMock(return_value=True))
    def test_psu_led(self):
        psu1 = Psu(0)
        psu2 = Psu(1)
        physical_led = Psu.get_shared_led()._led
        self._verify_shared_led(physical_led, psu1, psu2)

    @mock.patch('sonic_platform.led.Led._wait_files_ready', mock.MagicMock(return_value=True))
    def test_fixed_psu_led(self):
        psu = FixedPsu(0)
        physical_led = psu.led
        self._verify_non_shared_led(physical_led, psu)

    def test_get_actual_color(self):
        led = Led()
        assert led._get_actual_color('red') is None
        led.supported_colors.add('orange')
        assert led._get_actual_color('red') is 'orange'

    @mock.patch('os.path.exists')
    @mock.patch('time.sleep', mock.MagicMock())
    def test_wait_files_ready(self, mock_exists):
        mock_exists.side_effect = [True, True]
        led = Led()
        assert led._wait_files_ready(['a', 'b'])
        mock_exists.side_effect = [False, False, True, True]
        assert led._wait_files_ready(['a', 'b'])
        mock_exists.side_effect = None
        mock_exists.return_value = False
        assert not led._wait_files_ready(['a', 'b'])

    @mock.patch('sonic_platform.utils.write_file')
    @mock.patch('sonic_platform.led.Led.get_led_path', mock.MagicMock())
    @mock.patch('sonic_platform.led.Led._stop_blink', mock.MagicMock())
    @mock.patch('sonic_platform.led.Led.get_capability', mock.MagicMock())
    @mock.patch('sonic_platform.device_data.DeviceDataManager.is_simx_platform', mock.MagicMock(return_value=False))
    def test_get_set_led_status(self, mock_write):
        led = Led()
        led._led_id = 'fan'
        led.supported_colors.add('red')
        led.supported_colors.add('green')
        assert not led.set_status('black')
        assert led.set_status(led.STATUS_LED_COLOR_OFF)
        assert mock_write.call_count == 2
        mock_write.side_effect = ValueError('')
        assert not led.set_status(led.STATUS_LED_COLOR_OFF)

        led.supported_colors.clear()
        assert led.get_status() == led.STATUS_LED_COLOR_OFF