# -*- coding: utf-8 -*- from __future__ import absolute_import # Import Salt Testing libs from salttesting.helpers import ( destructiveTest, requires_network, requires_salt_modules, ensure_in_syspath ) ensure_in_syspath('../../') # Import salt libs import integration class PkgModuleTest(integration.ModuleCase, integration.SaltReturnAssertsMixIn): ''' Validate the pkg module ''' def test_list(self): ''' verify that packages are installed ''' ret = self.run_function('pkg.list_pkgs') self.assertNotEqual(len(ret.keys()), 0) def test_version_cmp(self): ''' test package version comparison on supported platforms ''' func = 'pkg.version_cmp' os_family = self.run_function('grains.item', ['os_family'])['os_family'] if os_family == 'Debian': lt = ['0.2.4-0ubuntu1', '0.2.4.1-0ubuntu1'] eq = ['0.2.4-0ubuntu1', '0.2.4-0ubuntu1'] gt = ['0.2.4.1-0ubuntu1', '0.2.4-0ubuntu1'] self.assertEqual(self.run_function(func, lt), -1) self.assertEqual(self.run_function(func, eq), 0) self.assertEqual(self.run_function(func, gt), 1) else: self.skipTest('{0} is unavailable on {1}'.format(func, os_family)) @requires_network() @destructiveTest def test_mod_del_repo(self): ''' test modifying and deleting a software repository ''' func = 'pkg.mod_repo' os_grain = self.run_function('grains.item', ['os'])['os'] os_release_info = tuple(self.run_function('grains.item', ['osrelease_info'])['osrelease_info']) if os_grain == 'Ubuntu' and os_release_info < (15, 10): repo = 'ppa:saltstack/salt' uri = 'http://ppa.launchpad.net/saltstack/salt/ubuntu' ret = self.run_function(func, [repo, 'comps=main']) self.assertNotEqual(ret, {}) if os_release_info[0] == 12: self.assertIn(repo, ret) else: self.assertIn(uri, ret.keys()[0]) self.run_function('pkg.del_repo', [repo]) else: self.skipTest('{0} is unavailable on {1}'.format(func, os_grain)) def test_owner(self): ''' test finding the package owning a file ''' func = 'pkg.owner' available = self.run_function('sys.doc', [func]) if available: ret = self.run_function(func, ['/bin/ls']) self.assertNotEqual(len(ret), 0) else: os_grain = self.run_function('grains.item', ['os'])['os'] self.skipTest('{0} is unavailable on {1}'.format(func, os_grain)) @requires_network() @destructiveTest def test_install_remove(self): ''' successfully install and uninstall a package ''' pkg = 'htop' version = self.run_function('pkg.version', [pkg]) os_grain = self.run_function('grains.item', ['os'])['os'] os_release = self.run_function('grains.item', ['osrelease'])['osrelease'] if os_grain == 'Ubuntu': if os_release.startswith('12.'): self.skipTest('pkg.install and pkg.remove do not work on ubuntu ' '12 when run from the test suite') def test_install(): install_ret = self.run_function('pkg.install', [pkg]) self.assertIn(pkg, install_ret) def test_remove(): remove_ret = self.run_function('pkg.remove', [pkg]) self.assertIn(pkg, remove_ret) if version: test_remove() test_install() else: test_install() test_remove() @requires_network() @destructiveTest def test_hold_unhold(self): ''' test holding and unholding a package ''' pkg = 'htop' os_family = self.run_function('grains.item', ['os_family'])['os_family'] os_major_release = self.run_function('grains.item', ['osmajorrelease'])['osmajorrelease'] available = self.run_function('sys.doc', ['pkg.hold']) if os_family == 'RedHat': if os_major_release == '5': self.skipTest('`yum versionlock` does not seem to work on RHEL/CentOS 5') if available: if os_family == 'RedHat': lock_pkg = 'yum-versionlock' if os_major_release == '5' else 'yum-plugin-versionlock' versionlock = self.run_function('pkg.version', [lock_pkg]) if not versionlock: self.run_function('pkg.install', [lock_pkg]) hold_ret = self.run_function('pkg.hold', [pkg]) self.assertIn(pkg, hold_ret) self.assertTrue(hold_ret[pkg]['result']) unhold_ret = self.run_function('pkg.unhold', [pkg]) self.assertIn(pkg, unhold_ret) self.assertTrue(hold_ret[pkg]['result']) if os_family == 'RedHat': if not versionlock: self.run_function('pkg.remove', [lock_pkg]) else: os_grain = self.run_function('grains.item', ['os'])['os'] self.skipTest('{0} is unavailable on {1}'.format('pkg.hold', os_grain)) @requires_network() @destructiveTest def test_refresh_db(self): ''' test refreshing the package database ''' func = 'pkg.refresh_db' os_family = self.run_function('grains.item', ['os_family'])['os_family'] if os_family == 'RedHat': ret = self.run_function(func) self.assertIn(ret, (True, None)) elif os_family == 'Debian': ret = self.run_function(func) self.assertNotEqual(ret, {}) if not isinstance(ret, dict): self.skipTest('Upstream repo did not return coherent results. Skipping test.') for source, state in ret.items(): self.assertIn(state, (True, False, None)) else: os_grain = self.run_function('grains.item', ['os'])['os'] self.skipTest('{0} is unavailable on {1}'.format(func, os_grain)) @requires_salt_modules('pkg.info_installed') def test_pkg_info(self): ''' Test returning useful information on Ubuntu systems. ''' func = 'pkg.info_installed' os_family = self.run_function('grains.item', ['os_family'])['os_family'] if os_family == 'Debian': ret = self.run_function(func, ['bash-completion', 'dpkg']) keys = ret.keys() self.assertIn('bash-completion', keys) self.assertIn('dpkg', keys) elif os_family == 'RedHat': ret = self.run_function(func, ['rpm', 'yum']) keys = ret.keys() self.assertIn('rpm', keys) self.assertIn('yum', keys) elif os_family == 'Suse': ret = self.run_function(func, ['less', 'zypper']) keys = ret.keys() self.assertIn('less', keys) self.assertIn('zypper', keys) if __name__ == '__main__': from integration import run_tests run_tests(PkgModuleTest)