mirror of
https://github.com/isledecomp/isle.git
synced 2025-10-22 16:04:17 +00:00
59 lines
1.6 KiB
Python
59 lines
1.6 KiB
Python
import pytest
|
|
from isledecomp.cvdump.demangler import (
|
|
demangle_string_const,
|
|
demangle_vtable,
|
|
parse_encoded_number,
|
|
InvalidEncodedNumberError,
|
|
)
|
|
|
|
string_demangle_cases = [
|
|
("??_C@_08LIDF@December?$AA@", 8, False),
|
|
("??_C@_0L@EGPP@english?9nz?$AA@", 11, False),
|
|
(
|
|
"??_C@_1O@POHA@?$AA?$CI?$AAn?$AAu?$AAl?$AAl?$AA?$CJ?$AA?$AA?$AA?$AA?$AA?$AH?$AA?$AA?$AA?$AA?$AA?$AA?$AA?$9A?$AE?$;I@",
|
|
14,
|
|
True,
|
|
),
|
|
("??_C@_00A@?$AA@", 0, False),
|
|
("??_C@_01A@?$AA?$AA@", 1, False),
|
|
]
|
|
|
|
|
|
@pytest.mark.parametrize("symbol, strlen, is_utf16", string_demangle_cases)
|
|
def test_strings(symbol, is_utf16, strlen):
|
|
s = demangle_string_const(symbol)
|
|
assert s.len == strlen
|
|
assert s.is_utf16 == is_utf16
|
|
|
|
|
|
encoded_numbers = [
|
|
("A@", 0),
|
|
("AA@", 0), # would never happen?
|
|
("P@", 15),
|
|
("BA@", 16),
|
|
("BCD@", 291),
|
|
]
|
|
|
|
|
|
@pytest.mark.parametrize("string, value", encoded_numbers)
|
|
def test_encoded_numbers(string, value):
|
|
assert parse_encoded_number(string) == value
|
|
|
|
|
|
def test_invalid_encoded_number():
|
|
with pytest.raises(InvalidEncodedNumberError):
|
|
parse_encoded_number("Hello")
|
|
|
|
|
|
vtable_cases = [
|
|
("??_7LegoCarBuildAnimPresenter@@6B@", "LegoCarBuildAnimPresenter"),
|
|
("??_7?$MxCollection@PAVLegoWorld@@@@6B@", "MxCollection<LegoWorld *>"),
|
|
("??_7?$MxPtrList@VLegoPathController@@@@6B@", "MxPtrList<LegoPathController>"),
|
|
("??_7Renderer@Tgl@@6B@", "Tgl::Renderer"),
|
|
]
|
|
|
|
|
|
@pytest.mark.parametrize("symbol, class_name", vtable_cases)
|
|
def test_vtable(symbol, class_name):
|
|
assert demangle_vtable(symbol) == class_name
|