mirror of
https://github.com/valitydev/salt.git
synced 2024-11-07 08:58:59 +00:00
135 lines
4.4 KiB
Python
135 lines
4.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
'''
|
|
:codeauthor: :email:`Eric Vz <eric@base10.org>`
|
|
'''
|
|
|
|
# Import Python Libs
|
|
from __future__ import absolute_import
|
|
|
|
# Import Salt Testing Libs
|
|
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.pacman as pacman
|
|
|
|
|
|
@skipIf(NO_MOCK, NO_MOCK_REASON)
|
|
class PacmanTestCase(TestCase):
|
|
'''
|
|
Test cases for salt.modules.pacman
|
|
'''
|
|
|
|
def setUp(self):
|
|
pacman.__salt__ = {}
|
|
pacman.__context__ = {}
|
|
|
|
def test_list_pkgs(self):
|
|
'''
|
|
Test if it list the packages currently installed in a dict
|
|
'''
|
|
cmdmock = MagicMock(return_value='A 1.0\nB 2.0')
|
|
sortmock = MagicMock()
|
|
stringifymock = MagicMock()
|
|
mock_ret = {'A': ['1.0'], 'B': ['2.0']}
|
|
with patch.dict(pacman.__salt__, {
|
|
'cmd.run': cmdmock,
|
|
'pkg_resource.add_pkg': lambda pkgs, name, version: pkgs.setdefault(name, []).append(version),
|
|
'pkg_resource.sort_pkglist': sortmock,
|
|
'pkg_resource.stringify': stringifymock
|
|
}):
|
|
self.assertDictEqual(pacman.list_pkgs(), mock_ret)
|
|
|
|
sortmock.assert_called_with(mock_ret)
|
|
stringifymock.assert_called_with(mock_ret)
|
|
|
|
def test_list_pkgs_as_list(self):
|
|
'''
|
|
Test if it lists the packages currently installed in a dict
|
|
'''
|
|
cmdmock = MagicMock(return_value='A 1.0\nB 2.0')
|
|
sortmock = MagicMock()
|
|
stringifymock = MagicMock()
|
|
mock_ret = {'A': ['1.0'], 'B': ['2.0']}
|
|
with patch.dict(pacman.__salt__, {
|
|
'cmd.run': cmdmock,
|
|
'pkg_resource.add_pkg': lambda pkgs, name, version: pkgs.setdefault(name, []).append(version),
|
|
'pkg_resource.sort_pkglist': sortmock,
|
|
'pkg_resource.stringify': stringifymock
|
|
}):
|
|
self.assertDictEqual(pacman.list_pkgs(True), mock_ret)
|
|
|
|
sortmock.assert_called_with(mock_ret)
|
|
self.assertTrue(stringifymock.call_count == 0)
|
|
|
|
def test_group_list(self):
|
|
'''
|
|
Test if it lists the available groups
|
|
'''
|
|
|
|
def cmdlist(cmd, **kwargs):
|
|
'''
|
|
Handle several different commands being run
|
|
'''
|
|
if cmd == ['pacman', '-Sgg']:
|
|
return 'group-a pkg1\ngroup-a pkg2\ngroup-f pkg9\ngroup-c pkg3\ngroup-b pkg4'
|
|
elif cmd == ['pacman', '-Qg']:
|
|
return 'group-a pkg1\ngroup-b pkg4'
|
|
else:
|
|
return 'Untested command ({0}, {1})!'.format(cmd, kwargs)
|
|
|
|
cmdmock = MagicMock(side_effect=cmdlist)
|
|
|
|
sortmock = MagicMock()
|
|
with patch.dict(pacman.__salt__, {
|
|
'cmd.run': cmdmock,
|
|
'pkg_resource.sort_pkglist': sortmock
|
|
}):
|
|
self.assertDictEqual(pacman.group_list(), {'available': ['group-c', 'group-f'], 'installed': ['group-b'], 'partially_installed': ['group-a']})
|
|
|
|
def test_group_info(self):
|
|
'''
|
|
Test if it shows the packages in a group
|
|
'''
|
|
|
|
def cmdlist(cmd, **kwargs):
|
|
'''
|
|
Handle several different commands being run
|
|
'''
|
|
if cmd == ['pacman', '-Sgg', 'testgroup']:
|
|
return 'testgroup pkg1\ntestgroup pkg2'
|
|
else:
|
|
return 'Untested command ({0}, {1})!'.format(cmd, kwargs)
|
|
|
|
cmdmock = MagicMock(side_effect=cmdlist)
|
|
|
|
sortmock = MagicMock()
|
|
with patch.dict(pacman.__salt__, {
|
|
'cmd.run': cmdmock,
|
|
'pkg_resource.sort_pkglist': sortmock
|
|
}):
|
|
self.assertEqual(pacman.group_info('testgroup')['default'], ['pkg1', 'pkg2'])
|
|
|
|
def test_group_diff(self):
|
|
'''
|
|
Test if it shows the difference between installed and target group contents
|
|
'''
|
|
|
|
listmock = MagicMock(return_value={'A': ['1.0'], 'B': ['2.0']})
|
|
groupmock = MagicMock(return_value={
|
|
'mandatory': [],
|
|
'optional': [],
|
|
'default': ['A', 'C'],
|
|
'conditional': []})
|
|
with patch.dict(pacman.__salt__, {
|
|
'pkg.list_pkgs': listmock,
|
|
'pkg.group_info': groupmock
|
|
}):
|
|
results = pacman.group_diff('testgroup')
|
|
self.assertEqual(results['default'], {'installed': ['A'], 'not installed': ['C']})
|