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)
Esempio n. 2
0
	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")