diff --git a/test/EntityTest.cs b/test/EntityTest.cs index a36f937..230e11c 100644 --- a/test/EntityTest.cs +++ b/test/EntityTest.cs @@ -1,6 +1,5 @@ -using System; -using System.Linq; using NUnit.Framework; +using System.Linq; namespace Encompass { @@ -25,8 +24,52 @@ namespace Encompass world.Update(); Assert.IsTrue(entity.HasComponent()); - Assert.AreEqual(3, entity.GetComponent().myInt); - Assert.AreEqual("hello", entity.GetComponent().myString); + Assert.AreEqual(mockComponent, entity.GetComponent()); + } + + [Test] + public void GetComponents() + { + var world = new World(); + var entity = world.CreateEntity(); + + var mockComponentA = entity.AddComponent(); + var mockComponentB = entity.AddComponent(); + var mockComponentC = entity.AddComponent(); + + world.Update(); + + Assert.Contains(mockComponentA, entity.GetComponents().ToList()); + Assert.Contains(mockComponentB, entity.GetComponents().ToList()); + Assert.Contains(mockComponentC, entity.GetComponents().ToList()); + } + + [Test] + public void GetComponent() + { + var world = new World(); + var entity = world.CreateEntity(); + + var mockComponent = entity.AddComponent(); + mockComponent.myString = "hello"; + mockComponent.myInt = 3; + + world.Update(); + + Assert.AreEqual(mockComponent, entity.GetComponent()); + } + + [Test] + public void HasComponent() + { + var world = new World(); + var entity = world.CreateEntity(); + + var mockComponent = entity.AddComponent(); + + world.Update(); + + Assert.IsTrue(entity.HasComponent()); } } }