def getProperModelByKey(self,pagetype, link): if (pagetype == "person-refs"): return Person.get(link) elif (pagetype == "organization-refs"): return Organization.get(link) else: return Crisis.get(link)
def test_getProperList2(self): pagetype = "organization-refs" key = db.get(db.Key.from_path("Organization","nyfd")).key() model = self.handler.getProperModelByKey(pagetype, key) pagetype = "crisis-refs" keys = self.handler.getProperList(pagetype, model) someModel = Crisis.get(keys[0]) self.assert_(someModel != None) self.assert_(someModel.city == "New York")