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_getProperList3(self): pagetype = "crisis-refs" key = db.get(db.Key.from_path("Crisis","beslan-school-siege")).key() model = self.handler.getProperModelByKey(pagetype, key) pagetype = "person-refs" keys = self.handler.getProperList(pagetype, model) someModel = Person.get(keys[0]) self.assert_(someModel != None) self.assert_(someModel.name == "Vladimir Putin") self.assert_(someModel.city == "Moscow")