mirror of
https://github.com/github/codeql.git
synced 2025-12-17 09:13:20 +01:00
109 lines
3.0 KiB
Python
109 lines
3.0 KiB
Python
import sys
|
|
from copy import deepcopy
|
|
|
|
from swift.codegen.lib import ql
|
|
from swift.codegen.test.utils import *
|
|
|
|
|
|
def test_property_has_first_table_param_marked():
|
|
tableparams = ["a", "b", "c"]
|
|
prop = ql.Property("Prop", "foo", "props", tableparams)
|
|
assert prop.tableparams[0].first
|
|
assert [p.param for p in prop.tableparams] == tableparams
|
|
|
|
|
|
@pytest.mark.parametrize("type,expected", [
|
|
("Foo", True),
|
|
("Bar", True),
|
|
("foo", False),
|
|
("bar", False),
|
|
(None, False),
|
|
])
|
|
def test_property_is_a_class(type, expected):
|
|
tableparams = ["a", "result", "b"]
|
|
expected_tableparams = ["a", "x" if expected else "result", "b"]
|
|
prop = ql.Property("Prop", type, tableparams=tableparams)
|
|
assert prop.type_is_class is expected
|
|
assert [p.param for p in prop.tableparams] == expected_tableparams
|
|
|
|
|
|
@pytest.mark.parametrize("name,expected_getter", [
|
|
("Argument", "getAnArgument"),
|
|
("Element", "getAnElement"),
|
|
("Integer", "getAnInteger"),
|
|
("Operator", "getAnOperator"),
|
|
("Unit", "getAUnit"),
|
|
("Whatever", "getAWhatever"),
|
|
])
|
|
def test_property_indefinite_article(name, expected_getter):
|
|
prop = ql.Property(name, plural="X")
|
|
assert prop.indefinite_getter == expected_getter
|
|
|
|
|
|
@pytest.mark.parametrize("plural,expected", [
|
|
(None, False),
|
|
("", False),
|
|
("X", True),
|
|
])
|
|
def test_property_is_repeated(plural, expected):
|
|
prop = ql.Property("foo", "Foo", "props", ["x"], plural=plural)
|
|
assert prop.is_repeated is expected
|
|
|
|
|
|
@pytest.mark.parametrize("is_optional,is_predicate,plural,expected", [
|
|
(False, False, None, True),
|
|
(False, False, "", True),
|
|
(False, False, "X", False),
|
|
(True, False, None, False),
|
|
(False, True, None, False),
|
|
])
|
|
def test_property_is_single(is_optional, is_predicate, plural, expected):
|
|
prop = ql.Property("foo", "Foo", "props", ["x"], plural=plural, is_predicate=is_predicate, is_optional=is_optional)
|
|
assert prop.is_single is expected
|
|
|
|
|
|
def test_property_no_plural_no_indefinite_getter():
|
|
prop = ql.Property("Prop", "Foo", "props", ["x"])
|
|
assert prop.indefinite_getter is None
|
|
|
|
|
|
def test_property_getter():
|
|
prop = ql.Property("Prop", "Foo")
|
|
assert prop.getter == "getProp"
|
|
|
|
|
|
def test_property_predicate_getter():
|
|
prop = ql.Property("prop", is_predicate=True)
|
|
assert prop.getter == "prop"
|
|
|
|
|
|
def test_class_sorts_bases():
|
|
bases = ["B", "Ab", "C", "Aa"]
|
|
expected = ["Aa", "Ab", "B", "C"]
|
|
cls = ql.Class("Foo", bases=bases)
|
|
assert cls.bases == expected
|
|
|
|
|
|
def test_class_has_first_property_marked():
|
|
props = [
|
|
ql.Property(f"Prop{x}", f"Foo{x}", f"props{x}", [f"{x}"]) for x in range(4)
|
|
]
|
|
expected = deepcopy(props)
|
|
expected[0].first = True
|
|
cls = ql.Class("Class", properties=props)
|
|
assert cls.properties == expected
|
|
|
|
|
|
def test_root_class():
|
|
cls = ql.Class("Class")
|
|
assert cls.root
|
|
|
|
|
|
def test_non_root_class():
|
|
cls = ql.Class("Class", bases=["A"])
|
|
assert not cls.root
|
|
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(pytest.main([__file__] + sys.argv[1:]))
|