def test_session_flash_data(self):
for driver in ('memory', 'cookie'):
session = self.app.make('SessionManager').driver(driver)
session._session = {}
session.flash('flash_username', 'pep')
session.flash('flash_password', 'secret')
assert session.get('flash_username') == 'pep'
assert session.get('flash_password') == 'secret'
def test_session_get_all_data(self):
for driver in ('memory', 'cookie'):
session = self.app.make('SessionManager').driver(driver)
session.environ['REMOTE_ADDR'] = 'get.all.data'
session.set('username', 'pep')
session.flash('password', 'secret')
assert session.all() == {'username': '******', 'password': '******'}
def test_session_flash_data_serializes_dict(self):
for driver in ('cookie', 'memory'):
session = self.app.make('SessionManager').driver(driver)
session._session = {}
session.flash('flash_dict', {'id': 1})
session.set('get_dict', {'id': 1})
assert session.get('flash_dict') == {'id': 1}
assert session.get('get_dict') == {'id': 1}
def test_reset_flash_session_driver(self):
session = self.app.make('SessionManager').driver('cookie')
session.flash('flash_', 'test_pep')
session.reset(flash_only=True)
assert session.get('flash_') is None
def test_reset_flash_session_memory(self):
session = self.app.make('SessionManager').driver('memory')
session.flash('flash_', 'test_pep')
session.reset(flash_only=True)
self.assertIsNone(session.get('flash_'))
def test_session_flash_data_serializes_list(self):
for driver in ('cookie', 'memory'):
session = self.app.make('SessionManager').driver(driver)
session._session = {}
session.flash('flash_dict', [1, 2, 3])
self.assertEqual(session.get('flash_dict'), [1, 2, 3])
