116 lines
4.1 KiB
Python
116 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])
|