Merge pull request #461 from upro/gridfs-fix

Fix GridFSLoader
liip · Sep 17, 2014 · 017bd45 · 017bd45
2 parents 334653c + aeb8d48
commit 017bd45
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/Binary/Loader/GridFSLoader.php b/Binary/Loader/GridFSLoader.php
@@ -40,6 +40,6 @@ public function find($id)
             throw new NotLoadableException(sprintf('Source image was not found with id "%s"', $id));
         }
 
-        return $image['file']->getBytes();
+        return $image->getFile()->getBytes();
     }
 }
diff --git a/Tests/Binary/Loader/GridFSLoaderTest.php b/Tests/Binary/Loader/GridFSLoaderTest.php
@@ -37,7 +37,15 @@ public function testFindWithValidDocument()
         $image = new GridFSFile();
         $image->setBytes('01010101');
 
-        $this->repo->expects($this->atLeastOnce())->method('find')->with($this->isInstanceOf('\MongoId'))->will($this->returnValue(array('file'=>$image)));
+        $imageDocument = $this->getMock('ImageDocument', array('getFile'));
+        $imageDocument
+            ->expects($this->any())
+            ->method('getFile')
+            ->with()
+            ->will($this->returnValue($image))
+        ;
+
+        $this->repo->expects($this->atLeastOnce())->method('find')->with($this->isInstanceOf('\MongoId'))->will($this->returnValue($imageDocument));
 
         $this->assertEquals('01010101', $this->loader->find('0123456789abcdef01234567'));
     }