diff options
Diffstat (limited to 'tests/middleware/test_open_database.py')
-rw-r--r-- | tests/middleware/test_open_database.py | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/tests/middleware/test_open_database.py b/tests/middleware/test_open_database.py index 43ebb07..1584ac0 100644 --- a/tests/middleware/test_open_database.py +++ b/tests/middleware/test_open_database.py @@ -15,10 +15,9 @@ def app(): @open_database @validator def app(environ, start_response): - assert "paste.db_conn" in environ - conn = environ["paste.db_conn"] - (ver,) = conn.execute("PRAGMA user_version").fetchone() - assert ver > 0 + assert "paste.store" in environ + db = environ["paste.store"] + assert db.version > 0 start_response("200 OK", [("Content-Type", "text/plain")]) return [b"Hello, World!"] |