From 31abe4e5da999d2412f7f68811ef7e68bc616427 Mon Sep 17 00:00:00 2001
From: Colin O'Dell <colinodell@gmail.com>
Date: Sat, 21 May 2016 13:16:44 -0400
Subject: [PATCH] Add setUrl() method (fixes #227)

---
 src/Inline/Element/AbstractWebResource.php    | 12 ++++++
 .../Element/AbstractWebResourceTest.php       | 41 +++++++++++++++++++
 2 files changed, 53 insertions(+)
 create mode 100644 tests/unit/Inline/Element/AbstractWebResourceTest.php

diff --git a/src/Inline/Element/AbstractWebResource.php b/src/Inline/Element/AbstractWebResource.php
index d9b3aa0835..1662fc0d06 100644
--- 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
new file mode 100644
index 0000000000..26d15e14d6
--- /dev/null
+++ 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());
+    }
+}