1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
import unittest
from tests.utils import (
setup_test_env,
)
setup_test_env()
from softwarecenter.distro.ubuntu import Ubuntu
class MockCache(object):
def __init__(self, mock_base_url):
self._baseurl = mock_base_url
self._cache = []
self.ready = True
def __getitem__(self, k):
return MockPackage(self, k)
def __contains__(self, k):
return True
class MockPackage(object):
def __init__(self, parent, pkgname):
self._parentcache = parent
self.name = pkgname
@property
def candidate(self):
return MockVersion(self)
class MockVersion(object):
def __init__(self, parent, version="1.0"):
self._parentpkg = parent
self._version = version
@property
def uri(self):
pkgname = self._parentpkg.name
baseurl = self._parentpkg._parentcache._baseurl
return "%s/pool/main/%s/%s/%s_%s_all.deb" % (
baseurl, pkgname[0], pkgname, pkgname, self._version)
class TestDistroUbuntu(unittest.TestCase):
def setUp(self):
self.distro = Ubuntu()
def test_icon_download_url(self):
mock_cache = MockCache("http://ppa.launchpad.net/mvo/ppa/ubuntu")
pkgname = "pkg"
iconname = "iconfilename"
pkg_uri = mock_cache[pkgname].candidate.uri
icon_url = self.distro.get_downloadable_icon_url(pkg_uri, iconname)
self.assertEqual(icon_url,
"http://ppa.launchpad.net/mvo/meta/ppa/iconfilename")
if __name__ == "__main__":
unittest.main()
|