From 1dbd73f17a98437d395dbe08b0c62132a8720fb7 Mon Sep 17 00:00:00 2001
From: Dennis Eichhorn <spl1nes.com@googlemail.com>
Date: Sun, 16 Dec 2018 11:25:35 +0100
Subject: [PATCH] Implement tests

---
 Models/DrawImage.php | 14 +++-----------
 1 file changed, 3 insertions(+), 11 deletions(-)

diff --git a/Models/DrawImage.php b/Models/DrawImage.php
index 4a655b4..cebc0aa 100644
--- a/Models/DrawImage.php
+++ b/Models/DrawImage.php
@@ -39,7 +39,7 @@ class DrawImage implements ArrayableInterface, \JsonSerializable
     /**
      * Media object.
      *
-     * @var Media
+     * @var int|Media
      * @since 1.0.0
      */
     private $media = null;
@@ -89,18 +89,10 @@ class DrawImage implements ArrayableInterface, \JsonSerializable
     {
         return [
             'id' => $this->id,
-            'media' => $this->media->toArray(),
+            'media' => \is_scalar($this->media) ? $this->media : $this->media->toArray(),
         ];
     }
 
-    /**
-     * {@inheritdoc}
-     */
-    public function __toString()
-    {
-        return (string) \json_encode($this->toArray());
-    }
-
     /**
      * {@inheritdoc}
      */
@@ -114,7 +106,7 @@ class DrawImage implements ArrayableInterface, \JsonSerializable
      *
      * @param Media $media Media object
      *
-     * @return void
+     * @return DrawImage
      *
      * @since  1.0.0
      */