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]})