fix: missing default tags

Author: yt-ms

src/structurizr/model/container_instance.py

@@ -24,6 +24,7 @@
     StaticStructureElementInstance,
     StaticStructureElementInstanceIO,
 )
+from .tags import Tags
 
 
 if TYPE_CHECKING:
@@ -47,6 +48,7 @@ def __init__(self, *, container: "Container", **kwargs) -> None:
         """Initialize a container instance."""
         super().__init__(element=container, **kwargs)
         self.container = container
+        self.tags.add(Tags.CONTAINER_INSTANCE)
 
     @property
     def container_id(self) -> str:

src/structurizr/model/software_system_instance.py

@@ -25,6 +25,7 @@
     StaticStructureElementInstance,
     StaticStructureElementInstanceIO,
 )
+from .tags import Tags
 
 
 if TYPE_CHECKING:
@@ -47,6 +48,7 @@ def __init__(self, *, software_system: SoftwareSystem, **kwargs) -> None:
         """Initialize a software system instance."""
         super().__init__(element=software_system, **kwargs)
         self.software_system = software_system
+        self.tags.add(Tags.SOFTWARE_SYSTEM_INSTANCE)
 
     @property
     def software_system_id(self) -> str:

tests/unit/model/test_container_instance.py

@@ -18,6 +18,7 @@
 
 from structurizr.model.container_instance import ContainerInstance, ContainerInstanceIO
 from structurizr.model.http_health_check import HTTPHealthCheck, HTTPHealthCheckIO
+from structurizr.model.tags import Tags
 
 
 class MockModel:
@@ -73,6 +74,15 @@ def test_container_instance_init(attributes):
         assert getattr(instance, attr) == expected
 
 
+def test_container_instance_tags(model_with_container):
+    """Check default tags."""
+    instance = ContainerInstance(
+        container=model_with_container.mock_container, instance_id="11"
+    )
+    assert Tags.ELEMENT in instance.tags
+    assert Tags.CONTAINER_INSTANCE in instance.tags
+
+
 def test_container_instance_hydration_retrieves_container_from_id(model_with_container):
     """Check that the container instance is retrieved from the model on hydration."""
     io = ContainerInstanceIO(container_id="19", instance_id=3, environment="Live")

tests/unit/model/test_software_system_instance.py

@@ -21,6 +21,7 @@
     SoftwareSystemInstance,
     SoftwareSystemInstanceIO,
 )
+from structurizr.model.tags import Tags
 
 
 class MockModel:
@@ -76,6 +77,15 @@ def test_software_system_instance_init(attributes):
         assert getattr(instance, attr) == expected
 
 
+def test_software_system_instance_tags(model_with_system):
+    """Check default tags."""
+    node = SoftwareSystemInstance(
+        software_system=model_with_system.mock_system, instance_id="9"
+    )
+    assert Tags.ELEMENT in node.tags
+    assert Tags.SOFTWARE_SYSTEM_INSTANCE in node.tags
+
+
 def test_software_system_instance_hydration_retrieves_software_system_from_id(
     model_with_system,
 ):