saltstack/salt
1
0
Fork 0
mirror of https://github.com/saltstack/salt.git synced 2025-04-17 10:10:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Adding states/vbox_guest.py unit tests

Browse source
This commit is contained in:
Rahul Handay 2015-05-12 22:31:03 +05:30
parent 9beb7bc529
commit 4cde78a58a
1 changed files with 131 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

131
tests/unit/states/vbox_guest_test.py Normal file
View file

@ -0,0 +1,131 @@
# -*- coding: utf-8 -*-
'''
:codeauthor: :email:`Rahul Handay <rahulha@saltstack.com>`
'''
# Import Python Libs
from __future__ import absolute_import
# Import Salt Testing Libs
from salttesting import TestCase, skipIf
from salttesting.helpers import ensure_in_syspath
from salttesting.mock import (
MagicMock,
patch,
NO_MOCK,
NO_MOCK_REASON
)
ensure_in_syspath('../../')
# Import Salt Libs
from salt.states import vbox_guest
# Globals
vbox_guest.__salt__ = {}
vbox_guest.__opts__ = {}
@skipIf(NO_MOCK, NO_MOCK_REASON)
class VboxGuestTestCase(TestCase):
'''
Validate the vbox_guest state
'''
def test_additions_installed(self):
'''
Test to ensure that the VirtualBox Guest Additions are installed
'''
ret = {'name': 'salt',
'changes': {},
'result': True,
'comment': ''}
mock = MagicMock(side_effect=[True, False, False, False])
with patch.dict(vbox_guest.__salt__,
{"vbox_guest.additions_version": mock,
"vbox_guest.additions_install": mock}):
ret.update({'comment': 'System already in the correct state'})
self.assertDictEqual(vbox_guest.additions_installed('salt'), ret)
with patch.dict(vbox_guest.__opts__, {"test": True}):
ret.update({'changes': {'new': True, 'old': False},
'comment': 'The state of VirtualBox Guest'
' Additions will be changed.', 'result': None})
self.assertDictEqual(vbox_guest.additions_installed('salt'),
ret)
with patch.dict(vbox_guest.__opts__, {"test": False}):
ret.update({'changes': {'new': False, 'old': False},
'comment': 'The state of VirtualBox Guest'
' Additions was changed!', 'result': False})
self.assertDictEqual(vbox_guest.additions_installed('salt'),
ret)
def test_additions_removed(self):
'''
Test to ensure that the VirtualBox Guest Additions are removed.
'''
ret = {'name': 'salt',
'changes': {},
'result': True,
'comment': ''}
mock = MagicMock(side_effect=[False, True, True, True])
with patch.dict(vbox_guest.__salt__,
{"vbox_guest.additions_version": mock,
"vbox_guest.additions_remove": mock}):
ret.update({'comment': 'System already in the correct state'})
self.assertDictEqual(vbox_guest.additions_removed('salt'), ret)
with patch.dict(vbox_guest.__opts__, {"test": True}):
ret.update({'changes': {'new': True, 'old': True},
'comment': 'The state of VirtualBox Guest'
' Additions will be changed.', 'result': None})
self.assertDictEqual(vbox_guest.additions_removed('salt'),
ret)
with patch.dict(vbox_guest.__opts__, {"test": False}):
ret.update({'comment': 'The state of VirtualBox Guest'
' Additions was changed!', 'result': True})
self.assertDictEqual(vbox_guest.additions_removed('salt'),
ret)
def test_grantaccess_to_sharedfolders(self):
'''
Test to grant access to auto-mounted shared folders to the users.
'''
ret = {'name': 'AB',
'changes': {},
'result': True,
'comment': ''}
mock = MagicMock(side_effect=[['AB'], 'salt', 'salt', 'salt'])
with patch.dict(vbox_guest.__salt__,
{"vbox_guest.list_shared_folders_users": mock,
"vbox_guest.grant_access_to_shared_folders_to": mock}
):
ret.update({'comment': 'System already in the correct state'})
self.assertMethod(ret)
with patch.dict(vbox_guest.__opts__, {"test": True}):
ret.update({'changes': {'new': ['AB'], 'old': 'salt'},
'comment': 'List of users who have access to'
' auto-mounted shared folders will be changed',
'result': None})
self.assertMethod(ret)
with patch.dict(vbox_guest.__opts__, {"test": False}):
ret.update({'changes': {'new': 'salt', 'old': 'salt'},
'comment': 'List of users who have access to'
' auto-mounted shared folders was changed',
'result': True})
self.assert_method(ret)
def assert_method(self, ret):
'''
Method call for assert statements
'''
self.assertDictEqual(vbox_guest.grant_access_to_shared_folders_to('AB'),
ret)
if __name__ == '__main__':
from integration import run_tests
run_tests(VboxGuestTestCase, needs_daemon=False)