def generate_response(self, results, queries, osHostURL, context): ospath = generate_autodiscovery_path(osHostURL, None, self.extension, rel = None) #Generates the ATOM document atomdoc = createAtomDocument(ospath + "atom", results.title, results.updated) #Generate feed's links self._generate_feed_links(atomdoc, ospath, results) #Inserts the OpenSearchResponse elements createOpenSearchRespose(atomdoc, results.totalResult, results.startIndex, results.count, queries) self.generateEntries(atomdoc, results.subresult, ospath) reparsed = minidom.parseString(tostring(atomdoc)) return reparsed.toprettyxml(indent=" ")
def entry_test(self): atom = createAtom() title = createTitle(root=atom, body="testEntry") iid = createID(1, root=atom) update = createUpdated("2012-0619T21:02:00.626Z", root=atom) entry = createEntry(iid, title, update, root=atom) atom.append(entry) self.assertEqual( tostring(atom), '<feed xmlns="http://www.w3.org/2005/Atom">\ <entry><id>1</id><title>testEntry</title><updated>2012-0619T21:02:00.626Z</updated></entry></feed>', ) root = Element("myCustomTag") title = createTitle(root=root, body="testEntry") iid = createID(1, root=root) update = createUpdated("2012-0619T21:02:00.626Z", root=atom) atom = createEntry(iid, title, update, root=root) root.append(atom) self.assertEqual( tostring(root), '<myCustomTag xmlns:atom="http://www.w3.org/2005/Atom">\ <atom:entry><atom:id>1</atom:id><atom:title>testEntry</atom:title><updated>2012-0619T21:02:00.626Z</updated>\ </atom:entry></myCustomTag>', ) root = createAtomDocument(1, "atomTitle", "2012-0619T21:02:00.626Z") iid = createID(2, root=root) title = createTitle(root=root, body="testEntry") update = createUpdated("2012-0619T21:02:00.626Z", root=root) entry = createEntry(iid, title, update, root=root) root.append(entry) self.assertEqual( tostring(root), '<feed xmlns="http://www.w3.org/2005/Atom"><id>1</id>\ <title>atomTitle</title><updated>2012-0619T21:02:00.626Z</updated><entry><id>2</id>\ <title>testEntry</title><updated>2012-0619T21:02:00.626Z</updated></entry></feed>', "Error", )