1 #!/usr/bin/env python2.4
2 from basetest
import BaseTest
3 import sys
, tempfile
, os
, shutil
, logging
4 from StringIO
import StringIO
6 from logging
import getLogger
, DEBUG
, INFO
8 sys
.path
.insert(0, '..')
10 from zeroinstall
import NeedDownload
11 from zeroinstall
.injector
import model
, basedir
, autopolicy
, gpg
, iface_cache
, namespaces
, reader
14 foo_iface_uri
= 'http://foo'
16 logger
= logging
.getLogger()
18 class TestAutoPolicy(BaseTest
):
21 stream
= tempfile
.TemporaryFile()
22 stream
.write(data
.thomas_key
)
24 gpg
.import_key(stream
)
26 def cache_iface(self
, name
, data
):
27 cached_ifaces
= basedir
.save_cache_path('0install.net',
30 f
= file(os
.path
.join(cached_ifaces
, model
.escape(name
)), 'w')
34 def testNoNeedDl(self
):
35 policy
= autopolicy
.AutoPolicy(foo_iface_uri
,
36 download_only
= False)
38 assert policy
.need_download()
39 self
.cache_iface(foo_iface_uri
,
40 """<?xml version="1.0" ?>
41 <interface last-modified="1110752708"
43 xmlns="http://zero-install.sourceforge.net/2004/injector/interface">
45 <summary>Foo</summary>
46 <description>Foo</description>
47 </interface>""" % foo_iface_uri
)
48 iface_cache
.iface_cache
._interfaces
= {}
49 assert not policy
.need_download()
51 def testUnknownAlg(self
):
52 self
.cache_iface(foo_iface_uri
,
53 """<?xml version="1.0" ?>
56 xmlns="http://zero-install.sourceforge.net/2004/injector/interface">
58 <summary>Foo</summary>
59 <description>Foo</description>
60 <implementation id='unknown=123' version='1.0'>
61 <archive href='http://foo/foo.tgz' size='100'/>
63 </interface>""" % foo_iface_uri
)
64 policy
= autopolicy
.AutoPolicy(foo_iface_uri
,
65 download_only
= False)
68 assert policy
.need_download()
70 except model
.SafeException
, ex
:
71 assert 'Unknown digest algorithm' in str(ex
)
73 def testDownload(self
):
74 tmp
= tempfile
.NamedTemporaryFile()
76 """<?xml version="1.0" ?>
78 main='ThisBetterNotExist'
79 xmlns="http://zero-install.sourceforge.net/2004/injector/interface">
81 <summary>Foo</summary>
82 <description>Foo</description>
83 <implementation version='1.0' id='/bin'/>
86 policy
= autopolicy
.AutoPolicy(tmp
.name
, False, False)
88 policy
.download_and_execute(['Hello'])
90 except model
.SafeException
, ex
:
91 assert "ThisBetterNotExist" in str(ex
)
95 tmp
= tempfile
.NamedTemporaryFile()
97 """<?xml version="1.0" ?>
99 xmlns="http://zero-install.sourceforge.net/2004/injector/interface">
101 <summary>Foo</summary>
102 <description>Foo</description>
103 <implementation version='1.0' id='/bin'/>
106 policy
= autopolicy
.AutoPolicy(tmp
.name
, False, False)
108 policy
.download_and_execute(['Hello'])
110 except model
.SafeException
, ex
:
111 assert "library" in str(ex
)
114 def testNeedDL(self
):
115 self
.cache_iface(foo_iface_uri
,
116 """<?xml version="1.0" ?>
117 <interface last-modified="0"
119 main='ThisBetterNotExist'
120 xmlns="http://zero-install.sourceforge.net/2004/injector/interface">
122 <summary>Foo</summary>
123 <description>Foo</description>
124 <implementation version='1.0' id='sha1=123'>
125 <archive href='http://foo/foo.tgz' size='100'/>
127 </interface>""" % foo_iface_uri
)
128 policy
= autopolicy
.AutoPolicy(foo_iface_uri
, False, True)
130 policy
.network_use
= model
.network_full
132 assert policy
.need_download()
134 policy
.start_downloading_impls()
136 except NeedDownload
, ex
:
139 def testBinding(self
):
140 local_impl
= os
.path
.dirname(os
.path
.abspath(__file__
))
141 tmp
= tempfile
.NamedTemporaryFile()
143 """<?xml version="1.0" ?>
145 main='testautopolicy.py'
146 xmlns="http://zero-install.sourceforge.net/2004/injector/interface">
148 <summary>Bar</summary>
149 <description>Bar</description>
151 <requires interface='%s'>
152 <environment name='FOO_PATH' insert='.'/>
153 <environment name='BAR_PATH' insert='.' default='/a:/b'/>
154 <environment name='XDG_DATA_DIRS' insert='.'/>
156 <environment name='SELF_GROUP' insert='group' mode='replace'/>
157 <implementation version='1.0' id='%s'>
158 <environment name='SELF_IMPL' insert='impl' mode='replace'/>
161 </interface>""" % (foo_iface_uri
, local_impl
))
163 self
.cache_iface(foo_iface_uri
,
164 """<?xml version="1.0" ?>
165 <interface last-modified="0"
167 xmlns="http://zero-install.sourceforge.net/2004/injector/interface">
169 <summary>Foo</summary>
170 <description>Foo</description>
171 <implementation version='1.0' id='sha1=123'/>
172 </interface>""" % foo_iface_uri
)
173 cached_impl
= basedir
.save_cache_path('0install.net',
176 policy
= autopolicy
.AutoPolicy(tmp
.name
, False,
178 policy
.network_use
= model
.network_offline
179 os
.environ
['FOO_PATH'] = "old"
180 old
, sys
.stdout
= sys
.stdout
, StringIO()
182 policy
.download_and_execute(['Hello'])
185 self
.assertEquals(cached_impl
+ '/.:old',
186 os
.environ
['FOO_PATH'])
187 self
.assertEquals(cached_impl
+ '/.:/a:/b',
188 os
.environ
['BAR_PATH'])
190 self
.assertEquals(os
.path
.join(local_impl
, 'group'), os
.environ
['SELF_GROUP'])
191 self
.assertEquals(os
.path
.join(local_impl
, 'impl'), os
.environ
['SELF_IMPL'])
193 del os
.environ
['FOO_PATH']
194 if 'XDG_DATA_DIRS' in os
.environ
:
195 del os
.environ
['XDG_DATA_DIRS']
196 os
.environ
['BAR_PATH'] = '/old'
197 old
, sys
.stdout
= sys
.stdout
, StringIO()
199 policy
.download_and_execute(['Hello'])
202 self
.assertEquals(cached_impl
+ '/.',
203 os
.environ
['FOO_PATH'])
204 self
.assertEquals(cached_impl
+ '/.:/old',
205 os
.environ
['BAR_PATH'])
206 self
.assertEquals(cached_impl
+ '/.:/usr/local/share:/usr/share',
207 os
.environ
['XDG_DATA_DIRS'])
210 self
.cache_iface(foo_iface_uri
,
211 """<?xml version="1.0" ?>
212 <interface last-modified="0"
214 xmlns="http://zero-install.sourceforge.net/2004/injector/interface">
216 <summary>Foo</summary>
217 <description>Foo</description>
218 <feed src='http://bar'/>
219 </interface>""" % foo_iface_uri
)
220 self
.cache_iface('http://bar',
221 """<?xml version="1.0" ?>
222 <interface last-modified="0"
224 xmlns="http://zero-install.sourceforge.net/2004/injector/interface">
225 <feed-for interface='%s'/>
227 <summary>Bar</summary>
228 <description>Bar</description>
229 <implementation version='1.0' id='sha1=123'/>
230 </interface>""" % foo_iface_uri
)
231 policy
= autopolicy
.AutoPolicy(foo_iface_uri
, False,
234 policy
.network_use
= model
.network_full
237 foo_iface
= policy
.get_interface(foo_iface_uri
)
238 self
.assertEquals('sha1=123', policy
.implementation
[foo_iface
].id)
240 def testBadConfig(self
):
241 path
= basedir
.save_config_path(namespaces
.config_site
,
242 namespaces
.config_prog
)
243 glob
= os
.path
.join(path
, 'global')
244 assert not os
.path
.exists(glob
)
245 stream
= file(glob
, 'w')
246 stream
.write('hello!')
249 logger
.setLevel(logging
.ERROR
)
250 policy
= autopolicy
.AutoPolicy(foo_iface_uri
,
251 download_only
= False)
252 logger
.setLevel(logging
.WARN
)
254 def testRanking(self
):
255 self
.cache_iface('http://bar',
256 """<?xml version="1.0" ?>
257 <interface last-modified="1110752708"
259 xmlns="http://zero-install.sourceforge.net/2004/injector/interface">
261 <summary>Bar</summary>
262 <description>Bar</description>
263 <implementation id='sha1=125' version='1.0' arch='odd-weird' stability='buggy'/>
264 <implementation id='sha1=126' version='1.0'/>
266 self
.cache_iface(foo_iface_uri
,
267 """<?xml version="1.0" ?>
268 <interface last-modified="1110752708"
270 xmlns="http://zero-install.sourceforge.net/2004/injector/interface">
272 <summary>Foo</summary>
273 <description>Foo</description>
274 <feed src='http://example.com' arch='odd-unknown'/>
275 <feed src='http://bar'/>
276 <implementation id='sha1=123' version='1.0' arch='odd-strange'/>
277 <implementation id='sha1=124' version='1.0' arch='odd-weird'/>
278 </interface>""" % foo_iface_uri
)
279 policy
= autopolicy
.AutoPolicy(foo_iface_uri
,
280 download_only
= False)
281 policy
.network_use
= model
.network_full
283 impls
= policy
.get_ranked_implementations(
284 policy
.get_interface(policy
.root
))
285 assert len(impls
) == 4
287 logger
.setLevel(logging
.ERROR
)
288 policy
.network_use
= model
.network_offline
# Changes sort order tests
289 policy
.recalculate() # Triggers feed-for warning
290 logger
.setLevel(logging
.WARN
)
292 def testNoLocal(self
):
293 self
.cache_iface(foo_iface_uri
,
294 """<?xml version="1.0" ?>
295 <interface last-modified="1110752708"
297 xmlns="http://zero-install.sourceforge.net/2004/injector/interface">
299 <summary>Foo</summary>
300 <description>Foo</description>
301 <feed src='/etc/passwd'/>
302 </interface>""" % foo_iface_uri
)
303 policy
= autopolicy
.AutoPolicy(foo_iface_uri
,
304 download_only
= False)
305 policy
.network_use
= model
.network_offline
307 policy
.get_interface(foo_iface_uri
)
309 except reader
.InvalidInterface
, ex
:
310 assert 'Invalid feed URL' in str(ex
)
312 def testDLfeed(self
):
313 self
.cache_iface(foo_iface_uri
,
314 """<?xml version="1.0" ?>
315 <interface last-modified="1110752708"
317 xmlns="http://zero-install.sourceforge.net/2004/injector/interface">
319 <summary>Foo</summary>
320 <description>Foo</description>
321 <feed src='http://example.com'/>
322 </interface>""" % foo_iface_uri
)
323 policy
= autopolicy
.AutoPolicy(foo_iface_uri
, dry_run
= True)
324 policy
.network_use
= model
.network_full
330 except NeedDownload
, ex
:
333 iface
= policy
.get_interface(foo_iface_uri
)
334 iface
._main
_feed
.feeds
= [model
.Feed('/BadFeed', None, False)]
336 logger
.setLevel(logging
.ERROR
)
337 policy
.recalculate() # Triggers warning
338 logger
.setLevel(logging
.WARN
)
340 def testBestUnusable(self
):
341 self
.cache_iface(foo_iface_uri
,
342 """<?xml version="1.0" ?>
343 <interface last-modified="1110752708"
345 xmlns="http://zero-install.sourceforge.net/2004/injector/interface">
347 <summary>Foo</summary>
348 <description>Foo</description>
349 <implementation id='sha1=123' version='1.0' arch='odd-weird'/>
350 </interface>""" % foo_iface_uri
)
351 policy
= autopolicy
.AutoPolicy(foo_iface_uri
,
352 download_only
= False)
353 policy
.network_use
= model
.network_offline
355 assert not policy
.ready
357 policy
.download_and_execute([])
359 except model
.SafeException
, ex
:
360 assert "Can't find all required implementations" in str(ex
)
362 def testNoArchives(self
):
363 self
.cache_iface(foo_iface_uri
,
364 """<?xml version="1.0" ?>
365 <interface last-modified="1110752708"
367 xmlns="http://zero-install.sourceforge.net/2004/injector/interface">
369 <summary>Foo</summary>
370 <description>Foo</description>
371 <implementation id='sha1=123' version='1.0'/>
372 </interface>""" % foo_iface_uri
)
373 policy
= autopolicy
.AutoPolicy(foo_iface_uri
,
374 download_only
= False)
379 policy
.download_and_execute([])
381 except model
.SafeException
, ex
:
382 assert 'no download locations' in str(ex
)
385 self
.cache_iface(foo_iface_uri
,
386 """<?xml version="1.0" ?>
387 <interface last-modified="1110752708"
389 xmlns="http://zero-install.sourceforge.net/2004/injector/interface">
391 <summary>Foo</summary>
392 <description>Foo</description>
394 <requires interface='%s'/>
395 <implementation id='sha1=123' version='1.0'/>
397 </interface>""" % (foo_iface_uri
, foo_iface_uri
))
398 policy
= autopolicy
.AutoPolicy(foo_iface_uri
,
399 download_only
= False)
403 def testConstraints(self
):
404 self
.cache_iface('http://bar',
405 """<?xml version="1.0" ?>
406 <interface last-modified="1110752708"
408 xmlns="http://zero-install.sourceforge.net/2004/injector/interface">
410 <summary>Bar</summary>
411 <description>Bar</description>
412 <implementation id='sha1=100' version='1.0'/>
413 <implementation id='sha1=150' stability='developer' version='1.5'/>
414 <implementation id='sha1=200' version='2.0'/>
416 self
.cache_iface(foo_iface_uri
,
417 """<?xml version="1.0" ?>
418 <interface last-modified="1110752708"
420 xmlns="http://zero-install.sourceforge.net/2004/injector/interface">
422 <summary>Foo</summary>
423 <description>Foo</description>
425 <requires interface='http://bar'>
428 <implementation id='sha1=123' version='1.0'/>
430 </interface>""" % foo_iface_uri
)
431 policy
= autopolicy
.AutoPolicy(foo_iface_uri
,
432 download_only
= False)
433 policy
.network_use
= model
.network_full
435 #logger.setLevel(logging.DEBUG)
437 #logger.setLevel(logging.WARN)
438 foo_iface
= policy
.get_interface(foo_iface_uri
)
439 bar_iface
= policy
.get_interface('http://bar')
440 assert policy
.implementation
[bar_iface
].id == 'sha1=200'
442 dep
= policy
.implementation
[foo_iface
].dependencies
['http://bar']
443 assert len(dep
.restrictions
) == 1
444 restriction
= dep
.restrictions
[0]
446 restriction
.before
= model
.parse_version('2.0')
448 assert policy
.implementation
[bar_iface
].id == 'sha1=100'
450 restriction
.not_before
= model
.parse_version('1.5')
452 assert policy
.implementation
[bar_iface
].id == 'sha1=150'
454 suite
= unittest
.makeSuite(TestAutoPolicy
)
455 if __name__
== '__main__':
456 sys
.argv
.append('-v')