Updated Policy tests to test Driver instead
[zeroinstall.git] / tests / testmodel.py
blob2c0b78063fef9118dca1c93b2f543d5503431bcb
1 #!/usr/bin/env python
2 # -*- coding: utf-8 -*-
3 from __future__ import unicode_literals
4 import basetest
5 from basetest import BaseTest, empty_feed
6 import sys, os
7 from xml.dom import minidom
8 import unittest
9 from io import BytesIO
11 sys.path.insert(0, '..')
12 from zeroinstall.injector import model, qdom, namespaces
14 mydir = os.path.dirname(os.path.abspath(__file__))
16 class TestModel(BaseTest):
17 def testLevels(self):
18 assert model.network_offline in model.network_levels
19 assert model.network_minimal in model.network_levels
20 assert model.network_full in model.network_levels
22 def testStabilities(self):
23 assert 'insecure' in model.stability_levels
24 assert 'buggy' in model.stability_levels
25 assert 'developer' in model.stability_levels
26 assert 'testing' in model.stability_levels
27 assert 'stable' in model.stability_levels
28 assert 'preferred' in model.stability_levels
29 str(model.insecure)
31 def testEscape(self):
32 self.assertEqual("", model.escape(""))
33 self.assertEqual("hello", model.escape("hello"))
34 self.assertEqual("%20", model.escape(" "))
36 self.assertEqual("file%3a%2f%2ffoo%7ebar",
37 model.escape("file://foo~bar"))
38 self.assertEqual("file%3a%2f%2ffoo%25bar",
39 model.escape("file://foo%bar"))
41 self.assertEqual("file:##foo%7ebar",
42 model._pretty_escape("file://foo~bar"))
43 self.assertEqual("file:##foo%25bar",
44 model._pretty_escape("file://foo%bar"))
46 def testUnescape(self):
47 self.assertEqual("", model.unescape(""))
48 self.assertEqual("hello", model.unescape("hello"))
49 self.assertEqual(" ", model.unescape("%20"))
51 self.assertEqual("file://foo~bar",
52 model.unescape("file%3a%2f%2ffoo%7ebar"))
53 self.assertEqual("file://foo%bar",
54 model.unescape("file%3a%2f%2ffoo%25bar"))
56 self.assertEqual("file://foo",
57 model.unescape("file:##foo"))
58 self.assertEqual("file://foo~bar",
59 model.unescape("file:##foo%7ebar"))
60 self.assertEqual("file://foo%bar",
61 model.unescape("file:##foo%25bar"))
63 def testEscaping(self):
64 def check(str):
65 self.assertEqual(str, model.unescape(model.escape(str)))
66 self.assertEqual(str, model.unescape(model._pretty_escape(str)))
68 check('http://example.com')
69 check('http://example%46com')
70 check('http:##example#com')
71 check('http://example.com/foo/bar.xml')
72 check('%20%21~&!"£ :@;,./{}$%^&()')
74 def testBadInterface(self):
75 try:
76 model.Interface('foo')
77 assert 0
78 except model.SafeException:
79 pass
81 def testInterface(self):
82 i = model.Interface('http://foo')
83 self.assertEqual('(foo)', i.get_name())
84 feed = model.ZeroInstallFeed(empty_feed, local_path = '/foo')
85 self.assertEqual('Empty', feed.get_name())
86 repr(i)
88 def testMetadata(self):
89 main_feed = model.ZeroInstallFeed(empty_feed, local_path = '/foo')
90 assert main_feed.local_path == "/foo"
91 e = qdom.parse(BytesIO(b'<ns:b xmlns:ns="a" foo="bar"/>'))
92 main_feed.metadata = [e]
93 assert main_feed.get_metadata('a', 'b') == [e]
94 assert main_feed.get_metadata('b', 'b') == []
95 assert main_feed.get_metadata('a', 'a') == []
96 assert e.getAttribute('foo') == 'bar'
97 self.assertEqual(None, main_feed.get_replaced_by())
99 def testLocale(self):
100 local_path = os.path.join(mydir, 'Local.xml')
101 dom = qdom.parse(open(local_path))
102 feed = model.ZeroInstallFeed(dom, local_path = local_path)
103 # (defaults to en-US if no language is set in the locale)
104 self.assertEqual("Local feed (English)", feed.summary)
105 self.assertEqual("English", feed.description)
107 self.assertEqual(4, len(feed.summaries))
108 self.assertEqual(2, len(feed.descriptions))
110 try:
111 basetest.test_locale = ('es_ES', 'UTF8')
113 self.assertEqual("Fuente local", feed.summary)
114 self.assertEqual("Español", feed.description)
116 basetest.test_locale = ('en_GB', 'UTF8')
118 self.assertEqual("Local feed (English GB)", feed.summary)
120 basetest.test_locale = ('fr_FR', 'UTF8')
122 self.assertEqual("Local feed (English)", feed.summary)
123 self.assertEqual("English", feed.description)
124 finally:
125 basetest.test_locale = (None, None)
127 def testCommand(self):
128 local_path = os.path.join(mydir, 'Command.xml')
129 dom = qdom.parse(open(local_path))
130 feed = model.ZeroInstallFeed(dom, local_path = local_path)
132 assert feed.implementations['a'].main == 'foo'
133 assert feed.implementations['a'].commands['run'].path == 'foo'
134 assert feed.implementations['a'].commands['test'].path == 'test-foo'
136 assert feed.implementations['b'].main == 'bar'
137 assert feed.implementations['b'].commands['run'].path == 'bar'
138 assert feed.implementations['b'].commands['test'].path == 'test-foo'
140 assert feed.implementations['c'].main == 'test-gui'
141 assert feed.implementations['c'].commands['run'].path == 'test-gui'
142 assert feed.implementations['c'].commands['test'].path == 'test-baz'
144 def testStabPolicy(self):
145 i = model.Interface('http://foo')
146 self.assertEqual(None, i.stability_policy)
147 i.set_stability_policy(model.buggy)
148 self.assertEqual(model.buggy, i.stability_policy)
150 def testImpl(self):
151 i = model.Interface('http://foo')
152 a = model.ZeroInstallImplementation(i, 'foo', None)
153 assert a.id == 'foo'
154 assert a.size == a.version == a.user_stability == None
155 assert a.arch == a.upstream_stability == None
156 assert a.dependencies == {}
157 assert a.download_sources == []
158 assert a.get_stability() is model.testing
159 a.upstream_stability = model.stable
160 assert a.get_stability() is model.stable
161 a.user_stability = model.buggy
162 assert a.get_stability() is model.buggy
163 a.version = model.parse_version('1.2.3')
164 self.assertEqual('1.2.3', a.get_version())
165 a.version = model.parse_version('1.2.3-rc2-post')
166 self.assertEqual('1.2.3-rc2-post', a.get_version())
167 assert str(a) == 'foo'
169 b = model.ZeroInstallImplementation(i, 'foo', None)
170 b.version = model.parse_version("1.2.1")
171 assert b > a
173 def testDownloadSource(self):
174 f = model.ZeroInstallFeed(empty_feed, local_path = '/foo')
175 a = model.ZeroInstallImplementation(f, 'foo', None)
176 a.add_download_source('ftp://foo', 1024, None)
177 a.add_download_source('ftp://foo.tgz', 1025, 'foo')
178 assert a.download_sources[0].url == 'ftp://foo'
179 assert a.download_sources[0].size == 1024
180 assert a.download_sources[0].extract == None
181 assert a.feed is f
183 def testEnvBind(self):
184 a = model.EnvironmentBinding('PYTHONPATH', 'path')
185 assert a.name == 'PYTHONPATH'
186 assert a.insert == 'path'
187 str(a)
189 def testEnvModes(self):
190 prepend = model.EnvironmentBinding('PYTHONPATH', 'lib', None, model.EnvironmentBinding.PREPEND)
191 assert prepend.name == 'PYTHONPATH'
192 assert prepend.insert == 'lib'
193 assert prepend.mode is model.EnvironmentBinding.PREPEND
195 self.assertEqual('/impl/lib:/usr/lib', prepend.get_value('/impl', '/usr/lib'))
196 self.assertEqual('/impl/lib', prepend.get_value('/impl', None))
198 append = model.EnvironmentBinding('PYTHONPATH', 'lib', '/opt/lib', model.EnvironmentBinding.APPEND)
199 assert append.name == 'PYTHONPATH'
200 assert append.insert == 'lib'
201 assert append.mode is model.EnvironmentBinding.APPEND
203 self.assertEqual('/usr/lib:/impl/lib', append.get_value('/impl', '/usr/lib'))
204 self.assertEqual('/opt/lib:/impl/lib', append.get_value('/impl', None))
206 append = model.EnvironmentBinding('PYTHONPATH', 'lib', None, model.EnvironmentBinding.REPLACE)
207 assert append.name == 'PYTHONPATH'
208 assert append.insert == 'lib'
209 assert append.mode is model.EnvironmentBinding.REPLACE
211 self.assertEqual('/impl/lib', append.get_value('/impl', '/usr/lib'))
212 self.assertEqual('/impl/lib', append.get_value('/impl', None))
214 assert model.EnvironmentBinding('PYTHONPATH', 'lib').mode == model.EnvironmentBinding.PREPEND
216 def testOverlay(self):
217 for xml, expected in [(b'<overlay/>', '<overlay . on />'),
218 (b'<overlay src="usr"/>', '<overlay usr on />'),
219 (b'<overlay src="package" mount-point="/usr/games"/>', '<overlay package on /usr/games>')]:
220 e = qdom.parse(BytesIO(xml))
221 ol = model.process_binding(e)
222 self.assertEqual(expected, str(ol))
224 doc = minidom.parseString('<doc/>')
225 new_xml = str(ol._toxml(doc, None).toxml())
226 new_e = qdom.parse(BytesIO(new_xml))
227 new_ol = model.process_binding(new_e)
228 self.assertEqual(expected, str(new_ol))
230 def testReplaced(self):
231 local_path = os.path.join(mydir, 'Replaced.xml')
232 with open(local_path) as stream:
233 dom = qdom.parse(stream)
234 feed = model.ZeroInstallFeed(dom, local_path = local_path)
235 self.assertEqual("http://localhost:8000/Hello", feed.get_replaced_by())
237 def testDep(self):
238 b = model.InterfaceDependency('http://foo', element = qdom.Element(namespaces.XMLNS_IFACE, 'requires', {}))
239 assert not b.restrictions
240 assert not b.bindings
241 str(b)
243 def testFeed(self):
244 f = model.Feed('http://feed', arch = None, user_override = False)
245 assert f.uri == 'http://feed'
246 assert f.os == None
247 assert f.machine == None
248 assert f.arch == None
249 assert f.user_override == False
251 f = model.Feed('http://feed', arch = 'Linux-*', user_override = True)
252 assert f.uri == 'http://feed'
253 assert f.os == 'Linux'
254 assert f.machine == None
255 assert f.arch == 'Linux-*'
256 assert f.user_override == True
258 f = model.Feed('http://feed', arch = '*-i386', user_override = True)
259 assert f.uri == 'http://feed'
260 assert f.os == None
261 assert f.machine == 'i386'
262 assert f.arch == '*-i386'
263 assert f.user_override == True
264 assert str(f).startswith('<Feed from')
266 try:
267 f = model.Feed('http://feed', arch = 'i386', user_override = True)
268 assert False
269 except model.SafeException as ex:
270 assert 'Malformed arch' in str(ex)
272 def testCanonical(self):
273 # HTTP
274 try:
275 model.canonical_iface_uri('http://foo')
276 assert False
277 except model.SafeException as ex:
278 assert 'Missing /' in str(ex)
280 self.assertEqual('http://foo/',
281 model.canonical_iface_uri('http://foo/'))
282 try:
283 model.canonical_iface_uri('bad-name')
284 assert False
285 except model.SafeException as ex:
286 assert 'Bad interface name' in str(ex)
288 # Bare relative path
289 model.canonical_iface_uri('Command.xml')
290 try:
291 model.canonical_iface_uri('CommandMissing.xml')
292 assert False
293 except model.SafeException as ex:
294 assert "Bad interface name 'CommandMissing.xml'" in str(ex), ex
296 # file:absolute
297 model.canonical_iface_uri('file://{path}/Command.xml'.format(path = mydir))
298 try:
299 print model.canonical_iface_uri('file://{path}/CommandMissing.xml'.format(path = mydir))
300 assert False
301 except model.SafeException as ex:
302 assert "Bad interface name 'file://" in str(ex), ex
304 # file:relative
305 model.canonical_iface_uri('file:Command.xml')
306 try:
307 model.canonical_iface_uri('file:CommandMissing.xml')
308 assert False
309 except model.SafeException as ex:
310 assert "Bad interface name 'file:CommandMissing.xml'" in str(ex), ex
313 def testVersions(self):
314 def pv(v):
315 parsed = model.parse_version(v)
316 assert model.format_version(parsed) == v
317 return parsed
319 assert pv('1.0') > pv('0.9')
320 assert pv('1.0') > pv('1')
321 assert pv('1.0') == pv('1.0')
322 assert pv('0.9.9') < pv('1.0')
323 assert pv('10') > pv('2')
325 def invalid(v):
326 try:
327 pv(v)
328 assert False
329 except model.SafeException:
330 pass
331 invalid('.')
332 invalid('hello')
333 invalid('2./1')
334 invalid('.1')
335 invalid('')
337 # Check parsing
338 assert pv('1') == [[1], 0]
339 assert pv('1.0') == [[1,0], 0]
340 assert pv('1.0-pre5') == [[1,0], -2, [5], 0]
341 assert pv('1.0-rc5') == [[1,0], -1, [5], 0]
342 assert pv('1.0-5') == [[1,0], 0, [5], 0]
343 assert pv('1.0-post5') == [[1,0], 1, [5], 0]
344 assert pv('1.0-post') == [[1,0], 1]
345 assert pv('1-rc2.0-pre2-post') == [[1], -1, [2,0], -2, [2], 1]
346 assert pv('1-rc2.0-pre-post') == [[1], -1, [2,0], -2, [], 1]
348 assert pv('1.0-0') > pv('1.0')
349 assert pv('1.0-1') > pv('1.0-0')
350 assert pv('1.0-0') < pv('1.0-1')
352 assert pv('1.0-pre99') > pv('1.0-pre1')
353 assert pv('1.0-pre99') < pv('1.0-rc1')
354 assert pv('1.0-rc1') < pv('1.0')
355 assert pv('1.0') < pv('1.0-0')
356 assert pv('1.0-0') < pv('1.0-post')
357 assert pv('2.1.9-pre-1') > pv('2.1.9-pre')
359 assert pv('2-post999') < pv('3-pre1')
361 if __name__ == '__main__':
362 unittest.main()