1 files changed, 57 insertions, 0 deletions
diff --git a/tests/units/ExternalLink/WebLinkTest.php b/tests/units/ExternalLink/WebLinkTest.php
new file mode 100644
index 00000000..0644620f
--- /dev/null
+++ b/tests/units/ExternalLink/WebLinkTest.php
@@ -0,0 +1,57 @@
+<?php
+
+require_once __DIR__.'/../Base.php';
+
+use Kanboard\ExternalLink\WebLink;
+
+class WebLinkTest extends Base
+{
+    public function testGetTitleFromHtml()
+    {
+        $url = 'http://kanboard.net/something';
+        $title = 'My title';
+        $html = '<!DOCTYPE html><html><head><title>  '.$title.'  </title></head><body>Test</body></html>';
+
+        $this->container['httpClient'] = $this
+            ->getMockBuilder('\Kanboard\Core\Http\Client')
+            ->setConstructorArgs(array($this->container))
+            ->setMethods(array('get'))
+            ->getMock();
+
+        $webLink = new WebLink($this->container);
+        $webLink->setUrl($url);
+        $this->assertEquals($url, $webLink->getUrl());
+
+        $this->container['httpClient']
+            ->expects($this->once())
+            ->method('get')
+            ->with($url)
+            ->will($this->returnValue($html));
+
+        $this->assertEquals($title, $webLink->getTitle());
+    }
+
+    public function testGetTitleFromUrl()
+    {
+        $url = 'http://kanboard.net/something';
+        $html = '<!DOCTYPE html><html><head></head><body>Test</body></html>';
+
+        $this->container['httpClient'] = $this
+            ->getMockBuilder('\Kanboard\Core\Http\Client')
+            ->setConstructorArgs(array($this->container))
+            ->setMethods(array('get'))
+            ->getMock();
+
+        $webLink = new WebLink($this->container);
+        $webLink->setUrl($url);
+        $this->assertEquals($url, $webLink->getUrl());
+
+        $this->container['httpClient']
+            ->expects($this->once())
+            ->method('get')
+            ->with($url)
+            ->will($this->returnValue($html));
+
+        $this->assertEquals('kanboard.net/something', $webLink->getTitle());
+    }
+}