adding get set cases (#23)

opencomputeproject · Sep 15, 2023 · 8e8dee8 · 8e8dee8
1 parent babf7d4
commit 8e8dee8
Show file tree

Hide file tree

Showing 6 changed files with 1,232 additions and 22 deletions.
diff --git a/tests/api/test_acl_counter.py b/tests/api/test_acl_counter.py
@@ -1,6 +1,9 @@
 from pprint import pprint
 
+import pytest
 
+
+@pytest.mark.npu
 class TestSaiAclCounter:
     # object with parent SAI_OBJECT_TYPE_ACL_TABLE
 
@@ -19,23 +22,33 @@ def test_acl_counter_create(self, npu):
                 'attributes': ['SAI_ACL_COUNTER_ATTR_TABLE_ID', '$acl_table_1'],
             },
         ]
-
+        npu.objects_discovery()
         results = [*npu.process_commands(commands)]
         print('======= SAI commands RETURN values create =======')
         pprint(results)
 
-    def test_acl_counter_remove(self, npu):
+
+    @pytest.mark.dependency(name='test_sai_acl_counter_attr_label_set')
+    def test_sai_acl_counter_attr_label_set(self, npu):
         commands = [
             {
                 'name': 'acl_counter_1',
-                'op': 'remove',
-            },
-            {
-                'name': 'acl_table_1',
-                'op': 'remove',
-            },
+                'op': 'set',
+                'attributes': ['SAI_ACL_COUNTER_ATTR_LABEL', '""'],
+            }
         ]
+        npu.objects_discovery()
+        results = [*npu.process_commands(commands)]
+        print('======= SAI commands RETURN values get =======')
+        pprint(results)
+
 
+    def test_acl_counter_remove(self, npu):
+        commands = [
+            {'name': 'acl_counter_1', 'op': 'remove'},
+            {'name': 'acl_table_1', 'op': 'remove'},
+        ]
+        npu.objects_discovery()
         results = [*npu.process_commands(commands)]
         print('======= SAI commands RETURN values remove =======')
         pprint(results)
diff --git a/tests/api/test_acl_range.py b/tests/api/test_acl_range.py
@@ -24,12 +24,7 @@ def test_acl_range_create(self, npu):
         pprint(results)
 
     def test_acl_range_remove(self, npu):
-        commands = [
-            {
-                'name': 'acl_range_1',
-                'op': 'remove',
-            }
-        ]
+        commands = [{'name': 'acl_range_1', 'op': 'remove'}]
 
         results = [*npu.process_commands(commands)]
         print('======= SAI commands RETURN values remove =======')

diff --git a/tests/api/test_hostif_trap_group.py b/tests/api/test_hostif_trap_group.py
@@ -1,5 +1,7 @@
 from pprint import pprint
 
+import pytest
+
 
 class TestSaiHostifTrapGroup:
     # object with no attributes
@@ -18,13 +20,70 @@ def test_hostif_trap_group_create(self, npu):
         print('======= SAI commands RETURN values create =======')
         pprint(results)
 
-    def test_hostif_trap_group_remove(self, npu):
+    @pytest.mark.dependency(name='test_sai_hostif_trap_group_attr_admin_state_set')
+    def test_sai_hostif_trap_group_attr_admin_state_set(self, npu):
+        commands = [
+            {
+                'name': 'hostif_trap_group_1',
+                'op': 'set',
+                'attributes': ['SAI_HOSTIF_TRAP_GROUP_ATTR_ADMIN_STATE', 'true'],
+            }
+        ]
+        results = [*npu.process_commands(commands)]
+        print('======= SAI commands RETURN values get =======')
+        pprint(results)
+
+    @pytest.mark.dependency(depends=['test_sai_hostif_trap_group_attr_admin_state_set'])
+    def test_sai_hostif_trap_group_attr_admin_state_get(self, npu):
+        commands = [
+            {
+                'name': 'hostif_trap_group_1',
+                'op': 'get',
+                'attributes': ['SAI_HOSTIF_TRAP_GROUP_ATTR_ADMIN_STATE'],
+            }
+        ]
+        results = [*npu.process_commands(commands)]
+        print('======= SAI commands RETURN values get =======')
+        for command in results:
+            for attribute in command:
+                pprint(attribute.raw())
+        r_value = results[0][0].value()
+        print(r_value)
+        assert r_value == 'true', 'Get error, expected true but got %s' % r_value
+
+    @pytest.mark.dependency(name='test_sai_hostif_trap_group_attr_queue_set')
+    def test_sai_hostif_trap_group_attr_queue_set(self, npu):
+        commands = [
+            {
+                'name': 'hostif_trap_group_1',
+                'op': 'set',
+                'attributes': ['SAI_HOSTIF_TRAP_GROUP_ATTR_QUEUE', '0'],
+            }
+        ]
+        results = [*npu.process_commands(commands)]
+        print('======= SAI commands RETURN values get =======')
+        pprint(results)
+
+    @pytest.mark.dependency(depends=['test_sai_hostif_trap_group_attr_queue_set'])
+    def test_sai_hostif_trap_group_attr_queue_get(self, npu):
         commands = [
             {
                 'name': 'hostif_trap_group_1',
-                'op': 'remove',
+                'op': 'get',
+                'attributes': ['SAI_HOSTIF_TRAP_GROUP_ATTR_QUEUE'],
             }
         ]
+        results = [*npu.process_commands(commands)]
+        print('======= SAI commands RETURN values get =======')
+        for command in results:
+            for attribute in command:
+                pprint(attribute.raw())
+        r_value = results[0][0].value()
+        print(r_value)
+        assert r_value == '0', 'Get error, expected 0 but got %s' % r_value
+
+    def test_hostif_trap_group_remove(self, npu):
+        commands = [{'name': 'hostif_trap_group_1', 'op': 'remove'}]
 
         results = [*npu.process_commands(commands)]
         print('======= SAI commands RETURN values remove =======')

diff --git a/tests/api/test_policer.py b/tests/api/test_policer.py
@@ -1,5 +1,7 @@
 from pprint import pprint
 
+import pytest
+
 
 class TestSaiPolicer:
     # object with no parents
@@ -23,13 +25,225 @@ def test_policer_create(self, npu):
         print('======= SAI commands RETURN values create =======')
         pprint(results)
 
-    def test_policer_remove(self, npu):
+    @pytest.mark.dependency(name='test_sai_policer_attr_cbs_set')
+    def test_sai_policer_attr_cbs_set(self, npu):
+        commands = [
+            {
+                'name': 'policer_1',
+                'op': 'set',
+                'attributes': ['SAI_POLICER_ATTR_CBS', '0'],
+            }
+        ]
+        results = [*npu.process_commands(commands)]
+        print('======= SAI commands RETURN values get =======')
+        pprint(results)
+
+    @pytest.mark.dependency(depends=['test_sai_policer_attr_cbs_set'])
+    def test_sai_policer_attr_cbs_get(self, npu):
+        commands = [
+            {'name': 'policer_1', 'op': 'get', 'attributes': ['SAI_POLICER_ATTR_CBS']}
+        ]
+        results = [*npu.process_commands(commands)]
+        print('======= SAI commands RETURN values get =======')
+        for command in results:
+            for attribute in command:
+                pprint(attribute.raw())
+        r_value = results[0][0].value()
+        print(r_value)
+        assert r_value == '0', 'Get error, expected 0 but got %s' % r_value
+
+    @pytest.mark.dependency(name='test_sai_policer_attr_cir_set')
+    def test_sai_policer_attr_cir_set(self, npu):
+        commands = [
+            {
+                'name': 'policer_1',
+                'op': 'set',
+                'attributes': ['SAI_POLICER_ATTR_CIR', '0'],
+            }
+        ]
+        results = [*npu.process_commands(commands)]
+        print('======= SAI commands RETURN values get =======')
+        pprint(results)
+
+    @pytest.mark.dependency(depends=['test_sai_policer_attr_cir_set'])
+    def test_sai_policer_attr_cir_get(self, npu):
+        commands = [
+            {'name': 'policer_1', 'op': 'get', 'attributes': ['SAI_POLICER_ATTR_CIR']}
+        ]
+        results = [*npu.process_commands(commands)]
+        print('======= SAI commands RETURN values get =======')
+        for command in results:
+            for attribute in command:
+                pprint(attribute.raw())
+        r_value = results[0][0].value()
+        print(r_value)
+        assert r_value == '0', 'Get error, expected 0 but got %s' % r_value
+
+    @pytest.mark.dependency(name='test_sai_policer_attr_pbs_set')
+    def test_sai_policer_attr_pbs_set(self, npu):
+        commands = [
+            {
+                'name': 'policer_1',
+                'op': 'set',
+                'attributes': ['SAI_POLICER_ATTR_PBS', '0'],
+            }
+        ]
+        results = [*npu.process_commands(commands)]
+        print('======= SAI commands RETURN values get =======')
+        pprint(results)
+
+    @pytest.mark.dependency(depends=['test_sai_policer_attr_pbs_set'])
+    def test_sai_policer_attr_pbs_get(self, npu):
+        commands = [
+            {'name': 'policer_1', 'op': 'get', 'attributes': ['SAI_POLICER_ATTR_PBS']}
+        ]
+        results = [*npu.process_commands(commands)]
+        print('======= SAI commands RETURN values get =======')
+        for command in results:
+            for attribute in command:
+                pprint(attribute.raw())
+        r_value = results[0][0].value()
+        print(r_value)
+        assert r_value == '0', 'Get error, expected 0 but got %s' % r_value
+
+    @pytest.mark.dependency(name='test_sai_policer_attr_pir_set')
+    def test_sai_policer_attr_pir_set(self, npu):
+        commands = [
+            {
+                'name': 'policer_1',
+                'op': 'set',
+                'attributes': ['SAI_POLICER_ATTR_PIR', '0'],
+            }
+        ]
+        results = [*npu.process_commands(commands)]
+        print('======= SAI commands RETURN values get =======')
+        pprint(results)
+
+    @pytest.mark.dependency(depends=['test_sai_policer_attr_pir_set'])
+    def test_sai_policer_attr_pir_get(self, npu):
+        commands = [
+            {'name': 'policer_1', 'op': 'get', 'attributes': ['SAI_POLICER_ATTR_PIR']}
+        ]
+        results = [*npu.process_commands(commands)]
+        print('======= SAI commands RETURN values get =======')
+        for command in results:
+            for attribute in command:
+                pprint(attribute.raw())
+        r_value = results[0][0].value()
+        print(r_value)
+        assert r_value == '0', 'Get error, expected 0 but got %s' % r_value
+
+    @pytest.mark.dependency(name='test_sai_policer_attr_green_packet_action_set')
+    def test_sai_policer_attr_green_packet_action_set(self, npu):
         commands = [
             {
                 'name': 'policer_1',
-                'op': 'remove',
+                'op': 'set',
+                'attributes': [
+                    'SAI_POLICER_ATTR_GREEN_PACKET_ACTION',
+                    'SAI_PACKET_ACTION_FORWARD',
+                ],
             }
         ]
+        results = [*npu.process_commands(commands)]
+        print('======= SAI commands RETURN values get =======')
+        pprint(results)
+
+    @pytest.mark.dependency(depends=['test_sai_policer_attr_green_packet_action_set'])
+    def test_sai_policer_attr_green_packet_action_get(self, npu):
+        commands = [
+            {
+                'name': 'policer_1',
+                'op': 'get',
+                'attributes': ['SAI_POLICER_ATTR_GREEN_PACKET_ACTION'],
+            }
+        ]
+        results = [*npu.process_commands(commands)]
+        print('======= SAI commands RETURN values get =======')
+        for command in results:
+            for attribute in command:
+                pprint(attribute.raw())
+        r_value = results[0][0].value()
+        print(r_value)
+        assert r_value == 'SAI_PACKET_ACTION_FORWARD', (
+            'Get error, expected SAI_PACKET_ACTION_FORWARD but got %s' % r_value
+        )
+
+    @pytest.mark.dependency(name='test_sai_policer_attr_yellow_packet_action_set')
+    def test_sai_policer_attr_yellow_packet_action_set(self, npu):
+        commands = [
+            {
+                'name': 'policer_1',
+                'op': 'set',
+                'attributes': [
+                    'SAI_POLICER_ATTR_YELLOW_PACKET_ACTION',
+                    'SAI_PACKET_ACTION_FORWARD',
+                ],
+            }
+        ]
+        results = [*npu.process_commands(commands)]
+        print('======= SAI commands RETURN values get =======')
+        pprint(results)
+
+    @pytest.mark.dependency(depends=['test_sai_policer_attr_yellow_packet_action_set'])
+    def test_sai_policer_attr_yellow_packet_action_get(self, npu):
+        commands = [
+            {
+                'name': 'policer_1',
+                'op': 'get',
+                'attributes': ['SAI_POLICER_ATTR_YELLOW_PACKET_ACTION'],
+            }
+        ]
+        results = [*npu.process_commands(commands)]
+        print('======= SAI commands RETURN values get =======')
+        for command in results:
+            for attribute in command:
+                pprint(attribute.raw())
+        r_value = results[0][0].value()
+        print(r_value)
+        assert r_value == 'SAI_PACKET_ACTION_FORWARD', (
+            'Get error, expected SAI_PACKET_ACTION_FORWARD but got %s' % r_value
+        )
+
+    @pytest.mark.dependency(name='test_sai_policer_attr_red_packet_action_set')
+    def test_sai_policer_attr_red_packet_action_set(self, npu):
+        commands = [
+            {
+                'name': 'policer_1',
+                'op': 'set',
+                'attributes': [
+                    'SAI_POLICER_ATTR_RED_PACKET_ACTION',
+                    'SAI_PACKET_ACTION_FORWARD',
+                ],
+            }
+        ]
+        results = [*npu.process_commands(commands)]
+        print('======= SAI commands RETURN values get =======')
+        pprint(results)
+
+    @pytest.mark.dependency(depends=['test_sai_policer_attr_red_packet_action_set'])
+    def test_sai_policer_attr_red_packet_action_get(self, npu):
+        commands = [
+            {
+                'name': 'policer_1',
+                'op': 'get',
+                'attributes': ['SAI_POLICER_ATTR_RED_PACKET_ACTION'],
+            }
+        ]
+        results = [*npu.process_commands(commands)]
+        print('======= SAI commands RETURN values get =======')
+        for command in results:
+            for attribute in command:
+                pprint(attribute.raw())
+        r_value = results[0][0].value()
+        print(r_value)
+        assert r_value == 'SAI_PACKET_ACTION_FORWARD', (
+            'Get error, expected SAI_PACKET_ACTION_FORWARD but got %s' % r_value
+        )
+
+
+    def test_policer_remove(self, npu):
+        commands = [{'name': 'policer_1', 'op': 'remove'}]
 
         results = [*npu.process_commands(commands)]
         print('======= SAI commands RETURN values remove =======')