def setUp(self):
self.jsonObj = JSONify()
class JSONifyTestCase(unittest.TestCase):
def setUp(self):
self.jsonObj = JSONify()
def test_objectToSQL(self):
tObj = {"facility":['auth', 'authpriv'],
"includeFacility":"include",
"priority":['debug', 'info', 'notice'],
"includePriority":"include",
"hosts":['www', 'mail'],
"includeHosts":"include",
"tags":['ssh', 'getty'],
"includeTags":"include",
"message":"loged in",
"includeMessage":"include",
"before":"2011-02-01 10:00:00",
"after":"2011-01-01 10:00:00","page":"50"}
querry = self.jsonObj.objectToSQL(tObj)
sql = "SELECT id, concat(datetime) AS datetime, host, facility, priority, tag, message, MATCH(`message`) AGAINST('loged in' IN BOOLEAN MODE) AS score FROM logs WHERE `facility` IN('auth', 'authpriv') AND `priority` IN('debug', 'info', 'notice') AND `host` IN('www', 'mail') AND `tag` IN('ssh', 'getty') AND MATCH(`message`) AGAINST('loged in' IN BOOLEAN MODE) ORDER BY score DESC, id DESC LIMIT 50;"
self.assertEquals(querry, sql, "SQL querry cretion with objectToSQL failed!")
def test_validityCheckClean(self):
tObj = {"facility":['auth', 'authpriv'],
"includeFacility":"include",
"priority":['debug', 'info', 'notice'],
"includePriority":"include",
"includeHosts":"include",
"includeTags":"include",
"message":"loged in",
"includeMessage":"include",
"before":"2011-02-01 10:00:00",
"after":"2011-01-01 10:00:00",
"page":"50"}
qObj = self.jsonObj.validityCheck(tObj)
self.assertTrue(set(tObj).intersection(set(qObj)), "JSON object validity clean check failed!")
def test_validityCheckUnCleanFacility(self):
tObj = {"facility":['auth', 'authpriv', 'UNCLEAN'],
"includeFacility":"include",
"priority":['debug', 'info', 'notice'],
"includePriority":"include",
"includeHosts":"include",
"includeTags":"include",
"message":"loged in",
"includeMessage":"include",
"before":"2011-02-01 10:00:00",
"after":"2011-01-01 10:00:00",
"page":"50"}
qObj = self.jsonObj.validityCheck(tObj)
self.assertEquals(qObj["facility"], ['authpriv', 'auth'], "JSON object validity unclean facility check failed!")
def test_validityCheckUnCleanPriority(self):
tObj = {"facility":['auth', 'authpriv'],
"includeFacility":"include",
"priority":['debug', 'info', 'notice', 'UNCLEAN'],
"includePriority":"include",
"includeHosts":"include",
"includeTags":"include",
"message":"loged in",
"includeMessage":"include",
"before":"2011-02-01 10:00:00",
"after":"2011-01-01 10:00:00",
"page":"50"}
qObj = self.jsonObj.validityCheck(tObj)
self.assertEquals(qObj["priority"], ['debug', 'info', 'notice'], "JSON object validity unclean priority check failed!")
def test_validityCheckUnCleanIncludePriority(self):
tObj = {"facility":['auth', 'authpriv'],
"includeFacility":"include",
"priority":['debug', 'info', 'notice'],
"includePriority":"UNCLEAN",
"includeHosts":"include",
"includeTags":"include",
"message":"loged in",
"includeMessage":"include",
"before":"2011-02-01 10:00:00",
"after":"2011-01-01 10:00:00",
"page":"50"}
qObj = self.jsonObj.validityCheck(tObj)
self.assertEquals(qObj["includePriority"], [], "JSON object validity unclean priority check failed!")
def test_validityCheckUnCleanBefore(self):
tObj = {"facility":['auth', 'authpriv'],
"includeFacility":"include",
"priority":['debug', 'info', 'notice'],
"includePriority":"include",
"includeHosts":"include",
"includeTags":"include",
"message":"loged in",
"includeMessage":"include",
"before":"UNCLEAN",
"after":"2011-01-01 10:00:00",
"page":"50"}
qObj = self.jsonObj.validityCheck(tObj)
self.assertEquals(qObj["before"], (time.strftime('%Y-%m-%d %H:%M:%S')), "JSON object validity unclean before check failed!")
def test_validityCheckUnCleanAfter(self):
tObj = {"facility":['auth', 'authpriv'],
"includeFacility":"include",
"priority":['debug', 'info', 'notice'],
"includePriority":"include",
"includeHosts":"include",
"includeTags":"include",
"message":"loged in",
"includeMessage":"include",
"before":"2011-02-01 10:00:00",
"after":"UNCLEAN",
"page":"50"}
qObj = self.jsonObj.validityCheck(tObj)
self.assertEquals(qObj["after"], "1970:01:01 00:00:00", "JSON object validity unclean after check failed!")
def test_validityCheckCleanMessage_1(self):
tObj = {"facility":['auth', 'authpriv'],
"includeFacility":"include",
"priority":['debug', 'info', 'notice'],
"includePriority":"include",
"includeHosts":"include",
"includeTags":"include",
"message":"' SELECT * FROM tags; -- ",
"includeMessage":"include",
"before":"2011-02-01 10:00:00",
"after":"2011-01-01 10:00:00",
"page":"50"}
qObj = self.jsonObj.validityCheck(tObj)
self.assertEquals(qObj["message"], r"\' SELECT * FROM tags\; \-\- ", "JSON object validity unclean message_1 failed!")
def test_validityCheckCleanMessage_2(self):
tObj = {"facility":['auth', 'authpriv'],
"includeFacility":"include",
"priority":['debug', 'info', 'notice'],
"includePriority":"include",
"includeHosts":"include",
"includeTags":"include",
"message":"\' SELECT * FROM tags; -- ",
"includeMessage":"include",
"before":"2011-02-01 10:00:00",
"after":"2011-01-01 10:00:00",
"page":"50"}
qObj = self.jsonObj.validityCheck(tObj)
self.assertEquals(qObj["message"], r"\' SELECT * FROM tags\; \-\- ", "JSON object validity unclean message_1 failed!")