def test_get_identity(self):
from cleaver.identity.beaker import BeakerSessionProvider
class FakeSession(object):
id = 'ABC123'
environ = {'beaker.session': FakeSession()}
provider = BeakerSessionProvider()
assert provider.get_identity(environ) == 'ABC123'
def test_get_identity_with_custom_environ_key(self):
from cleaver.identity.beaker import BeakerSessionProvider
class FakeSession(object):
id = 'ABC123'
environ = {'beaker.special_key': FakeSession()}
provider = BeakerSessionProvider(environ_key='beaker.special_key')
assert provider.get_identity(environ) == 'ABC123'
