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
session.finalize()
self.assertIsNone(session.local)
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
session.finalize()
self.assertEqual(session.local, None)
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 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()
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
session.finalize()
self.assertEqual(session.local, None)
conn = memcache.Client(["127.0.0.1:11211"], debug=0)
session = Session(expiration=10, storage=conn)
request.cookies[a] = b
session.on_request()
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"
session.on_request()
self.assertEqual(session.get("key"), None)
finally:
if memcache_process is None:
print("memcached not availabl, test skipped")
elif memcache_process:
memcache_process.kill()
