add 'kind' field to user and group models

This commit is contained in:
Min RK
2017-01-26 11:48:20 +01:00
parent d0813cc736
commit 4b6c58292b
2 changed files with 7 additions and 0 deletions

View File

@@ -89,6 +89,7 @@ class APIHandler(BaseHandler):
def user_model(self, user): def user_model(self, user):
"""Get the JSON model for a User object""" """Get the JSON model for a User object"""
model = { model = {
'kind': 'user',
'name': user.name, 'name': user.name,
'admin': user.admin, 'admin': user.admin,
'groups': [ g.name for g in user.groups ], 'groups': [ g.name for g in user.groups ],
@@ -105,6 +106,7 @@ class APIHandler(BaseHandler):
def group_model(self, group): def group_model(self, group):
"""Get the JSON model for a Group object""" """Get the JSON model for a Group object"""
return { return {
'kind': 'group',
'name': group.name, 'name': group.name,
'users': [ u.name for u in group.users ], 'users': [ u.name for u in group.users ],
} }

View File

@@ -168,6 +168,7 @@ def test_get_users(app):
u.pop('last_activity') u.pop('last_activity')
assert users == [ assert users == [
{ {
'kind': 'user',
'name': 'admin', 'name': 'admin',
'groups': [], 'groups': [],
'admin': True, 'admin': True,
@@ -175,6 +176,7 @@ def test_get_users(app):
'pending': None, 'pending': None,
}, },
{ {
'kind': 'user',
'name': 'user', 'name': 'user',
'groups': [], 'groups': [],
'admin': False, 'admin': False,
@@ -209,6 +211,7 @@ def test_get_user(app):
user = r.json() user = r.json()
user.pop('last_activity') user.pop('last_activity')
assert user == { assert user == {
'kind': 'user',
'name': name, 'name': name,
'groups': [], 'groups': [],
'admin': False, 'admin': False,
@@ -570,6 +573,7 @@ def test_groups_list(app):
r.raise_for_status() r.raise_for_status()
reply = r.json() reply = r.json()
assert reply == [{ assert reply == [{
'kind': 'group',
'name': 'alphaflight', 'name': 'alphaflight',
'users': [] 'users': []
}] }]
@@ -589,6 +593,7 @@ def test_group_get(app):
r.raise_for_status() r.raise_for_status()
reply = r.json() reply = r.json()
assert reply == { assert reply == {
'kind': 'group',
'name': 'alphaflight', 'name': 'alphaflight',
'users': ['sasquatch'] 'users': ['sasquatch']
} }