Added check in doUpdate() to make sure the document containing the fi…

…le is returned when updating an existing document with a new file.
doctrine · Jul 10, 2012 · 0cc2a9a · 0cc2a9a
1 parent bf542f0
commit 0cc2a9a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/lib/Doctrine/MongoDB/GridFS.php b/lib/Doctrine/MongoDB/GridFS.php
@@ -70,7 +70,10 @@ protected function doUpdate($query, array $newObj, array $options = array())
 
             // First do a find and remove query to remove the file metadata and chunks so
             // we can restore the file below
-            $document = $this->findAndRemove($query, $options);
+            if (null === $document = $this->findAndRemove($query, $options)) {
+                throw new \Exception("Could not find original document containing file");
+            }
+
             unset(
                 $document['filename'],
                 $document['length'],