Mercurial > repos > yufei-luo > s_mart
diff commons/tools/tests/Test_F_CheckMysqlConnect.py @ 18:94ab73e8a190
Uploaded
author | m-zytnicki |
---|---|
date | Mon, 29 Apr 2013 03:20:15 -0400 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/commons/tools/tests/Test_F_CheckMysqlConnect.py Mon Apr 29 03:20:15 2013 -0400 @@ -0,0 +1,64 @@ +import unittest +import os +import socket + +class Test_F_CheckMysqlConnect(unittest.TestCase): + + HOST_NAME = "compute-2-46.local" + + def setUp(self): + self._host = socket.gethostname() + self._cmd = "" + + def tearDown(self): + os.remove("testOutputCheckMysqlConnect.txt") + + def test_CheckMysqlConnect_as_script_True(self): + if self.HOST_NAME in self._host: + self._cmd = "CheckMysqlConnect.py -p /share/apps/bin/python > testOutputCheckMysqlConnect.txt" + else: + self._cmd = "CheckMysqlConnect.py > testOutputCheckMysqlConnect.txt" + os.system(self._cmd) + + self.assertFalse(self._isFailedInOutputFile()) + + def test_CheckMysqlConnect_as_script_False(self): + configFile = "%s/repet_tools/tests/config.cfg" % os.environ.get("REPET_PATH") + host = "unknownServer" + self._writeConfig(configFile, host) + + if self.HOST_NAME in self._host: + self._cmd = "CheckMysqlConnect.py -p /share/apps/bin/python -C %s > testOutputCheckMysqlConnect.txt" % configFile + else: + self._cmd = "CheckMysqlConnect.py -C %s > testOutputCheckMysqlConnect.txt " % configFile + os.system(self._cmd) + os.remove(configFile) + + self.assertTrue(self._isFailedInOutputFile()) + + def _isFailedInOutputFile(self): + f = open("testOutputCheckMysqlConnect.txt", "r") + line = f.readline() + + while line: + if "FAILED" in line: + f.close() + return True + line = f.readline() + + f.close() + return False + + def _writeConfig(self, configFileName, repetHost = os.environ.get("REPET_HOST")): + f = open( configFileName, "w" ) + f.write("[repet_env]\n") + f.write("repet_host: %s\n" % repetHost) + f.write("repet_user: %s\n" % os.environ.get("REPET_USER")) + f.write("repet_pw: %s\n" % os.environ.get("REPET_PW")) + f.write("repet_db: %s\n" % os.environ.get("REPET_DB")) + f.write("repet_port: %s\n" % os.environ.get("REPET_PORT")) + f.close() + +if __name__ == "__main__": + unittest.main() + \ No newline at end of file