def test_purge(self): source_a = 'A' source_b = 'B' manager = ContentCatalogManager() for unit_key, url in self.units(0, 10): manager.add_entry(source_a, EXPIRATION, TYPE_ID, unit_key, url) for unit_key, url in self.units(0, 10): manager.add_entry(source_b, EXPIRATION, TYPE_ID, unit_key, url) collection = ContentCatalog.get_collection() self.assertEqual(20, collection.find().count()) manager = ContentCatalogManager() manager.purge(source_a) self.assertEqual(collection.find({'source_id': source_a}).count(), 0) self.assertEqual(collection.find({'source_id': source_b}).count(), 10)
def test_purge(self): source_a = 'A' source_b = 'B' manager = ContentCatalogManager() for unit_key, url in self.units(0, 10): manager.add_entry(source_a, EXPIRATION, TYPE_ID, unit_key, url) for unit_key, url in self.units(0, 10): manager.add_entry(source_b, EXPIRATION, TYPE_ID, unit_key, url) collection = ContentCatalog.get_collection() self.assertEqual(20, collection.find().count()) manager = ContentCatalogManager() manager.purge(source_a) self.assertEqual(collection.find({'source_id': source_a}).count(), 0) self.assertEqual(collection.find({'source_id': source_b}).count(), 10)
def test_has_entries(self): source_a = 'A' source_b = 'B' source_c = 'C' manager = ContentCatalogManager() for unit_key, url in self.units(0, 10): manager.add_entry(source_a, EXPIRATION, TYPE_ID, unit_key, url) for unit_key, url in self.units(0, 10): manager.add_entry(source_b, -1, TYPE_ID, unit_key, url) for unit_key, url in self.units(0, 10): manager.add_entry(source_c, EXPIRATION, TYPE_ID, unit_key, url) collection = ContentCatalog.get_collection() self.assertEqual(30, collection.find().count()) manager = ContentCatalogManager() self.assertTrue(manager.has_entries(source_a)) self.assertFalse(manager.has_entries(source_b)) self.assertTrue(manager.has_entries(source_c)) manager.purge(source_c) self.assertTrue(manager.has_entries(source_a)) self.assertFalse(manager.has_entries(source_b)) self.assertFalse(manager.has_entries(source_c))
def test_has_entries(self): source_a = 'A' source_b = 'B' source_c = 'C' manager = ContentCatalogManager() for unit_key, url in self.units(0, 10): manager.add_entry(source_a, EXPIRATION, TYPE_ID, unit_key, url) for unit_key, url in self.units(0, 10): manager.add_entry(source_b, -1, TYPE_ID, unit_key, url) for unit_key, url in self.units(0, 10): manager.add_entry(source_c, EXPIRATION, TYPE_ID, unit_key, url) collection = ContentCatalog.get_collection() self.assertEqual(30, collection.find().count()) manager = ContentCatalogManager() self.assertTrue(manager.has_entries(source_a)) self.assertFalse(manager.has_entries(source_b)) self.assertTrue(manager.has_entries(source_c)) manager.purge(source_c) self.assertTrue(manager.has_entries(source_a)) self.assertFalse(manager.has_entries(source_b)) self.assertFalse(manager.has_entries(source_c))