mirror of
https://github.com/valitydev/salt.git
synced 2024-11-07 17:09:03 +00:00
97 lines
3.5 KiB
Python
97 lines
3.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Import Python libs
|
|
from __future__ import absolute_import, unicode_literals, print_function
|
|
|
|
# Import Salt Libs
|
|
import salt.modules.win_license as win_license
|
|
|
|
# Import Salt Testing Libs
|
|
from tests.support.mixins import LoaderModuleMockMixin
|
|
from tests.support.unit import TestCase
|
|
from tests.support.mock import (
|
|
MagicMock,
|
|
patch
|
|
)
|
|
|
|
|
|
class LicenseTestCase(TestCase, LoaderModuleMockMixin):
|
|
|
|
def setup_loader_modules(self):
|
|
return {win_license: {}}
|
|
|
|
def test_installed(self):
|
|
'''
|
|
Test to see if the given license key is installed
|
|
'''
|
|
mock = MagicMock(return_value='Partial Product Key: ABCDE')
|
|
with patch.dict(win_license.__salt__, {'cmd.run': mock}):
|
|
out = win_license.installed('AAAAA-AAAAA-AAAAA-AAAA-AAAAA-ABCDE')
|
|
mock.assert_called_once_with(r'cscript C:\Windows\System32\slmgr.vbs /dli')
|
|
self.assertTrue(out)
|
|
|
|
def test_installed_diff(self):
|
|
'''
|
|
Test to see if the given license key is installed when the key is different
|
|
'''
|
|
mock = MagicMock(return_value='Partial Product Key: 12345')
|
|
with patch.dict(win_license.__salt__, {'cmd.run': mock}):
|
|
out = win_license.installed('AAAAA-AAAAA-AAAAA-AAAA-AAAAA-ABCDE')
|
|
mock.assert_called_once_with(r'cscript C:\Windows\System32\slmgr.vbs /dli')
|
|
self.assertFalse(out)
|
|
|
|
def test_install(self):
|
|
'''
|
|
Test installing the given product key
|
|
'''
|
|
mock = MagicMock()
|
|
with patch.dict(win_license.__salt__, {'cmd.run': mock}):
|
|
win_license.install('AAAAA-AAAAA-AAAAA-AAAA-AAAAA-ABCDE')
|
|
mock.assert_called_once_with(r'cscript C:\Windows\System32\slmgr.vbs /ipk '
|
|
'AAAAA-AAAAA-AAAAA-AAAA-AAAAA-ABCDE')
|
|
|
|
def test_uninstall(self):
|
|
'''
|
|
Test uninstalling the given product key
|
|
'''
|
|
mock = MagicMock()
|
|
with patch.dict(win_license.__salt__, {'cmd.run': mock}):
|
|
win_license.uninstall()
|
|
mock.assert_called_once_with(r'cscript C:\Windows\System32\slmgr.vbs /upk')
|
|
|
|
def test_activate(self):
|
|
'''
|
|
Test activating the current product key
|
|
'''
|
|
mock = MagicMock()
|
|
with patch.dict(win_license.__salt__, {'cmd.run': mock}):
|
|
win_license.activate()
|
|
mock.assert_called_once_with(r'cscript C:\Windows\System32\slmgr.vbs /ato')
|
|
|
|
def test_licensed(self):
|
|
'''
|
|
Test checking if the minion is licensed
|
|
'''
|
|
mock = MagicMock(return_value='License Status: Licensed')
|
|
with patch.dict(win_license.__salt__, {'cmd.run': mock}):
|
|
win_license.licensed()
|
|
mock.assert_called_once_with(r'cscript C:\Windows\System32\slmgr.vbs /dli')
|
|
|
|
def test_info(self):
|
|
'''
|
|
Test getting the info about the current license key
|
|
'''
|
|
expected = {
|
|
'description': 'Prof',
|
|
'licensed': True,
|
|
'name': 'Win7',
|
|
'partial_key': '12345'
|
|
}
|
|
|
|
mock = MagicMock(return_value='Name: Win7\r\nDescription: Prof\r\nPartial Product Key: 12345\r\n'
|
|
'License Status: Licensed')
|
|
with patch.dict(win_license.__salt__, {'cmd.run': mock}):
|
|
out = win_license.info()
|
|
mock.assert_called_once_with(r'cscript C:\Windows\System32\slmgr.vbs /dli')
|
|
self.assertEqual(out, expected)
|