tests/testunpack.py

   1 #!/usr/bin/env python
   2 from basetest import BaseTest
   3 import sys, tempfile, os
   4 import unittest, logging
   5
   6 sys.path.insert(0, '..')
   7 from zeroinstall.zerostore import unpack, manifest, Store, BadDigest
   8 from zeroinstall import SafeException, support
   9 from zeroinstall.support import find_in_path
  10
  11 def skipIf(condition, reason):
  12         def wrapped(underlying):
  13                 if condition:
  14                         print "Skipped %s: %s" % (underlying.func_name, reason)
  15                         def run(self): pass
  16                         return run
  17                 else:
  18                         return underlying
  19         return wrapped
  20
  21 class AbstractTestUnpack():
  22         def setUp(self):
  23                 BaseTest.setUp(self)
  24
  25                 self.tmpdir = tempfile.mkdtemp('-testunpack')
  26
  27                 os.umask(0022)
  28
  29         def tearDown(self):
  30                 BaseTest.tearDown(self)
  31
  32                 support.ro_rmtree(self.tmpdir)
  33
  34                 assert os.umask(0022) == 0022
  35
  36         def testBadExt(self):
  37                 try:
  38                         unpack.unpack_archive('ftp://foo/file.foo', file('HelloWorld.tgz'), self.tmpdir)
  39                         assert False
  40                 except SafeException, ex:
  41                         assert 'Unknown extension' in str(ex)
  42
  43         def testTgz(self):
  44                 unpack.unpack_archive('ftp://foo/file.tgz', file('HelloWorld.tgz'), self.tmpdir)
  45                 self.assert_manifest('sha1=3ce644dc725f1d21cfcf02562c76f375944b266a')
  46
  47         @skipIf(not find_in_path('hdiutil'), "not running on MacOS X; no hdiutil")
  48         def testDmg(self):
  49                 unpack.unpack_archive('ftp://foo/file.dmg', file('HelloWorld.dmg'), self.tmpdir)
  50                 self.assert_manifest('sha1=3ce644dc725f1d21cfcf02562c76f375944b266a')
  51
  52         def testZip(self):
  53                 unpack.unpack_archive('ftp://foo/file.zip', file('HelloWorld.zip'), self.tmpdir)
  54                 self.assert_manifest('sha1=3ce644dc725f1d21cfcf02562c76f375944b266a')
  55
  56         def testExtract(self):
  57                 unpack.unpack_archive('ftp://foo/file.tgz', file('HelloWorld.tgz'), self.tmpdir, extract = 'HelloWorld')
  58                 self.assert_manifest('sha1=3ce644dc725f1d21cfcf02562c76f375944b266a')
  59
  60         def testExtractOver(self):
  61                 unpack.unpack_archive_over('ftp://foo/file.tgz', file('HelloWorld.tgz'), self.tmpdir, extract = 'HelloWorld')
  62                 self.assert_manifest('sha1=491678c37f77fadafbaae66b13d48d237773a68f')
  63
  64         def testExtractZip(self):
  65                 unpack.unpack_archive('ftp://foo/file.zip', file('HelloWorld.zip'), self.tmpdir, extract = 'HelloWorld')
  66                 self.assert_manifest('sha1=3ce644dc725f1d21cfcf02562c76f375944b266a')
  67
  68         def testExtractIllegal(self):
  69                 try:
  70                         unpack.unpack_archive('ftp://foo/file.tgz', file('HelloWorld.tgz'), self.tmpdir, extract = 'Hello`World`')
  71                         assert False
  72                 except SafeException, ex:
  73                         assert 'Illegal' in str(ex)
  74
  75         def testExtractFails(self):
  76                 stderr = os.dup(2)
  77                 try:
  78                         null = os.open('/dev/null', os.O_WRONLY)
  79                         os.close(2)
  80                         os.dup2(null, 2)
  81                         try:
  82                                 unpack.unpack_archive('ftp://foo/file.tgz', file('HelloWorld.tgz'), self.tmpdir, extract = 'HelloWorld2')
  83                                 assert False
  84                         except SafeException, ex:
  85                                 if ('Failed to extract' not in str(ex) and      # GNU tar
  86                                     'Unable to find' not in str(ex)):           # Python tar
  87                                         raise ex
  88                 finally:
  89                         os.dup2(stderr, 2)
  90
  91         def testTargz(self):
  92                 unpack.unpack_archive('ftp://foo/file.tar.GZ', file('HelloWorld.tgz'), self.tmpdir)
  93                 self.assert_manifest('sha1=3ce644dc725f1d21cfcf02562c76f375944b266a')
  94
  95         def testTbz(self):
  96                 unpack.unpack_archive('ftp://foo/file.tar.bz2', file('HelloWorld.tar.bz2'), self.tmpdir)
  97                 self.assert_manifest('sha1=3ce644dc725f1d21cfcf02562c76f375944b266a')
  98
  99         def testTar(self):
 100                 unpack.unpack_archive('ftp://foo/file.tar', file('HelloWorld.tar'), self.tmpdir)
 101                 self.assert_manifest('sha1new=290eb133e146635fe37713fd58174324a16d595f')
 102
 103         def testRPM(self):
 104                 unpack.unpack_archive('ftp://foo/file.rpm', file('dummy-1-1.noarch.rpm'), self.tmpdir)
 105                 self.assert_manifest('sha1=7be9228c8fe2a1434d4d448c4cf130e3c8a4f53d')
 106
 107         def testDeb(self):
 108                 unpack.unpack_archive('ftp://foo/file.deb', file('dummy_1-1_all.deb'), self.tmpdir)
 109                 self.assert_manifest('sha1new=2c725156ec3832b7980a3de2270b3d8d85d4e3ea')
 110
 111         def testSpecial(self):
 112                 os.chmod(self.tmpdir, 02755)
 113                 store = Store(self.tmpdir)
 114                 store.add_archive_to_cache('sha1=3ce644dc725f1d21cfcf02562c76f375944b266a',
 115                                            file('HelloWorld.tgz'),
 116                                            'http://foo/foo.tgz')
 117
 118         def testBad(self):
 119                 logging.getLogger('').setLevel(logging.ERROR)
 120
 121                 store = Store(self.tmpdir)
 122                 try:
 123                         store.add_archive_to_cache('sha1=3ce644dc725f1d21cfcf02562c76f375944b266b',
 124                                                    file('HelloWorld.tgz'),
 125                                                    'http://foo/foo.tgz')
 126                         assert 0
 127                 except BadDigest:
 128                         pass
 129
 130                 logging.getLogger('').setLevel(logging.INFO)
 131
 132         def assert_manifest(self, required):
 133                 alg_name = required.split('=', 1)[0]
 134                 manifest.fixup_permissions(self.tmpdir)
 135                 sha1 = alg_name + '=' + manifest.add_manifest_file(self.tmpdir, manifest.get_algorithm(alg_name)).hexdigest()
 136                 self.assertEquals(sha1, required)
 137
 138                 # Check permissions are sensible
 139                 for root, dirs, files in os.walk(self.tmpdir):
 140                         for f in files + dirs:
 141                                 full = os.path.join(root, f)
 142                                 if os.path.islink(full): continue
 143                                 full_mode = os.stat(full).st_mode
 144                                 self.assertEquals(0444, full_mode & 0666)       # Must be r-?r-?r-?
 145
 146 class TestUnpackPython(AbstractTestUnpack, BaseTest):
 147         def setUp(self):
 148                 AbstractTestUnpack.setUp(self)
 149                 unpack._tar_version = 'Solaris tar'
 150                 assert not unpack._gnu_tar()
 151
 152 class TestUnpackGNU(AbstractTestUnpack, BaseTest):
 153         def setUp(self):
 154                 AbstractTestUnpack.setUp(self)
 155                 unpack._tar_version = None
 156                 assert unpack._gnu_tar()
 157
 158         # Only available with GNU tar
 159         def testLzma(self):
 160                 unpack.unpack_archive('ftp://foo/file.tar.lzma', file('HelloWorld.tar.lzma'), self.tmpdir)
 161                 self.assert_manifest('sha1new=290eb133e146635fe37713fd58174324a16d595f')
 162
 163 if not unpack._gnu_tar():
 164         print "No GNU tar: SKIPPING tests"
 165         del globals()['TestUnpackGNU']
 166
 167 if __name__ == '__main__':
 168         unittest.main()