在Behave和Flask中模拟

# app.py
from flask import Flask, jsonify
import sqlite3

app = Flask(__name__)

def get_user_from_db(user_id):
    conn = sqlite3.connect('example.db')
    cursor = conn.cursor()
    cursor.execute("SELECT * FROM users WHERE id=?", (user_id,))
    user = cursor.fetchone()
    conn.close()
    return user

@app.route('/user/<int:user_id>')
def user_profile(user_id):
    user = get_user_from_db(user_id)
    if user:
        return jsonify({'id': user[0], 'name': user[1]})
    else:
        return jsonify({'error': 'User not found'}), 404

if __name__ == '__main__':
    app.run(debug=True)

# features/steps/user_steps.py
from behave import given, when, then
from unittest.mock import patch
from app import app

@given('a user with id {user_id}')
def step_impl(context, user_id):
    context.user_id = user_id

@when('I request the user profile')
def step_impl(context):
    with app.test_client() as client:
        with patch('__main__.get_user_from_db') as mock_get_user:
            mock_get_user.return_value = (1, 'John Doe')
            context.response = client.get(f'/user/{context.user_id}')

@then('I should see the user profile')
def step_impl(context):
    assert context.response.status_code == 200
    assert context.response.json == {'id': 1, 'name': 'John Doe'}