def test_basic_load(self): image_directory = 'tests/images/' db = TinyDB(storage=storages.MemoryStorage) content = utils.load_content(db, image_directory) full_image_path = content[0] image_file = full_image_path.replace(image_directory, '') status_text = content[1] self.assertTrue(image_file in IMAGE_NAMES) self.assertEqual(image_file.replace('.jpg', ''), status_text)
def test_storage_in_db(self): image_directory = 'tests/images/' # let's load a list of tweets into the db db = TinyDB(storage=storages.MemoryStorage) image_names = [ 'goldfinch1.jpg', 'goldfinch2.jpg', 'goldfinch3.jpg', 'goldfinch4.jpg' ] for image_name in image_names: delivery_timestamp = datetime.now(tz=timezone.utc).isoformat() tweet = {'image': image_name, 'delivered_on': delivery_timestamp} db.insert(tweet) content = utils.load_content(db, image_directory) self.assertEqual(content[2], 'goldfinch5.jpg') tweets = db.all() self.assertEqual(len(tweets), 4, msg=tweets)