def test_session_in_db(self):
request.app_name = "myapp"
db = DAL("sqlite:memory")
session = Session(secret="a", expiration=10, storage=DBStore(db))
request.cookies.clear()
session.on_request()
session["key"] = "value"
cookie_name = session.local.session_cookie_name
session.on_success(200)
a, b = str(
response._cookies)[len("Set-Cookie: "):].split(";")[0].split(
"=", 1)
request.cookies[a] = b
_before_request()
with pytest.raises(RuntimeError) as err:
session.local
self.assertTrue('py4web hint' in str(err.value))
session = Session(expiration=10, storage=DBStore(db))
request.cookies[a] = b
session.on_request()
self.assertEqual(session.get("key"), "value")
session = Session(expiration=10, storage=DBStore(db))
request.cookies[a] = "wrong_cookie"
session.on_request()
self.assertEqual(session.get("key"), None)
def test_session(self):
request.app_name = "myapp"
session = Session(secret="a", expiration=10)
session.on_request()
session["key"] = "value"
cookie_name = session.local.session_cookie_name
session.on_success(200)
a, b = str(
response._cookies)[len("Set-Cookie: "):].split(";")[0].split(
"=", 1)
request.cookies[a] = b
_before_request()
with pytest.raises(RuntimeError) as err:
session.local
self.assertTrue('py4web hint' in str(err.value))
session = Session(secret="b", expiration=10)
request.cookies[a] = b
session.on_request()
self.assertEqual(session.get("key"), None)
session = Session(secret="a", expiration=10)
request.cookies[a] = b
session.on_request()
self.assertEqual(session.get("key"), "value")
def on_request(self, keep_session=False):
storage = self.session._safe_local
# mimic before_request bottle-hook
_before_request()
# mimic action.uses()
self.session.initialize()
self.auth.flash.on_request()
self.auth.on_request()
if keep_session:
self.session._safe_local = storage
def setUp(self):
os.environ["PY4WEB_APPS_FOLDER"] = "apps"
_before_request() # mimic before_request bottle-hook
self.db = DAL("sqlite:memory")
self.session = Session(secret="a", expiration=10)
self.session.initialize()
self.auth = Auth(self.session,
self.db,
define_tables=True,
password_complexity=None)
self.auth.enable()
self.auth.action = self.action
request.app_name = "_scaffold"
def test_session_in_memcache(self):
memcache_process = None
try:
memcache_process = subprocess.Popen(["memcached", "-p", "11211"])
time.sleep(1)
request.app_name = "myapp"
conn = memcache.Client(["127.0.0.1:11211"], debug=0)
session = Session(secret="a", expiration=10, storage=conn)
request.cookies.clear()
context = dict(status=200)
session.on_request(context)
session["key"] = "value"
cookie_name = session.local.session_cookie_name
session.on_success(context)
a, b = (
str(response._cookies)[len("Set-Cookie: ") :]
.split(";")[0]
.split("=", 1)
)
request.cookies[a] = b
_before_request()
with pytest.raises(RuntimeError) as err:
session.local
self.assertTrue('py4web hint' in str(err.value))
conn = memcache.Client(["127.0.0.1:11211"], debug=0)
session = Session(expiration=10, storage=conn)
request.cookies[a] = b
context = dict(status=200)
session.on_request(context)
self.assertEqual(session.get("key"), "value")
conn = memcache.Client(["127.0.0.1:11211"], debug=0)
session = Session(expiration=10, storage=conn)
request.cookies[a] = "wrong_cookie"
context = dict(status=200)
session.on_request(context)
self.assertEqual(session.get("key"), None)
finally:
if memcache_process is None:
print("memcached not availabl, test skipped")
elif memcache_process:
memcache_process.kill()
def setUp(self):
request.environ["wsgi.input"] = io.StringIO()
_before_request() # mimic before_request bottle-hook
request.cookies.clear()
response._cookies = ""