def append(self, xml_element): self.num_items += 1 if isinstance(xml_element, bytes): xml_element, num = get_cached_items_with_count(xml_element) self.num_items += num - 1 self.response_body.write(xml_element) else: self.response_body.write(xml_util.tostring(xml_element))
def append(self, xml_element): self.num_items += 1 if isinstance(xml_element, bytes): xml_element, num = get_cached_items_with_count(xml_element) self.num_items += num - 1 self.response_body.write(xml_element) else: self.response_body.write(xml_util.tostring(xml_element))
def test_get_cached_items_with_count_bad_format(self): XML = b'<!--items=2JUNK--><fixture>...</fixture>' xml, num = mod.get_cached_items_with_count(XML) self.assertEqual(xml, XML) self.assertEqual(num, 1)
def test_get_cached_items_with_count(self): XML = b'<!--items=42--><fixture>...</fixture>' xml, num = mod.get_cached_items_with_count(XML) self.assertEqual(xml, b'<fixture>...</fixture>') self.assertEqual(num, 42)
def test_get_cached_items_with_count_no_count(self): XML = b'<fixture />' xml, num = mod.get_cached_items_with_count(XML) self.assertEqual(xml, XML) self.assertEqual(num, 1)