def test_generate_collectd_snmp_config_raises_SystemExit_for_model_error(
self):
self.session.get_model.side_effect = models.Error('whoops')
with self.assertRaises(SystemExit):
disco_config.generate_collectd_snmp_config(self.session, 'hostname',
'community1')
def test_generate_collectd_snmp_config_raises_SystemExit_for_ifindex_error(
self):
self.session.get_model.return_value = self.model_cisco
self.session.auto_discover_ifindex.side_effect = discovery.Error(
'whoops')
with self.assertRaises(SystemExit):
disco_config.generate_collectd_snmp_config(self.session, 'hostname',
'community1')
def test_generate_collectd_snmp_config(self):
expected_plugin_config = """LoadPlugin snmp
<Plugin snmp>
<Data "uplink_discards">
Type "ifx_discards"
Table false
Instance "uplink"
Values "rx.10" "tx.10"
</Data>
<Host "hostname">
Address "hostname"
Version 2
Community "community1"
Collect "uplink_discards"
Interval 10
</Host>
</Plugin>"""
self.session.get_model.return_value = self.model_cisco
self.session.auto_discover_ifindex.return_value = [('uplink', '10')]
actual = disco_config.generate_collectd_snmp_config(
self.session, 'hostname', 'community1')
self.assertEqual(expected_plugin_config, actual)