def _drop_database(self):
Database.connection().drop_database(self.TEST_DATABASE_NAME)
def set_collection(cls, collection_name):
return Database.db()[collection_name]
def _create_database(self):
Database.db(self.TEST_DATABASE_NAME)
async def db_session_middleware(request: Request, call_next):
request.state.db = Database.Session()
response = await call_next(request)
request.state.db.close()
return response
