示例#1
0
    def test_get_allowed_service(self):
        token = generate_token()

        with self.assertRaises(ValueError):
            get_allowed_service(None)
        
        result = get_allowed_service(token)
        self.assertIsNotNone(result)
        self.assertEqual(result, 'admin')

        with self.assertRaises(ValueError):
            get_allowed_service('Is.Not_A_Valid_Token')
示例#2
0
def init_tenant_context(request, db):
    try:
        token = request.headers['authorization']
    except KeyError:
        raise HTTPRequestError(401, "No authorization token has been supplied")

    tenant = get_allowed_service(token)
    init_tenant(tenant, db)
    return tenant
示例#3
0
def init_tenant_context(token, db):

    tenant = get_allowed_service(token)
    init_tenant(tenant, db)
    return tenant