Skip to content

Commit 75f0aa9

Browse files
committed
Migrate test_session to pytest engine
1 parent d073037 commit 75f0aa9

File tree

1 file changed

+40
-0
lines changed

1 file changed

+40
-0
lines changed

tests/test_session.py

Lines changed: 40 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,40 @@
1+
import pytest
2+
from flask import session
3+
4+
from flask_mongoengine import MongoEngineSessionInterface
5+
6+
7+
@pytest.fixture(autouse=True)
8+
def setup_endpoints(app, db):
9+
10+
app.session_interface = MongoEngineSessionInterface(db)
11+
12+
@app.route("/")
13+
def index():
14+
session["a"] = "hello session"
15+
return session["a"]
16+
17+
@app.route("/check-session")
18+
def check_session():
19+
return "session: %s" % session["a"]
20+
21+
@app.route("/check-session-database")
22+
def check_session_database():
23+
sessions = app.session_interface.cls.objects.count()
24+
return "sessions: %s" % sessions
25+
26+
27+
def test_setting_session(app):
28+
client = app.test_client()
29+
30+
response = client.get("/")
31+
assert response.status_code == 200
32+
assert response.data.decode("utf-8") == "hello session"
33+
34+
response = client.get("/check-session")
35+
assert response.status_code == 200
36+
assert response.data.decode("utf-8") == "session: hello session"
37+
38+
response = client.get("/check-session-database")
39+
assert response.status_code == 200
40+
assert response.data.decode("utf-8") == "sessions: 1"

0 commit comments

Comments
 (0)