tests/testdistro.py

   1 #!/usr/bin/env python
   2 from basetest import BaseTest, empty_feed
   3 import sys, os
   4 from StringIO import StringIO
   5 import unittest
   6
   7 sys.path.insert(0, '..')
   8 from zeroinstall.injector import distro, model, qdom
   9
  10 def parse_impls(impls, test_distro):
  11         xml = """<?xml version="1.0" ?>
  12                  <interface xmlns="http://zero-install.sourceforge.net/2004/injector/interface">
  13                    <name>Foo</name>
  14                    <summary>Foo</summary>
  15                    <description>Foo</description>
  16                    %s
  17                 </interface>""" % impls
  18         element = qdom.parse(StringIO(xml))
  19         return model.ZeroInstallFeed(element, "myfeed.xml", test_distro)
  20
  21 class TestDistro(BaseTest):
  22         def setUp(self):
  23                 BaseTest.setUp(self)
  24                 self.feed = model.ZeroInstallFeed(empty_feed, local_path = '/empty.xml')
  25
  26         def tearDown(self):
  27                 BaseTest.tearDown(self)
  28
  29         def factory(self, id):
  30                 impl = model.DistributionImplementation(self.feed, id)
  31                 assert id not in self.feed.implementations
  32                 self.feed.implementations[id] = impl
  33                 return impl
  34
  35         def testDefault(self):
  36                 host = distro.Distribution()
  37
  38                 host.get_package_info('gimp', self.factory)
  39                 self.assertEquals(self.feed.implementations, {})
  40
  41         def testDebian(self):
  42                 dpkgdir = os.path.join(os.path.dirname(__file__), 'dpkg')
  43                 host = distro.DebianDistribution(os.path.join(dpkgdir, 'status'))
  44
  45                 self.assertEquals(2, len(host.versions))
  46
  47                 host.get_package_info('gimp', self.factory)
  48                 self.assertEquals({}, self.feed.implementations)
  49
  50                 host.get_package_info('python-bittorrent', self.factory)
  51                 self.assertEquals(1, len(self.feed.implementations))
  52                 bittorrent = self.feed.implementations['package:deb:python-bittorrent:3.4.2-10']
  53                 self.assertEquals('3.4.2-10', bittorrent.get_version())
  54
  55                 host.get_package_info('libxcomposite-dev', self.factory)
  56                 self.assertEquals(2, len(self.feed.implementations))
  57                 libxcomposite = self.feed.implementations['package:deb:libxcomposite-dev:0.3.1-1']
  58                 self.assertEquals('0.3.1-1', libxcomposite.get_version())
  59
  60         def testRPM(self):
  61                 rpmdir = os.path.join(os.path.dirname(__file__), 'rpm')
  62                 os.environ['PATH'] = rpmdir + ':' + self.old_path
  63                 rpm = distro.RPMDistribution(os.path.join(rpmdir, 'status'))
  64
  65                 self.assertEquals(2, len(rpm.versions))
  66
  67                 rpm.get_package_info('yast2-update', self.factory)
  68                 self.assertEquals(1, len(self.feed.implementations))
  69                 yast = self.feed.implementations['package:rpm:yast2-update:2.15.23-21:i586']
  70                 self.assertEquals('2.15.23-21', yast.get_version())
  71                 self.assertEquals('*-i586', yast.arch)
  72
  73                 impls = parse_impls("""
  74                                 <package-implementation distributions="Debian" package="yast2-mail"/>
  75                                 <package-implementation distributions="RPM" package="yast2-update"/>
  76                                 """, rpm).implementations
  77                 assert len(impls) == 1, impls
  78                 impl, = impls
  79                 assert impl == 'package:rpm:yast2-update:2.15.23-21:i586'
  80
  81                 impls = parse_impls("""
  82                                 <package-implementation distributions="RPM" package="yast2-mail"/>
  83                                 <package-implementation distributions="RPM" package="yast2-update"/>
  84                                 """, rpm).implementations
  85                 assert len(impls) == 2, impls
  86
  87                 impls = parse_impls("""
  88                                 <package-implementation distributions="" package="yast2-mail"/>
  89                                 <package-implementation package="yast2-update"/>
  90                                 """, rpm).implementations
  91                 assert len(impls) == 2, impls
  92
  93         def testGentoo(self):
  94                 pkgdir = os.path.join(os.path.dirname(__file__), 'gentoo')
  95                 ebuilds = distro.GentooDistribution(pkgdir)
  96
  97                 ebuilds.get_package_info('media-gfx/gimp', self.factory)
  98                 self.assertEquals({}, self.feed.implementations)
  99
 100                 ebuilds.get_package_info('sys-apps/portage', self.factory)
 101                 self.assertEquals(1, len(self.feed.implementations))
 102                 impl = self.feed.implementations['package:gentoo:sys-apps/portage:2.1.7.16:x86_64']
 103                 self.assertEquals('2.1.7.16', impl.get_version())
 104
 105                 ebuilds.get_package_info('sys-kernel/gentoo-sources', self.factory)
 106                 self.assertEquals(3, len(self.feed.implementations))
 107                 impl = self.feed.implementations['package:gentoo:sys-kernel/gentoo-sources:2.6.30-4:i686']
 108                 self.assertEquals('2.6.30-4', impl.get_version())
 109                 impl = self.feed.implementations['package:gentoo:sys-kernel/gentoo-sources:2.6.32:x86_64']
 110                 self.assertEquals('2.6.32', impl.get_version())
 111
 112         def testCleanVersion(self):
 113                 self.assertEquals('1', distro.try_cleanup_distro_version('1:0.3.1-1'))
 114                 self.assertEquals('0.3.1-1', distro.try_cleanup_distro_version('0.3.1-1ubuntu0'))
 115                 self.assertEquals('0.3-post1-rc2', distro.try_cleanup_distro_version('0.3-post1-rc2'))
 116                 self.assertEquals('0.3.1-2', distro.try_cleanup_distro_version('0.3.1-r2-r3'))
 117
 118 suite = unittest.makeSuite(TestDistro)
 119 if __name__ == '__main__':
 120         sys.argv.append('-v')
 121         unittest.main()