def test_find(self): col = self.db['cache'] anime = AniDBTitle('Some Anime') anime.tags = [AniDBTag('fanservice', 200)] for obj in [Cache.create_cache_object(anime)]: id = col.insert_one(obj).inserted_id found_anime = Cache(self.db).find(anime) self.assertIsNotNone(found_anime) self.assertEqual(found_anime._id, id)
def test_update(self): col = self.db['cache'] anime = AniDBTitle('Some Anime 123') anime.anidb_id = 2 anime.tags = [AniDBTag('fanservice', 200)] id = col.insert_one(Cache.create_cache_object(anime)).inserted_id cache = Cache(self.db) anime.anidb_id = 4 updated_anime = cache.update(anime) self.assertEqual(4, updated_anime.anidb_id)