File tree Expand file tree Collapse file tree 1 file changed +40
-0
lines changed
Expand file tree Collapse file tree 1 file changed +40
-0
lines changed Original file line number Diff line number Diff line change 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"
You can’t perform that action at this time.
0 commit comments