Updating CollectionRepositoryTest

johannesx75 · Feb 23, 2014 · 97d2232 · 97d2232
1 parent c1a2e68
commit 97d2232
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/test/Tmdb/Tests/Repository/CollectionRepositoryTest.php b/test/Tmdb/Tests/Repository/CollectionRepositoryTest.php
@@ -12,6 +12,8 @@
  */
 namespace Tmdb\Tests\Repository;
 
+use Tmdb\Repository\CollectionRepository;
+
 class CollectionRepositoryTest extends TestCase
 {
     const COLLECTION_ID = 120;
@@ -36,6 +38,22 @@ public function shouldGetImages()
         $repository->getImages(self::COLLECTION_ID);
     }
 
+    /**
+     * @test
+     */
+    public function shouldBeAbleToSetFactories()
+    {
+        /**
+         * @var CollectionRepository $repository
+         */
+        $repository = $this->getRepositoryWithMockedHttpClient();
+        $class      = new \stdClass();
+
+        $repository->setImageFactory($class);
+
+        $this->assertInstanceOf('stdClass', $repository->getImageFactory());
+    }
+
     protected function getApiClass()
     {
         return 'Tmdb\Api\Collections';