AbstractMember coverage

johannesx75 · Feb 8, 2014 · eb2a91a · eb2a91a
1 parent adfc8dd
commit eb2a91a
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 0 deletions.
diff --git a/lib/Tmdb/Model/Person/AbstractMember.php b/lib/Tmdb/Model/Person/AbstractMember.php
@@ -114,4 +114,14 @@ public function getProfile()
     {
         return $this->profile;
     }
+
+    /**
+     * Assert if there is an profile image object
+     *
+     * @return bool
+     */
+    public function hasProfileImage()
+    {
+        return $this->profile instanceof Image;
+    }
 }
diff --git a/test/Tmdb/Tests/Factory/MovieFactoryTest.php b/test/Tmdb/Tests/Factory/MovieFactoryTest.php
@@ -14,6 +14,7 @@
 
 use Tmdb\Factory\MovieFactory;
 use Tmdb\Model\Movie;
+use Tmdb\Model\Person\CastMember;
 
 class MovieFactoryTest extends TestCase
 {
@@ -122,6 +123,26 @@ public function shouldBeAbleToDissectResults()
         $this->assertEquals(2, count($collection));
     }
 
+    /**
+     * @test
+     */
+    public function shouldGetProfileImages()
+    {
+        $cast = $this->movie->getCredits()->getCast();
+
+        /**
+         * @var CastMember $c
+         */
+        foreach($cast as $c) {
+            if ($c->hasProfileImage()) {
+                $filePath = $c->getProfile()->getFilePath();
+                $this->assertEquals(false, empty($filePath));
+            }else{
+                $this->assertEquals(null, $c->getProfile());
+            }
+        }
+    }
+
     protected function getFactoryClass()
     {
         return 'Tmdb\Factory\MovieFactory';