#
#    Copyright (c) 2023 Project CHIP Authors
#    All rights reserved.
#
#    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 logging

import chip.clusters as Clusters
from chip.interaction_model import Status
from matter_testing_support import MatterBaseTest, TestStep, async_test_body, default_matter_test_main
from mobly import asserts

logger = logging.getLogger(__name__)


class TC_FAN_3_3(MatterBaseTest):

    async def read_fc_attribute_expect_success(self, endpoint, attribute):
        cluster = Clusters.Objects.FanControl
        return await self.read_single_attribute_check_success(endpoint=endpoint, cluster=cluster, attribute=attribute)

    async def read_rock_support(self, endpoint):
        return await self.read_fc_attribute_expect_success(endpoint, Clusters.FanControl.Attributes.RockSupport)

    async def read_rock_setting(self, endpoint):
        return await self.read_fc_attribute_expect_success(endpoint, Clusters.FanControl.Attributes.RockSetting)

    async def write_rock_setting(self, endpoint, rock_setting):
        result = await self.default_controller.WriteAttribute(self.dut_node_id, [(endpoint, Clusters.FanControl.Attributes.RockSetting(rock_setting))])
        asserts.assert_equal(result[0].Status, Status.Success, "RockSetting write failed")

    async def write_rock_setting_expect_failure(self, endpoint, rock_setting):
        result = await self.default_controller.WriteAttribute(self.dut_node_id, [(endpoint, Clusters.FanControl.Attributes.RockSetting(rock_setting))])
        asserts.assert_equal(result[0].Status, Status.ConstraintError, "Expected ConstraintError but received a different error.")

    def desc_TC_FAN_3_3(self) -> str:
        return "[TC-FAN-3.3] Optional rock functionality with DUT as Server"

    def steps_TC_FAN_3_3(self) -> list[TestStep]:
        steps = [
            TestStep(1, "Commissioning, already done", is_commissioning=True),
            TestStep(2, "Read from the DUT the RockSupport attribute and store"),
            TestStep(3, "RockLeftRight is supported, so write 0x01 to RockSetting"),
            TestStep(4, "Read from the DUT the RockSetting attribute"),
            TestStep(5, "RockLeftRight is not supported, so write 0x01 to RockSetting to check for constraint error"),
            TestStep(6, "RockUpDown is supported, so write 0x02 to RockSetting"),
            TestStep(7, "Read from the DUT the RockSetting attribute"),
            TestStep(8, "RockUpDown is not supported, so write 0x02 to RockSetting to check for constraint error"),
            TestStep(9, "RockRound is supported, so write 0x04 to RockSetting"),
            TestStep(10, "Read from the DUT the RockSetting attribute"),
            TestStep(11, "RockRound is not supported, so write 0x04 to RockSetting to check for constraint error"),
            TestStep(12, "Write RockSetting to 0x00"),
        ]
        return steps

    def pics_TC_FAN_3_3(self) -> list[str]:
        return ["FAN.S"]

    @async_test_body
    async def test_TC_FAN_3_3(self):
        if not self.check_pics("FAN.S.F02"):
            logger.info("Test skipped because PICS FAN.S.F02 is not set")
            self.skip_all_remaining_steps(1)
            return

        endpoint = self.user_params.get("endpoint", 1)

        self.step(1)

        self.step(2)
        rock_support = await self.read_rock_support(endpoint=endpoint)

        if rock_support & Clusters.FanControl.Bitmaps.RockBitmap.kRockLeftRight:
            self.step(3)
            await self.write_rock_setting(endpoint=endpoint, rock_setting=Clusters.FanControl.Bitmaps.RockBitmap.kRockLeftRight)

            self.step(4)
            rock_setting = await self.read_rock_setting(endpoint=endpoint)
            asserts.assert_equal(rock_setting, Clusters.FanControl.Bitmaps.RockBitmap.kRockLeftRight, "RockSetting is not 0x01")

            self.skip_step(5)
        else:
            self.skip_step(3)
            self.skip_step(4)

            self.step(5)
            self.write_rock_setting_expect_failure(
                endpoint=endpoint, rock_setting=Clusters.FanControl.Bitmaps.RockBitmap.kRockLeftRight)

        if rock_support & Clusters.FanControl.Bitmaps.RockBitmap.kRockUpDown:
            self.step(6)
            await self.write_rock_setting(endpoint=endpoint, rock_setting=Clusters.FanControl.Bitmaps.RockBitmap.kRockUpDown)

            self.step(7)
            rock_setting = await self.read_rock_setting(endpoint=endpoint)
            asserts.assert_equal(rock_setting, Clusters.FanControl.Bitmaps.RockBitmap.kRockUpDown, "RockSetting is not 0x02")

            self.skip_step(8)
        else:
            self.skip_step(6)
            self.skip_step(7)

            self.step(8)
            await self.write_rock_setting_expect_failure(endpoint=endpoint, rock_setting=Clusters.FanControl.Bitmaps.RockBitmap.kRockUpDown)

        if rock_support & Clusters.FanControl.Bitmaps.RockBitmap.kRockRound:
            self.step(9)
            await self.write_rock_setting(endpoint=endpoint, rock_setting=Clusters.FanControl.Bitmaps.RockBitmap.kRockRound)

            self.step(10)
            rock_setting = await self.read_rock_setting(endpoint=endpoint)
            asserts.assert_equal(rock_setting, Clusters.FanControl.Bitmaps.RockBitmap.kRockRound, "RockSetting is not 0x04")
            self.skip_step(11)
        else:
            self.skip_step(9)
            self.skip_step(10)

            self.step(11)
            await self.write_rock_setting_expect_failure(endpoint=endpoint, rock_setting=Clusters.FanControl.Bitmaps.RockBitmap.kRockRound)

        self.step(12)
        await self.write_rock_setting(endpoint=endpoint, rock_setting=0x00)


if __name__ == "__main__":
    default_matter_test_main()