def test_write_element_raises_on_dup_itemuri(self):
"""ZipFileSystem.write_element() raises on duplicate itemURI"""
# setup -----------------------
elm = etree.fromstring(self.xml_in)
itemURI = "/ppt/test.xml"
zipfs = ZipFileSystem(test_save_pptx_path, "w")
# exercise --------------------
zipfs.write_element(elm, itemURI)
# verify ----------------------
with self.assertRaises(DuplicateKeyError):
zipfs.write_element(elm, itemURI)
def test_write_element_round_trips(self):
"""ZipFileSystem.write_element() round-trips intact"""
# setup -----------------------
elm = etree.fromstring(self.xml_in)
itemURI = "/ppt/test.xml"
zipfs = ZipFileSystem(test_save_pptx_path, "w")
# exercise --------------------
zipfs.write_element(elm, itemURI)
# verify ----------------------
stream = zipfs.getstream(itemURI)
xml_out = stream.read()
stream.close()
expected = self.xml_out
actual = xml_out
msg = "expected \n%s\n, got\n%s" % (expected, actual)
self.assertEqual(expected, actual, msg)
