mirror of https://github.com/mitsuhiko/flask.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
28 lines
617 B
28 lines
617 B
import sqlite3 |
|
|
|
import pytest |
|
from flaskr.db import get_db |
|
|
|
|
|
def test_get_close_db(app): |
|
with app.app_context(): |
|
db = get_db() |
|
assert db is get_db() |
|
|
|
with pytest.raises(sqlite3.ProgrammingError) as e: |
|
db.execute('SELECT 1') |
|
|
|
assert 'closed' in str(e) |
|
|
|
|
|
def test_init_db_command(runner, monkeypatch): |
|
class Recorder(object): |
|
called = False |
|
|
|
def fake_init_db(): |
|
Recorder.called = True |
|
|
|
monkeypatch.setattr('flaskr.db.init_db', fake_init_db) |
|
result = runner.invoke(args=['init-db']) |
|
assert 'Initialized' in result.output |
|
assert Recorder.called
|
|
|