Skip to content

Commit 5080119

Browse files
authored
Update test_enable_snmp_flow.py
1 parent 7d4a1ef commit 5080119

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

tests/networking/cisco/flows/test_enable_snmp_flow.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@ def test_enable_snmp_v3(self, enable_actions_mock):
3131
enable_snmp_view_mock = MagicMock()
3232
enable_actions_mock.return_value.get_current_snmp_user.side_effect = ["", self.SNMP_USER]
3333
enable_actions_mock.return_value.enable_snmp_v3 = enable_snmp_mock
34+
enable_actions_mock.return_value.get_current_snmp_config.return_value = ""
3435
enable_actions_mock.return_value.enable_snmp_group = enable_snmp_group_mock
3536
enable_actions_mock.return_value.enable_snmp_view = enable_snmp_view_mock
3637

@@ -81,8 +82,9 @@ def test_enable_snmp_v2_read(self, enable_actions_mock):
8182
enable_flow = self._get_handler()
8283
enable_snmp_mock = MagicMock()
8384
is_read_only = True
84-
enable_actions_mock.return_value.get_current_snmp_communities.side_effect = ["",
85-
self.snmp_v2_read_parameters.snmp_community]
85+
enable_actions_mock.return_value.get_current_snmp_config.side_effect = ["",
86+
"snmp-server community {}".format(
87+
self.snmp_v2_read_parameters.snmp_community)]
8688
enable_snmp_mock.return_value.enable_snmp.return_value = enable_snmp_mock
8789
enable_actions_mock.return_value.enable_snmp = enable_snmp_mock
8890

@@ -95,8 +97,9 @@ def test_enable_snmp_v2_write(self, enable_actions_mock):
9597
enable_flow = self._get_handler()
9698
enable_snmp_mock = MagicMock()
9799
is_read_only = False
98-
enable_actions_mock.return_value.get_current_snmp_communities.side_effect = ["",
99-
self.snmp_v2_write_parameters.snmp_community]
100+
enable_actions_mock.return_value.get_current_snmp_config.side_effect = ["",
101+
"snmp-server community {}".format(
102+
self.snmp_v2_write_parameters.snmp_community)]
100103
enable_snmp_mock.return_value.enable_snmp.return_value = enable_snmp_mock
101104
enable_actions_mock.return_value.enable_snmp = enable_snmp_mock
102105

0 commit comments

Comments
 (0)