Update test case accordingly

liip · Aug 18, 2014 · aeb8d48 · aeb8d48
1 parent 03205e4
commit aeb8d48
Showing 1 changed file with 9 additions and 2 deletions.
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'));
     }
@@ -52,4 +60,3 @@ public function testFindWithInValidDocument()
         $this->loader->find('0123456789abcdef01234567');
     }
 }
-