def setUp(self):
self.dbdict = {}
# get the information from configuration file
self.dbdict['host'] = "localhost"
self.dbdict['user'] = "******"
self.dbdict['password'] = "******"
self.dbdict['db'] = "buffer"
self.dbdict['poll_interval'] = 5
self.dbdict['poll_limit'] = 100
self.dbdict['keep_buffer_data'] = 0
# setup storagemanager
self.sm = StorageManager(2)
self.interpreter = Interpreter(self.sm)
self.bufferLis = BufferListener(self.dbdict, self.interpreter)
class TestBufferListener(unittest.TestCase):
def setUp(self):
self.dbdict = {}
# get the information from configuration file
self.dbdict['host'] = "localhost"
self.dbdict['user'] = "******"
self.dbdict['password'] = "******"
self.dbdict['db'] = "buffer"
self.dbdict['poll_interval'] = 5
self.dbdict['poll_limit'] = 100
self.dbdict['keep_buffer_data'] = 0
# setup storagemanager
self.sm = StorageManager(2)
self.interpreter = Interpreter(self.sm)
self.bufferLis = BufferListener(self.dbdict, self.interpreter)
def tearDown(self):
# get the number of events in the buffer
# self.TotalEvents = self.bufferLis.getMessageCount()
self.TotalEvents1 = 0
def testConnection(self):
self.conn = self.bufferLis.getDBConnection(self.dbdict)
self.assert_(self.conn is not None)
def testGetMessages(self):
self.msgs = self.bufferLis.getBufferMessages()
self.msgs1 = self.bufferLis.getBufferMessages()
# test the numbers
self.assertEqual(len(self.msgs), self.dbdict['poll_limit'])
self.assertEqual(len(self.msgs1), self.dbdict['poll_limit'])
def testClearMessages(self):
self.event_ids = []
self.msgs = self.bufferLis.getBufferMessages()
for msg in self.msgs:
self.event_ids.append(msg[2])
self.TotalEvents = self.bufferLis.getMessageCount()
self.bufferLis.clearBufferMessages(self.event_ids)
self.TotalEvents1 = self.bufferLis.getMessageCount()
self.assertEqual(len(self.event_ids),
self.TotalEvents - self.TotalEvents1)
