TC-TCCM Python Migration

src/python_testing/TC_TCCM_1_2.py

@@ -0,0 +1,101 @@
+#
+#    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.
+#
+
+# === BEGIN CI TEST ARGUMENTS ===
+# test-runner-runs:
+#   run1:
+#     app: ${ENERGY_MANAGEMENT_APP}
+#     app-args: >
+#       --discriminator 1234
+#       --KVS kvs1
+#       --trace-to json:${TRACE_APP}.json
+#       --enable-key 000102030405060708090a0b0c0d0e0f
+#       --application water-heater
+#     script-args: >
+#       --storage-path admin_storage.json
+#       --commissioning-method on-network
+#       --discriminator 1234
+#       --passcode 20202021
+#       --hex-arg enableKey:000102030405060708090a0b0c0d0e0f
+#       --endpoint 2
+#       --trace-to json:${TRACE_TEST_JSON}.json
+#       --trace-to perfetto:${TRACE_TEST_PERFETTO}.perfetto
+#     factory-reset: true
+#     quiet: true
+# === END CI TEST ARGUMENTS ===
+
+import logging
+
+from tc_mode_base import TC_MODE_BASE
+
+import chip.clusters as Clusters
+from chip.testing.matter_testing import MatterBaseTest, TestStep, async_test_body, default_matter_test_main
+from mobly import asserts
+
+
+class TC_TCCM_1_2(MatterBaseTest, TC_MODE_BASE):
+
+    def desc_TC_TCCM_1_2(self) -> str:
+        return "[TC-TCCM-1.2] Cluster attributes with DUT as Server"
+
+    def steps_TC_TCCM_1_2(self) -> list[TestStep]:
+        steps = [
+            TestStep(1, "Commissioning, already done", is_commissioning=True),
+            TestStep(2, "TH reads from the DUT the SupportedModes attribute."),
+            TestStep(3, "TH reads from the DUT the CurrentMode attribute."),
+            TestStep(4, "TH reads from the DUT the OnMode attribute."),
+            TestStep(5, "TH reads from the DUT the StartUpMode attribute.")
+        ]
+        return steps
+
+    def pics_TC_TCCM_1_2(self) -> list[str]:
+        pics = [
+            "TCCM.S"
+        ]
+        return pics
+
+    @async_test_body
+    async def test_TC_TCCM_1_2(self):
+
+        endpoint = self.get_endpoint(default=1)
+
+        attributes = Clusters.RefrigeratorAndTemperatureControlledCabinetMode.Attributes
+
+        self.initialize_tc_base(requested_cluster=Clusters.RefrigeratorAndTemperatureControlledCabinetMode,
+                                cluster_objects=Clusters.Objects.RefrigeratorAndTemperatureControlledCabinetMode,
+                                endpoint=endpoint, attributes=attributes)
+
+        self.step(1)
+
+        self.step(2)
+        await self.check_supported_modes_and_labels()
+        additional_tags = [Clusters.RefrigeratorAndTemperatureControlledCabinetMode.Enums.ModeTag.kRapidCool,
+                           Clusters.RefrigeratorAndTemperatureControlledCabinetMode.Enums.ModeTag.kRapidFreeze]
+        await self.check_if_labels_in_lists(requiredtags=additional_tags)
+
+        self.step(3)
+        await self.read_and_check_mode(mode=attributes.CurrentMode)
+
+        self.step(4)
+        await self.read_and_check_mode(mode=attributes.OnMode, is_nullable=True)
+
+        self.step(5)
+        await self.read_and_check_mode(mode=attributes.StartUpMode, is_nullable=True)
+
+
+if __name__ == "__main__":
+    default_matter_test_main()

src/python_testing/tc_mode_base.py

@@ -0,0 +1,112 @@
+#
+#    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
+
+from mobly import asserts
+from chip.clusters.Types import NullValue
+
+logger = logging.getLogger(__name__)
+
+
+class TC_MODE_BASE:
+
+    def initialize_tc_base(self, endpoint, attributes, requested_cluster, cluster_objects, *args):
+        self.commonTags = {0x0: 'Auto',
+                           0x1: 'Quick',
+                           0x2: 'Quiet',
+                           0x3: 'LowNoise',
+                           0x4: 'LowEnergy',
+                           0x5: 'Vacation',
+                           0x6: 'Min',
+                           0x7: 'Max',
+                           0x8: 'Night',
+                           0x9: 'Day'}
+        self.specificTags = [tag.value for tag in requested_cluster.Enums.ModeTag]
+        self.supported_modes_dut = set()
+        self.cluster_objects = cluster_objects
+        self.supported_modes = None
+        self.endpoint = endpoint
+        self.attributes = attributes
+
+    async def check_supported_modes_and_labels(self):
+        # Get the supported modes
+        self.supported_modes = await self.read_single_attribute_check_success(endpoint=self.endpoint,
+                                                                              cluster=self.cluster_objects,
+                                                                              attribute=self.attributes.SupportedModes)
+
+        # Check if the list of supported modes is larger than 2
+        asserts.assert_greater_equal(len(self.supported_modes), 2, "SupportedModes must have at least 2 entries!")
+        # Check that supported modes are less than 255
+        asserts.assert_less_equal(len(self.supported_modes), 255, "SupportedModes must have at most 255 entries!")
+
+        # Check for repeated labels or modes
+        labels = set()
+        for mode_options_struct in self.supported_modes:
+            # Verify that the modes in all ModeOptionsStruct in SupportedModes are unique.
+            if mode_options_struct.mode in self.supported_modes_dut:
+                asserts.fail("SupportedModes can't have repeated Mode values")
+            else:
+                self.supported_modes_dut.add(mode_options_struct.mode)
+            # Verify that the labels in all ModeOptionsStruct in SupportedModes are unique.
+            if mode_options_struct.label in labels:
+                asserts.fail("SupportedModes can't have repeated Label values")
+            else:
+                labels.add(mode_options_struct.label)
+
+    async def check_if_labels_in_lists(self, requiredtags=None):
+        # Verify the ModeTags on each ModeOptionsStruct
+        for mode_options_struct in self.supported_modes:
+            # Shuld have at least one entry
+            if len(mode_options_struct.modeTags) == 0:
+                asserts.fail("The ModeTags field should have at least one entry.")
+
+            # Check each ModelTag
+            at_least_one_common_or_derived = False
+            for tag in mode_options_struct.modeTags:
+                # Value should not larger than 16bits
+                if tag.value > 0xFFFF or tag.value < 0:
+                    asserts.fail("Tag should not be larger than 16bits.")
+
+                # Check if is tag is common, derived or mfg.
+                is_mfg = (0x8000 <= tag.value <= 0xBFFF)
+                if (tag.value not in self.commonTags and
+                    tag.value not in self.specificTags and
+                        not is_mfg):
+                    asserts.fail("Mode tag value is not a common, derived or vendor tag.")
+
+                # Confirm if tag is common or derived.
+                if not is_mfg:
+                    at_least_one_common_or_derived = True
+
+            if not at_least_one_common_or_derived:
+                asserts.fail("There should be at least one common or derived tag on each ModeOptionsStruct")
+
+        if requiredtags:
+            has_required_tags = False
+            for mode_options_struct in self.supported_modes:
+                has_required_tags = any(tag.value in requiredtags for tag in mode_options_struct.modeTags)
+                if has_required_tags:
+                    break
+            asserts.assert_true(has_required_tags, "No ModeOptionsStruct has the required tags.")
+
+    async def read_and_check_mode(self, mode, is_nullable=False):
+        mode_value = await self.read_single_attribute_check_success(endpoint=self.endpoint, cluster=self.cluster_objects, attribute=mode)
+        is_valid = mode_value in self.supported_modes_dut
+        if is_nullable and mode_value == NullValue:
+            is_valid = True
+        asserts.assert_true(is_valid, f"{mode} not supported")