Add method to fetch a collection of all components on an entity.

richardlord · Oct 4, 2012 · baf6cd4 · baf6cd4
1 parent b933e7d
commit baf6cd4
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/src/net/richardlord/ash/core/Entity.as b/src/net/richardlord/ash/core/Entity.as
@@ -106,6 +106,21 @@ package net.richardlord.ash.core
 		{
 			return components[ componentClass ];
 		}
+
+		/**
+		 * Get all components from the entity.
+		 * 
+		 * @return An array containing all the components that are on the entity.
+		 */
+		public function getAll() : Array
+		{
+			var componentArray : Array = new Array();
+			for each( var component : * in components )
+			{
+				componentArray.push( component );
+			}
+			return componentArray;
+		}
 
 		/**
 		 * Does the entity have a component of a particular type.

diff --git a/test/src/net/richardlord/ash/core/EntityTests.as b/test/src/net/richardlord/ash/core/EntityTests.as
@@ -3,6 +3,7 @@ package net.richardlord.ash.core
 	import asunit.framework.IAsync;
 
 	import org.hamcrest.assertThat;
+	import org.hamcrest.collection.hasItems;
 	import org.hamcrest.object.equalTo;
 	import org.hamcrest.object.isFalse;
 	import org.hamcrest.object.isTrue;
@@ -90,6 +91,19 @@ package net.richardlord.ash.core
 			assertThat( entity.get( MockComponent ), nullValue() );
 		}
 
+		[Test]
+		public function willRetrieveAllComponents() : void
+		{
+			var component1 : MockComponent = new MockComponent();
+			entity.add( component1 );
+			var component2 : MockComponent2 = new MockComponent2();
+			entity.add( component2 );
+			var all : Array = entity.getAll();
+			assertThat( all.length, equalTo( 2 ) );
+			assertThat( all, hasItems( component1, component2 ) );
+
+		}
+
 		[Test]
 		public function hasComponentIsFalseIfComponentTypeNotPresent() : void
 		{