pymongo_tests.py 文件源码

def mongo_database():
    connection_params = {'host': os.environ.get('MONGODB_HOST', 'localhost'),
                         'port': int(os.environ.get('MONGODB_PORT', 27017))}
    if pymongo.version_tuple < (3, 0):
        connection_params['safe'] = True
    mongo = pymongo.MongoClient(**connection_params)
    db = mongo.elasticapm_test
    yield db
    mongo.drop_database('elasticapm_test')
    mongo.close()