added EntityManager::verifyExistenceOf

alecmce · Mar 23, 2011 · f247002 · f247002
1 parent f5ae36e
commit f247002
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 2 deletions.
diff --git a/src/com/tomseysdavies/ember/base/EntityManager.as b/src/com/tomseysdavies/ember/base/EntityManager.as
@@ -52,6 +52,11 @@ package com.tomseysdavies.ember.base {
 			return entity;
 		}
 
+		public function verifyExistenceOf(id:String):Boolean
+		{
+			return _components[id] != null;
+		}
+
 		/**
 		 * @inheritDoc
 		 */

diff --git a/src/com/tomseysdavies/ember/core/IEntityManager.as b/src/com/tomseysdavies/ember/core/IEntityManager.as
@@ -10,13 +10,22 @@ package com.tomseysdavies.ember.core{
 	public interface IEntityManager extends IDisposable	{
 
 		/**
-		 * creates a new entity with the id provided. If no id is provided a unique id will be auto generated
+		 * Creates a new entity with the id provided. If no id is provided a unique id will be auto generated.
+		 * 
+		 * If an id is provided but the entityManager already has an entity
+		 * with the same id, no entity will be created.
 		 * 
 		 * @return the new entity
 		 * 
 		 */		
 		function createEntity(Id:String = null):IEntity;	
 
+		/**
+		 * 
+		 * @return Boolean, true if an entity with the provided id exists.
+		 */
+		function verifyExistenceOf(id:String):Boolean;
+
 		/**
 		 * unregisters an entity
 		 * 

diff --git a/test/ember/EntityManagerTest.as b/test/ember/EntityManagerTest.as
@@ -40,7 +40,14 @@ package ember
 			var id:String = "SOME_ID";
 			_entityManager.createEntity(id);
 			assertNull("Entity with duplicate ID won't be created.", _entityManager.createEntity(id));
-
+		}
+
+		public function testEntityIsDeleted():void
+		{
+			var id:String = "SOME_ID";
+			_entityManager.createEntity(id);
+			_entityManager.removeEntity(id);
+			assertFalse("Entity was removed.", _entityManager.verifyExistenceOf(id));
 		}
 
 		//_________________PRIVATE