如何在Flask中对HTTP摘要认证进行单元测试？

import pytest
from flask import Flask
from flask_httpauth import HTTPDigestAuth
from your_app import app

@pytest.fixture
def client():
    app.config['TESTING'] = True
    app.config['SECRET_KEY'] = 'your_secret_key'
    app.config['USERNAME'] = 'your_username'
    app.config['PASSWORD'] = 'your_password'
    client = app.test_client()
    return client

def test_http_digest_auth(client):
    response = client.get('/protected')
    assert response.status_code == 401  # 未授权，返回401状态码

    response = client.get('/protected', headers={
        'Authorization': HTTPDigestAuth.generate_header(
            app.config['USERNAME'], app.config['PASSWORD'], 'GET', '/protected')
    })
    assert response.status_code == 200  # 授权成功，返回200状态码

pytest test_auth.py