[MODM-33] Adding test

doctrine · Aug 2, 2010 · af9aa0f · af9aa0f
1 parent 44bf780
commit af9aa0f
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 2 deletions.
diff --git a/tests/Doctrine/ODM/MongoDB/Tests/Functional/InheritanceTest.php b/tests/Doctrine/ODM/MongoDB/Tests/Functional/InheritanceTest.php
@@ -68,4 +68,13 @@ public function testSingleCollectionInhertiance()
         $document = $this->dm->findOne('Documents\SubProject', array('name' => 'Sub Project'));
         $this->assertInstanceOf('Documents\SubProject', $document);
     }
+
+    public function testPrePersistIsCalledFromMappedSuperClass()
+    {
+        $user = new \Documents\User();
+        $user->setUsername('test');
+        $this->dm->persist($user);
+        $this->dm->flush();
+        $this->assertTrue($user->persisted);
+    }
 }
diff --git a/tests/Documents/BaseDocument.php b/tests/Documents/BaseDocument.php
@@ -2,9 +2,14 @@
 
 namespace Documents;
 
-/** @MappedSuperClass */
-class BaseDocument
+/**
+ * @MappedSuperClass
+ * @HasLifecycleCallbacks
+ */
+abstract class BaseDocument
 {
+    public $persisted = false;
+
     /** @String */
     protected $inheritedProperty;
 
@@ -17,4 +22,10 @@ public function getInheritedProperty()
     {
         return $this->inheritedProperty;
     }
+
+    /** @PrePersist */
+    public function prePersist()
+    {
+        $this->persisted = true;
+    }
 }