#
#    Copyright (c) 2024 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 chip.clusters as Clusters
from matter_testing_support import MatterBaseTest, TestStep, async_test_body, default_matter_test_main
from TC_OpstateCommon import TC_OPSTATE_BASE, TestInfo


class TC_OPSTATE_2_5(MatterBaseTest, TC_OPSTATE_BASE):
    def __init__(self, *args):
        super().__init__(*args)

        test_info = TestInfo(
            pics_code="OPSTATE",
            cluster=Clusters.OperationalState
        )

        super().setup_base(test_info=test_info)

    def steps_TC_OPSTATE_2_5(self) -> list[TestStep]:
        return self.STEPS_TC_OPSTATE_BASE_2_5()

    def pics_TC_OPSTATE_2_5(self) -> list[str]:
        return ["OPSTATE.S"]

    @async_test_body
    async def test_TC_OPSTATE_2_5(self):
        endpoint = self.matter_test_config.endpoint

        await self.TEST_TC_OPSTATE_BASE_2_5(endpoint=endpoint)


if __name__ == "__main__":
    default_matter_test_main()