mirror of
https://github.com/yzernik/squeaknode.git
synced 2026-08-17 13:07:32 +02:00
88108f7810
* Rename server entrypoint * Updated manifest * Moved db init.sql file * Rename module to squeakserver
29 lines
650 B
Python
29 lines
650 B
Python
import sqlite3
|
|
|
|
import pytest
|
|
|
|
from squeakserver.server.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.value)
|
|
|
|
|
|
def test_init_db_command(runner, monkeypatch):
|
|
class Recorder(object):
|
|
called = False
|
|
|
|
def fake_init_db():
|
|
Recorder.called = True
|
|
|
|
monkeypatch.setattr("squeakserver.server.db.init_db", fake_init_db)
|
|
result = runner.invoke(args=["init-db"])
|
|
assert "Initialized" in result.output
|
|
assert Recorder.called
|