Skip to content

Commit a460246

Browse files
yt-msMidnighter
authored andcommitted
refactor: PR feedback
1 parent 0a01c04 commit a460246

File tree

3 files changed

+7
-7
lines changed

3 files changed

+7
-7
lines changed

src/structurizr/model/element.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ def add_relationship(
123123
f"Cannot add relationship {relationship} to element {self} that is not its source."
124124
)
125125
self.relationships.add(relationship)
126-
self.get_model().add_relationship(
126+
self.model.add_relationship(
127127
relationship, create_implied_relationships=create_implied_relationships
128128
)
129129
return relationship

src/structurizr/model/implied_relationship_strategies.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@
3030
from .software_system import SoftwareSystem
3131

3232

33-
def default_implied_relationship_strategy(relationship: Relationship):
33+
def ignore_implied_relationship_strategy(relationship: Relationship):
3434
"""Don't create any implied relationships."""
3535
pass
3636

tests/integration/test_implied_relationship_strategies.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818
from structurizr.model.implied_relationship_strategies import (
1919
create_implied_relationships_unless_any_exist,
2020
create_implied_relationships_unless_same_exists,
21-
default_implied_relationship_strategy,
21+
ignore_implied_relationship_strategy,
2222
)
2323

2424

@@ -40,9 +40,9 @@ def test_by_default_model_doesnt_create_implied_relationships():
4040
assert set(system1.get_relationships()) == set()
4141

4242

43-
def test_default_implied_relationship_strategy():
43+
def test_ignore_implied_relationship_strategy():
4444
"""Check that by default no implied relationships are added."""
45-
model = Model(implied_relationship_strategy=default_implied_relationship_strategy)
45+
model = Model(implied_relationship_strategy=ignore_implied_relationship_strategy)
4646
system1 = model.add_software_system(name="system1")
4747
container1 = system1.add_container(name="container1", description="test")
4848
system2 = model.add_software_system(name="system2")
@@ -95,6 +95,7 @@ def test_create_implied_relationships_unless_same_exists():
9595
model.implied_relationship_strategy = (
9696
create_implied_relationships_unless_same_exists
9797
)
98+
9899
system1 = model.add_software_system(name="system1")
99100
container1 = system1.add_container(name="container1", description="test")
100101
system2 = model.add_software_system(name="system2")
@@ -133,8 +134,7 @@ def test_suppressing_implied_relationships():
133134
)
134135
def test_self_references_are_not_implied(strategy):
135136
"""Ensure references from an element to itself don't get implied to parents."""
136-
model = Model()
137-
model.implied_relationship_strategy = strategy
137+
model = Model(implied_relationship_strategy=strategy)
138138
system1 = model.add_software_system(name="system1")
139139
container1 = system1.add_container(name="container1", description="test")
140140

0 commit comments

Comments
 (0)