From f3fa44058321d34b9418a5727ad8da533add1492 Mon Sep 17 00:00:00 2001
From: Evan Hemsley <evan@moonside.games>
Date: Thu, 13 Jun 2019 20:41:21 -0700
Subject: [PATCH] more tests

---
 test/EntityTest.cs | 51 ++++++++++++++++++++++++++++++++++++++++++----
 1 file changed, 47 insertions(+), 4 deletions(-)

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<MockComponent>());
-                Assert.AreEqual(3, entity.GetComponent<MockComponent>().myInt);
-                Assert.AreEqual("hello", entity.GetComponent<MockComponent>().myString);
+                Assert.AreEqual(mockComponent, entity.GetComponent<MockComponent>());
+            }
+
+            [Test]
+            public void GetComponents()
+            {
+                var world = new World();
+                var entity = world.CreateEntity();
+
+                var mockComponentA = entity.AddComponent<MockComponent>();
+                var mockComponentB = entity.AddComponent<MockComponent>();
+                var mockComponentC = entity.AddComponent<MockComponent>();
+
+                world.Update();
+
+                Assert.Contains(mockComponentA, entity.GetComponents<MockComponent>().ToList());
+                Assert.Contains(mockComponentB, entity.GetComponents<MockComponent>().ToList());
+                Assert.Contains(mockComponentC, entity.GetComponents<MockComponent>().ToList());
+            }
+
+            [Test]
+            public void GetComponent()
+            {
+                var world = new World();
+                var entity = world.CreateEntity();
+
+                var mockComponent = entity.AddComponent<MockComponent>();
+                mockComponent.myString = "hello";
+                mockComponent.myInt = 3;
+
+                world.Update();
+
+                Assert.AreEqual(mockComponent, entity.GetComponent<MockComponent>());
+            }
+
+            [Test]
+            public void HasComponent()
+            {
+                var world = new World();
+                var entity = world.CreateEntity();
+
+                var mockComponent = entity.AddComponent<MockComponent>();
+
+                world.Update();
+
+                Assert.IsTrue(entity.HasComponent<MockComponent>());
             }
         }
     }