class TestDatabaseHandlerEvents(unittest.TestCase):
def setUp(self):
self.db = DatabaseHandler('wids0')
#self.db.storeEvent({'module':'UnitTestModule', 'datetime':0, 'name':'TestEvent', 'details':{}, 'uuids':['1234-1234-1234-1234'], 'packets':[0,1,2]})
def cleanup(self):
self.db.close()
def test_event_get_all(self):
count = self.db.getEvents(count=True)
self.assertEqual(count, total_count)
for event in self.db.getEvents():
print(event.id, event.datetime, event.module, event.name,
event.details, event.uuids, event.packets)
def test_event_get_filter_module(self):
count = self.db.session.query(Event).filter(
Event.module == 'DisassociationStormMonitor').count()
self.assertEqual(count, 24)
def test_event_get_mixfilter(self):
count = self.db.session.query(Event).filter(
Event.module == 'DisassociationStormMonitor').filter(
Event.name ==
'ZigbeeNWKCommandPayload Frame Detected').count()
self.assertEqual(count, 24)
class TestDatabaseHandlerEvents(unittest.TestCase):
def setUp(self):
self.db = DatabaseHandler('event2', path='./')
self.db.storeEvent({'module':'UnitTestModule', 'datetime':0, 'name':'TestEvent', 'details':{}, 'uuids':['1234-1234-1234-1234'], 'packets':[0,1,2]})
def cleanup(self):
self.db.close()
def test_event_get_all(self):
events = self.db.getEvents()
for event in events:
print(event.id, event.datetime, event.module, event.name, event.details, event.uuids, event.packets)
class TestDatabaseHandlerEvents(unittest.TestCase):
def setUp(self):
self.db = DatabaseHandler('event2', path='./')
self.db.storeEvent({
'module': 'UnitTestModule',
'datetime': 0,
'name': 'TestEvent',
'details': {},
'uuids': ['1234-1234-1234-1234'],
'packets': [0, 1, 2]
})
def cleanup(self):
self.db.close()
def test_event_get_all(self):
events = self.db.getEvents()
for event in events:
print(event.id, event.datetime, event.module, event.name,
event.details, event.uuids, event.packets)
class TestDatabaseHandlerEvents(unittest.TestCase):
def setUp(self):
self.db = DatabaseHandler('wids0')
#self.db.storeEvent({'module':'UnitTestModule', 'datetime':0, 'name':'TestEvent', 'details':{}, 'uuids':['1234-1234-1234-1234'], 'packets':[0,1,2]})
def cleanup(self):
self.db.close()
def test_event_get_all(self):
count = self.db.getEvents(count=True)
self.assertEqual(count, total_count)
for event in self.db.getEvents():
print(event.id, event.datetime, event.module, event.name, event.details, event.uuids, event.packets)
def test_event_get_filter_module(self):
count = self.db.session.query(Event).filter(Event.module == 'DisassociationStormMonitor').count()
self.assertEqual(count, 24)
def test_event_get_mixfilter(self):
count = self.db.session.query(Event).filter(Event.module == 'DisassociationStormMonitor').filter(Event.name == 'ZigbeeNWKCommandPayload Frame Detected').count()
self.assertEqual(count, 24)
class TestDatabaseHandlerPackets(unittest.TestCase):
def setUp(self):
self.db = DatabaseHandler('packet', path='./')
def cleanup(self):
self.db.close()
def test_packet__get_all(self):
count = self.db.getPackets(count=True)
self.assertEqual(count, total_count)
def test_packet_get_new_(self):
# query packets WITHOUT new flag
packets_1 = self.db.getPackets(maxcount=100)
index_1a = packets_1[0].id
index_1b = packets_1[-1].id
packets_2 = self.db.getPackets(maxcount=100)
index_2a = packets_2[0].id
index_2b = packets_2[-1].id
self.assertEqual(index_1a, 1)
self.assertEqual(index_1b, 100)
self.assertEqual(index_2a, 1)
self.assertEqual(index_2b, 100)
# query packets WITH new flag
packets_1 = self.db.getPackets(maxcount=100, new=True)
index_1a = packets_1[0].id
index_1b = packets_1[-1].id
packets_2 = self.db.getPackets(maxcount=100, new=True)
index_2a = packets_2[0].id
index_2b = packets_2[-1].id
self.assertEqual(index_1a, 1)
self.assertEqual(index_1b, 100)
self.assertEqual(index_2a, 101)
self.assertEqual(index_2b, 200)
def test_packet_get_maxcount_(self):
count = self.db.getPackets(maxcount=100, count=True)
self.assertEqual(count, 100)
def test_packet_get_t1_count(self):
count = self.db.getPackets(uuidFilterList=[t1_uuid], count=True)
self.assertEqual(count, t1_count)
def test_packet_get_t2_count(self):
count = self.db.getPackets(uuidFilterList=[t2_uuid], count=True)
self.assertEqual(count, t2_count)
def test_packet_filter_rssi_gt(self):
count = self.db.getPackets(valueFilterList=[('rssi','>',78)], count=True)
self.assertEqual(count, rssi_gt_78)
def test_packet_filter_rssi_eq(self):
count = self.db.getPackets(valueFilterList=[('rssi','==',78)], count=True)
self.assertEqual(count, rssi_eq_78)
def test_packet_filter_rssi_lt(self):
count = self.db.getPackets(valueFilterList=[('rssi','<',78)], count=True)
self.assertEqual(count, rssi_lt_78)
def test_packet_filter_datetime(self):
pass
def test_packet_mix_filters(self):
count = self.db.getPackets(valueFilterList=[('rssi','>',78)], uuidFilterList=[t1_uuid], count=True)
self.assertEqual(count, 76)
class TestDatabaseHandlerPackets(unittest.TestCase):
def setUp(self):
self.db = DatabaseHandler('packet', path='./')
def cleanup(self):
self.db.close()
def test_packet__get_all(self):
count = self.db.getPackets(count=True)
self.assertEqual(count, total_count)
def test_packet_get_new_(self):
# query packets WITHOUT new flag
packets_1 = self.db.getPackets(maxcount=100)
index_1a = packets_1[0].id
index_1b = packets_1[-1].id
packets_2 = self.db.getPackets(maxcount=100)
index_2a = packets_2[0].id
index_2b = packets_2[-1].id
self.assertEqual(index_1a, 1)
self.assertEqual(index_1b, 100)
self.assertEqual(index_2a, 1)
self.assertEqual(index_2b, 100)
# query packets WITH new flag
packets_1 = self.db.getPackets(maxcount=100, new=True)
index_1a = packets_1[0].id
index_1b = packets_1[-1].id
packets_2 = self.db.getPackets(maxcount=100, new=True)
index_2a = packets_2[0].id
index_2b = packets_2[-1].id
self.assertEqual(index_1a, 1)
self.assertEqual(index_1b, 100)
self.assertEqual(index_2a, 101)
self.assertEqual(index_2b, 200)
def test_packet_get_maxcount_(self):
count = self.db.getPackets(maxcount=100, count=True)
self.assertEqual(count, 100)
def test_packet_get_t1_count(self):
count = self.db.getPackets(uuidFilterList=[t1_uuid], count=True)
self.assertEqual(count, t1_count)
def test_packet_get_t2_count(self):
count = self.db.getPackets(uuidFilterList=[t2_uuid], count=True)
self.assertEqual(count, t2_count)
def test_packet_filter_rssi_gt(self):
count = self.db.getPackets(valueFilterList=[('rssi', '>', 78)],
count=True)
self.assertEqual(count, rssi_gt_78)
def test_packet_filter_rssi_eq(self):
count = self.db.getPackets(valueFilterList=[('rssi', '==', 78)],
count=True)
self.assertEqual(count, rssi_eq_78)
def test_packet_filter_rssi_lt(self):
count = self.db.getPackets(valueFilterList=[('rssi', '<', 78)],
count=True)
self.assertEqual(count, rssi_lt_78)
def test_packet_filter_datetime(self):
pass
def test_packet_mix_filters(self):
count = self.db.getPackets(valueFilterList=[('rssi', '>', 78)],
uuidFilterList=[t1_uuid],
count=True)
self.assertEqual(count, 76)
