Fix MockMapper findOne if id doesn't exist.

Gignite · Jul 11, 2012 · f32b305 · f32b305
1 parent bd9e34a
commit f32b305
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/classes/TheCure/Mappers/MockMapper.php b/classes/TheCure/Mappers/MockMapper.php
@@ -122,7 +122,10 @@ function ($where) use ($collection)
 			{
 				if (isset($where['_id']))
 				{
-					return $collection[$where['_id']];
+					if (isset($collection[$where['_id']]))
+					{
+						return $collection[$where['_id']];
+					}
 				}
 				elseif ($where)
 				{

diff --git a/test/unit/classes/TheCure/Specs/Mappers/Mapper.php b/test/unit/classes/TheCure/Specs/Mappers/Mapper.php
@@ -89,6 +89,12 @@ public function testFindNone()
 		$this->assertNull($collection->current());
 	}
 
+	public function testFindOneWithNonExistentId()
+	{
+		$result = static::mapper()->findOne(123);
+		$this->assertNull($result);
+	}
+
 	public function provideModel()
 	{
 		$model = new Models\User;