test_debug_message.py
No OneTemporary
Actions

Subscribers

None

File Metadata

	import unittest
	from os import sep, path
	import lxml.etree as ET
	import sys

	sys.path.append('svgscripts')
	from datatypes.debug_message import DebugMessage

	class TestDebugMessage(unittest.TestCase):
	def test_init(self):
	debug_tag = 'end'
	dmsg = DebugMessage(message='test', tag=debug_tag)
	self.assertEqual(dmsg.message, 'test')
	self.assertEqual(dmsg.tag, debug_tag)

	def test_attach_object_to_tree(self):
	dmsg = DebugMessage(message='test')
	empty_tree = ET.ElementTree(ET.Element('page'))
	dmsg.attach_object_to_tree(empty_tree)
	for node in empty_tree.getroot().xpath('//' + dmsg.tag):
	self.assertEqual(node.get('id'), '0')
	self.assertEqual(node.get('message'), 'test')

	def test_init_node(self):
	dmsg = DebugMessage(message='test')
	empty_tree = ET.ElementTree(ET.Element('page'))
	dmsg.attach_object_to_tree(empty_tree)
	dmsg2 = DebugMessage(node=empty_tree.getroot().find('./' + dmsg.tag))
	self.assertEqual(dmsg2.id, dmsg.id)
	self.assertEqual(dmsg2.message, dmsg.message)

	if __name__ == "__main__":
	unittest.main()