def test_db_purge_when_all_posted(self): available_files = list() for index in range(1,101): image_name = 'image{0}.png'.format(index) available_files.append(image_name) db = TinyDB(storage=storages.MemoryStorage) for id in range(1,106): image_name = 'image{0}.png'.format(id) db.insert({'image': image_name}) self.assertEqual(len(db.all()), 105) unused_files = utils.get_unused_files(db, available_files) self.assertEqual(len(unused_files), 100) self.assertEqual(unused_files[0], 'image1.png') self.assertEqual(unused_files[5], 'image6.png') self.assertEqual(unused_files[10], 'image11.png') self.assertEqual(unused_files[33], 'image34.png') self.assertEqual(unused_files[50], 'image51.png')
def test_get_unused_files(self): available_files = list() for index in range(1,101): image_name = 'image{0}.png'.format(index) available_files.append(image_name) db = TinyDB(storage=storages.MemoryStorage) for id in range(1,52): image_name = 'image{0}.png'.format(id) db.insert({'image': image_name}) unused_files = utils.get_unused_files(db, available_files) self.assertEqual(len(unused_files), 49) self.assertEqual(unused_files[0], 'image52.png') self.assertEqual(unused_files[5], 'image57.png') self.assertEqual(unused_files[10], 'image62.png') self.assertEqual(unused_files[15], 'image67.png') self.assertEqual(unused_files[20], 'image72.png') self.assertEqual(unused_files[33], 'image85.png') self.assertEqual(unused_files[48], 'image100.png')