mirror of
https://github.com/valitydev/salt.git
synced 2024-11-07 08:58:59 +00:00
e57f170001
while here, adjust the sysctl state documentation as it can also be used for configuring other kernels
75 lines
2.5 KiB
Python
75 lines
2.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
'''
|
|
:codeauthor: :email:`Jasper Lievisse Adriaanse <j@jasper.la>`
|
|
'''
|
|
|
|
# Import Python libs
|
|
from __future__ import absolute_import
|
|
|
|
# Import Salt Libs
|
|
import salt.modules.openbsd_sysctl as openbsd_sysctl
|
|
from salt.exceptions import CommandExecutionError
|
|
|
|
# Import Salt Testing Libs
|
|
from tests.support.mixins import LoaderModuleMockMixin
|
|
from tests.support.unit import skipIf, TestCase
|
|
from tests.support.mock import (
|
|
MagicMock,
|
|
patch,
|
|
NO_MOCK,
|
|
NO_MOCK_REASON
|
|
)
|
|
|
|
|
|
@skipIf(NO_MOCK, NO_MOCK_REASON)
|
|
class OpenBSDSysctlTestCase(TestCase, LoaderModuleMockMixin):
|
|
'''
|
|
TestCase for salt.modules.openbsd_sysctl module
|
|
'''
|
|
def setup_loader_modules(self):
|
|
return {openbsd_sysctl: {}}
|
|
|
|
def test_get(self):
|
|
'''
|
|
Tests the return of get function
|
|
'''
|
|
mock_cmd = MagicMock(return_value='OpenBSD')
|
|
with patch.dict(openbsd_sysctl.__salt__, {'cmd.run': mock_cmd}):
|
|
self.assertEqual(openbsd_sysctl.get('kern.ostype'), 'OpenBSD')
|
|
|
|
def test_assign_cmd_failed(self):
|
|
'''
|
|
Tests if the assignment was successful or not
|
|
'''
|
|
cmd = {'pid': 1234, 'retcode': 1, 'stderr': '',
|
|
'stdout': 'kern.securelevel: 1 -> 9000'}
|
|
mock_cmd = MagicMock(return_value=cmd)
|
|
with patch.dict(openbsd_sysctl.__salt__, {'cmd.run_all': mock_cmd}):
|
|
self.assertRaises(CommandExecutionError,
|
|
openbsd_sysctl.assign,
|
|
'kern.securelevel', 9000)
|
|
|
|
def test_assign_cmd_eperm(self):
|
|
'''
|
|
Tests if the assignment was not permitted.
|
|
'''
|
|
cmd = {'pid': 1234, 'retcode': 0, 'stdout': '',
|
|
'stderr': 'sysctl: ddb.console: Operation not permitted'}
|
|
mock_cmd = MagicMock(return_value=cmd)
|
|
with patch.dict(openbsd_sysctl.__salt__, {'cmd.run_all': mock_cmd}):
|
|
self.assertRaises(CommandExecutionError,
|
|
openbsd_sysctl.assign,
|
|
'ddb.console', 1)
|
|
|
|
def test_assign(self):
|
|
'''
|
|
Tests the return of successful assign function
|
|
'''
|
|
cmd = {'pid': 1234, 'retcode': 0, 'stderr': '',
|
|
'stdout': 'net.inet.ip.forwarding: 0 -> 1'}
|
|
ret = {'net.inet.ip.forwarding': '1'}
|
|
mock_cmd = MagicMock(return_value=cmd)
|
|
with patch.dict(openbsd_sysctl.__salt__, {'cmd.run_all': mock_cmd}):
|
|
self.assertEqual(openbsd_sysctl.assign(
|
|
'net.inet.ip.forwarding', 1), ret)
|