Method cleanup

test/test_mongodb_integration.py:62–81

()

    def tearDownClass(cls):
        if cls.mongodb_available:
            async def cleanup():
                conn = MongoDBConnection()
                db = await conn.get_db()

                test_collections = [
                    "test_xhs_contents",
                    "test_xhs_comments",
                    "test_xhs_creators",
                    "test_douyin_contents",
                    "test_douyin_comments",
                    "test_douyin_creators"
                ]

                for collection_name in test_collections:
                    try:
                        await db[collection_name].drop()
                    except:
                        pass

                await conn.close()

            try:
                asyncio.run(cleanup())