Skip to content

Commit 9cecfa2

Browse files
committed
fix UT
1 parent 6a347a0 commit 9cecfa2

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

tests/generic_config_updater/field_operation_validator_test.py

+9-1
Original file line numberDiff line numberDiff line change
@@ -31,9 +31,17 @@ def test_rdma_config_update_validator_td3_asic_invalid_version(self):
3131
@patch("generic_config_updater.field_operation_validators.get_asic_name", mock.Mock(return_value="spc1"))
3232
@patch("os.path.exists", mock.Mock(return_value=True))
3333
@patch("builtins.open", mock_open(read_data='{"tables": {"PFC_WD": {"validator_data": {"rdma_config_update_validator": {"PFCWD enable/disable": {"fields": ["detection_time", "action"], "operations": ["remove", "replace", "add"], "platforms": {"spc1": "20181100"}}}}}}}'))
34-
def test_rdma_config_update_validator_spc_asic_valid_version(self):
34+
def test_rdma_config_update_validator_spc_asic_valid_version_remove(self):
3535
patch_element = {"path": "/PFC_WD/Ethernet8/detection_time", "op": "remove"}
3636
assert generic_config_updater.field_operation_validators.rdma_config_update_validator(patch_element) == True
37+
38+
@patch("sonic_py_common.device_info.get_sonic_version_info", mock.Mock(return_value={"build_version": "SONiC.20220530"}))
39+
@patch("generic_config_updater.field_operation_validators.get_asic_name", mock.Mock(return_value="spc1"))
40+
@patch("os.path.exists", mock.Mock(return_value=True))
41+
@patch("builtins.open", mock_open(read_data='{"tables": {"PFC_WD": {"validator_data": {"rdma_config_update_validator": {"PFCWD enable/disable": {"fields": ["detection_time", "restoration_time", "action"], "operations": ["remove", "replace", "add"], "platforms": {"spc1": "20181100"}}}}}}}'))
42+
def test_rdma_config_update_validator_spc_asic_valid_version_add_pfcwd(self):
43+
patch_element = {"path": "/PFC_WD/Ethernet8", "op": "add", "value": {"action": "drop", "detection_time": "300", "restoration_time": "200"}}
44+
assert generic_config_updater.field_operation_validators.rdma_config_update_validator(patch_element) == True
3745

3846
@patch("sonic_py_common.device_info.get_sonic_version_info", mock.Mock(return_value={"build_version": "SONiC.20220530"}))
3947
@patch("generic_config_updater.field_operation_validators.get_asic_name", mock.Mock(return_value="spc1"))

0 commit comments

Comments
 (0)