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)
Example #2
0
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)