def test_getIds(self):
people = [{"ID": "1","Last":"Batman"},
{"ID": "2","Last":"Shrek"},
{"ID": "3","Last":"Batman"}]
ids = ["1","3"]
#code under test
response = functionality.getIdsByLastName(people, 'Batman')
self.assertEqual(response, ids, 'List does not match expected ids')
def getIdsByLastNamePath(lastname):
'''
Returns Json block of ids found for the given last name
Using path params
'''
if lastname == None:
return 'Must provide path param "lastname"'
else:
data = functionality.getIdsByLastName(GLOB['people'], lastname)
return jsonify(**{'ids': data[0]})
def getIdsByLastName():
'''
Returns Json block of ids found for the given last name
Using Query Params for this call
'''
lastName = request.args.get('lastname')
if lastName == None:
return 'Must provide query param "lastname"'
else:
data = functionality.getIdsByLastName(GLOB['people'], lastName)
return jsonify(**{'ids': data[0]})
