def test_parse_type_single_class_with_composition():
domain_items_by_fqdn: Dict[str, UmlItem] = {}
domain_relations: List[UmlRelation] = []
parse_type(Worker, 'tests.modules.withcomposition', domain_items_by_fqdn,
domain_relations)
umlitems_by_fqdn = list(domain_items_by_fqdn.items())
assert len(umlitems_by_fqdn) == 1, 'one class has been parsed'
assert len(domain_relations) == 1, 'class has 1 domain component'
assert domain_relations[
0].source_fqdn == 'tests.modules.withcomposition.Worker'
assert domain_relations[
0].target_fqdn == 'tests.modules.withcomposition.Address'
def test_parse_enum_type():
domain_items_by_fqdn: Dict[str, UmlItem] = {}
domain_relations: List[UmlRelation] = []
parse_type(TimeUnit, 'tests.modules.withenum', domain_items_by_fqdn,
domain_relations)
umlitems_by_fqdn = list(domain_items_by_fqdn.items())
assert len(umlitems_by_fqdn) == 1, 'one enum has been parsed'
umlenum: UmlEnum
fqdn, umlenum = umlitems_by_fqdn[0]
assert fqdn == 'tests.modules.withenum.TimeUnit'
assert umlenum.fqdn == fqdn
assert umlenum.name == 'TimeUnit'
members = umlenum.members
assert len(members) == 3, 'enum has 3 members'
assert_member(members[0], 'DAYS', 'd')
assert_member(members[1], 'HOURS', 'h')
assert_member(members[2], 'MINUTE', 'm')
assert len(domain_relations) == 0, 'parsing enum adds no relation'
def test_parse_namedtupled_class():
domain_items_by_fqdn: Dict[str, UmlItem] = {}
domain_relations: List[UmlRelation] = []
parse_type(Circle, 'tests.modules.withnamedtuple', domain_items_by_fqdn,
domain_relations)
umlitems_by_fqdn = list(domain_items_by_fqdn.items())
assert len(umlitems_by_fqdn) == 1, 'one namedtupled class has been parsed'
namedtupled_class: UmlClass
fqdn, namedtupled_class = umlitems_by_fqdn[0]
assert fqdn == 'tests.modules.withnamedtuple.Circle'
assert namedtupled_class.fqdn == fqdn
assert namedtupled_class.name == 'Circle'
attributes = namedtupled_class.attributes
assert len(attributes) == 3, 'namedtupled class has 3 attributes'
assert_attribute(attributes[0], 'x', 'any')
assert_attribute(attributes[1], 'y', 'any')
assert_attribute(attributes[2], 'radius', 'any')
assert len(domain_relations) == 0, 'parsing enum adds no relation'
def test_parse_type_single_class_without_composition():
domain_items_by_fqdn: Dict[str, UmlItem] = {}
domain_relations: List[UmlRelation] = []
parse_type(Contact, 'tests.modules.withbasictypes', domain_items_by_fqdn,
domain_relations)
umlitems_by_fqdn = list(domain_items_by_fqdn.items())
assert len(umlitems_by_fqdn) == 1, 'one class has been parsed'
umlclass: UmlClass
fqdn, umlclass = umlitems_by_fqdn[0]
assert fqdn == 'tests.modules.withbasictypes.Contact'
assert umlclass.fqdn == fqdn
assert umlclass.name == 'Contact'
attributes = umlclass.attributes
assert len(attributes) == 4, 'class has 4 attributes'
assert_attribute(attributes[0], 'full_name', 'str')
assert_attribute(attributes[1], 'age', 'int')
assert_attribute(attributes[2], 'weight', 'float')
assert_attribute(attributes[3], 'can_twist_tongue', 'bool')
assert len(domain_relations) == 0, 'class has no component'
def test_parse_inheritance_within_module():
domain_items_by_fqdn: Dict[str, UmlItem] = {}
domain_relations: List[UmlRelation] = []
parse_type(GlowingFish, 'tests.modules.withinheritancewithinmodule',
domain_items_by_fqdn, domain_relations)
umlitems_by_fqdn = list(domain_items_by_fqdn.values())
assert len(umlitems_by_fqdn
) == 1, 'the class with multiple inheritance has been parsed'
child_class: UmlClass = umlitems_by_fqdn[0]
assert child_class.name == 'GlowingFish'
assert child_class.fqdn == 'tests.modules.withinheritancewithinmodule.GlowingFish'
assert len(
domain_relations) == 2, '2 inheritance relations must have been parsed'
inheritance: UmlRelation = domain_relations[0]
assert inheritance.type == RelType.INHERITANCE
assert inheritance.source_fqdn == 'tests.modules.withinheritancewithinmodule.Fish', 'parent class'
assert inheritance.target_fqdn == 'tests.modules.withinheritancewithinmodule.GlowingFish', 'child class'
inheritance: UmlRelation = domain_relations[1]
assert inheritance.type == RelType.INHERITANCE
assert inheritance.source_fqdn == 'tests.modules.withinheritancewithinmodule.Light', 'parent class'
assert inheritance.target_fqdn == 'tests.modules.withinheritancewithinmodule.GlowingFish', 'child class'