def test_add(self): units = self.units(0, 10) conduit = CatalogerConduit(SOURCE_ID, EXPIRES) for unit_key, url in units: conduit.add_entry(TYPE_ID, unit_key, url) collection = ContentCatalog.get_collection() self.assertEqual(conduit.source_id, SOURCE_ID) self.assertEqual(conduit.expires, EXPIRES) self.assertEqual(len(units), collection.find().count()) self.assertEqual(conduit.added_count, len(units)) self.assertEqual(conduit.deleted_count, 0) for unit_key, url in units: locator = ContentCatalog.get_locator(TYPE_ID, unit_key) entry = collection.find_one({"locator": locator}) self.assertEqual(entry["type_id"], TYPE_ID) self.assertEqual(entry["unit_key"], unit_key) self.assertEqual(entry["url"], url)
def test_add(self): units = self.units(0, 10) conduit = CatalogerConduit(SOURCE_ID, EXPIRES) for unit_key, url in units: conduit.add_entry(TYPE_ID, unit_key, url) collection = ContentCatalog.get_collection() self.assertEqual(conduit.source_id, SOURCE_ID) self.assertEqual(conduit.expires, EXPIRES) self.assertEqual(len(units), collection.find().count()) self.assertEqual(conduit.added_count, len(units)) self.assertEqual(conduit.deleted_count, 0) for unit_key, url in units: locator = ContentCatalog.get_locator(TYPE_ID, unit_key) entry = collection.find_one({'locator': locator}) self.assertEqual(entry['type_id'], TYPE_ID) self.assertEqual(entry['unit_key'], unit_key) self.assertEqual(entry['url'], url)
def test_delete(self): units = self.units(0, 10) conduit = CatalogerConduit(SOURCE_ID, EXPIRES) for unit_key, url in units: conduit.add_entry(TYPE_ID, unit_key, url) collection = ContentCatalog.get_collection() self.assertEqual(len(units), collection.find().count()) unit_key, url = units[5] locator = ContentCatalog.get_locator(TYPE_ID, unit_key) entry = collection.find_one({"locator": locator}) self.assertEqual(entry["type_id"], TYPE_ID) self.assertEqual(entry["unit_key"], unit_key) self.assertEqual(entry["url"], url) conduit.delete_entry(TYPE_ID, unit_key) self.assertEqual(len(units) - 1, collection.find().count()) self.assertEqual(conduit.added_count, len(units)) self.assertEqual(conduit.deleted_count, 1) entry = collection.find_one({"locator": locator}) self.assertTrue(entry is None)
def test_delete(self): units = self.units(0, 10) conduit = CatalogerConduit(SOURCE_ID, EXPIRES) for unit_key, url in units: conduit.add_entry(TYPE_ID, unit_key, url) collection = ContentCatalog.get_collection() self.assertEqual(len(units), collection.find().count()) unit_key, url = units[5] locator = ContentCatalog.get_locator(TYPE_ID, unit_key) entry = collection.find_one({'locator': locator}) self.assertEqual(entry['type_id'], TYPE_ID) self.assertEqual(entry['unit_key'], unit_key) self.assertEqual(entry['url'], url) conduit.delete_entry(TYPE_ID, unit_key) self.assertEqual(len(units) - 1, collection.find().count()) self.assertEqual(conduit.added_count, len(units)) self.assertEqual(conduit.deleted_count, 1) entry = collection.find_one({'locator': locator}) self.assertTrue(entry is None)