def test_bad_argument(self): """Test bad args : -H webserver -S apache -t INT64 -l 0 -s 0 -e 5 -H webserver -S apache -t INT64 -l 0 -s 6 -e 2 -H webserver -S apache -t INT -l 0 -s 0 -e 2 -H webserver -S apache -g INT64 -l 1 -s 0 -e 2 """ # -H webserver -S apache -t INT64 -l 0 -s 0 -e 5 sys.argv = [sys.argv[0]] sys.argv.append('-H') sys.argv.append('webserver') sys.argv.append('-S') sys.argv.append('apache') sys.argv.append('-t') sys.argv.append('INT64') sys.argv.append('-l') sys.argv.append('0') sys.argv.append('-s') sys.argv.append('0') sys.argv.append('-e') sys.argv.append('5') try: out = StringIO() sys.stdout = out check_fake.main() except SystemExit, e: self.assertEquals(type(e), type(SystemExit())) self.assertEquals(e.code, 3) output = out.getvalue().strip() assert output == """Error state must be 0, 1, 2 or 3"""
def test_error_return(self): """Test plugin with the following args : -H webserver -S apache -t INT64 -l 0 -s 0 -e 1 """ f = open("/tmp/webserver_apache", "w") f.close() check_fake.POOL_FOLDER = "/tmp" sys.argv = [sys.argv[0]] sys.argv.append('-H') sys.argv.append('webserver') sys.argv.append('-S') sys.argv.append('apache') sys.argv.append('-t') sys.argv.append('INT16') sys.argv.append('-l') sys.argv.append('0') sys.argv.append('-s') sys.argv.append('0') sys.argv.append('-e') sys.argv.append('2') try: out = StringIO() sys.stdout = out check_fake.main() except SystemExit, e: self.assertEquals(type(e), type(SystemExit())) self.assertEquals(e.code, 2) output = out.getvalue().strip() assert output.startswith("CRITICAL: ")
def test_ok_return(self): """Test plugin with the following args : -H webserver -S apache -t INT64 -l 0 -s 0 -e 1 """ check_fake.POOL_FOLDER = "/fake_folder" sys.argv = [sys.argv[0]] sys.argv.append('-H') sys.argv.append('webserver') sys.argv.append('-S') sys.argv.append('apache1') sys.argv.append('-t') sys.argv.append('INT64') sys.argv.append('-l') sys.argv.append('0') sys.argv.append('-s') sys.argv.append('0') sys.argv.append('-e') sys.argv.append('1') try: out = StringIO() sys.stdout = out check_fake.main() except SystemExit, e: self.assertEquals(type(e), type(SystemExit())) self.assertEquals(e.code, 0) output = out.getvalue().strip() assert output.startswith("OK: ")
def test_help(self): """Test help output : -h """ sys.argv = [sys.argv[0]] sys.argv.append('-h') try: out = StringIO() sys.stdout = out check_fake.main() except SystemExit, e: self.assertEquals(type(e), type(SystemExit())) self.assertEquals(e.code, 3) output = out.getvalue().strip() assert output.startswith("""check_fake """)
def test_version(self): """Test version output : -V """ sys.argv = [sys.argv[0]] sys.argv.append('-V') try: out = StringIO() sys.stdout = out check_fake.main() except SystemExit, e: self.assertEquals(type(e), type(SystemExit())) self.assertEquals(e.code, 3) output = out.getvalue().strip() assert output.startswith("check_fake v%s" % check_fake.PLUGIN_VERSION)
def test_missing_args(self): """Test missings args: -S apache -t TEXT """ sys.argv = [sys.argv[0]] sys.argv.append('-S') sys.argv.append('apache1') sys.argv.append('-t') sys.argv.append('TEXT') try: out = StringIO() sys.stdout = out check_fake.main() except SystemExit, e: self.assertEquals(type(e), type(SystemExit())) self.assertEquals(e.code, 3) output = out.getvalue().strip() assert output.startswith("""Argument `hostname' is missing !""")
def test_default_args(self): """Test default values: -H webserver -S apache -t TEXT """ sys.argv = [sys.argv[0]] sys.argv.append('-H') sys.argv.append('webserver') sys.argv.append('-S') sys.argv.append('apache1') sys.argv.append('-t') sys.argv.append('TEXT') try: out = StringIO() sys.stdout = out check_fake.main() except SystemExit, e: self.assertEquals(type(e), type(SystemExit())) self.assertEquals(e.code, 0)
sys.argv.append('-H') sys.argv.append('webserver') sys.argv.append('-S') sys.argv.append('apache') sys.argv.append('-t') sys.argv.append('INT64') sys.argv.append('-l') sys.argv.append('0') sys.argv.append('-s') sys.argv.append('6') sys.argv.append('-e') sys.argv.append('2') try: out = StringIO() sys.stdout = out check_fake.main() except SystemExit, e: self.assertEquals(type(e), type(SystemExit())) self.assertEquals(e.code, 3) output = out.getvalue().strip() assert output == """State must be 0, 1, 2 or 3""" # -H webserver -S apache -t INT -l 0 -s 0 -e 2 sys.argv = [sys.argv[0]] sys.argv.append('-H') sys.argv.append('webserver') sys.argv.append('-S') sys.argv.append('apache') sys.argv.append('-t') sys.argv.append('INT') sys.argv.append('-l')