family: fix family members retrieval (#22585)

This commit is contained in:
Serghei Mihai 2018-03-22 17:43:12 +01:00
parent 9539f8e432
commit cc57162f8b
2 changed files with 14 additions and 4 deletions

View File

@ -299,13 +299,17 @@ class GenericFamily(BaseResource):
@endpoint(name='family', perm='can_access', pattern='^adults/$')
def adults_infos(self, request, NameID):
data = self.family_infos(request, NameID)
return {'data': data['data']['adults']}
family_infos = self.family_infos(request, NameID)['data']
if not family_infos:
return {'data': []}
return {'data': family_infos['adults']}
@endpoint(name='family', perm='can_access', pattern='^children/$')
def children_infos(self, request, NameID, **kwargs):
data = self.family_infos(request, NameID)
return {'data': data['data']['children']}
family_infos = self.family_infos(request, NameID)['data']
if not family_infos:
return {'data': []}
return {'data': family_infos['children']}
def get_invoices(self, NameID, paid=False):
family = self.get_family_by_nameid(NameID)

View File

@ -105,6 +105,12 @@ def test_family_members(app, resource):
assert person['city'] is not None
assert person['address'] is not None
r = app.get(reverse('generic-endpoint', kwargs={'connector': 'family',
'slug': resource.slug, 'endpoint': 'family', 'rest': '%s/' % typ}),
params={'NameID': 'unknown', 'apikey': API_KEY})
data = r.json['data']
assert len(data) == 0
def test_get_family_invoices(app, resource):
r = app.get(reverse('generic-endpoint', kwargs={'connector': 'family',
'slug': resource.slug, 'endpoint': 'family', 'rest': 'link/'}),