test: add get test

andygeiss · Aug 16, 2024 · 10830aa · 10830aa
1 parent 917d1a0
commit 10830aa
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/core/entity_manager_default_test.go b/core/entity_manager_default_test.go
@@ -173,3 +173,22 @@ func TestEntityManager_FilterByNames_Should_Return_Three_Entities_Out_Of_Three(t
 		t.Errorf("Entity should have correct Id, but got %s", filtered[2].Id)
 	}
 }
+
+func TestEntityManager_Get_Should_Return_Entity(t *testing.T) {
+	em := core.NewEntityManager()
+	e1 := core.NewEntity("e1", []core.Component{
+		&mockComponent{name: "position", mask: 1},
+		&mockComponent{name: "size", mask: 2},
+	})
+	e2 := core.NewEntity("e2", []core.Component{
+		&mockComponent{name: "position", mask: 1},
+		&mockComponent{name: "size", mask: 2},
+	})
+	em.Add(e1, e2)
+	if e := em.Get("e1"); e == nil {
+		t.Error("Entity should not be nil")
+	}
+	if e := em.Get("e2"); e == nil {
+		t.Error("Entity should not be nil")
+	}
+}