def test_future_simple(self):
d = Doc('a')
f = Future()
d.put(f)
self.assertEqual(d.to_string(),
"""<?xml version='1.0' encoding='utf-8'?>\n<a/>""")
f.set_result('test')
self.assertEqual(
d.to_string(),
"""<?xml version='1.0' encoding='utf-8'?>\n<a>test</a>""")
def test_other_types(self):
a = Doc('a')
a.put(1)
a.put(2.0)
a.put((3, 4, 5))
self.assertEqual(a.to_string(), """<?xml version='1.0' encoding='utf-8'?>\n<a>12.0(3, 4, 5)</a>""")
def test_root_node(self):
d = Doc(root_node=etree.Element('doc'))
d.put(etree.Element('test1'))
self.assertEqual(
d.to_string(),
"""<?xml version='1.0' encoding='utf-8'?>\n<doc><test1/></doc>""")
def test_doc_nested(self):
a = Doc('a')
b = Doc('b')
b.put('test')
a.put(b)
self.assertEqual(
a.to_string(),
"""<?xml version='1.0' encoding='utf-8'?>\n<a><b>test</b></a>""")
def test_future_etree_element(self):
d = Doc('a')
f = Future()
f.set_result(etree.Element('b'))
d.put(f)
self.assertEqual(
d.to_string(),
"""<?xml version='1.0' encoding='utf-8'?>\n<a><b/></a>""")
def test_other_types(self):
a = Doc('a')
a.put(1)
a.put(2.0)
a.put((3, 4, 5))
self.assertEqual(
a.to_string(),
"""<?xml version='1.0' encoding='utf-8'?>\n<a>12.0(3, 4, 5)</a>""")
def test_future_list(self):
d = Doc('a')
f = Future()
f.set_result([etree.Comment('ccc'), etree.Element('bbb')])
d.put(f)
self.assertEqual(
d.to_string(),
"""<?xml version='1.0' encoding='utf-8'?>\n<a><!--ccc--><bbb/></a>"""
)
def test_nodes_and_text(self):
a = Doc('a')
a.put('1')
a.put(Doc('b'))
a.put('2')
a.put(Doc('c'))
a.put('3')
self.assertEqual(
a.to_string(),
"""<?xml version='1.0' encoding='utf-8'?>\n<a>1<b/>2<c/>3</a>""")
def test_simple(self):
d = Doc('a')
self.assertTrue(d.is_empty())
d.put('test')
self.assertFalse(d.is_empty())
self.assertEqual(
d.to_string(),
"""<?xml version='1.0' encoding='utf-8'?>\n<a>test</a>""")
def test_root_node_doc(self):
d1 = Doc('a')
d1.put(etree.Comment('1'))
d2 = Doc(root_node=d1)
d2.put(etree.Comment('2'))
self.assertEqual(
d2.to_string(),
"""<?xml version='1.0' encoding='utf-8'?>\n<a><!--1--><!--2--></a>"""
)
def test_failed_future(self):
d = Doc('a')
f = Future()
result = RequestResult()
result.set_exception(
FailedRequestException(reason='error', code='code'))
f.set_result(result)
d.put(f)
self.assertEqual(
d.to_string(), """<?xml version='1.0' encoding='utf-8'?>\n"""
"""<a><error reason="error" code="code"/></a>""")
def test_serializable(self):
class Serializable(object):
def __init__(self, tag, value):
self.tag = tag
self.value = value
def to_etree_element(self):
result = etree.Element(self.tag)
result.text = self.value
return result
a = Doc('a')
a.put(Serializable('testNode', 'vally'))
self.assertEqual(a.to_string(), """<?xml version='1.0' encoding='utf-8'?>\n<a><testNode>vally</testNode></a>""")
def test_serializable(self):
class Serializable(object):
def __init__(self, tag, value):
self.tag = tag
self.value = value
def to_etree_element(self):
result = etree.Element(self.tag)
result.text = self.value
return result
a = Doc('a')
a.put(Serializable('testNode', 'vally'))
self.assertEqual(
a.to_string(),
"""<?xml version='1.0' encoding='utf-8'?>\n<a><testNode>vally</testNode></a>"""
)