def test_remove_simple(self):
store = SavedSearchStore()
search = store.new('Some @tag', 'Looking for some tag')
self.assertEqual(len(store.data), 1)
store.remove(search.id)
self.assertEqual(len(store.data), 0)
with self.assertRaises(KeyError):
store.remove('000')
def test_count(self):
store = SavedSearchStore()
root_1 = store.new('Root search', '@tag')
root_2 = store.new('Root search 2', '@tag2')
child_2 = store.new('Child search 2', '@another_tag', root_2.id)
self.assertEqual(store.count(), 3)
self.assertEqual(store.count(root_only=True), 2)
store.remove(child_2.id)
self.assertEqual(store.count(), 2)
self.assertEqual(store.count(root_only=True), 2)
def test_remove_tree(self):
store = SavedSearchStore()
root_1 = store.new('Root search', '@tag')
root_2 = store.new('Root search 2', '@tag2')
child_1 = store.new('Child search 1', '@another_tag', root_1.id)
child_2 = store.new('Child search 2', '@another_tag', root_2.id)
child_3 = store.new('Child search 3', '@another_tag', child_2.id)
self.assertEqual(len(store.lookup), 5)
store.remove(child_3.id)
self.assertEqual(len(store.lookup), 4)
self.assertEqual(len(child_2.children), 0)
store.remove(root_1.id)
store.remove(root_2.id)
self.assertEqual(len(store.lookup), 0)
self.assertEqual(len(store.data), 0)