Merge branch 'feature/4751' into develop

Close zendframework/zendframework#4751
zendframework · Jul 1, 2013 · 296eaf1 · 296eaf1
commit 296eaf1
Show file tree

Hide file tree

Showing 2 changed files with 94 additions and 0 deletions.
diff --git a/src/Hydrator/HydratorAwareTrait.php b/src/Hydrator/HydratorAwareTrait.php
@@ -0,0 +1,49 @@
+<?php
+/**
+ * Zend Framework (http://framework.zend.com/)
+ *
+ * @link      http://github.com/zendframework/zf2 for the canonical source repository
+ * @copyright Copyright (c) 2005-2013 Zend Technologies USA Inc. (http://www.zend.com)
+ * @license   http://framework.zend.com/license/new-bsd New BSD License
+ */
+
+namespace Zend\Stdlib\Hydrator;
+
+use Zend\Stdlib\hydrator\HydratorInterface;
+
+trait HydratorAwareTrait
+{
+    /**
+     * Hydrator instance
+     *
+     * @var HydratorInterface
+     * @access protected
+     */
+    protected $hydrator = null;
+
+    /**
+     * Set hydrator
+     *
+     * @param  HydratorInterface $hydrator
+     * @return self
+     * @access public
+     */
+    public function setHydrator(HydratorInterface $hydrator)
+    {
+        $this->hydrator = $hydrator;
+
+        return $this;
+    }
+
+    /**
+     * Retrieve hydrator
+     *
+     * @param void
+     * @return null|HydratorInterface
+     * @access public
+     */
+    public function getHydrator()
+    {
+        return $this->hydrator;
+    }
+}
diff --git a/test/Hydrator/HydratorAwareTraitTest.php b/test/Hydrator/HydratorAwareTraitTest.php
@@ -0,0 +1,45 @@
+<?php
+/**
+ * Zend Framework (http://framework.zend.com/)
+ *
+ * @link      http://github.com/zendframework/zf2 for the canonical source repository
+ * @copyright Copyright (c) 2005-2013 Zend Technologies USA Inc. (http://www.zend.com)
+ * @license   http://framework.zend.com/license/new-bsd New BSD License
+ * @package   Zend_Hydrator
+ */
+
+namespace ZendTest\Stdlib\Hydrator;
+
+use \PHPUnit_Framework_TestCase as TestCase;
+
+/**
+ * @requires PHP 5.4
+ */
+class HydratorAwareTraitTest extends TestCase
+{
+    public function testSetHydrator()
+    {
+        $object = $this->getObjectForTrait('\Zend\Stdlib\Hydrator\HydratorAwareTrait');
+
+        $this->assertAttributeEquals(null, 'hydrator', $object);
+
+        $hydrator = $this->getMockForAbstractClass('\Zend\Stdlib\Hydrator\AbstractHydrator');
+
+        $object->setHydrator($hydrator);
+
+        $this->assertAttributeEquals($hydrator, 'hydrator', $object);
+    }
+
+    public function testGetHydrator()
+    {
+        $object = $this->getObjectForTrait('\Zend\Stdlib\Hydrator\HydratorAwareTrait');
+
+        $this->assertNull($object->getHydrator());
+
+        $hydrator = $this->getMockForAbstractClass('\Zend\Stdlib\Hydrator\AbstractHydrator');
+
+        $object->setHydrator($hydrator);
+
+        $this->assertEquals($hydrator, $object->getHydrator());
+    }
+}