def setUp(self):
self.x = XMLUploadWorker()
self.handler = ExportHandler()
xsdFileInput = open("test/test.xsd", 'r')
xmlfile = open('test/test2.xml','r')
xmleval = xsv.parseAndValidateString(re.sub('&', '&', xmlfile.read()), xsdFileInput.read(), xmlIfClass=xsv.XMLIF_ELEMENTTREE)
etree = xmleval.getTree()
root = etree.getroot()
self.x.createModels(root.find("people"))
self.x.createModels(root.find("crises"))
self.x.createModels(root.find("organizations"))
self.x.createLinks(root.find("crises"), False)
self.x.createLinks(root.find("people"), False)
self.x.createLinks(root.find("organizations"), False)
class TestExportHandler(unittest.TestCase):
def setUp(self):
self.x = XMLUploadWorker()
self.handler = ExportHandler()
xsdFileInput = open("test/test.xsd", 'r')
xmlfile = open('test/test2.xml','r')
xmleval = xsv.parseAndValidateString(re.sub('&', '&', xmlfile.read()), xsdFileInput.read(), xmlIfClass=xsv.XMLIF_ELEMENTTREE)
etree = xmleval.getTree()
root = etree.getroot()
self.x.createModels(root.find("people"))
self.x.createModels(root.find("crises"))
self.x.createModels(root.find("organizations"))
self.x.createLinks(root.find("crises"), False)
self.x.createLinks(root.find("people"), False)
self.x.createLinks(root.find("organizations"), False)
def test_getProperModelByKey1(self):
pagetype = "person-refs"
key = db.get(db.Key.from_path("Person","george-w-bush")).key()
model = self.handler.getProperModelByKey(pagetype, key)
self.assert_(model != None)
self.assert_(model.name == "George W. Bush")
self.assert_(model.city == "Washington D.C.")
def test_getProperModelByKey2(self):
pagetype = "organization-refs"
key = db.get(db.Key.from_path("Organization","nyfd")).key()
model = self.handler.getProperModelByKey(pagetype, key)
self.assert_(model != None)
self.assert_(model.name == "New York City Fire Department")
self.assert_(model.city == "Staten Island")
def test_getProperModelByKey3(self):
pagetype = "crisis-refs"
key = db.get(db.Key.from_path("Crisis","beslan-school-siege")).key()
model = self.handler.getProperModelByKey(pagetype, key)
self.assert_(model != None)
self.assert_(model.name == "Beslan School Siege")
self.assert_(model.city == "Beslan")
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")
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")
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")
def test_organization_details(self) :
pagetype = "organization-refs"
key = db.get(db.Key.from_path("Organization","nyfd")).key()
model = self.handler.getProperModelByKey(pagetype, key)
element = Element("main")
self.handler.organization_details(element, model)
self.assert_(element.find("address") != None)
self.assert_(element.find("address").text == "New York City, NY, United States")
self.assert_(element.find("email") != None)
self.assert_(element.find("email").text == "*****@*****.**")
self.assert_(element.find("phone") != None)
self.assert_(element.find("phone").text)
def test_crisis_details(self) :
pagetype = "crisis-refs"
key = db.get(db.Key.from_path("Crisis","beslan-school-siege")).key()
model = self.handler.getProperModelByKey(pagetype, key)
element = Element("main")
self.handler.crisis_details(element, model)
impact = element.find("human-impact")
self.assert_(element.find("start-date") != None)
self.assert_(element.find("start-date").text == "2004-09-01T09:00:00")
self.assert_(impact.find("deaths") != None)
self.assert_(impact.find("deaths").text == "380")
self.assert_(impact.find("missing") != None)
self.assert_(impact.find("missing").text == "0")
self.assert_(impact.find("injured") != None)
self.assert_(impact.find("injured").text == "783")
self.assert_(impact.find("displaced") != None)
self.assert_(impact.find("displaced").text == "0")
