def test_delete(self):
skiplist = SkipList(config=self.config)
self.assertRaises(MissingArgumentError, skiplist.delete)
self.assertRaises(MissingArgumentError,
skiplist.delete,
category='something')
self.assertRaises(MissingArgumentError,
skiplist.delete,
rule='something')
cursor = self.connection.cursor()
cursor.execute("select count(*) from skiplist")
first, = cursor.fetchall()
count = first[0]
self.assertEqual(count, 5)
self.assertTrue(skiplist.delete(category='irrelevant', rule='ashmem'))
cursor.execute("select count(*) from skiplist")
first, = cursor.fetchall()
count = first[0]
self.assertEqual(count, 4)
cursor.execute(
"""
select count(*) from skiplist
where category=%s and rule=%s
""", ('irrelevant', 'ashmem'))
first, = cursor.fetchall()
count = first[0]
self.assertEqual(count, 0)
self.assertTrue(not skiplist.delete(category='neverheard', rule='of'))
def test_delete(self):
skiplist = SkipList(config=self.config)
self.assertRaises(MissingArgumentError, skiplist.delete)
self.assertRaises(
MissingArgumentError,
skiplist.delete,
category='something'
)
self.assertRaises(
MissingArgumentError,
skiplist.delete,
rule='something'
)
cursor = self.connection.cursor()
cursor.execute("select count(*) from skiplist")
first, = cursor.fetchall()
count = first[0]
self.assertEqual(count, 5)
self.assertTrue(skiplist.delete(category='irrelevant', rule='ashmem'))
cursor.execute("select count(*) from skiplist")
first, = cursor.fetchall()
count = first[0]
self.assertEqual(count, 4)
cursor.execute("""
select count(*) from skiplist
where category=%s and rule=%s
""", ('irrelevant', 'ashmem'))
first, = cursor.fetchall()
count = first[0]
self.assertEqual(count, 0)
self.assertTrue(not skiplist.delete(category='neverheard', rule='of'))
