def test_xml_write_tree(self):
store = SavedSearchStore()
search1 = store.new('Some @tag', 'Looking for some tag')
search2 = store.new('Some @other @tag', 'Looking for more')
xml_root = store.to_xml()
self.assertEqual(len(xml_root), 2)
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_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_unparent(self):
store = SavedSearchStore()
search1 = store.new('Some @tag', 'Looking for some tag')
search2 = store.new('Some @other @tag', 'Looking for more')
self.assertEqual(len(store.lookup), 2)
self.assertEqual(len(store.data), 2)
store.parent(search1.id, search2.id)
self.assertEqual(len(store.data), 1)
self.assertEqual(len(search2.children), 1)
self.assertEqual(len(store.lookup), 2)
store.unparent(search1.id, search2.id)
self.assertEqual(len(store.data), 2)
self.assertEqual(len(search2.children), 0)
self.assertEqual(len(store.lookup), 2)
def test_new_simple(self):
store = SavedSearchStore()
search = store.new('Some @tag', 'Looking for some tag')
# Make sure we have added a new search
self.assertEqual(len(store.data), 1)
# Make sure it's the same one we received
self.assertEqual(store.lookup[search.id], search)
# Make sure it's an instance of SavedSearch
self.assertIsInstance(search, SavedSearch)
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)
def test_new_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)
self.assertEqual(len(store.data), 2)
self.assertEqual(len(root_1.children), 1)
self.assertEqual(len(child_2.children), 1)
with self.assertRaises(KeyError):
store.new('An error', '@error', '000')