mirror of https://github.com/Icinga/icinga2.git
Johannes Meyer
parent
87e00a1b84
commit
a4ba11bd13
|
|
@ -18,6 +18,6 @@ file { '/etc/motd':
|
||||||
}
|
}
|
||||||
|
|
||||||
user { 'vagrant':
|
user { 'vagrant':
|
||||||
groups => 'icingacmd',
|
groups => ['icinga', 'icingacmd'],
|
||||||
require => Group['icingacmd']
|
require => [User['icinga'], Group['icingacmd']]
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,210 @@
|
||||||
|
#!/usr/bin/env python
|
||||||
|
from __future__ import unicode_literals
|
||||||
|
|
||||||
|
import os
|
||||||
|
import sys
|
||||||
|
import time
|
||||||
|
|
||||||
|
import utils
|
||||||
|
|
||||||
|
|
||||||
|
LS_HOST_COLUMNS = [
|
||||||
|
'name',
|
||||||
|
'name',
|
||||||
|
'display_name',
|
||||||
|
'display_name',
|
||||||
|
None,
|
||||||
|
'state',
|
||||||
|
'state_type',
|
||||||
|
'current_attempt',
|
||||||
|
'max_check_attempts',
|
||||||
|
None,
|
||||||
|
'last_state',
|
||||||
|
None,
|
||||||
|
'last_state_change',
|
||||||
|
None,
|
||||||
|
'latency',
|
||||||
|
'execution_time',
|
||||||
|
'plugin_output',
|
||||||
|
None,
|
||||||
|
'last_check',
|
||||||
|
'address',
|
||||||
|
'address6'
|
||||||
|
]
|
||||||
|
|
||||||
|
LS_SVC_COLUMNS = [
|
||||||
|
'description',
|
||||||
|
'display_name',
|
||||||
|
'display_name',
|
||||||
|
None,
|
||||||
|
'state',
|
||||||
|
'state_type',
|
||||||
|
'current_attempt',
|
||||||
|
'max_check_attempts',
|
||||||
|
None,
|
||||||
|
'last_state',
|
||||||
|
None,
|
||||||
|
'last_state_change',
|
||||||
|
None,
|
||||||
|
'latency',
|
||||||
|
'execution_time',
|
||||||
|
'plugin_output',
|
||||||
|
'perf_data',
|
||||||
|
'last_check',
|
||||||
|
'host_num_services',
|
||||||
|
'host_num_services_ok',
|
||||||
|
'host_num_services_warn',
|
||||||
|
'host_num_services_unknown',
|
||||||
|
'host_num_services_crit'
|
||||||
|
]
|
||||||
|
|
||||||
|
STATE_MAP = {
|
||||||
|
'SOFT': 0,
|
||||||
|
'HARD': 1
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def send_command(command):
|
||||||
|
try:
|
||||||
|
return send_query('COMMAND [{0}] {1}'.format(int(time.time()), command))
|
||||||
|
except utils.LiveStatusError, error:
|
||||||
|
sys.stderr.write('Failed to execute command: {0}\n\n{1}\n'
|
||||||
|
''.format(command, error))
|
||||||
|
|
||||||
|
|
||||||
|
def send_query(query):
|
||||||
|
response = LIVESTATUS.query(query + '\nColumnHeaders: on')
|
||||||
|
if response:
|
||||||
|
header, result = response.pop(0), {}
|
||||||
|
return [dict((header[i], v) for i, v in enumerate(r)) for r in response]
|
||||||
|
return []
|
||||||
|
|
||||||
|
|
||||||
|
def get_one(query):
|
||||||
|
return next(iter(send_query(query)), {})
|
||||||
|
|
||||||
|
|
||||||
|
def get_event_output():
|
||||||
|
try:
|
||||||
|
with open('/tmp/test_event.out') as f:
|
||||||
|
remove = True
|
||||||
|
return f.read().rstrip().split('|')
|
||||||
|
except (IOError, OSError):
|
||||||
|
remove = False
|
||||||
|
finally:
|
||||||
|
if remove:
|
||||||
|
os.system('sudo rm -f /tmp/test_event.out')
|
||||||
|
|
||||||
|
|
||||||
|
def convert_output(value):
|
||||||
|
try:
|
||||||
|
return int(value)
|
||||||
|
except ValueError:
|
||||||
|
try:
|
||||||
|
return float(value)
|
||||||
|
except ValueError:
|
||||||
|
return STATE_MAP.get(value, value)
|
||||||
|
|
||||||
|
|
||||||
|
def success(msg):
|
||||||
|
print '[OK] {0}'.format(msg).encode('utf-8')
|
||||||
|
|
||||||
|
|
||||||
|
def fail(msg):
|
||||||
|
print '[FAIL] {0}'.format(msg).encode('utf-8')
|
||||||
|
|
||||||
|
|
||||||
|
def info(msg):
|
||||||
|
print '[INFO] {0}'.format(msg).encode('utf-8')
|
||||||
|
|
||||||
|
|
||||||
|
def main():
|
||||||
|
send_command('CHANGE_HOST_EVENT_HANDLER;localhost;test_event')
|
||||||
|
host_info = get_one('GET hosts\nFilter: name = localhost'
|
||||||
|
'\nColumns: event_handler')
|
||||||
|
if host_info.get('event_handler') != 'test_event':
|
||||||
|
fail('Could not assign eventcommand "test_event" to host "localhost"')
|
||||||
|
return 1
|
||||||
|
success('Successfully assigned an eventcommand to host "localhost"')
|
||||||
|
|
||||||
|
send_command('PROCESS_HOST_CHECK_RESULT;localhost;1;A negative result to'
|
||||||
|
' trigger an eventhandler|some interesting perfdata!')
|
||||||
|
event_output = get_event_output()
|
||||||
|
if not event_output:
|
||||||
|
send_command('CHANGE_HOST_EVENT_HANDLER;localhost;')
|
||||||
|
fail('Could not trigger the eventcommand')
|
||||||
|
return 1
|
||||||
|
success('Successfully triggered the eventcommand')
|
||||||
|
failure = False
|
||||||
|
|
||||||
|
info('Checking host macros...')
|
||||||
|
host_info = get_one('GET hosts\nFilter: name = localhost\nColumns: {0}'
|
||||||
|
''.format(' '.join(c for c in LS_HOST_COLUMNS if c)))
|
||||||
|
if event_output[0] != host_info['name']*2:
|
||||||
|
failure = True
|
||||||
|
fail('Escaping environment variables seems not to properly working')
|
||||||
|
fail(' Expected: {0!r} Got: {1!r}'.format(host_info['name']*2,
|
||||||
|
event_output[0]))
|
||||||
|
for i, column in enumerate(LS_HOST_COLUMNS[1:], 1):
|
||||||
|
if column is not None:
|
||||||
|
output_value = convert_output(event_output[i])
|
||||||
|
if output_value != host_info[column]:
|
||||||
|
failure = True
|
||||||
|
fail('{0!r} is unequal to {1!r} ({2})'.format(output_value,
|
||||||
|
host_info[column],
|
||||||
|
column))
|
||||||
|
|
||||||
|
info('Checking service macros...')
|
||||||
|
svc_info = get_one('GET services\nFilter: description = ping4\nColumns: {0}'
|
||||||
|
''.format(' '.join(c for c in LS_SVC_COLUMNS if c)))
|
||||||
|
for i, column in enumerate(LS_SVC_COLUMNS, len(LS_HOST_COLUMNS)):
|
||||||
|
if column is not None:
|
||||||
|
output_value = convert_output(event_output[i])
|
||||||
|
if output_value != svc_info[column]:
|
||||||
|
failure = True
|
||||||
|
fail('{0!r} is unequal to {1!r} ({2})'.format(output_value,
|
||||||
|
svc_info[column],
|
||||||
|
column))
|
||||||
|
|
||||||
|
info('Checking global macros...')
|
||||||
|
|
||||||
|
info('Checking command macros...')
|
||||||
|
if convert_output(event_output[-1]) != 1337:
|
||||||
|
failure = True
|
||||||
|
fail('The command macro "custom_macro" is not being substituted')
|
||||||
|
|
||||||
|
send_command('DISABLE_HOST_EVENT_HANDLER;localhost')
|
||||||
|
send_command('PROCESS_HOST_CHECK_RESULT;localhost;0;A positive result that'
|
||||||
|
' should not trigger an eventhandler')
|
||||||
|
if get_event_output():
|
||||||
|
failure = True
|
||||||
|
fail('Could not disable the eventcommand')
|
||||||
|
else:
|
||||||
|
success('Successfully disabled the eventcommand')
|
||||||
|
|
||||||
|
send_command('ENABLE_HOST_EVENT_HANDLER;localhost')
|
||||||
|
host_info = get_one('GET hosts\nFilter: name = localhost'
|
||||||
|
'\nColumns: event_handler_enabled')
|
||||||
|
if host_info['event_handler_enabled'] != 1:
|
||||||
|
failure = True
|
||||||
|
fail('Could not re-enable the eventcommand')
|
||||||
|
else:
|
||||||
|
success('Successfully re-enabled the eventcommand')
|
||||||
|
|
||||||
|
send_command('CHANGE_HOST_EVENT_HANDLER;localhost;')
|
||||||
|
host_info = get_one('GET hosts\nFilter: name = localhost'
|
||||||
|
'\nColumns: event_handler')
|
||||||
|
if host_info['event_handler']:
|
||||||
|
failure = True
|
||||||
|
fail('Could not remove eventcommand "test_event"'
|
||||||
|
' assigned to host "localhost"')
|
||||||
|
else:
|
||||||
|
success('Successfully removed the eventcommand'
|
||||||
|
' assigned to host "localhost"')
|
||||||
|
|
||||||
|
return 1 if failure else 0
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
with utils.LiveStatusSocket('/var/run/icinga2/cmd/livestatus') as LIVESTATUS:
|
||||||
|
sys.exit(main())
|
||||||
|
|
@ -0,0 +1,59 @@
|
||||||
|
object EventCommand "test_event" inherits "plugin-event-command" {
|
||||||
|
command = {{{echo "\
|
||||||
|
$$HOSTNAME$HOSTNAME$\
|
||||||
|
|$HOSTNAME$\
|
||||||
|
|$HOSTDISPLAYNAME$\
|
||||||
|
|$HOSTALIAS$\
|
||||||
|
|$HOSTSTATE$\
|
||||||
|
|$HOSTSTATEID$\
|
||||||
|
|$HOSTSTATETYPE$\
|
||||||
|
|$HOSTATTEMPT$\
|
||||||
|
|$MAXHOSTATTEMPT$\
|
||||||
|
|$LASTHOSTSTATE$\
|
||||||
|
|$LASTHOSTSTATEID$\
|
||||||
|
|$LASTHOSTSTATETYPE$\
|
||||||
|
|$LASTHOSTSTATECHANGE$\
|
||||||
|
|$HOSTDURATIONSEC$\
|
||||||
|
|$HOSTLATENCY$\
|
||||||
|
|$HOSTEXECUTIONTIME$\
|
||||||
|
|$HOSTOUTPUT$\
|
||||||
|
|$HOSTPERFDATA$\
|
||||||
|
|$LASTHOSTCHECK$\
|
||||||
|
|$HOSTADDRESS$\
|
||||||
|
|$HOSTADDRESS6$\
|
||||||
|
|$SERVICEDESC$\
|
||||||
|
|$SERVICEDISPLAYNAME$\
|
||||||
|
|$SERVICECHECKCOMMAND$\
|
||||||
|
|$SERVICESTATE$\
|
||||||
|
|$SERVICESTATEID$\
|
||||||
|
|$SERVICESTATETYPE$\
|
||||||
|
|$SERVICEATTEMPT$\
|
||||||
|
|$MAXSERVICEATTEMPT$\
|
||||||
|
|$LASTSERVICESTATE$\
|
||||||
|
|$LASTSERVICESTATEID$\
|
||||||
|
|$LASTSERVICESTATETYPE$\
|
||||||
|
|$LASTSERVICESTATECHANGE$\
|
||||||
|
|$SERVICEDURATIONSEC$\
|
||||||
|
|$SERVICELATENCY$\
|
||||||
|
|$SERVICEEXECUTIONTIME$\
|
||||||
|
|$SERVICEOUTPUT$\
|
||||||
|
|$SERVICEPERFDATA$\
|
||||||
|
|$LASTSERVICECHECK$\
|
||||||
|
|$TOTALHOSTSERVICES$\
|
||||||
|
|$TOTALHOSTSERVICESOK$\
|
||||||
|
|$TOTALHOSTSERVICESWARNING$\
|
||||||
|
|$TOTALHOSTSERVICESUNKNOWN$\
|
||||||
|
|$TOTALHOSTSERVICESCRITICAL$\
|
||||||
|
|$TIMET$\
|
||||||
|
|$LONGDATETIME$\
|
||||||
|
|$SHORTDATETIME$\
|
||||||
|
|$DATE$\
|
||||||
|
|$TIME$\
|
||||||
|
|$custom_macro$\
|
||||||
|
" > /tmp/test_event.out}}},
|
||||||
|
|
||||||
|
export_macros = ["HOSTNAME"],
|
||||||
|
macros = {
|
||||||
|
custom_macro = 1337
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -55,6 +55,25 @@
|
||||||
"clean": ["/tmp/utils.py*"],
|
"clean": ["/tmp/utils.py*"],
|
||||||
"exec": ["sudo service icinga2 restart"]
|
"exec": ["sudo service icinga2 restart"]
|
||||||
}
|
}
|
||||||
|
},
|
||||||
|
"eventhandler.test": {
|
||||||
|
"setup": {
|
||||||
|
"copy": [
|
||||||
|
"files/utils.py >> /tmp/utils.py",
|
||||||
|
"files/configs/eventhandler.conf >> /tmp/eventhandler.conf"
|
||||||
|
],
|
||||||
|
"exec": [
|
||||||
|
"sudo mv /tmp/eventhandler.conf /etc/icinga2/conf.d/",
|
||||||
|
"sudo service icinga2 restart"
|
||||||
|
]
|
||||||
|
},
|
||||||
|
"teardown": {
|
||||||
|
"clean": ["/tmp/utils.py*"],
|
||||||
|
"exec": [
|
||||||
|
"sudo rm /etc/icinga2/conf.d/eventhandler.conf",
|
||||||
|
"sudo service icinga2 restart"
|
||||||
|
]
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue