Skip to content

Commit 065340f

Browse files
committed
clean UT
1 parent 79ed8df commit 065340f

File tree

1 file changed

+8
-16
lines changed

1 file changed

+8
-16
lines changed

tests/generic_config_updater/field_operation_validator_test.py

+8-16
Original file line numberDiff line numberDiff line change
@@ -24,45 +24,37 @@ def test_rdma_config_update_validator_unknown_asic(self):
2424
@patch("sonic_py_common.device_info.get_sonic_version_info", mock.Mock(return_value={"build_version": "SONiC.20171131"}))
2525
@patch("generic_config_updater.field_operation_validators.get_asic_name", mock.Mock(return_value="spc1"))
2626
@patch("os.path.exists", mock.Mock(return_value=True))
27-
@patch("builtins.open", mock_open(read_data="sample"))
28-
@patch('json.load')
29-
def test_rdma_config_update_validator_spc_asic_invalid_version(self, mock_json_config):
27+
@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"}}}}}}}'))
28+
def test_rdma_config_update_validator_spc_asic_invalid_version(self):
3029
path = "/PFC_WD/Ethernet8/action"
3130
operation = "replace"
32-
mock_json_config.return_value = {"tables": {"pfc_wd": {"validator_data": {"rdma_config_update_validator": {"PFCWD enable/disable": {"fields": ["detection_time", "action"], "operations": ["remove", "replace", "add"], "platforms": {"spc1": "20181100"}}}}}}}
3331
assert generic_config_updater.field_operation_validators.rdma_config_update_validator(path, operation) == False
3432

3533
@patch("sonic_py_common.device_info.get_sonic_version_info", mock.Mock(return_value={"build_version": "SONiC.20181131"}))
3634
@patch("generic_config_updater.field_operation_validators.get_asic_name", mock.Mock(return_value="spc1"))
3735
@patch("os.path.exists", mock.Mock(return_value=True))
38-
@patch("builtins.open", mock_open(read_data="sample"))
39-
@patch('json.load')
40-
def test_rdma_config_update_validator_spc_asic_valid_version(self, mock_json_config):
36+
@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"}}}}}}}'))
37+
def test_rdma_config_update_validator_spc_asic_valid_version(self):
4138
path = "/PFC_WD/Ethernet8/action"
4239
operation = "replace"
43-
mock_json_config.return_value = {"tables": {"pfc_wd": {"validator_data": {"rdma_config_update_validator": {"PFCWD enable/disable": {"fields": ["detection_time", "action"], "operations": ["remove", "replace", "add"], "platforms": {"spc1": "20181100"}}}}}}}
4440
assert generic_config_updater.field_operation_validators.rdma_config_update_validator(path, operation) == True
4541

4642
@patch("sonic_py_common.device_info.get_sonic_version_info", mock.Mock(return_value={"build_version": "SONiC.20181131"}))
4743
@patch("generic_config_updater.field_operation_validators.get_asic_name", mock.Mock(return_value="spc1"))
4844
@patch("os.path.exists", mock.Mock(return_value=True))
49-
@patch("builtins.open", mock_open(read_data="sample"))
50-
@patch('json.load')
51-
def test_rdma_config_update_validator_spc_asic_invalid_op(self, mock_json_config):
45+
@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"}}}}}}}'))
46+
def test_rdma_config_update_validator_spc_asic_invalid_op(self):
5247
path = "/PFC_WD/Ethernet8/action"
5348
operation = "invalid-op"
54-
mock_json_config.return_value = {"tables": {"pfc_wd": {"validator_data": {"rdma_config_update_validator": {"PFCWD enable/disable": {"fields": ["detection_time", "action"], "operations": ["remove", "replace", "add"], "platforms": {"spc1": "20181100"}}}}}}}
5549
assert generic_config_updater.field_operation_validators.rdma_config_update_validator(path, operation) == False
5650

5751
@patch("sonic_py_common.device_info.get_sonic_version_info", mock.Mock(return_value={"build_version": "SONiC.20181131"}))
5852
@patch("generic_config_updater.field_operation_validators.get_asic_name", mock.Mock(return_value="spc1"))
5953
@patch("os.path.exists", mock.Mock(return_value=True))
60-
@patch("builtins.open", mock_open(read_data="sample"))
61-
@patch('json.load')
62-
def test_rdma_config_update_validator_spc_asic_other_field(self, mock_json_config):
54+
@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"}}}}}}}'))
55+
def test_rdma_config_update_validator_spc_asic_other_field(self):
6356
path = "/PFC_WD/Ethernet8/other_field"
6457
operation = "invalid-op"
65-
mock_json_config.return_value = {"tables": {"pfc_wd": {"validator_data": {"rdma_config_update_validator": {"PFCWD enable/disable": {"fields": ["detection_time", "action"], "operations": ["remove", "replace", "add"], "platforms": {"spc1": "20181100"}}}}}}}
6658
assert generic_config_updater.field_operation_validators.rdma_config_update_validator(path, operation) == False
6759

6860
@patch("sonic_py_common.device_info.get_sonic_version_info", mock.Mock(return_value={"build_version": "SONiC.20181131"}))

0 commit comments

Comments
 (0)