API changes: use Feed rather than Interface in many places
[zeroinstall/zeroinstall-afb.git] / tests / testmodel.py
blob181ab1f2713afbc6767ffef39de9b2d9d2d7e04c
1 #!/usr/bin/env python
2 # -*- coding: utf-8 -*-
3 from basetest import BaseTest, empty_feed
4 import sys
5 from xml.dom import minidom
6 import unittest
7 from StringIO import StringIO
9 sys.path.insert(0, '..')
10 from zeroinstall.injector import model, qdom, iface_cache
12 class TestModel(BaseTest):
13 def testLevels(self):
14 assert model.network_offline in model.network_levels
15 assert model.network_minimal in model.network_levels
16 assert model.network_full in model.network_levels
18 def testStabilities(self):
19 assert 'insecure' in model.stability_levels
20 assert 'buggy' in model.stability_levels
21 assert 'developer' in model.stability_levels
22 assert 'testing' in model.stability_levels
23 assert 'stable' in model.stability_levels
24 assert 'preferred' in model.stability_levels
25 str(model.insecure)
27 def testEscape(self):
28 self.assertEquals("", model.escape(""))
29 self.assertEquals("hello", model.escape("hello"))
30 self.assertEquals("%20", model.escape(" "))
32 self.assertEquals("file%3a%2f%2ffoo%7ebar",
33 model.escape("file://foo~bar"))
34 self.assertEquals("file%3a%2f%2ffoo%25bar",
35 model.escape("file://foo%bar"))
37 self.assertEquals("file:##foo%7ebar",
38 model._pretty_escape("file://foo~bar"))
39 self.assertEquals("file:##foo%25bar",
40 model._pretty_escape("file://foo%bar"))
42 def testUnescape(self):
43 self.assertEquals("", model.unescape(""))
44 self.assertEquals("hello", model.unescape("hello"))
45 self.assertEquals(" ", model.unescape("%20"))
47 self.assertEquals("file://foo~bar",
48 model.unescape("file%3a%2f%2ffoo%7ebar"))
49 self.assertEquals("file://foo%bar",
50 model.unescape("file%3a%2f%2ffoo%25bar"))
52 self.assertEquals("file://foo",
53 model.unescape("file:##foo"))
54 self.assertEquals("file://foo~bar",
55 model.unescape("file:##foo%7ebar"))
56 self.assertEquals("file://foo%bar",
57 model.unescape("file:##foo%25bar"))
59 def testEscaping(self):
60 def check(str):
61 self.assertEquals(str, model.unescape(model.escape(str)))
62 self.assertEquals(str, model.unescape(model._pretty_escape(str)))
64 check(u'http://example.com')
65 check(u'http://example%46com')
66 check(u'http:##example#com')
67 check(u'http://example.com/foo/bar.xml')
68 check(u'%20%21~&!"£ :@;,./{}$%^&()')
70 def testBadInterface(self):
71 try:
72 model.Interface('foo')
73 assert 0
74 except model.SafeException:
75 pass
77 def testInterface(self):
78 i = model.Interface('http://foo')
79 self.assertEquals('(foo)', i.get_name())
80 feed = model.ZeroInstallFeed(empty_feed, local_path = '/foo')
81 self.assertEquals('Empty', feed.get_name())
82 repr(i)
84 def testMetadata(self):
85 main_feed = model.ZeroInstallFeed(empty_feed, local_path = '/foo')
86 e = qdom.parse(StringIO('<ns:b xmlns:ns="a" foo="bar"/>'))
87 main_feed.metadata = [e]
88 assert main_feed.get_metadata('a', 'b') == [e]
89 assert main_feed.get_metadata('b', 'b') == []
90 assert main_feed.get_metadata('a', 'a') == []
91 assert e.getAttribute('foo') == 'bar'
93 def testStabPolicy(self):
94 i = model.Interface('http://foo')
95 self.assertEquals(None, i.stability_policy)
96 i.set_stability_policy(model.buggy)
97 self.assertEquals(model.buggy, i.stability_policy)
99 def testImpl(self):
100 i = model.Interface('http://foo')
101 a = model.ZeroInstallImplementation(i, 'foo', None)
102 assert a.id == 'foo'
103 assert a.size == a.version == a.user_stability == None
104 assert a.arch == a.upstream_stability == None
105 assert a.dependencies == {}
106 assert a.download_sources == []
107 assert a.get_stability() is model.testing
108 a.upstream_stability = model.stable
109 assert a.get_stability() is model.stable
110 a.user_stability = model.buggy
111 assert a.get_stability() is model.buggy
112 a.version = model.parse_version('1.2.3')
113 self.assertEquals('1.2.3', a.get_version())
114 a.version = model.parse_version('1.2.3-rc2-post')
115 self.assertEquals('1.2.3-rc2-post', a.get_version())
116 assert str(a) == 'foo'
118 b = model.ZeroInstallImplementation(i, 'foo', None)
119 b.version = [1,2,1]
120 assert b > a
122 def testDownloadSource(self):
123 f = model.ZeroInstallFeed(empty_feed, local_path = '/foo')
124 a = model.ZeroInstallImplementation(f, 'foo', None)
125 a.add_download_source('ftp://foo', 1024, None)
126 a.add_download_source('ftp://foo.tgz', 1025, 'foo')
127 assert a.download_sources[0].url == 'ftp://foo'
128 assert a.download_sources[0].size == 1024
129 assert a.download_sources[0].extract == None
130 assert a.feed is f
132 def testEnvBind(self):
133 a = model.EnvironmentBinding('PYTHONPATH', 'path')
134 assert a.name == 'PYTHONPATH'
135 assert a.insert == 'path'
136 str(a)
138 def testEnvModes(self):
139 prepend = model.EnvironmentBinding('PYTHONPATH', 'lib', None, model.EnvironmentBinding.PREPEND)
140 assert prepend.name == 'PYTHONPATH'
141 assert prepend.insert == 'lib'
142 assert prepend.mode is model.EnvironmentBinding.PREPEND
144 self.assertEquals('/impl/lib:/usr/lib', prepend.get_value('/impl', '/usr/lib'))
145 self.assertEquals('/impl/lib', prepend.get_value('/impl', None))
147 append = model.EnvironmentBinding('PYTHONPATH', 'lib', '/opt/lib', model.EnvironmentBinding.APPEND)
148 assert append.name == 'PYTHONPATH'
149 assert append.insert == 'lib'
150 assert append.mode is model.EnvironmentBinding.APPEND
152 self.assertEquals('/usr/lib:/impl/lib', append.get_value('/impl', '/usr/lib'))
153 self.assertEquals('/opt/lib:/impl/lib', append.get_value('/impl', None))
155 append = model.EnvironmentBinding('PYTHONPATH', 'lib', None, model.EnvironmentBinding.REPLACE)
156 assert append.name == 'PYTHONPATH'
157 assert append.insert == 'lib'
158 assert append.mode is model.EnvironmentBinding.REPLACE
160 self.assertEquals('/impl/lib', append.get_value('/impl', '/usr/lib'))
161 self.assertEquals('/impl/lib', append.get_value('/impl', None))
163 assert model.EnvironmentBinding('PYTHONPATH', 'lib').mode == model.EnvironmentBinding.PREPEND
165 def testOverlay(self):
166 for xml, expected in [('<overlay/>', '<overlay . on />'),
167 ('<overlay src="usr"/>', '<overlay usr on />'),
168 ('<overlay src="package" mount-point="/usr/games"/>', '<overlay package on /usr/games>')]:
169 e = qdom.parse(StringIO(xml))
170 ol = model.process_binding(e)
171 self.assertEquals(expected, str(ol))
173 doc = minidom.parseString('<doc/>')
174 new_xml = str(ol._toxml(doc).toxml())
175 new_e = qdom.parse(StringIO(new_xml))
176 new_ol = model.process_binding(new_e)
177 self.assertEquals(expected, str(new_ol))
179 def testDep(self):
180 b = model.InterfaceDependency('http://foo')
181 assert not b.restrictions
182 assert not b.bindings
183 str(b)
185 def testFeed(self):
186 f = model.Feed('http://feed', arch = None, user_override = False)
187 assert f.uri == 'http://feed'
188 assert f.os == None
189 assert f.machine == None
190 assert f.arch == None
191 assert f.user_override == False
193 f = model.Feed('http://feed', arch = 'Linux-*', user_override = True)
194 assert f.uri == 'http://feed'
195 assert f.os == 'Linux'
196 assert f.machine == None
197 assert f.arch == 'Linux-*'
198 assert f.user_override == True
200 f = model.Feed('http://feed', arch = '*-i386', user_override = True)
201 assert f.uri == 'http://feed'
202 assert f.os == None
203 assert f.machine == 'i386'
204 assert f.arch == '*-i386'
205 assert f.user_override == True
206 assert str(f).startswith('<Feed from')
208 try:
209 f = model.Feed('http://feed', arch = 'i386', user_override = True)
210 assert False
211 except model.SafeException, ex:
212 assert 'Malformed arch' in str(ex)
214 def testCanonical(self):
215 try:
216 model.canonical_iface_uri('http://foo')
217 assert False
218 except model.SafeException, ex:
219 assert 'Missing /' in str(ex)
221 self.assertEquals('http://foo/',
222 model.canonical_iface_uri('http://foo/'))
223 try:
224 model.canonical_iface_uri('bad-name')
225 assert False
226 except model.SafeException, ex:
227 assert 'Bad interface name' in str(ex)
229 def testVersions(self):
230 def pv(v):
231 parsed = model.parse_version(v)
232 assert model.format_version(parsed) == v
233 return parsed
235 assert pv('1.0') > pv('0.9')
236 assert pv('1.0') > pv('1')
237 assert pv('1.0') == pv('1.0')
238 assert pv('0.9.9') < pv('1.0')
239 assert pv('10') > pv('2')
241 def invalid(v):
242 try:
243 pv(v)
244 assert False
245 except model.SafeException, ex:
246 pass
247 invalid('.')
248 invalid('hello')
249 invalid('2./1')
250 invalid('.1')
251 invalid('')
253 # Check parsing
254 assert pv('1') == [[1], 0]
255 assert pv('1.0') == [[1,0], 0]
256 assert pv('1.0-pre5') == [[1,0], -2, [5], 0]
257 assert pv('1.0-rc5') == [[1,0], -1, [5], 0]
258 assert pv('1.0-5') == [[1,0], 0, [5], 0]
259 assert pv('1.0-post5') == [[1,0], 1, [5], 0]
260 assert pv('1.0-post') == [[1,0], 1]
261 assert pv('1-rc2.0-pre2-post') == [[1], -1, [2,0], -2, [2], 1]
262 assert pv('1-rc2.0-pre-post') == [[1], -1, [2,0], -2, [], 1]
264 assert pv('1.0-0') > pv('1.0')
265 assert pv('1.0-1') > pv('1.0-0')
266 assert pv('1.0-0') < pv('1.0-1')
268 assert pv('1.0-pre99') > pv('1.0-pre1')
269 assert pv('1.0-pre99') < pv('1.0-rc1')
270 assert pv('1.0-rc1') < pv('1.0')
271 assert pv('1.0') < pv('1.0-0')
272 assert pv('1.0-0') < pv('1.0-post')
273 assert pv('2.1.9-pre-1') > pv('2.1.9-pre')
275 assert pv('2-post999') < pv('3-pre1')
277 if __name__ == '__main__':
278 unittest.main()