115 lines
4.1 KiB
Python
115 lines
4.1 KiB
Python
|
# vim: set fileencoding=utf-8
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
from bemani.protocol.xml import XmlDecoder
|
||
|
|
|
||
|
|
|
||
|
|
class TestXmlDecoder(unittest.TestCase):
|
||
|
|
|
||
|
|
def test_detect_encoding(self) -> None:
|
||
|
|
xml = XmlDecoder(b'<?xml version="1.0" encoding="utf-8"?>', 'ascii')
|
||
|
|
tree = xml.get_tree()
|
||
|
|
|
||
|
|
self.assertEqual(xml.encoding, 'utf-8')
|
||
|
|
self.assertEqual(tree, None)
|
||
|
|
|
||
|
|
xml = XmlDecoder(b'<?xml\nversion = "1.0"\tencoding = "utf-8"?>', 'ascii')
|
||
|
|
tree = xml.get_tree()
|
||
|
|
|
||
|
|
self.assertEqual(xml.encoding, 'utf-8')
|
||
|
|
self.assertEqual(tree, None)
|
||
|
|
|
||
|
|
def test_decode_void(self) -> None:
|
||
|
|
xml = XmlDecoder(b'<node></node>', 'ascii')
|
||
|
|
tree = xml.get_tree()
|
||
|
|
|
||
|
|
self.assertEqual(tree.children, [])
|
||
|
|
self.assertEqual(tree.name, 'node')
|
||
|
|
self.assertEqual(tree.attributes, {})
|
||
|
|
self.assertEqual(tree.data_type, 'void')
|
||
|
|
self.assertEqual(tree.value, None)
|
||
|
|
|
||
|
|
xml = XmlDecoder(b'<node />', 'ascii')
|
||
|
|
tree = xml.get_tree()
|
||
|
|
|
||
|
|
self.assertEqual(tree.children, [])
|
||
|
|
self.assertEqual(tree.name, 'node')
|
||
|
|
self.assertEqual(tree.attributes, {})
|
||
|
|
self.assertEqual(tree.data_type, 'void')
|
||
|
|
self.assertEqual(tree.value, None)
|
||
|
|
|
||
|
|
def test_decode_attributes(self) -> None:
|
||
|
|
xml = XmlDecoder(b'<node attr1="foo" attr2="bar"></node>', 'ascii')
|
||
|
|
tree = xml.get_tree()
|
||
|
|
|
||
|
|
self.assertEqual(tree.children, [])
|
||
|
|
self.assertEqual(tree.name, 'node')
|
||
|
|
self.assertEqual(tree.attributes, {'attr1': 'foo', 'attr2': 'bar'})
|
||
|
|
self.assertEqual(tree.data_type, 'void')
|
||
|
|
self.assertEqual(tree.value, None)
|
||
|
|
|
||
|
|
xml = XmlDecoder(b'<node attr1="foo" attr2="bar" />', 'ascii')
|
||
|
|
tree = xml.get_tree()
|
||
|
|
|
||
|
|
self.assertEqual(tree.children, [])
|
||
|
|
self.assertEqual(tree.name, 'node')
|
||
|
|
self.assertEqual(tree.attributes, {'attr1': 'foo', 'attr2': 'bar'})
|
||
|
|
self.assertEqual(tree.data_type, 'void')
|
||
|
|
self.assertEqual(tree.value, None)
|
||
|
|
|
||
|
|
xml = XmlDecoder(b'<node\nattr1="foo"\tattr2="bar"/>', 'ascii')
|
||
|
|
tree = xml.get_tree()
|
||
|
|
|
||
|
|
self.assertEqual(tree.children, [])
|
||
|
|
self.assertEqual(tree.name, 'node')
|
||
|
|
self.assertEqual(tree.attributes, {'attr1': 'foo', 'attr2': 'bar'})
|
||
|
|
self.assertEqual(tree.data_type, 'void')
|
||
|
|
self.assertEqual(tree.value, None)
|
||
|
|
|
||
|
|
def test_decode_bin(self) -> None:
|
||
|
|
xml = XmlDecoder(b'<node __type="bin">DEADBEEF</node>', 'ascii')
|
||
|
|
tree = xml.get_tree()
|
||
|
|
|
||
|
|
self.assertEqual(tree.children, [])
|
||
|
|
self.assertEqual(tree.name, 'node')
|
||
|
|
self.assertEqual(tree.attributes, {})
|
||
|
|
self.assertEqual(tree.data_type, 'bin')
|
||
|
|
self.assertEqual(tree.value, b'\xDE\xAD\xBE\xEF')
|
||
|
|
|
||
|
|
xml = XmlDecoder(b'<node __type="bin">\nDEADBEEF\n</node>', 'ascii')
|
||
|
|
tree = xml.get_tree()
|
||
|
|
|
||
|
|
self.assertEqual(tree.children, [])
|
||
|
|
self.assertEqual(tree.name, 'node')
|
||
|
|
self.assertEqual(tree.attributes, {})
|
||
|
|
self.assertEqual(tree.data_type, 'bin')
|
||
|
|
self.assertEqual(tree.value, b'\xDE\xAD\xBE\xEF')
|
||
|
|
|
||
|
|
xml = XmlDecoder(b'<node __type="bin"> D E A D B E E F </node>', 'ascii')
|
||
|
|
tree = xml.get_tree()
|
||
|
|
|
||
|
|
self.assertEqual(tree.children, [])
|
||
|
|
self.assertEqual(tree.name, 'node')
|
||
|
|
self.assertEqual(tree.attributes, {})
|
||
|
|
self.assertEqual(tree.data_type, 'bin')
|
||
|
|
self.assertEqual(tree.value, b'\xDE\xAD\xBE\xEF')
|
||
|
|
|
||
|
|
def test_decode_array(self) -> None:
|
||
|
|
xml = XmlDecoder(b'<node __type="u32" __count="4">1 2 3 4</node>', 'ascii')
|
||
|
|
tree = xml.get_tree()
|
||
|
|
|
||
|
|
self.assertEqual(tree.children, [])
|
||
|
|
self.assertEqual(tree.name, 'node')
|
||
|
|
self.assertEqual(tree.attributes, {})
|
||
|
|
self.assertEqual(tree.data_type, 'u32')
|
||
|
|
self.assertEqual(tree.value, [1, 2, 3, 4])
|
||
|
|
|
||
|
|
xml = XmlDecoder(b'<node __type="u32" __count="4">\n1\n2\n3\n4\n</node>', 'ascii')
|
||
|
|
tree = xml.get_tree()
|
||
|
|
|
||
|
|
self.assertEqual(tree.children, [])
|
||
|
|
self.assertEqual(tree.name, 'node')
|
||
|
|
self.assertEqual(tree.attributes, {})
|
||
|
|
self.assertEqual(tree.data_type, 'u32')
|
||
|
|
self.assertEqual(tree.value, [1, 2, 3, 4])
|