Initial commit
This commit is contained in:
Dustin Spicuzza
649
tests/test_enum.py
Normal file
649
tests/test_enum.py
Normal file
@@ -0,0 +1,649 @@
|
||||
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():
|
||||
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():
|
||||
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():
|
||||
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="\\"),
|
||||
Token(value="'A'"),
|
||||
Token(value="<<"),
|
||||
Token(value="16"),
|
||||
Token(value="|"),
|
||||
Token(value="\\"),
|
||||
Token(value="'S'"),
|
||||
Token(value="<<"),
|
||||
Token(value="8"),
|
||||
Token(value="|"),
|
||||
Token(value="\\"),
|
||||
Token(value="'H'"),
|
||||
Token(value=")"),
|
||||
]
|
||||
),
|
||||
)
|
||||
],
|
||||
)
|
||||
]
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
def test_basic_enum_class():
|
||||
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():
|
||||
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():
|
||||
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():
|
||||
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():
|
||||
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():
|
||||
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():
|
||||
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():
|
||||
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():
|
||||
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():
|
||||
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():
|
||||
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():
|
||||
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,
|
||||
),
|
||||
],
|
||||
)
|
||||
)
|
||||
Reference in New Issue
Block a user