2

我有一个 Flask 应用程序，其中 REST API 是使用带有 MongoDB 后端的 flask_restful 构建的。我想使用 pytest 和 mongomock 编写功能测试来模拟 MongoDB 以测试 API，但无法对其进行配置。谁能指导我提供一个可以实现相同目标的示例？这是我在 conftest.py 文件中使用的夹具：

@pytest.fixture(scope='module')
   def test_client():
    # flask_app = Flask(__name__)
 
    # Flask provides a way to test your application by exposing the  Werkzeug test Client
    # and handling the context locals for you.
    testing_client = app.test_client()
 
    # Establish an application context before running the tests.
    ctx = app.app_context()
    ctx.push()
 
    yield testing_client  # this is where the testing happens!
 
    ctx.pop()
@pytest.fixture(autouse=True)
def patch_mongo():
    db = connect('testdb', host='mongomock://localhost')
    
    yield db
    db.drop_database('testdb')
    disconnect()
    db.close()

这是用于测试创建用户的发布请求的测试功能：

def test_mongo(test_client,patch_mongo):
    headers={
        "Content-Type": "application/json",
        "Authorization": "token"
    }
    response=test_client.post('/users',headers=headers,data=json.dumps(data))
    print(response.get_json())

    assert response.status_code == 200

问题是 pytest 不是使用 testdb，而是在生产数据库中创建用户。配置中是否缺少某些内容？

4

0 回答 0