Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with HTTPS or Subversion.

Download ZIP
Fetching contributors…

Cannot retrieve contributors at this time

executable file 261 lines (217 sloc) 8.541 kb
#!/usr/bin/env python
# Copyright (C) 2009-2010:
# Gabes Jean, naparuba@gmail.com
# Gerhard Lausser, Gerhard.Lausser@consol.de
#
# This file is part of Shinken.
#
# Shinken is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Shinken is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with Shinken. If not, see <http://www.gnu.org/licenses/>.
#
# This file is used to test reading and processing of config files
#
import os
import sys
from shinken_test import *
from shinken.action import Action
time.time = original_time_time
time.sleep = original_time_sleep
class TestAction(ShinkenTest):
# setUp is inherited from ShinkenTest
def wait_finished(self, a, size=8012):
start = time.time()
while True:
# Do the job
if a.status == 'launched':
#print a.process.poll()
a.check_finished(size)
time.sleep(0.01)
#print a.status
if a.status != 'launched':
#print "Finish", a.status
return
# 20s timeout
if time.time() - start > 20:
print "COMMAND TIMEOUT AT 20s"
return
def test_action(self):
a = Action()
a.timeout = 10
a.env = {}
if os.name == 'nt':
a.command = r'libexec\\dummy_command.cmd'
else:
a.command = "libexec/dummy_command.sh"
self.assert_(a.got_shell_characters() == False)
a.execute()
self.assert_(a.status == 'launched')
# Give also the max output we want for the command
self.wait_finished(a)
self.assert_(a.exit_status == 0)
self.assert_(a.status == 'done')
print a.output
self.assert_(a.output == "Hi, I'm for testing only. Please do not use me directly, really")
self.assert_(a.perf_data == "Hip=99% Bob=34mm")
def test_echo_environment_variables(self):
if os.name == 'nt':
return
a = Action()
a.timeout = 10
a.env = {} # :fixme: this sould be pre-set in Action.__init__()
a.command = "echo $TITI"
self.assertNotIn('TITI', a.get_local_environnement())
a.env = {'TITI': 'est en vacance'}
self.assertIn('TITI', a.get_local_environnement())
self.assertEqual(a.get_local_environnement()['TITI'],
'est en vacance' )
a.execute()
self.wait_finished(a)
self.assertEqual(a.output, 'est en vacance')
def test_grep_for_environment_variables(self):
if os.name == 'nt':
return
a = Action()
a.timeout = 10
a.env = {} # :fixme: this sould be pre-set in Action.__init__()
a.command = "/usr/bin/env | grep TITI"
self.assertNotIn('TITI', a.get_local_environnement())
a.env = {'TITI': 'est en vacance'}
self.assertIn('TITI', a.get_local_environnement())
self.assertEqual(a.get_local_environnement()['TITI'],
'est en vacance' )
a.execute()
self.wait_finished(a)
self.assertEqual(a.output, 'TITI=est en vacance')
def test_environment_variables(self):
class ActionWithoutPerfData(Action):
def get_outputs(self, out, max_len):
# do not cut the outputs into perf_data to avoid
# problems with enviroments containing a dash like in
# `LESSOPEN=|/usr/bin/lesspipe.sh %s`
self.output = out
if os.name == 'nt':
return
a = ActionWithoutPerfData()
a.timeout = 10
a.command = "/usr/bin/env"
a.env = {} # :fixme: this sould be pre-set in Action.__init__()
self.assertNotIn('TITI', a.get_local_environnement())
a.env = {'TITI': 'est en vacance'}
self.assert_(a.got_shell_characters() == False)
self.assertIn('TITI', a.get_local_environnement())
self.assertEqual(a.get_local_environnement()['TITI'],
'est en vacance' )
a.execute()
self.assert_(a.status == 'launched')
# Give also the max output we want for the command
self.wait_finished(a, size=20*1024)
titi_found = False
for l in a.output.splitlines():
if l == 'TITI=est en vacance':
titi_found = True
self.assertTrue(titi_found)
# Some commands are shell without bangs! (like in Centreon...)
# We can show it in the launch, and it should be managed
def test_noshell_bang_command(self):
a = Action()
a.timeout = 10
a.command = "libexec/dummy_command_nobang.sh"
a.env = {}
if os.name == 'nt':
return
self.assert_(a.got_shell_characters() == False)
a.execute()
self.assert_(a.status == 'launched')
self.wait_finished(a)
print "FUck", a.status, a.output
self.assert_(a.exit_status == 0)
self.assert_(a.status == 'done')
def test_got_shell_characters(self):
a = Action()
a.timeout = 10
a.command = "libexec/dummy_command_nobang.sh && echo finished ok"
a.env = {}
if os.name == 'nt':
return
self.assert_(a.got_shell_characters() == True)
a.execute()
self.assert_(a.status == 'launched')
self.wait_finished(a)
print "FUck", a.status, a.output
self.assert_(a.exit_status == 0)
self.assert_(a.status == 'done')
def test_got_pipe_shell_characters(self):
a = Action()
a.timeout = 10
a.command = "libexec/dummy_command_nobang.sh | grep 'Please do not use me directly'"
a.env = {}
if os.name == 'nt':
return
self.assert_(a.got_shell_characters() == True)
a.execute()
self.assert_(a.status == 'launched')
self.wait_finished(a)
print "FUck", a.status, a.output
self.assert_(a.exit_status == 0)
self.assert_(a.status == 'done')
def test_got_unclosed_quote(self):
# https://github.com/naparuba/shinken/issues/155
a = Action()
a.timeout = 10
a.command = "libexec/dummy_command_nobang.sh -a 'wwwwzzzzeeee"
a.env = {}
if os.name == 'nt':
return
a.execute()
self.wait_finished(a)
self.assert_(a.status == 'done')
print "FUck", a.status, a.output
if sys.version_info < (2, 7):
# cygwin: /bin/sh: -c: line 0: unexpected EOF while looking for matching'
# ubuntu: /bin/sh: Syntax error: Unterminated quoted string
self.assert_(a.output.startswith("/bin/sh"))
self.assert_(a.exit_status == 3)
else:
self.assert_(a.output == 'Not a valid shell command: No closing quotation')
self.assert_(a.exit_status == 3)
# We got problems on LARGE output, more than 64K in fact.
# We try to solve it with the fcntl and non blocking read
# instead of "communicate" mode. So here we try to get a 100K
# output. Should NOT be in a timeout
def test_huge_output(self):
a = Action()
a.timeout = 5
a.env = {}
if os.name == 'nt':
a.command = r"""python -c 'print "A"*1000000'"""
# FROM NOW IT4S FAIL ON WINDOWS :(
return
else:
a.command = r"""python -u -c 'print "A"*100000'"""
print "EXECUTE"
a.execute()
print "EXECUTE FINISE"
self.assert_(a.status == 'launched')
# Give also the max output we want for the command
self.wait_finished(a, 10000000000)
print "Status?", a.exit_status
self.assert_(a.exit_status == 0)
print "Output", len(a.output)
self.assert_(a.exit_status == 0)
self.assert_(a.status == 'done')
self.assert_(a.output == "A"*100000)
self.assert_(a.perf_data == "")
if __name__ == '__main__':
import sys
#os.chdir(os.path.dirname(sys.argv[0]))
unittest.main()
Jump to Line
Something went wrong with that request. Please try again.