def test_update_snmp_community_raises_SystemExit_when_no_vlans_found( self, mock_discovery_session, mock_session): mock_discovery_session.return_value = self.session self.session.auto_discover_vlan.return_value = ['123'] self.session.auto_discover_ifindex.return_value = [] with self.assertRaises(SystemExit): disco_config.update_snmp_community(self.session, 'hostname', self.switch_config, 'community1') mock_session.assert_called_with(DestHost='hostname', Version=2, Community='community1@123')
def test_update_snmp_community_raises_SystemExit_for_ifindex_error( self, mock_discovery_session, mock_session): mock_discovery_session.return_value = self.session self.session.auto_discover_vlan.return_value = ['123'] self.session.auto_discover_ifindex.side_effect = discovery.Error( 'whoops') with self.assertRaises(SystemExit): disco_config.update_snmp_community(self.session, 'hostname', self.switch_config, 'community1') mock_session.assert_called_with(DestHost='hostname', Version=2, Community='community1@123')
def test_update_snmp_community(self, mock_discovery_session, mock_session): mock_discovery_session.return_value = self.session self.session.auto_discover_vlan.return_value = ['123'] self.session.auto_discover_ifindex.return_value = [('uplink', '10')] actual = disco_config.update_snmp_community( self.session, 'hostname', self.switch_config, 'community1') self.assertEqual('community1@123', actual) mock_session.assert_called_with(DestHost='hostname', Version=2, Community='community1@123')