Flask单元测试 #

pip install pytest pytest-cov

from app import create_app, db

import pytest

@pytest.fixture
def app():
    app = create_app('testing')
    app.config['TESTING'] = True
    app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///:memory:'
    
    with app.app_context():
        db.create_all()
        yield app
        db.drop_all()

@pytest.fixture
def client(app):
    return app.test_client()

def test_index(client):
    response = client.get('/')
    assert response.status_code == 200
    assert b'Welcome' in response.data

def test_login(client):
    response = client.post('/login', data={
        'username': 'test',
        'password': 'test'
    })
    assert response.status_code == 302

import json

def test_api_get_users(client):
    response = client.get('/api/users')
    assert response.status_code == 200
    data = json.loads(response.data)
    assert isinstance(data, list)

def test_api_create_user(client):
    response = client.post('/api/users',
        data=json.dumps({'username': 'test', 'email': 'test@example.com'}),
        content_type='application/json'
    )
    assert response.status_code == 201

pytest
pytest --cov=app