Merge pull request #61 from pgodel/patch-1

Add check in doUpdate() to make sure the document containing the file is returned
doctrine · Sep 9, 2012 · de0c453 · de0c453
2 parents b5ab03b + 0cc2a9a
commit de0c453
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'],