Merge pull request willdurand#7 from K-Phoen/post-hydrate

Added a post_hydrate hook
marcj · Dec 19, 2013 · 572bfc4 · 572bfc4
2 parents 0d76e7b + debe531
commit 572bfc4
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/src/EventDispatcherObjectBuilderModifier.php b/src/EventDispatcherObjectBuilderModifier.php
@@ -17,6 +17,7 @@ public function objectAttributes($builder)
         $events = array();
         foreach (array(
             'construct',
+            'post_hydrate',
             'pre_save', 'post_save',
             'pre_update', 'post_update',
             'pre_insert', 'post_insert',
@@ -71,6 +72,13 @@ public function addConstructHook()
         )) . '    ';
     }
 
+    public function postHydrate()
+    {
+        return $this->behavior->renderTemplate('objectHook', array(
+            'eventName' => $this->getEventName('post_hydrate'),
+        ));
+    }
+
     public function preSave()
     {
         return $this->behavior->renderTemplate('objectHook', array(

diff --git a/tests/EventDispatcherBehaviorTest.php b/tests/EventDispatcherBehaviorTest.php
@@ -60,6 +60,7 @@ public function testObjectMethods()
         $this->assertTrue(defined('Post::EVENT_PRE_DELETE'));
         $this->assertTrue(defined('Post::EVENT_POST_DELETE'));
         $this->assertTrue(defined('Post::EVENT_CONSTRUCT'));
+        $this->assertTrue(defined('Post::EVENT_POST_HYDRATE'));
     }
 
     public function testGetDispatcher()
@@ -76,10 +77,18 @@ public function testFireEvent()
         $preSaveFired  = false;
         $postSaveFired = false;
         $postConstructFired = false;
+        $postHydrateFired  = false;
         $threadConstructFired = false;
 
         $that = $this;
 
+        Post::getEventDispatcher()->addListener(Post::EVENT_POST_HYDRATE, function (Event $event) use (& $postHydrateFired, $that) {
+            $postHydrateFired = true;
+
+            $that->assertInstanceOf('Symfony\Component\EventDispatcher\GenericEvent', $event);
+            $that->assertInstanceOf('Post', $event->getSubject());
+        });
+
         Post::getEventDispatcher()->addListener(Post::EVENT_CONSTRUCT, function (Event $event) use (& $postConstructFired, $that) {
             $postConstructFired = true;
 
@@ -111,12 +120,15 @@ public function testFireEvent()
         new Thread();
         $this->assertTrue($threadConstructFired);
 
+
         $post = new Post();
         $this->assertTrue($postConstructFired);
 
         $post->setName('a-name');
         $post->save();
+        $post->reload();
 
+        $this->assertTrue($postHydrateFired);
         $this->assertTrue($preSaveFired);
         $this->assertTrue($postSaveFired);
     }