fix up the tracker icon tests
This commit is contained in:
parent
6fd4b298f3
commit
3b6bad2f13
|
@ -1,3 +1,5 @@
|
|||
import os
|
||||
|
||||
from twisted.trial import unittest
|
||||
|
||||
from deluge.ui.tracker_icons import TrackerIcons, TrackerIcon
|
||||
|
@ -7,10 +9,13 @@ import common
|
|||
common.set_tmp_config_dir()
|
||||
icons = TrackerIcons()
|
||||
|
||||
dirname = os.path.dirname(__file__)
|
||||
|
||||
class TrackerIconsTestCase(unittest.TestCase):
|
||||
|
||||
def test_get_deluge_png(self):
|
||||
# Deluge has a png favicon link
|
||||
icon = TrackerIcon("../deluge.png")
|
||||
icon = TrackerIcon(os.path.join(dirname, "deluge.png"))
|
||||
d = icons.get("deluge-torrent.org")
|
||||
d.addCallback(self.assertNotIdentical, None)
|
||||
d.addCallback(self.assertEquals, icon)
|
||||
|
@ -19,7 +24,7 @@ class TrackerIconsTestCase(unittest.TestCase):
|
|||
def test_get_google_ico(self):
|
||||
# Google doesn't have any icon links
|
||||
# So instead we'll grab its favicon.ico
|
||||
icon = TrackerIcon("../google.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)
|
||||
|
@ -27,7 +32,7 @@ class TrackerIconsTestCase(unittest.TestCase):
|
|||
|
||||
def test_get_google_ico_with_redirect(self):
|
||||
# google.com redirects to www.google.com
|
||||
icon = TrackerIcon("../google.ico")
|
||||
icon = TrackerIcon(os.path.join(dirname, "google.ico"))
|
||||
d = icons.get("google.com")
|
||||
d.addCallback(self.assertNotIdentical, None)
|
||||
d.addCallback(self.assertEquals, icon)
|
||||
|
@ -48,14 +53,14 @@ class TrackerIconsTestCase(unittest.TestCase):
|
|||
|
||||
def test_get_openbt_png(self):
|
||||
# openbittorrent.com has an incorrect type (image/gif)
|
||||
icon = TrackerIcon("../openbt.png")
|
||||
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("../publicbt.ico")
|
||||
icon = TrackerIcon(os.path.join(dirname, "publicbt.ico"))
|
||||
d = icons.get("publicbt.org")
|
||||
d.addCallback(self.assertNotIdentical, None)
|
||||
d.addCallback(self.assertEquals, icon)
|
||||
|
|
Loading…
Reference in New Issue