def test_init_bare(self, _mock_generate_request_id):
ctx = context.Context('user', 'tenant')
self.assertEqual(ctx.user, 'user')
self.assertEqual(ctx.tenant, 'tenant')
self.assertEqual(ctx.roles, [])
self.assertEqual(ctx.request_id, 'request_id')
self.assertEqual(ctx.is_admin, False)
self.assertEqual(ctx.session, None)
def test_init_admin(self):
ctx = context.Context('user',
'tenant',
roles=['one', 'aDmIn', 'two'],
request_id='request_id')
self.assertEqual(ctx.user, 'user')
self.assertEqual(ctx.tenant, 'tenant')
self.assertEqual(ctx.roles, ['one', 'aDmIn', 'two'])
self.assertEqual(ctx.request_id, 'request_id')
self.assertEqual(ctx.is_admin, True)
self.assertEqual(ctx.session, None)
def test_elevated_admin(self):
ctx = context.Context('user',
'tenant',
roles=['one', 'aDmIn', 'two'],
request_id='request_id',
is_admin=True)
ctx.session = 'session'
elev_ctx = ctx.elevated()
self.assertNotEqual(id(ctx), id(elev_ctx))
self.assertEqual(elev_ctx.user, 'user')
self.assertEqual(elev_ctx.tenant, 'tenant')
self.assertEqual(elev_ctx.roles, ['one', 'aDmIn', 'two'])
self.assertEqual(elev_ctx.request_id, 'request_id')
self.assertEqual(elev_ctx.is_admin, True)
self.assertEqual(id(elev_ctx.session), id(ctx.session))
def test_to_dict(self):
ctx = context.Context('user',
'tenant',
roles=['one', 'two'],
request_id='request_id',
is_admin=True)
result = ctx.to_dict()
self.assertEqual(
result,
dict(
user='******',
tenant='tenant',
roles=['one', 'two'],
request_id='request_id',
is_admin=True,
))