Add setUrl() method (fixes #227)

thephpleague · May 21, 2016 · 31abe4e · 31abe4e
1 parent 4ff8dd0
commit 31abe4e
Show file tree

Hide file tree

Showing 2 changed files with 53 additions and 0 deletions.
diff --git a/src/Inline/Element/AbstractWebResource.php b/src/Inline/Element/AbstractWebResource.php
@@ -33,4 +33,16 @@ public function getUrl()
     {
         return $this->url;
     }
+
+    /**
+     * @param string $url
+     *
+     * @return $this
+     */
+    public function setUrl($url)
+    {
+        $this->url = $url;
+
+        return $this;
+    }
 }
diff --git a/tests/unit/Inline/Element/AbstractWebResourceTest.php b/tests/unit/Inline/Element/AbstractWebResourceTest.php
@@ -0,0 +1,41 @@
+<?php
+
+namespace League\CommonMark\Tests\Unit\Inline\Element;
+
+use League\CommonMark\Inline\Element\AbstractWebResource;
+
+class AbstractWebResourceTest extends \PHPUnit_Framework_TestCase
+{
+    /**
+     * Tests the URL constructor parameter and getUrl() method
+     */
+    public function testConstructorAndGetUrl()
+    {
+        $url = 'https://www.example.com/foo';
+
+        /** @var AbstractWebResource $element */
+        $element = $this->getMockBuilder('League\\CommonMark\\Inline\\Element\\AbstractWebResource')
+            ->setConstructorArgs([$url])
+            ->getMockForAbstractClass();
+
+        $this->assertEquals($url, $element->getUrl());
+    }
+
+    /**
+     * Tests the setUrl() method
+     */
+    public function testSetUrl()
+    {
+        $url1 = 'https://www.example.com/foo';
+        $url2 = 'https://www.example.com/bar';
+
+        /** @var AbstractWebResource $element */
+        $element = $this->getMockBuilder('League\\CommonMark\\Inline\\Element\\AbstractWebResource')
+            ->setConstructorArgs([$url1])
+            ->getMockForAbstractClass();
+
+        $element->setUrl($url2);
+
+        $this->assertEquals($url2, $element->getUrl());
+    }
+}