mirror of
https://github.com/xforcered/SoaPy
synced 2026-06-21 14:14:03 +00:00
shellph1sh
125 lines
4.2 KiB
Python
125 lines
4.2 KiB
Python
import unittest
|
|
|
|
from src.encoder.records import record, dump_records, print_records
|
|
from src.encoder.records.text import Chars8TextRecord
|
|
|
|
import textwrap
|
|
|
|
from io import BytesIO, StringIO
|
|
|
|
from base64 import b64decode
|
|
|
|
|
|
class TestRecord(unittest.TestCase):
|
|
def test_collect_records(self):
|
|
"""All the record types should be assembled
|
|
when the Record object is initalized
|
|
"""
|
|
r = record()
|
|
self.assertEqual(len(r.records), 150)
|
|
|
|
def test_singlton_records(self):
|
|
"""There should only be one version
|
|
of the records dictionary around for multiple
|
|
Record objects
|
|
"""
|
|
|
|
r1 = record()
|
|
r2 = record()
|
|
|
|
self.assertTrue(r1.records is r2.records)
|
|
|
|
def test_parse_bin_to_record_type(self):
|
|
value = b"A\x01a\x04test\x01"
|
|
r = record.parse(BytesIO(value))
|
|
self.assertEqual(r[0].type, 0x41)
|
|
|
|
def test_parse_bin_to_record_value(self):
|
|
value = b"A\x01a\x04test\x01"
|
|
r = record.parse(BytesIO(value))
|
|
self.assertEqual(str(r[0]), "<a:test>")
|
|
|
|
def test_parse_Chars8TextRecord_value(self):
|
|
value = b64decode("CnhzZDpzdHJpbmeZ")
|
|
expected_result = "xsd:string"
|
|
r = Chars8TextRecord.parse(BytesIO(value))
|
|
self.assertEqual(r.value, expected_result)
|
|
|
|
def test_to_int_complex2(self):
|
|
"""This caused breaking issues
|
|
by miscalculating the size of some of the elements
|
|
"""
|
|
value = b64decode(
|
|
"mAp4c2Q6c3RyaW5nmSQ0MDg5MTk4Yi00NmZkLTQyZjEtYmM1YS03NzA5ZGZlZmUzYzcBQQZhZGRhdGELZGVzY3JpcHRpb24ECkxkYXBTeW50YXiYDVVuaWNvZGVTdHJpbmdBAmFkBXZhbHVlBQN4c2kEdHlwZZg="
|
|
)
|
|
value = BytesIO(value)
|
|
value.read(1)
|
|
result = Chars8TextRecord.parse(value)
|
|
expected = "xsd:string"
|
|
|
|
self.assertEqual(result.value, expected)
|
|
|
|
|
|
class TestDumpRecords(unittest.TestCase):
|
|
def test_parse_bin_to_record_dump(self):
|
|
value = b"A\x01a\x04test\x01"
|
|
r = record.parse(BytesIO(value))
|
|
|
|
s = dump_records(r)
|
|
|
|
self.assertEqual(s, b"A\x01a\x04test\x01")
|
|
|
|
|
|
class TestPrintRecords(unittest.TestCase):
|
|
def test_dict_basic(self):
|
|
value = b"\x56\x08\x01"
|
|
expected_result = "<s:Header></s:Header>"
|
|
r = record.parse(BytesIO(value))
|
|
result = print_records(r)
|
|
self.assertEqual(result, expected_result)
|
|
|
|
def test_dict_complex(self):
|
|
value = b"\x44\x0a\x1e\x00\x82\x99\x06\x61\x63\x74\x69\x6f\x6e"
|
|
expected_result = '<a:Action s:mustUnderstand="1">action</a:Action>'
|
|
|
|
r = record.parse(BytesIO(value))
|
|
result = print_records(r)
|
|
self.assertEqual(result.replace("\r\n", ""), expected_result)
|
|
|
|
def test_dict_xmlns_complex(self):
|
|
value = b"\x56\x02\x0b\x01\x61\x06\x0b\x01\x73\x04\x01"
|
|
expected_result = '<s:Envelope xmlns:a="http://www.w3.org/2005/08/addressing" xmlns:s="http://www.w3.org/2003/05/soap-envelope"></s:Envelope>'
|
|
|
|
r = record.parse(BytesIO(value))
|
|
result = print_records(r)
|
|
self.assertEqual(result.replace("\r\n", ""), expected_result)
|
|
|
|
def test_basic(self):
|
|
value = b"\x40\x09\x49\x6e\x76\x65\x6e\x74\x6f\x72\x79\x01"
|
|
expected_result = "<Inventory></Inventory>"
|
|
|
|
r = record.parse(BytesIO(value))
|
|
result = print_records(r)
|
|
self.assertEqual(result.replace("\r\n", ""), expected_result)
|
|
|
|
def test_zerowithend_complex(self):
|
|
value = b"\x40\x09\x49\x6e\x76\x65\x6e\x74\x6f\x72\x79\x81"
|
|
expected_result = "<Inventory>0</Inventory>"
|
|
|
|
r = record.parse(BytesIO(value))
|
|
result = print_records(r)
|
|
self.assertEqual(result.replace("\r\n", ""), expected_result)
|
|
|
|
def test_xmlns_large_complex(self):
|
|
self.maxDiff = None
|
|
|
|
value = b"V\x02\x0b\x01a\x06\x0b\x01s\x04V\x08D\n\x1e\x00\x82\x99\x06action\x01V\x0e@\tInventory\x81\x01\x01"
|
|
|
|
expected_result = """<s:Envelope xmlns:a="http://www.w3.org/2005/08/addressing" xmlns:s="http://www.w3.org/2003/05/soap-envelope"><s:Header><a:Action s:mustUnderstand="1">action</a:Action></s:Header><s:Body><Inventory>0</Inventory></s:Body></s:Envelope>"""
|
|
|
|
r = record.parse(BytesIO(value))
|
|
|
|
result = print_records(r)
|
|
|
|
self.assertEqual(textwrap.dedent(result), textwrap.dedent(expected_result))
|