Update EntityTest.java

iluwatar · Dec 13, 2024 · 804d0e3 · 804d0e3
1 parent 98c0abf
commit 804d0e3
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/entity-component-system/src/test/java/com/iluwatar/EntityTest.java b/entity-component-system/src/test/java/com/iluwatar/EntityTest.java
@@ -169,14 +169,17 @@ public void testGetName() {
   @Test
   public void testSetIsEnabled() {
     Entity entity = new Entity("MyEntity");
-
+    Entity entity2 = new Entity("child");
+    entity.addChild(entity2);
     entity.setIsEnabled(false);
 
     assertFalse(entity.isEnabled(), "The entity should be disabled after calling setIsEnabled(false).");
-
+    assertFalse(entity2.isEnabled(), "The entity child should be disabled after calling setIsEnabled(false).");
+
     entity.setIsEnabled(true);
 
     assertTrue(entity.isEnabled(), "The entity should be enabled after calling setIsEnabled(true).");
+    assertTrue(entity2.isEnabled(), "The entity child should be enabled after calling setIsEnabled(true).");
   }
 
   @Test