Add missing UnitOfWork::getScheduledDocumentUpserts() method. Fixes i…

…ssue reported by @mlively
doctrine · Jan 27, 2014 · 505ad37 · 505ad37
1 parent 86cd699
commit 505ad37
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/lib/Doctrine/ODM/MongoDB/UnitOfWork.php b/lib/Doctrine/ODM/MongoDB/UnitOfWork.php
@@ -2988,6 +2988,16 @@ public function getScheduledDocumentInsertions()
         return $this->documentInsertions;
     }
 
+    /**
+     * Gets the currently scheduled document upserts in this UnitOfWork.
+     *
+     * @return array
+     */
+    public function getScheduledDocumentUpserts()
+    {
+        return $this->documentUpserts;
+    }
+
     /**
      * Gets the currently scheduled document updates in this UnitOfWork.
      *

diff --git a/tests/Doctrine/ODM/MongoDB/Tests/UnitOfWorkTest.php b/tests/Doctrine/ODM/MongoDB/Tests/UnitOfWorkTest.php
@@ -47,6 +47,16 @@ public function testScheduleForUpsert()
         $this->assertTrue($this->uow->isScheduledForUpsert($user));
     }
 
+    public function testGetScheduledDocumentUpserts()
+    {
+        $class = $this->dm->getClassMetadata('Documents\ForumUser');
+        $user = new ForumUser();
+        $user->id = new \MongoId();
+        $this->assertEmpty($this->uow->getScheduledDocumentUpserts());
+        $this->uow->scheduleForUpsert($class, $user);
+        $this->assertEquals(array(spl_object_hash($user) => $user), $this->uow->getScheduledDocumentUpserts());
+    }
+
     public function testScheduleForEmbeddedUpsert()
     {
         $class = $this->dm->getClassMetadata('Documents\ForumUser');