deluge/deluge/tests/test_tracker_icons.py
bendikro a60dc95fed Fixed tests: sessionproxy/torrent/tracker_icon
* Fixed sessionproxy tests
* Fixed test_torrent messing up component for other tests
* Updated tracker_icon test to use unresized google icon
2013-05-02 01:20:05 +02:00

71 lines
2.3 KiB
Python

import os
from twisted.trial import unittest
from deluge.ui.tracker_icons import TrackerIcons, TrackerIcon
import common
common.set_tmp_config_dir()
icons = TrackerIcons()
dirname = os.path.dirname(__file__)
import deluge.ui.tracker_icons
deluge.ui.tracker_icons.PIL_INSTALLED = False
class TrackerIconsTestCase(unittest.TestCase):
def test_get_deluge_png(self):
# Deluge has a png favicon link
icon = TrackerIcon(os.path.join(dirname, "deluge.png"))
d = icons.get("deluge-torrent.org")
d.addCallback(self.assertNotIdentical, None)
d.addCallback(self.assertEquals, icon)
return d
def test_get_google_ico(self):
# Google doesn't have any icon links
# So instead we'll grab its favicon.ico
icon = TrackerIcon(os.path.join(dirname, "google.ico"))
d = icons.get("www.google.com")
d.addCallback(self.assertNotIdentical, None)
d.addCallback(self.assertEquals, icon)
return d
def test_get_google_ico_with_redirect(self):
# google.com redirects to www.google.com
icon = TrackerIcon(os.path.join(dirname, "google.ico"))
d = icons.get("google.com")
d.addCallback(self.assertNotIdentical, None)
d.addCallback(self.assertEquals, icon)
return d
def test_get_ubuntu_ico(self):
# ubuntu.com has inline css which causes HTMLParser issues
icon = TrackerIcon(os.path.join(dirname, "ubuntu.ico"))
d = icons.get("www.ubuntu.com")
d.addCallback(self.assertNotIdentical, None)
d.addCallback(self.assertEquals, icon)
return d
def test_get_openbt_png(self):
# openbittorrent.com has an incorrect type (image/gif)
icon = TrackerIcon(os.path.join(dirname, "openbt.png"))
d = icons.get("openbittorrent.com")
d.addCallback(self.assertNotIdentical, None)
d.addCallback(self.assertEquals, icon)
return d
def test_get_publicbt_ico(self):
icon = TrackerIcon(os.path.join(dirname, "publicbt.ico"))
d = icons.get("publicbt.org")
d.addCallback(self.assertNotIdentical, None)
d.addCallback(self.assertEquals, icon)
return d
def test_get_empty_string_tracker(self):
d = icons.get("")
d.addCallback(self.assertIdentical, None)
return d