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_getProperList1(self): pagetype = "person-refs" key = db.get(db.Key.from_path("Person","george-w-bush")).key() model = self.handler.getProperModelByKey(pagetype, key) pagetype = "organization-refs" keys = self.handler.getProperList(pagetype, model) someModel = Organization.get(keys[0]) self.assert_(someModel != None) self.assert_(someModel.name == "New York City Fire Department") self.assert_(someModel.city == "Staten Island")