76 lines
1.7 KiB
Python
76 lines
1.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
tests.sessions
|
|
~~~~~~~~~~~~~~
|
|
|
|
Added tests for the sessions.
|
|
|
|
:copyright: (c) 2014 by Armin Ronacher.
|
|
:license: BSD, see LICENSE for more details.
|
|
"""
|
|
import os
|
|
from tempfile import gettempdir
|
|
|
|
from werkzeug.contrib.sessions import FilesystemSessionStore
|
|
|
|
|
|
def test_default_tempdir():
|
|
store = FilesystemSessionStore()
|
|
assert store.path == gettempdir()
|
|
|
|
|
|
def test_basic_fs_sessions(tmpdir):
|
|
store = FilesystemSessionStore(str(tmpdir))
|
|
x = store.new()
|
|
assert x.new
|
|
assert not x.modified
|
|
x['foo'] = [1, 2, 3]
|
|
assert x.modified
|
|
store.save(x)
|
|
|
|
x2 = store.get(x.sid)
|
|
assert not x2.new
|
|
assert not x2.modified
|
|
assert x2 is not x
|
|
assert x2 == x
|
|
x2['test'] = 3
|
|
assert x2.modified
|
|
assert not x2.new
|
|
store.save(x2)
|
|
|
|
x = store.get(x.sid)
|
|
store.delete(x)
|
|
x2 = store.get(x.sid)
|
|
# the session is not new when it was used previously.
|
|
assert not x2.new
|
|
|
|
|
|
def test_non_urandom(tmpdir):
|
|
urandom = os.urandom
|
|
del os.urandom
|
|
try:
|
|
store = FilesystemSessionStore(str(tmpdir))
|
|
store.new()
|
|
finally:
|
|
os.urandom = urandom
|
|
|
|
|
|
def test_renewing_fs_session(tmpdir):
|
|
store = FilesystemSessionStore(str(tmpdir), renew_missing=True)
|
|
x = store.new()
|
|
store.save(x)
|
|
store.delete(x)
|
|
x2 = store.get(x.sid)
|
|
assert x2.new
|
|
|
|
|
|
def test_fs_session_lising(tmpdir):
|
|
store = FilesystemSessionStore(str(tmpdir), renew_missing=True)
|
|
sessions = set()
|
|
for x in range(10):
|
|
sess = store.new()
|
|
store.save(sess)
|
|
sessions.add(sess.sid)
|
|
|
|
listed_sessions = set(store.list())
|
|
assert sessions == listed_sessions
|