From 88929d4821cd1cf0de688cc4eb38a62f3749afa6 Mon Sep 17 00:00:00 2001
From: John Garland <johnnybg+deluge@gmail.com>
Date: Sun, 21 Feb 2010 17:20:34 +1100
Subject: [PATCH] Rewrite tracker_icons.py (#995)

---
 ChangeLog                   |   1 +
 deluge/ui/tracker_icons.py  | 606 +++++++++++++++++++++++++++---------
 tests/deluge.png            | Bin 0 -> 722 bytes
 tests/google.ico            | Bin 0 -> 1150 bytes
 tests/test_tracker_icons.py |  38 +++
 5 files changed, 497 insertions(+), 148 deletions(-)
 create mode 100644 tests/deluge.png
 create mode 100644 tests/google.ico
 create mode 100644 tests/test_tracker_icons.py

diff --git a/ChangeLog b/ChangeLog
index 3a9a743af..4b546ecf8 100644
--- a/ChangeLog
+++ b/ChangeLog
@@ -6,6 +6,7 @@
 	* #496: Remove deprecated functions in favour of get_session_status()
 	* #1112: Fix renaming files in add torrent dialog
 	* #1247: Fix deluge-gtk from hanging on shutdown
+	* #995: Rewrote tracker_icons
 
 ==== Blocklist ====
 	* Implement local blocklist support
diff --git a/deluge/ui/tracker_icons.py b/deluge/ui/tracker_icons.py
index 8d38c9c2c..168ebaa10 100644
--- a/deluge/ui/tracker_icons.py
+++ b/deluge/ui/tracker_icons.py
@@ -1,7 +1,7 @@
 #
 # tracker_icons.py
 #
-# Copyright (C) 2008 Martijn Voncken <mvoncken@gmail.com>
+# Copyright (C) 2010 John Garland <johnnybg+deluge@gmail.com>
 #
 # Deluge is free software.
 #
@@ -33,174 +33,484 @@
 #
 #
 
-
-
-import threading
-from twisted.internet import reactor
-
-from urllib import urlopen
-from deluge.log import LOG as log
-from deluge.common import get_pixmap
 import os
-import deluge.configmanager
-import deluge.component as component
+from HTMLParser import HTMLParser
+from urlparse import urljoin, urlparse
+from tempfile import mkstemp
 
-#some servers don't have their favicon at the expected location
-RENAMES = {
-    "legaltorrents.com":"beta.legaltorrents.com",
-    "aelitis.com":"www.vuze.com"
-    }
+from twisted.internet import defer, threads
+from twisted.web import error
 
-VALID_ICO_TYPES = ["octet-stream", "x-icon", "image/vnd.microsoft.icon", "vnd.microsoft.icon", "plain"]
-VALID_PNG_TYPES = ["octet-stream", "png"]
+from deluge.component import Component
+from deluge.configmanager import get_config_dir
+from deluge.httpdownloader import download_file
+from deluge.decorators import proxy
+from deluge.log import LOG as log
 
-def fetch_url(url, valid_subtypes=None):
+try:
+    import PIL.Image as Image
+    import deluge.ui.Win32IconImagePlugin
+except ImportError:
+    PIL_INSTALLED = False
+else:
+    PIL_INSTALLED = True
+
+class TrackerIcon(object):
     """
-    returns: data or None
+    Represents a tracker's icon
     """
-    try:
-        url_file = urlopen(url)
-        data = url_file.read()
-
-        #validate:
-        if valid_subtypes and (url_file.info().getsubtype() not in valid_subtypes):
-            raise Exception("Unexpected type for %s : %s" % (url, url_file.info().getsubtype()))
-        if not data:
-            raise Exception("No data")
-    except Exception, e:
-        log.debug("%s %s" % (url, e))
-        return None
-
-    return data
-
-class TrackerIcons(component.Component):
-    def __init__(self):
-        component.Component.__init__(self, "TrackerIcons")
-        #set image cache dir
-        self.image_dir = os.path.join(deluge.configmanager.get_config_dir(), "icons")
-        if not os.path.exists(self.image_dir):
-            os.makedirs(self.image_dir)
-
-        #self.images : {tracker_host:filename}
-        self.images = {"DHT":get_pixmap("dht16.png" )}
-
-        #load image-names in cache-dir
-        for icon in os.listdir(self.image_dir):
-            if icon.endswith(".ico"):
-                self.images[icon[:-4]] = os.path.join(self.image_dir, icon)
-            if icon.endswith(".png"):
-                self.images[icon[:-4]] = os.path.join(self.image_dir, icon)
-
-    def _fetch_icon(self, tracker_host):
+    def __init__(self, filename):
         """
-        returns (ext, data)
+        Initialises a new TrackerIcon object
+
+        :param filename: the filename of the icon
+        :type filename: string
         """
-        host_name = RENAMES.get(tracker_host, tracker_host) #HACK!
-
-        ico = fetch_url("http://%s/favicon.ico" % host_name, VALID_ICO_TYPES)
-        if ico:
-            return ("ico", ico)
-
-        png = fetch_url("http://%s/favicon.png" % host_name, VALID_PNG_TYPES)
-        if png:
-            return ("png", png)
-
-        # FIXME: This should be cleaned up and not copy the top code
-
-        try:
-            html = urlopen("http://%s/" % (host_name,))
-        except Exception, e:
-            log.debug(e)
-            html = None
-
-        if html:
-            icon_path = ""
-            line = html.readline()
-            while line:
-                if '<link rel="icon"' in line or '<link rel="shortcut icon"' in line:
-                    log.debug("line: %s", line)
-                    icon_path = line[line.find("href"):].split("\"")[1]
-                    break
-                line = html.readline()
-            if icon_path:
-                ico = fetch_url(("http://%s/" + icon_path) % host_name, VALID_ICO_TYPES)
-                if ico:
-                    return ("ico", ico)
-                png = fetch_url(("http://%s/" + icon_path) % host_name, VALID_PNG_TYPES)
-                if png:
-                    return ("png", png)
+        self.filename = os.path.abspath(filename)
+        self.mimetype = ext_to_mimetype(self.filename.rpartition('.')[2])
+        self.data = None
 
+    def __eq__(self, other):
         """
-        TODO: need a test-site first...
-        html = fetch_url("http://%s/" % (host_name,))
-        if html:
-            for line in html:
-                print line
-        """
-        return (None, None)
+        Compares this TrackerIcon with another to determine if they're equal
 
-    def _fetch_icon_thread(self, tracker_host, callback):
+        :param other: the TrackerIcon to compare to
+        :type other: TrackerIcon
+        :returns: whether or not they're equal
+        :rtype: boolean
         """
-        gets new icon from the internet.
-        used by get().
-        calls callback on sucess
-        assumes dicts,urllib and logging are threadsafe.
+        return os.path.samefile(self.filename, other.filename) or \
+               self.get_mimetype() == other.get_mimetype() and \
+               self.get_data() == other.get_data()
+
+    def get_mimetype(self):
         """
+        Returns the mimetype of this TrackerIcon's image
 
-        ext, icon_data  = self._fetch_icon(tracker_host)
+        :returns: the mimetype of the image
+        :rtype: string
+        """
+        return self.mimetype
 
-        if icon_data:
-            filename = os.path.join(self.image_dir, "%s.%s" % (tracker_host, ext))
-            f = open(filename,"wb")
-            f.write(icon_data)
+    def get_data(self):
+        """
+        Returns the TrackerIcon's image data as a string
+
+        :returns: the image data
+        :rtype: string
+        """
+        if not self.data:
+            f = open(self.filename, "rb")
+            self.data = f.read()
             f.close()
-        else:
-            filename = None
+        return self.data
 
-        self.images[tracker_host] = filename
-
-        if callback:
-            reactor.callFromThread(callback, filename)
-
-    def get_async(self, tracker_host, callback):
-        if tracker_host in self.images:
-            callback(self.images[tracker_host])
-        elif "." in tracker_host:
-            #only find icon if there's a dot in the name.
-            self.images[tracker_host] = None
-            threading.Thread(target=self. _fetch_icon_thread,
-                args=(tracker_host, callback)).start()
-
-    def get(self, tracker_host):
+    def get_filename(self, full=True):
         """
-        returns None if the icon is not fetched(yet) or not fond.
+        Returns the TrackerIcon image's filename
+
+        :param full: an (optional) arg to indicate whether or not to
+                     return the full path
+        :type full: boolean
+        :returns: the path of the TrackerIcon's image
+        :rtype: string
         """
-        if not tracker_host:
-            return None
+        return self.filename if full else os.path.basename(self.filename)
 
-        if tracker_host in self.images:
-            return self.images[tracker_host]
+class TrackerIcons(Component):
+    """
+    A TrackerIcon factory class
+    """
+    def __init__(self, dir=None, noIcon=None):
+        """
+        Initialises a new TrackerIcons object
+
+        :param dir: the (optional) directory of where to store the icons
+        :type dir: string
+        :param noIcon: the (optional) path name of the icon to show when no icon
+                       can be fetched
+        :type noIcon: string
+        """
+        Component.__init__(self, "TrackerIcons")
+        if not dir:
+            dir = get_config_dir("icons")
+        self.dir = dir
+        if not os.path.isdir(self.dir):
+            os.makedirs(self.dir)
+
+        self.icons = {}
+        for icon in os.listdir(self.dir):
+            if icon != noIcon:
+                host = icon_name_to_host(icon)
+                self.icons[host] = TrackerIcon(os.path.join(self.dir, icon))
+        if noIcon:
+            self.icons[None] = TrackerIcon(noIcon)
         else:
-            self.get_async(tracker_host, None)
-            return None
+            self.icons[None] = None
 
-if __name__ == "__main__":
-    import time
-    def del_old():
-        filename = os.path.join(deluge.configmanager.get_config_dir(),"legaltorrents.com.ico")
-        if os.path.exists(filename):
-            os.remove(filename)
+        self.pending = {}
 
-    def test_get():
-        del_old()
-        trackericons  = TrackerIcons()
-        print trackericons.images
-        print trackericons.get("unknown2")
-        print trackericons.get("ntorrents.net")
-        print trackericons.get("google.com")
-        print trackericons.get("legaltorrents.com")
-        time.sleep(5.0)
-        print trackericons.get("legaltorrents.com")
+    def get(self, host):
+        """
+        Returns a TrackerIcon for the given tracker's host
 
-    test_get()
-    #test_async()
+        :param host: the host to obtain the TrackerIcon for
+        :type host: string
+        :returns: a Deferred which fires with the TrackerIcon for the given host
+        :rtype: Deferred
+        """
+        host = host.lower()
+        if host in self.icons:
+            # We already have it, so let's return it
+            d = defer.succeed(self.icons[host])
+        elif host in self.pending:
+            # We're in the middle of getting it
+            # Add ourselves to the waiting list
+            d = defer.Deferred()
+            self.pending[host].append(d)
+        else:
+            # We need to fetch it
+            self.pending[host] = []
+            # Start callback chain
+            d = self.download_page(host)
+            d.addCallbacks(self.on_download_page_complete, self.on_download_page_fail,
+                           errbackArgs=(host,))
+            d.addCallback(self.parse_html_page)
+            d.addCallbacks(self.on_parse_complete, self.on_parse_fail,
+                           callbackArgs=(host,))
+            d.addCallback(self.download_icon, host)
+            d.addCallbacks(self.on_download_icon_complete, self.on_download_icon_fail,
+                           callbackArgs=(host,), errbackArgs=(host,))
+            if PIL_INSTALLED:
+                d.addCallback(self.resize_icon)
+            d.addCallback(self.store_icon, host)
+        return d
+
+    def download_page(self, host, url=None):
+        """
+        Downloads a tracker host's page
+        If no url is provided, it bases the url on the host
+
+        :param host: the tracker host
+        :type host: string
+        :param url: the (optional) url of the host
+        :type url: string
+        :returns: the filename of the tracker host's page
+        :rtype: Deferred
+        """
+        if not url:
+            url = host_to_url(host)
+        log.debug("Downloading %s", url)
+        return download_file(url, mkstemp()[1], force_filename=True)
+
+    def on_download_page_complete(self, page):
+        """
+        Runs any download clean up functions
+
+        :param page: the page that finished downloading
+        :type page: string
+        :returns: the page that finished downloading
+        :rtype: string
+        """
+        log.debug("Finished downloading %s", page)
+        return page
+
+    def on_download_page_fail(self, f, host):
+        """
+        Recovers from download error
+
+        :param f: the failure that occured
+        :type f: Failure
+        :param host: the name of the host whose page failed to download
+        :type host: string
+        :returns: a Deferred if recovery was possible
+                  else the original failure
+        :rtype: Deferred or Failure
+        """
+        error_msg = f.getErrorMessage()
+        log.debug("Error downloading page: %s", error_msg)
+        d = f
+        if f.check(error.PageRedirect):
+            # Handle redirect errors
+            location = error_msg.split(" to ")[1]
+            d = self.download_page(host, url=location)
+            d.addCallbacks(self.on_download_page_complete, self.on_download_page_fail,
+                           errbackArgs=(host,))
+
+        return d
+
+    @proxy(threads.deferToThread)
+    def parse_html_page(self, page):
+        """
+        Parses the html page for favicons
+
+        :param page: the page to parse
+        :type page: string
+        :returns: a Deferred which callbacks a list of available favicons (url, type)
+        :rtype: Deferred
+        """
+        f = open(page, "r")
+        parser = FaviconParser()
+        for line in f:
+            parser.feed(line)
+        parser.close()
+        f.close()
+        os.remove(page)
+        return parser.get_icons()
+
+    def on_parse_complete(self, icons, host):
+        """
+        Runs any parse clean up functions
+
+        :param icons: the icons that were extracted from the page
+        :type icons: list
+        :param host: the host the icons are for
+        :type host: string
+        :returns: the icons that were extracted from the page
+        :rtype: list
+        """
+        log.debug("Got icons for %s: %s", host, icons)
+        url = host_to_url(host)
+        icons = [(urljoin(url, icon), mimetype) for icon, mimetype in icons]
+        log.debug("Icon urls: %s", icons)
+        return icons
+
+    def on_parse_fail(self, f):
+        """
+        Recovers from a parse error
+
+        :param f: the failure that occured
+        :type f: Failure
+        :returns: a Deferred if recovery was possible
+                  else the original failure
+        :rtype: Deferred or Failure
+        """
+        log.debug("Error parsing page: %s", f.getErrorMessage())
+        return f
+
+    def download_icon(self, icons, host):
+        """
+        Downloads the first available icon from icons
+
+        :param icons: a list of icons
+        :type icons: list
+        :param host: the tracker's host name
+        :type host: string
+        :returns: a Deferred which fires with the downloaded icon's filename
+        :rtype: Deferred
+        """
+        (url, mimetype) = icons.pop(0)
+        d = download_file(url, os.path.join(self.dir, host_to_icon_name(host, mimetype)),
+                          force_filename=True)
+        if icons:
+            d.addErrback(self.on_download_icon_fail, host, icons)
+        return d
+
+    def on_download_icon_complete(self, icon_name, host):
+        """
+        Runs any download cleanup functions
+
+        :param icon_name: the filename of the icon that finished downloading
+        :type icon_name: string
+        :param host: the host the icon completed to download for
+        :type host: string
+        :returns: the icon that finished downloading
+        :rtype: TrackerIcon
+        """
+        log.debug("Successfully downloaded: %s", icon_name)
+        icon = TrackerIcon(icon_name)
+        return icon
+
+    def on_download_icon_fail(self, f, host, icons=[]):
+        """
+        Recovers from a download error
+
+        :param f: the failure that occured
+        :type f: Failure
+        :param host: the host the icon failed to download for
+        :type host: string
+        :param icons: the (optional) list of remaining icons
+        :type icons: list
+        :returns: a Deferred if recovery was possible
+                  else the original failure
+        :rtype: Deferred or Failure
+        """
+        error_msg = f.getErrorMessage()
+        log.debug("Error downloading icon: %s", error_msg)
+        d = f
+        if f.check(error.PageRedirect):
+            # Handle redirect errors
+            location = error_msg.split(" to ")[1]
+            d = self.download_icon([(location, ext_to_mimetype(location.rpartition('.')[2]))] + icons, host)
+            if not icons:
+                d.addCallbacks(self.on_download_icon_complete, self.on_download_icon_fail,
+                               callbackArgs=(host,), errbackArgs=(host,))
+        elif f.check(error.NoResource, error.ForbiddenResource) and icons:
+            d = self.download_icon(icons, host)
+        elif f.check(IndexError):
+            # No icons, try favicon.ico as an act of desperation
+            d = self.download_icon([(urljoin(host_to_url(host), "favicon.ico"), ext_to_mimetype("ico"))], host)
+            d.addCallbacks(self.on_download_icon_complete, self.on_download_icon_fail,
+                           callbackArgs=(host,), errbackArgs=(host,))
+        else:
+            # No icons :(
+            # Return the None Icon
+            d = self.icons[None]
+
+        return d
+
+    @proxy(threads.deferToThread)
+    def resize_icon(self, icon):
+        """
+        Resizes the given icon to be 16x16 pixels
+
+        :param icon: the icon to resize
+        :type icon: TrackerIcon
+        :returns: the resized icon
+        :rtype: TrackerIcon
+        """
+        if icon:
+            filename = icon.get_filename()
+            img = Image.open(filename)
+            if img.size > (16, 16):
+                new_filename = filename.rpartition('.')[0]+".png"
+                img = img.resize((16, 16), Image.ANTIALIAS)
+                img.save(new_filename)
+                if new_filename != filename:
+                    os.remove(filename)
+                    icon = TrackerIcon(new_filename)
+        return icon
+
+    def store_icon(self, icon, host):
+        """
+        Stores the icon for the given host
+        Callbacks any pending deferreds waiting on this icon
+
+        :param icon: the icon to store
+        :type icon: TrackerIcon or None
+        :param host: the host to store it for
+        :type host: string
+        :returns: the stored icon
+        :rtype: TrackerIcon or None
+        """
+        self.icons[host] = icon
+        for d in self.pending[host]:
+            d.callback(icon)
+        del self.pending[host]
+        return icon
+
+################################ HELPER CLASSES ###############################
+
+class FaviconParser(HTMLParser):
+    """
+    A HTMLParser which extracts favicons from a HTML page
+    """
+    def __init__(self):
+        self.icons = []
+        HTMLParser.__init__(self)
+
+    def handle_starttag(self, tag, attrs):
+        if tag == "link" and ("rel", "icon") in attrs or ("rel", "shortcut icon") in attrs:
+            href = None
+            type = None
+            for attr, value in attrs:
+                if attr == "href":
+                    href = value
+                elif attr == "type":
+                    type = value
+            if href and type:
+                self.icons.append((href, type))
+
+    def get_icons(self):
+        """
+        Returns a list of favicons extracted from the HTML page
+
+        :returns: a list of favicons
+        :rtype: list
+        """
+        return self.icons
+
+
+############################### HELPER FUNCTIONS ##############################
+
+def host_to_url(host):
+    """
+    Given a host, returns the URL to fetch
+
+    :param host: the tracker host
+    :type host: string
+    :returns: the url of the tracker
+    :rtype: string
+    """
+    return "http://%s/" % host
+
+def url_to_host(url):
+    """
+    Given a URL, returns the host it belongs to
+
+    :param url: the URL in question
+    :type url: string
+    :returns: the host of the given URL
+    :rtype:string
+    """
+    return urlparse(url).hostname
+
+def host_to_icon_name(host, mimetype):
+    """
+    Given a host, returns the appropriate icon name
+
+    :param host: the host in question
+    :type host: string
+    :param mimetype: the mimetype of the icon
+    :type mimetype: string
+    :returns: the icon's filename
+    :rtype: string
+    """
+    return host+'.'+mimetype_to_ext(mimetype)
+
+def icon_name_to_host(icon):
+    """
+    Given a host's icon name, returns the host name
+
+    :param icon: the icon name
+    :type icon: string
+    :returns: the host name
+    :rtype: string
+    """
+    return icon.rpartition('.')[0]
+
+def mimetype_to_ext(mimetype):
+    """
+    Given a mimetype, returns the appropriate filename extension
+
+    :param mimetype: the mimetype
+    :type mimetype: string
+    :returns: the filename extension for the given mimetype
+    :rtype: string
+    :raises KeyError: if given an invalid mimetype
+    """
+    return {
+        "image/gif" : "gif",
+        "image/jpeg" : "jpg",
+        "image/png" : "png",
+        "image/vnd.microsoft.icon" : "ico",
+        "image/x-icon" : "ico"
+    }[mimetype]
+
+def ext_to_mimetype(ext):
+    """
+    Given a filename extension, returns the appropriate mimetype
+
+    :param ext: the filename extension
+    :type ext: string
+    :returns: the mimetype for the given filename extension
+    :rtype: string
+    :raises KeyError: if given an invalid filename extension
+    """
+    return {
+        "gif" : "image/gif",
+        "jpg" : "image/jpeg",
+        "jpeg" : "image/jpeg",
+        "png" : "image/png",
+        "ico" : "image/vnd.microsoft.icon"
+    }[ext.lower()]
diff --git a/tests/deluge.png b/tests/deluge.png
new file mode 100644
index 0000000000000000000000000000000000000000..e39cd0c7eebdf68688339b82aebdeecfe9f5fc9c
GIT binary patch
literal 722
zcmV;@0xkWCP)<h;3K|Lk000e1NJLTq000mG000mO1^@s6AM^iV00006VoOIv0RI60
z0RN!9r;`8x00(qQO+^RS2@e)E4N|w2*#H0nbV)=(R5*=&lS^n5Q51&%%$>}ncbZHx
z(_~CE3L3E%3zlkKR5$9vQm6~b&euv%i*DR>*M*26Sg=KRA|i?uH(iJeMMX+|K&ez(
z=)<(u+D_86X_A>tW+oXo7HpIBEbsZw$2sTT3v3hJJ^k(1dj^he$GB=GF5J2&v>n_(
z&AHuVY+}Cg(cNon)i|b>pus0S(QPKIl}RYFJ33U0t19T~ei-Nd%DE+tK-F0=_5R~G
z`d;6z#*(cEC$98`WjXZr+jxRk)m6CQz$A5$B(Xixa(4K~hZl+3g5F*xR*=W%rV}AQ
zbwd;gY|{XYEOG`IWry5B_nv;5`0`5IC}IDvKbnvR%2?lXxG{o|>;o520G7fs@>pG5
zfSM%1=&sOn0H><%z|Xrb^DIHQK}6G`ChUwxp@aem1Qi6LF_iOZu!K2z=H|2W6@q<T
zy|H{QU*{Egh@yaSM1dlCA^9l0ln1RB;1Ol4<P6A?aJ51pQrZdthGheegP`Pv?57aw
zIJj8?hH-(VB223c+qRBWj^HZm05HF}hNV;nRzZj0;XoJ$j^n^Cm0{W?n3fI4Dw23S
zPBsaWsyYe)t62@R3oCf{egx@s8b#BBUbJA_B`|P6*%F+hJ~cQvSlV=r*~Dk=KwIa<
zf??MGRx`+CHKbE1<kmB=Z3|_Gfo@n3NfDD@Ki>Z{J374e9nid1{IxsQ-09)nWI@ef
zYEH#+at+H%Nvx-nU`s{JOniScGW4JyKnke*OhGZ!*wWf@=??D^nsb`LQI8uQo^uLT
zDfe@7?B(pI$9({109e%o8^Hna0igf7GyvoQ<hFu;0haIXW3*i%vj6}907*qoM6N<$
Eg0oypRR910

literal 0
HcmV?d00001

diff --git a/tests/google.ico b/tests/google.ico
new file mode 100644
index 0000000000000000000000000000000000000000..ee7c943abf1064e3110161deb3a6477fcc7789b5
GIT binary patch
literal 1150
zcmai!ZEO@(6o&8IDMXq|w?7~;fHp><qKz>@yX|&cgxJ_d0|o(W0wkiuMEv;W=W3v(
z)q+3~th6X-V-^ZT+iD9Nw5Wh0X<8r_wUvGWQrV^LZkvVebm#HR_?tMBvv>E*ec$uG
zXU<)v47oiXC7+q<#T2DxDy6a{Bqf>u^0M6Wii}$XPK8<0I>KX3QJ$1_XGfg4m7sUv
z9K~<WX2IKv!hmLBK#~8B;!*kS5ZaR#YmW4hwm-tmrcq`zN164_ICmb55;!>w*c>_2
zz(cQ__{t3m%S;O1RTKz*@mq@CNQ_zaf8Yt-AT1Ony(LC^^Qg=5;GwIGC4|9>Q&KU9
z+)Alip~;tZ;RcOgbc|a!$qBZjz8yiIvPnH*F|93*I(Ute#vW|jhDn>{JMLvcmBP1G
z;TQbkjZ;L20ZBOf&luju4pJNYP%W3xUkx(9;R1g~q7Ib)q2DR2yq(-?@&2KSze-b3
zAs#j=&K?Vr91goayGAEyyf{QnXD`>rZb%NS$q8N#7BhdVfp@3W7mnO5;^70~5{_-f
zck|h5FU{|*;D_p081C+rjwR)`3CF^`bfAcA!RFTp*1=2ow@Yl6*rcd0oyV8UXS28D
zHg-RrL3F6k^?W%pz*BW|$=)UPKGFCjyz<@mk@Ri5fnPLur5^`Y%;3`{9)hnfqOE?D
z>(>(9zjhzzt~WFf?@{E`O4NzoU7|z6>(t+3Q1HINzSZ-%&~_MG`k07~(0lxIt`By*
z9%_$P@IaZ=tTV~mZIUA%awYP1$Q+1|MS{OR&?D7s;?W^m)@O0-X+>z&bS`&>-29%V
zS~6doL(a!ivqok}aF0|?Bm0Fs((m8wygoP1H<S@tra1F_3g17YIbB`qYTzbh-S-Pq
z@A!#~xlLqbwlc*d^P&e(Mwft{`GB^7kDpfwZk^`jvzm@=D{=bZTuYZ-Bt?zOelS&1
z)-jp=E10_5eQsG6t?TdM^jbydI>nErit}HWJGR`}D=O;8PzjVdiD4v!V^lc8!qMjp
zh-|VsTE3LfN=5rx&3DgDC-(PW?z`yvElz6MRoR=wG;F~+_2Z%^A|CqPGl-nQ!Jc-4
vYcl!bi4@Kqt`tj2MxqJw3Qx%XF0`I-Ikc-^&W9Ryu^WtA{w@6P0|)yrC_;$3

literal 0
HcmV?d00001

diff --git a/tests/test_tracker_icons.py b/tests/test_tracker_icons.py
new file mode 100644
index 000000000..9a68dc935
--- /dev/null
+++ b/tests/test_tracker_icons.py
@@ -0,0 +1,38 @@
+from twisted.trial import unittest
+
+from deluge.ui.tracker_icons import TrackerIcons, TrackerIcon
+from deluge.log import setupLogger
+
+# Must come before import common
+setupLogger("debug", "debug.log")
+
+import common
+
+common.set_tmp_config_dir()
+icons = TrackerIcons()
+
+class TrackerIconsTestCase(unittest.TestCase):
+    def test_get_png(self):
+        # Deluge has a png favicon link
+        icon = TrackerIcon("../deluge.png")
+        d = icons.get("deluge-torrent.org")
+        d.addCallback(self.assertNotIdentical, None)
+        d.addCallback(self.assertEquals, icon)
+        return d
+
+    def test_get_ico(self):
+        # Google doesn't have any icon links
+        # So instead we'll grab its favicon.ico
+        icon = TrackerIcon("../google.ico")
+        d = icons.get("www.google.com")
+        d.addCallback(self.assertNotIdentical, None)
+        d.addCallback(self.assertEquals, icon)
+        return d
+
+    def test_get_ico_with_redirect(self):
+        # google.com redirects to www.google.com
+        icon = TrackerIcon("../google.ico")
+        d = icons.get("google.com")
+        d.addCallback(self.assertNotIdentical, None)
+        d.addCallback(self.assertEquals, icon)
+        return d