tests/testautopolicy.py

   1 #!/usr/bin/env python2.3
   2 import sys, tempfile, os, shutil
   3 import unittest
   4 from logging import getLogger, DEBUG, INFO
   5 #getLogger().setLevel(DEBUG)
   6
   7 sys.path.insert(0, '..')
   8
   9 config_home = tempfile.mktemp()
  10 cache_home = tempfile.mktemp()
  11 os.environ['XDG_CONFIG_HOME'] = config_home
  12 os.environ['XDG_CACHE_HOME'] = cache_home
  13
  14 from zeroinstall.injector import model, basedir, autopolicy, gpg, iface_cache
  15 import data
  16 reload(basedir)
  17
  18 foo_iface_uri = 'http://foo'
  19
  20 class TestAutoPolicy(unittest.TestCase):
  21         def setUp(self):
  22                 os.mkdir(config_home, 0700)
  23                 os.mkdir(cache_home, 0700)
  24                 if os.environ.has_key('DISPLAY'):
  25                         del os.environ['DISPLAY']
  26                 self.gnupg_home = tempfile.mktemp()
  27                 os.environ['GNUPGHOME'] = self.gnupg_home
  28                 os.mkdir(self.gnupg_home, 0700)
  29                 stream = tempfile.TemporaryFile()
  30                 stream.write(data.thomas_key)
  31                 stream.seek(0)
  32                 gpg.import_key(stream)
  33                 iface_cache.iface_cache._interfaces = {}
  34
  35         def tearDown(self):
  36                 shutil.rmtree(config_home)
  37                 shutil.rmtree(cache_home)
  38                 shutil.rmtree(self.gnupg_home)
  39
  40         def cache_iface(self, name, data):
  41                 cached_ifaces = basedir.save_cache_path('0install.net',
  42                                                         'interfaces')
  43
  44                 f = file(os.path.join(cached_ifaces, model.escape(name)), 'w')
  45                 f.write(data)
  46                 f.close()
  47
  48         def testNoNeedDl(self):
  49                 policy = autopolicy.AutoPolicy(foo_iface_uri,
  50                                                 download_only = False)
  51                 policy.freshness = 0
  52                 assert policy.need_download()
  53                 self.cache_iface(foo_iface_uri,
  54 """<?xml version="1.0" ?>
  55 <interface last-modified="1110752708"
  56  uri="%s"
  57  xmlns="http://zero-install.sourceforge.net/2004/injector/interface">
  58   <name>Foo</name>
  59   <summary>Foo</summary>
  60   <description>Foo</description>
  61 </interface>""" % foo_iface_uri)
  62                 assert not policy.need_download()
  63
  64         def testDownload(self):
  65                 policy = autopolicy.AutoPolicy(foo_iface_uri, False, False)
  66                 policy.freshness = 0
  67                 self.cache_iface(foo_iface_uri,
  68 """<?xml version="1.0" ?>
  69 <interface last-modified="1110752708"
  70  uri="%s"
  71  main='ThisBetterNotExist'
  72  xmlns="http://zero-install.sourceforge.net/2004/injector/interface">
  73   <name>Foo</name>
  74   <summary>Foo</summary>
  75   <description>Foo</description>
  76   <implementation version='1.0' id='/bin'/>
  77 </interface>""" % foo_iface_uri)
  78                 try:
  79                         policy.download_and_execute(['Hello'])
  80                         assert 0
  81                 except model.SafeException, ex:
  82                         assert "ThisBetterNotExist" in str(ex)
  83
  84 suite = unittest.makeSuite(TestAutoPolicy)
  85 if __name__ == '__main__':
  86         sys.argv.append('-v')
  87         unittest.main()