mirror of
https://github.com/valitydev/salt.git
synced 2024-11-07 17:09:03 +00:00
133 lines
4.4 KiB
Python
133 lines
4.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
'''
|
|
:codeauthor: :email:`Jayesh Kariya <jayeshk@saltstack.com>`
|
|
'''
|
|
|
|
# Import Python Libs
|
|
from __future__ import absolute_import
|
|
|
|
# Import Salt Testing Libs
|
|
from tests.support.mixins import LoaderModuleMockMixin
|
|
from tests.support.unit import TestCase, skipIf
|
|
from tests.support.mock import (
|
|
MagicMock,
|
|
patch,
|
|
NO_MOCK,
|
|
NO_MOCK_REASON
|
|
)
|
|
|
|
# Import Salt Libs
|
|
import salt.modules.rpm as rpm
|
|
|
|
|
|
@skipIf(NO_MOCK, NO_MOCK_REASON)
|
|
class RpmTestCase(TestCase, LoaderModuleMockMixin):
|
|
'''
|
|
Test cases for salt.modules.rpm
|
|
'''
|
|
def setup_loader_modules(self):
|
|
return {rpm: {'rpm': MagicMock(return_value=MagicMock)}}
|
|
|
|
# 'list_pkgs' function tests: 1
|
|
|
|
def test_list_pkgs(self):
|
|
'''
|
|
Test if it list the packages currently installed in a dict
|
|
'''
|
|
mock = MagicMock(return_value='')
|
|
with patch.dict(rpm.__salt__, {'cmd.run': mock}):
|
|
self.assertDictEqual(rpm.list_pkgs(), {})
|
|
|
|
# 'verify' function tests: 1
|
|
|
|
def test_verify(self):
|
|
'''
|
|
Test if it runs an rpm -Va on a system,
|
|
and returns the results in a dict
|
|
'''
|
|
mock = MagicMock(return_value={'stdout': '',
|
|
'stderr': '',
|
|
'retcode': 0,
|
|
'pid': 12345})
|
|
with patch.dict(rpm.__salt__, {'cmd.run_all': mock}):
|
|
self.assertDictEqual(rpm.verify('httpd'), {})
|
|
|
|
# 'file_list' function tests: 1
|
|
|
|
def test_file_list(self):
|
|
'''
|
|
Test if it list the files that belong to a package.
|
|
'''
|
|
mock = MagicMock(return_value='')
|
|
with patch.dict(rpm.__salt__, {'cmd.run': mock}):
|
|
self.assertDictEqual(rpm.file_list('httpd'),
|
|
{'errors': [], 'files': []})
|
|
|
|
# 'file_dict' function tests: 1
|
|
|
|
def test_file_dict(self):
|
|
'''
|
|
Test if it list the files that belong to a package
|
|
'''
|
|
mock = MagicMock(return_value='')
|
|
with patch.dict(rpm.__salt__, {'cmd.run': mock}):
|
|
self.assertDictEqual(rpm.file_dict('httpd'),
|
|
{'errors': [], 'packages': {}})
|
|
|
|
# 'owner' function tests: 1
|
|
|
|
def test_owner(self):
|
|
'''
|
|
Test if it return the name of the package that owns the file.
|
|
'''
|
|
self.assertEqual(rpm.owner(), '')
|
|
|
|
ret = 'file /usr/bin/salt-jenkins-build is not owned by any package'
|
|
mock = MagicMock(return_value=ret)
|
|
with patch.dict(rpm.__salt__, {'cmd.run_stdout': mock}):
|
|
self.assertEqual(rpm.owner('/usr/bin/salt-jenkins-build'), '')
|
|
|
|
ret = {'/usr/bin/vim': 'vim-enhanced-7.4.160-1.e17.x86_64',
|
|
'/usr/bin/python': 'python-2.7.5-16.e17.x86_64'}
|
|
mock = MagicMock(side_effect=['python-2.7.5-16.e17.x86_64',
|
|
'vim-enhanced-7.4.160-1.e17.x86_64'])
|
|
with patch.dict(rpm.__salt__, {'cmd.run_stdout': mock}):
|
|
self.assertDictEqual(rpm.owner('/usr/bin/python', '/usr/bin/vim'),
|
|
ret)
|
|
|
|
# 'checksum' function tests: 1
|
|
|
|
def test_checksum(self):
|
|
'''
|
|
Test if checksum validate as expected
|
|
'''
|
|
ret = {
|
|
"file1.rpm": True,
|
|
"file2.rpm": False,
|
|
"file3.rpm": False,
|
|
}
|
|
|
|
mock = MagicMock(side_effect=[True, 0, True, 1, False, 0])
|
|
with patch.dict(rpm.__salt__, {'file.file_exists': mock, 'cmd.retcode': mock}):
|
|
self.assertDictEqual(rpm.checksum("file1.rpm", "file2.rpm", "file3.rpm"), ret)
|
|
|
|
@patch('salt.modules.rpm.HAS_RPM', True)
|
|
def test_version_cmp_rpm(self):
|
|
'''
|
|
Test package version is called RPM version if RPM-Python is installed
|
|
|
|
:return:
|
|
'''
|
|
with patch('salt.modules.rpm.rpm.labelCompare', MagicMock(return_value=0)):
|
|
self.assertEqual(0, rpm.version_cmp('1', '2')) # mock returns 0, which means RPM was called
|
|
|
|
@patch('salt.modules.rpm.HAS_RPM', False)
|
|
def test_version_cmp_fallback(self):
|
|
'''
|
|
Test package version is called RPM version if RPM-Python is installed
|
|
|
|
:return:
|
|
'''
|
|
with patch('salt.modules.rpm.rpm.labelCompare', MagicMock(return_value=0)):
|
|
self.assertEqual(-1, rpm.version_cmp('1', '2')) # mock returns -1, a python implementation was called
|