Skip to content

Commit a8c6958

Browse files
yt-msMidnighter
authored andcommitted
refactor: PR feedback
1 parent c611f34 commit a8c6958

File tree

2 files changed

+7
-19
lines changed

2 files changed

+7
-19
lines changed

src/structurizr/model/model.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,9 @@
2929
from .container_instance import ContainerInstance
3030
from .element import Element
3131
from .enterprise import Enterprise, EnterpriseIO
32+
from .implied_relationship_strategies import (
33+
ignore_implied_relationship_strategy as ignore_implied,
34+
)
3235
from .person import Person, PersonIO
3336
from .relationship import Relationship
3437
from .sequential_integer_id_generator import SequentialIntegerIDGenerator
@@ -96,7 +99,7 @@ def __init__(
9699
people: Optional[Iterable[Person]] = (),
97100
software_systems: Iterable[SoftwareSystem] = (),
98101
deployment_nodes: Iterable[DeploymentNode] = (),
99-
implied_relationship_strategy: Optional[Callable[[Relationship], None]] = None,
102+
implied_relationship_strategy: Callable[[Relationship], None] = ignore_implied,
100103
**kwargs,
101104
) -> None:
102105
"""
@@ -397,10 +400,7 @@ def _add_relationship(
397400
)
398401
self._add_relationship_to_internal_structures(relationship)
399402

400-
if (
401-
create_implied_relationships
402-
and self.implied_relationship_strategy is not None
403-
):
403+
if create_implied_relationships:
404404
self.implied_relationship_strategy(relationship)
405405
return True
406406

tests/integration/test_implied_relationship_strategies.py

Lines changed: 2 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -26,22 +26,10 @@
2626
)
2727

2828

29-
def test_by_default_model_doesnt_create_implied_relationships():
29+
def test_by_default_model_ignores_implied_relationships():
3030
"""Check if not set on Model, no implied relationships are added."""
3131
model = Model()
32-
assert model.implied_relationship_strategy is None
33-
34-
system1 = model.add_software_system(name="system1")
35-
container1 = system1.add_container(name="container1", description="test")
36-
system2 = model.add_software_system(name="system2")
37-
container2 = system2.add_container(name="container2", description="test")
38-
39-
rel = container1.uses(container2, "Uses")
40-
41-
assert len(list(container1.get_relationships())) == 1
42-
assert len(list(system1.get_relationships())) == 0
43-
assert set(container1.get_relationships()) == {rel}
44-
assert set(system1.get_relationships()) == set()
32+
assert model.implied_relationship_strategy is ignore
4533

4634

4735
def test_ignore_implied_relationship_strategy():

0 commit comments

Comments
 (0)