add test

laravolt · May 3, 2020 · 4650dd5 · 4650dd5
1 parent dc8cb3f
commit 4650dd5
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/src/Concerns/AttributeSetter.php b/src/Concerns/AttributeSetter.php
@@ -8,11 +8,11 @@ trait AttributeSetter
 {
     public function setTheme($theme)
     {
-        if (!array_key_exists($theme, $this->themes)) {
-            return $this;
-        }
-
         if (is_string($theme) || is_array($theme)) {
+            if (is_string($theme) && !array_key_exists($theme, $this->themes)) {
+                return $this;
+            }
+
             $this->theme = $theme;
         }
 

diff --git a/tests/AvatarPhpTest.php b/tests/AvatarPhpTest.php
@@ -124,7 +124,7 @@ public function it_can_resolve_random_themes_and_then_overrides()
     /**
      * @test
      */
-    public function it_cannot_set_non_existant_theme()
+    public function it_can_handle_invalid_theme()
     {
         $config = [
             'foregrounds' => ['#000000', '#111111'],
@@ -137,6 +137,9 @@ public function it_cannot_set_non_existant_theme()
 
         $this->assertAttributeEquals('#000000', 'background', $avatar);
         $this->assertAttributeEquals('#111111', 'foreground', $avatar);
+
+        $avatar->setTheme(new stdClass())->buildAvatar();
+        $avatar->setTheme(['satu', 'dua'])->buildAvatar();
     }
 
     /**