def testLogger_getLogs(self):
logger = Logger('sqlite:///testDB.db')
data = {'agentHostName':'test_getLogs_HostName',
'agentHostIp':'test_getLogs_HostIp',
'agentHostTime':'test_getLogs_HostTime',
'agentType':'test_getLogs_AgentType',
'agentHostData':'test_getLogs_HostData'}
logger.addLog(data)
logger.stop()
try:
logs = logger.getLogs()
except:
self.fail("Unexpected Exception!")
self.assertIsInstance(logs, list)
self.assertTrue('test_getLogs_HostName' == logs[-1].agentHostName)
def testLogger_addLog(self):
logger = Logger('sqlite:///testDB.db')
data = {'agentHostName':'test_addLog_HostName',
'agentHostIp':'test_addLog_HostIp',
'agentHostTime':'test_addLog_HostTime',
'agentType':'test_addLog_AgentType',
'agentHostData':'test_addLog_HostData'}
try:
logger.addLog(data)
except:
self.fail("Unexpected Exception!")
logger.session.commit()
logs = logger.session.query(Log).order_by(Log.id).all()
self.assertTrue('test_addLog_HostName' == logs[-1].agentHostName)
self.assertTrue('test_addLog_HostIp' == logs[-1].agentHostIp)
self.assertTrue('test_addLog_HostTime' == logs[-1].agentHostTime)
self.assertTrue('test_addLog_AgentType' == logs[-1].agentType)
self.assertTrue('test_addLog_HostData' == logs[-1].agentHostData)
logger.stop()
