647 lines
20 KiB
Python
647 lines
20 KiB
Python
from cxxheaderparser.types import (
|
|
AnonymousName,
|
|
ClassDecl,
|
|
EnumDecl,
|
|
Enumerator,
|
|
Field,
|
|
ForwardDecl,
|
|
Function,
|
|
FundamentalSpecifier,
|
|
NameSpecifier,
|
|
PQName,
|
|
Parameter,
|
|
Pointer,
|
|
Token,
|
|
Type,
|
|
Typedef,
|
|
Value,
|
|
Variable,
|
|
)
|
|
from cxxheaderparser.simple import (
|
|
ClassScope,
|
|
NamespaceScope,
|
|
parse_string,
|
|
ParsedData,
|
|
)
|
|
|
|
|
|
def test_basic_enum() -> None:
|
|
content = """
|
|
enum Foo {
|
|
A,
|
|
B,
|
|
};
|
|
"""
|
|
data = parse_string(content, cleandoc=True)
|
|
|
|
assert data == ParsedData(
|
|
namespace=NamespaceScope(
|
|
enums=[
|
|
EnumDecl(
|
|
typename=PQName(
|
|
segments=[NameSpecifier(name="Foo")], classkey="enum"
|
|
),
|
|
values=[Enumerator(name="A"), Enumerator(name="B")],
|
|
)
|
|
]
|
|
)
|
|
)
|
|
|
|
|
|
def test_enum_w_expr() -> None:
|
|
content = """
|
|
enum Foo {
|
|
A = (1 / 2),
|
|
B = 3,
|
|
};
|
|
"""
|
|
data = parse_string(content, cleandoc=True)
|
|
|
|
assert data == ParsedData(
|
|
namespace=NamespaceScope(
|
|
enums=[
|
|
EnumDecl(
|
|
typename=PQName(
|
|
segments=[NameSpecifier(name="Foo")], classkey="enum"
|
|
),
|
|
values=[
|
|
Enumerator(
|
|
name="A",
|
|
value=Value(
|
|
tokens=[
|
|
Token(value="("),
|
|
Token(value="1"),
|
|
Token(value="/"),
|
|
Token(value="2"),
|
|
Token(value=")"),
|
|
]
|
|
),
|
|
),
|
|
Enumerator(name="B", value=Value(tokens=[Token(value="3")])),
|
|
],
|
|
)
|
|
]
|
|
)
|
|
)
|
|
|
|
|
|
def test_enum_w_multiline_expr() -> None:
|
|
content = r"""
|
|
// clang-format off
|
|
enum Author
|
|
{
|
|
NAME = ('J' << 24 | \
|
|
'A' << 16 | \
|
|
'S' << 8 | \
|
|
'H'),
|
|
};
|
|
|
|
"""
|
|
data = parse_string(content, cleandoc=True)
|
|
|
|
assert data == ParsedData(
|
|
namespace=NamespaceScope(
|
|
enums=[
|
|
EnumDecl(
|
|
typename=PQName(
|
|
segments=[NameSpecifier(name="Author")], classkey="enum"
|
|
),
|
|
values=[
|
|
Enumerator(
|
|
name="NAME",
|
|
value=Value(
|
|
tokens=[
|
|
Token(value="("),
|
|
Token(value="'J'"),
|
|
Token(value="<<"),
|
|
Token(value="24"),
|
|
Token(value="|"),
|
|
Token(value="'A'"),
|
|
Token(value="<<"),
|
|
Token(value="16"),
|
|
Token(value="|"),
|
|
Token(value="'S'"),
|
|
Token(value="<<"),
|
|
Token(value="8"),
|
|
Token(value="|"),
|
|
Token(value="'H'"),
|
|
Token(value=")"),
|
|
]
|
|
),
|
|
)
|
|
],
|
|
)
|
|
]
|
|
)
|
|
)
|
|
|
|
|
|
def test_basic_enum_class() -> None:
|
|
content = """
|
|
enum class BE { BEX };
|
|
"""
|
|
data = parse_string(content, cleandoc=True)
|
|
|
|
assert data == ParsedData(
|
|
namespace=NamespaceScope(
|
|
enums=[
|
|
EnumDecl(
|
|
typename=PQName(
|
|
segments=[NameSpecifier(name="BE")], classkey="enum class"
|
|
),
|
|
values=[Enumerator(name="BEX")],
|
|
)
|
|
]
|
|
)
|
|
)
|
|
|
|
|
|
def test_basic_enum_struct() -> None:
|
|
content = """
|
|
enum struct BE { BEX };
|
|
"""
|
|
data = parse_string(content, cleandoc=True)
|
|
|
|
assert data == ParsedData(
|
|
namespace=NamespaceScope(
|
|
enums=[
|
|
EnumDecl(
|
|
typename=PQName(
|
|
segments=[NameSpecifier(name="BE")], classkey="enum struct"
|
|
),
|
|
values=[Enumerator(name="BEX")],
|
|
)
|
|
]
|
|
)
|
|
)
|
|
|
|
|
|
def test_enum_base() -> None:
|
|
content = """
|
|
enum class E : int {};
|
|
"""
|
|
data = parse_string(content, cleandoc=True)
|
|
|
|
assert data == ParsedData(
|
|
namespace=NamespaceScope(
|
|
enums=[
|
|
EnumDecl(
|
|
typename=PQName(
|
|
segments=[NameSpecifier(name="E")], classkey="enum class"
|
|
),
|
|
values=[],
|
|
base=PQName(segments=[FundamentalSpecifier(name="int")]),
|
|
)
|
|
]
|
|
)
|
|
)
|
|
|
|
|
|
# instances
|
|
|
|
|
|
def test_enum_instance_1() -> None:
|
|
content = """
|
|
enum class BE { BEX } be1;
|
|
"""
|
|
data = parse_string(content, cleandoc=True)
|
|
|
|
assert data == ParsedData(
|
|
namespace=NamespaceScope(
|
|
enums=[
|
|
EnumDecl(
|
|
typename=PQName(
|
|
segments=[NameSpecifier(name="BE")], classkey="enum class"
|
|
),
|
|
values=[Enumerator(name="BEX")],
|
|
)
|
|
],
|
|
variables=[
|
|
Variable(
|
|
name=PQName(segments=[NameSpecifier(name="be1")]),
|
|
type=Type(
|
|
typename=PQName(
|
|
segments=[NameSpecifier(name="BE")], classkey="enum class"
|
|
)
|
|
),
|
|
)
|
|
],
|
|
)
|
|
)
|
|
|
|
|
|
def test_enum_instance_2() -> None:
|
|
content = """
|
|
enum class BE { BEX } be1, *be2;
|
|
"""
|
|
data = parse_string(content, cleandoc=True)
|
|
|
|
assert data == ParsedData(
|
|
namespace=NamespaceScope(
|
|
enums=[
|
|
EnumDecl(
|
|
typename=PQName(
|
|
segments=[NameSpecifier(name="BE")], classkey="enum class"
|
|
),
|
|
values=[Enumerator(name="BEX")],
|
|
)
|
|
],
|
|
variables=[
|
|
Variable(
|
|
name=PQName(segments=[NameSpecifier(name="be1")]),
|
|
type=Type(
|
|
typename=PQName(
|
|
segments=[NameSpecifier(name="BE")], classkey="enum class"
|
|
)
|
|
),
|
|
),
|
|
Variable(
|
|
name=PQName(segments=[NameSpecifier(name="be2")]),
|
|
type=Pointer(
|
|
ptr_to=Type(
|
|
typename=PQName(
|
|
segments=[NameSpecifier(name="BE")],
|
|
classkey="enum class",
|
|
)
|
|
)
|
|
),
|
|
),
|
|
],
|
|
)
|
|
)
|
|
|
|
|
|
# bases in namespaces
|
|
|
|
|
|
def test_enum_base_in_ns() -> None:
|
|
content = """
|
|
namespace EN {
|
|
typedef int EINT;
|
|
};
|
|
|
|
enum class BE : EN::EINT { BEX };
|
|
"""
|
|
data = parse_string(content, cleandoc=True)
|
|
|
|
assert data == ParsedData(
|
|
namespace=NamespaceScope(
|
|
enums=[
|
|
EnumDecl(
|
|
typename=PQName(
|
|
segments=[NameSpecifier(name="BE")], classkey="enum class"
|
|
),
|
|
values=[Enumerator(name="BEX")],
|
|
base=PQName(
|
|
segments=[NameSpecifier(name="EN"), NameSpecifier(name="EINT")]
|
|
),
|
|
)
|
|
],
|
|
namespaces={
|
|
"EN": NamespaceScope(
|
|
name="EN",
|
|
typedefs=[
|
|
Typedef(
|
|
type=Type(
|
|
typename=PQName(
|
|
segments=[FundamentalSpecifier(name="int")]
|
|
)
|
|
),
|
|
name="EINT",
|
|
)
|
|
],
|
|
)
|
|
},
|
|
)
|
|
)
|
|
|
|
|
|
# forward declarations
|
|
|
|
|
|
def test_enum_fwd() -> None:
|
|
content = """
|
|
enum class BE1;
|
|
enum class BE2 : EN::EINT;
|
|
"""
|
|
data = parse_string(content, cleandoc=True)
|
|
|
|
assert data == ParsedData(
|
|
namespace=NamespaceScope(
|
|
forward_decls=[
|
|
ForwardDecl(
|
|
typename=PQName(
|
|
segments=[NameSpecifier(name="BE1")], classkey="enum class"
|
|
)
|
|
),
|
|
ForwardDecl(
|
|
typename=PQName(
|
|
segments=[NameSpecifier(name="BE2")], classkey="enum class"
|
|
),
|
|
enum_base=PQName(
|
|
segments=[NameSpecifier(name="EN"), NameSpecifier(name="EINT")]
|
|
),
|
|
),
|
|
]
|
|
)
|
|
)
|
|
|
|
|
|
def test_enum_private_in_class() -> None:
|
|
content = """
|
|
|
|
class C {
|
|
enum E { E1 };
|
|
};
|
|
"""
|
|
data = parse_string(content, cleandoc=True)
|
|
|
|
assert data == ParsedData(
|
|
namespace=NamespaceScope(
|
|
classes=[
|
|
ClassScope(
|
|
class_decl=ClassDecl(
|
|
typename=PQName(
|
|
segments=[NameSpecifier(name="C")], classkey="class"
|
|
)
|
|
),
|
|
enums=[
|
|
EnumDecl(
|
|
typename=PQName(
|
|
segments=[NameSpecifier(name="E")], classkey="enum"
|
|
),
|
|
values=[Enumerator(name="E1")],
|
|
access="private",
|
|
)
|
|
],
|
|
)
|
|
]
|
|
)
|
|
)
|
|
|
|
|
|
def test_enum_public_in_class() -> None:
|
|
content = """
|
|
|
|
class C {
|
|
public:
|
|
enum E { E1 };
|
|
};
|
|
"""
|
|
data = parse_string(content, cleandoc=True)
|
|
|
|
assert data == ParsedData(
|
|
namespace=NamespaceScope(
|
|
classes=[
|
|
ClassScope(
|
|
class_decl=ClassDecl(
|
|
typename=PQName(
|
|
segments=[NameSpecifier(name="C")], classkey="class"
|
|
)
|
|
),
|
|
enums=[
|
|
EnumDecl(
|
|
typename=PQName(
|
|
segments=[NameSpecifier(name="E")], classkey="enum"
|
|
),
|
|
values=[Enumerator(name="E1")],
|
|
access="public",
|
|
)
|
|
],
|
|
)
|
|
]
|
|
)
|
|
)
|
|
|
|
|
|
def test_default_enum() -> None:
|
|
content = """
|
|
class A {
|
|
enum {
|
|
v1,
|
|
v2,
|
|
} m_v1 = v1;
|
|
|
|
enum { vv1, vv2, vv3 } m_v2 = vv2, m_v3 = vv3;
|
|
};
|
|
"""
|
|
data = parse_string(content, cleandoc=True)
|
|
|
|
assert data == ParsedData(
|
|
namespace=NamespaceScope(
|
|
classes=[
|
|
ClassScope(
|
|
class_decl=ClassDecl(
|
|
typename=PQName(
|
|
segments=[NameSpecifier(name="A")], classkey="class"
|
|
)
|
|
),
|
|
enums=[
|
|
EnumDecl(
|
|
typename=PQName(
|
|
segments=[AnonymousName(id=1)], classkey="enum"
|
|
),
|
|
values=[Enumerator(name="v1"), Enumerator(name="v2")],
|
|
access="private",
|
|
),
|
|
EnumDecl(
|
|
typename=PQName(
|
|
segments=[AnonymousName(id=2)], classkey="enum"
|
|
),
|
|
values=[
|
|
Enumerator(name="vv1"),
|
|
Enumerator(name="vv2"),
|
|
Enumerator(name="vv3"),
|
|
],
|
|
access="private",
|
|
),
|
|
],
|
|
fields=[
|
|
Field(
|
|
access="private",
|
|
type=Type(
|
|
typename=PQName(
|
|
segments=[AnonymousName(id=1)], classkey="enum"
|
|
)
|
|
),
|
|
name="m_v1",
|
|
value=Value(tokens=[Token(value="v1")]),
|
|
),
|
|
Field(
|
|
access="private",
|
|
type=Type(
|
|
typename=PQName(
|
|
segments=[AnonymousName(id=2)], classkey="enum"
|
|
)
|
|
),
|
|
name="m_v2",
|
|
value=Value(tokens=[Token(value="vv2")]),
|
|
),
|
|
Field(
|
|
access="private",
|
|
type=Type(
|
|
typename=PQName(
|
|
segments=[AnonymousName(id=2)], classkey="enum"
|
|
)
|
|
),
|
|
name="m_v3",
|
|
value=Value(tokens=[Token(value="vv3")]),
|
|
),
|
|
],
|
|
)
|
|
]
|
|
)
|
|
)
|
|
|
|
|
|
def test_enum_template_vals() -> None:
|
|
content = """
|
|
enum {
|
|
IsRandomAccess = std::is_base_of<std::random_access_iterator_tag,
|
|
IteratorCategoryT>::value,
|
|
IsBidirectional = std::is_base_of<std::bidirectional_iterator_tag,
|
|
IteratorCategoryT>::value,
|
|
};
|
|
"""
|
|
data = parse_string(content, cleandoc=True)
|
|
|
|
assert data == ParsedData(
|
|
namespace=NamespaceScope(
|
|
enums=[
|
|
EnumDecl(
|
|
typename=PQName(segments=[AnonymousName(id=1)], classkey="enum"),
|
|
values=[
|
|
Enumerator(
|
|
name="IsRandomAccess",
|
|
value=Value(
|
|
tokens=[
|
|
Token(value="std"),
|
|
Token(value="::"),
|
|
Token(value="is_base_of"),
|
|
Token(value="<"),
|
|
Token(value="std"),
|
|
Token(value="::"),
|
|
Token(value="random_access_iterator_tag"),
|
|
Token(value=","),
|
|
Token(value="IteratorCategoryT"),
|
|
Token(value=">"),
|
|
Token(value="::"),
|
|
Token(value="value"),
|
|
]
|
|
),
|
|
),
|
|
Enumerator(
|
|
name="IsBidirectional",
|
|
value=Value(
|
|
tokens=[
|
|
Token(value="std"),
|
|
Token(value="::"),
|
|
Token(value="is_base_of"),
|
|
Token(value="<"),
|
|
Token(value="std"),
|
|
Token(value="::"),
|
|
Token(value="bidirectional_iterator_tag"),
|
|
Token(value=","),
|
|
Token(value="IteratorCategoryT"),
|
|
Token(value=">"),
|
|
Token(value="::"),
|
|
Token(value="value"),
|
|
]
|
|
),
|
|
),
|
|
],
|
|
)
|
|
]
|
|
)
|
|
)
|
|
|
|
|
|
def test_enum_fn() -> None:
|
|
content = """
|
|
enum E {
|
|
VALUE,
|
|
};
|
|
|
|
void fn_with_enum_param1(const enum E e);
|
|
|
|
void fn_with_enum_param2(const enum E e) {
|
|
// code here
|
|
}
|
|
|
|
enum E fn_with_enum_retval1(void);
|
|
|
|
enum E fn_with_enum_retval2(void) {
|
|
// code here
|
|
}
|
|
"""
|
|
data = parse_string(content, cleandoc=True)
|
|
|
|
assert data == ParsedData(
|
|
namespace=NamespaceScope(
|
|
enums=[
|
|
EnumDecl(
|
|
typename=PQName(
|
|
segments=[NameSpecifier(name="E")], classkey="enum"
|
|
),
|
|
values=[Enumerator(name="VALUE")],
|
|
)
|
|
],
|
|
functions=[
|
|
Function(
|
|
return_type=Type(
|
|
typename=PQName(segments=[FundamentalSpecifier(name="void")])
|
|
),
|
|
name=PQName(segments=[NameSpecifier(name="fn_with_enum_param1")]),
|
|
parameters=[
|
|
Parameter(
|
|
type=Type(
|
|
typename=PQName(
|
|
segments=[NameSpecifier(name="E")], classkey="enum"
|
|
),
|
|
const=True,
|
|
),
|
|
name="e",
|
|
)
|
|
],
|
|
),
|
|
Function(
|
|
return_type=Type(
|
|
typename=PQName(segments=[FundamentalSpecifier(name="void")])
|
|
),
|
|
name=PQName(segments=[NameSpecifier(name="fn_with_enum_param2")]),
|
|
parameters=[
|
|
Parameter(
|
|
type=Type(
|
|
typename=PQName(
|
|
segments=[NameSpecifier(name="E")], classkey="enum"
|
|
),
|
|
const=True,
|
|
),
|
|
name="e",
|
|
)
|
|
],
|
|
has_body=True,
|
|
),
|
|
Function(
|
|
return_type=Type(
|
|
typename=PQName(
|
|
segments=[NameSpecifier(name="E")], classkey="enum"
|
|
)
|
|
),
|
|
name=PQName(segments=[NameSpecifier(name="fn_with_enum_retval1")]),
|
|
parameters=[],
|
|
),
|
|
Function(
|
|
return_type=Type(
|
|
typename=PQName(
|
|
segments=[NameSpecifier(name="E")], classkey="enum"
|
|
)
|
|
),
|
|
name=PQName(segments=[NameSpecifier(name="fn_with_enum_retval2")]),
|
|
parameters=[],
|
|
has_body=True,
|
|
),
|
|
],
|
|
)
|
|
)
|