valitydev/salt
14
0
Fork 0
mirror of https://github.com/valitydev/salt.git synced 2024-11-07 08:58:59 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity
5fb9e8b22d
salt/tests/unit/beacons/test_btmp_beacon.py

209 lines
13 KiB
Python
Raw Normal View History

Adding test_btmp_beacon.
2017-09-22 18:57:34 +00:00
# coding: utf-8
# Python libs
from __future__ import absolute_import
Swapping out time.time for datetime.datetime.now for finding the current time. Fixing failing lists with a more accurate and portable mocking of the current time.
2018-06-01 22:59:17 +00:00
import datetime
Adding test_btmp_beacon.
2017-09-22 18:57:34 +00:00
import logging
import sys
# Salt testing libs
from tests.support.unit import skipIf, TestCase
from tests.support.mock import NO_MOCK, NO_MOCK_REASON, patch, MagicMock, mock_open
from tests.support.mixins import LoaderModuleMockMixin
# Salt libs
import salt.beacons.btmp as btmp
Swapping out time.time for datetime.datetime.now for finding the current time. Fixing failing lists with a more accurate and portable mocking of the current time.
2018-06-01 22:59:17 +00:00
# pylint: disable=import-error
try:
add disable=unused-import for dateutil.parser
2018-06-02 02:37:05 +00:00
import dateutil.parser as dateutil_parser # pylint: disable=unused-import
Swapping out time.time for datetime.datetime.now for finding the current time. Fixing failing lists with a more accurate and portable mocking of the current time.
2018-06-01 22:59:17 +00:00
_TIME_SUPPORTED = True
except ImportError:
_TIME_SUPPORTED = False
Adding test_btmp_beacon.
2017-09-22 18:57:34 +00:00
if sys.version_info >= (3,):
raw = bytes('\x06\x00\x00\x00Nt\x00\x00ssh:notty\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00garet\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00::1\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xdd\xc7\xc2Y\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00', 'utf-8')
pack = (6, 29774, b'ssh:notty\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00', b'\x00\x00\x00\x00', b'garet\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00', b'::1\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00', 0, 0, 0, 1505937373, 0, 0, 0, 0, 16777216)
else:
raw = b'\x06\x00\x00\x00Nt\x00\x00ssh:notty\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00garet\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00::1\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xdd\xc7\xc2Y\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'
pack = (6, 29774, 'ssh:notty\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00', '\x00\x00\x00\x00', 'garet\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00', '::1\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00', 0, 0, 0, 1505937373, 0, 0, 0, 0, 16777216)
log = logging.getLogger(__name__)
@skipIf(NO_MOCK, NO_MOCK_REASON)
class BTMPBeaconTestCase(TestCase, LoaderModuleMockMixin):
'''
Test case for salt.beacons.[s]
'''
def setup_loader_modules(self):
return {
btmp: {
'__context__': {'btmp.loc': 2},
'__salt__': {},
}
}
def test_non_list_config(self):
config = {}
ret = btmp.validate(config)
self.assertEqual(ret, (False, 'Configuration for btmp beacon must'
' be a list.'))
def test_empty_config(self):
config = [{}]
ret = btmp.validate(config)
self.assertEqual(ret, (True, 'Valid beacon configuration'))
def test_no_match(self):
Consolidating duplicate code. Adding additional tests. Fixing some btmp vs wtmp typos.
2018-05-24 02:31:46 +00:00
config = [{'users': {'gareth': {'time_range': {'end': '09-22-2017 5pm',
'start': '09-22-2017 3pm'}}}}
Adding test_btmp_beacon.
2017-09-22 18:57:34 +00:00
]
ret = btmp.validate(config)
self.assertEqual(ret, (True, 'Valid beacon configuration'))
Mock open to /var/log/btmp and /var/log/wtmp - btmp may not exist - wtmp may not exist, or it may not contain ASCII data Unbreaks tests on OpenBSD 6.2
2017-10-21 04:43:45 +00:00
with patch('salt.utils.files.fopen', mock_open()) as m_open:
ret = btmp.beacon(config)
m_open.assert_called_with(btmp.BTMP, 'rb')
self.assertEqual(ret, [])
Adding test_btmp_beacon.
2017-09-22 18:57:34 +00:00
Consolidating duplicate code. Adding additional tests. Fixing some btmp vs wtmp typos.
2018-05-24 02:31:46 +00:00
def test_invalid_users(self):
config = [{'users': ['gareth']}]
ret = btmp.validate(config)
self.assertEqual(ret, (False, 'User configuration for btmp beacon must be a dictionary.'))
def test_invalid_groups(self):
config = [{'groups': ['docker']}]
ret = btmp.validate(config)
self.assertEqual(ret, (False, 'Group configuration for btmp beacon must be a dictionary.'))
def test_default_invalid_time_range(self):
config = [{'defaults': {'time_range': {'start': '3pm'}}}]
ret = btmp.validate(config)
self.assertEqual(ret, (False, 'The time_range parameter for btmp beacon must contain start & end options.'))
def test_users_invalid_time_range(self):
config = [{'users': {'gareth': {'time_range': {'start': '3pm'}}}}]
ret = btmp.validate(config)
self.assertEqual(ret, (False, 'The time_range parameter for btmp beacon must contain start & end options.'))
def test_groups_invalid_time_range(self):
config = [{'groups': {'docker': {'time_range': {'start': '3pm'}}}}]
ret = btmp.validate(config)
self.assertEqual(ret, (False, 'The time_range parameter for btmp beacon must contain start & end options.'))
Adding test_btmp_beacon.
2017-09-22 18:57:34 +00:00
def test_match(self):
with patch('salt.utils.files.fopen',
mock_open(read_data=raw)):
with patch('struct.unpack',
MagicMock(return_value=pack)):
config = [{'users': {'garet': {}}}]
ret = btmp.validate(config)
self.assertEqual(ret, (True, 'Valid beacon configuration'))
_expected = [{'addr': 1505937373,
'exit_status': 0,
'inittab': '',
'hostname': '::1',
'PID': 29774,
'session': 0,
'user':
'garet',
'time': 0,
'line': 'ssh:notty',
'type': 6}]
ret = btmp.beacon(config)
self.assertEqual(ret, _expected)
Fixing skipIf in tests.
2018-06-02 19:14:50 +00:00
@skipIf(not _TIME_SUPPORTED, 'dateutil.parser is missing.')
Adding test_btmp_beacon.
2017-09-22 18:57:34 +00:00
def test_match_time(self):
with patch('salt.utils.files.fopen',
mock_open(read_data=raw)):
Swapping out time.time for datetime.datetime.now for finding the current time. Fixing failing lists with a more accurate and portable mocking of the current time.
2018-06-01 22:59:17 +00:00
mock_now = datetime.datetime(2017, 9, 22, 16, 0, 0, 0)
with patch('datetime.datetime', MagicMock()), \
patch('datetime.datetime.now',
MagicMock(return_value=mock_now)):
Adding test_btmp_beacon.
2017-09-22 18:57:34 +00:00
with patch('struct.unpack',
MagicMock(return_value=pack)):
Consolidating duplicate code. Adding additional tests. Fixing some btmp vs wtmp typos.
2018-05-24 02:31:46 +00:00
config = [{'users': {'garet': {'time_range': {'end': '09-22-2017 5pm',
Fixing date in unit.beacons.test_btmp_beacon.BTMPBeaconTestCase.test_match_time
2018-05-24 16:56:24 +00:00
'start': '09-22-2017 3pm'}}}}
Adding test_btmp_beacon.
2017-09-22 18:57:34 +00:00
]
ret = btmp.validate(config)
self.assertEqual(ret, (True, 'Valid beacon configuration'))
_expected = [{'addr': 1505937373,
'exit_status': 0,
'inittab': '',
'hostname': '::1',
'PID': 29774,
'session': 0,
'user':
'garet',
'time': 0,
'line': 'ssh:notty',
'type': 6}]
ret = btmp.beacon(config)
self.assertEqual(ret, _expected)
Adding group support in the btmp and wtmp beacons.
2018-05-17 21:46:00 +00:00
def test_match_group(self):
for groupadd in ('salt.modules.aix_group',
fixing beacons group tests.
2018-05-20 04:35:37 +00:00
'salt.modules.mac_group',
Adding group support in the btmp and wtmp beacons.
2018-05-17 21:46:00 +00:00
'salt.modules.pw_group',
'salt.modules.solaris_group',
fixing beacons group tests.
2018-05-20 04:35:37 +00:00
'salt.modules.win_groupadd'):
Adding group support in the btmp and wtmp beacons.
2018-05-17 21:46:00 +00:00
mock_group_info = {'passwd': 'x',
'gid': 100,
'name': 'users',
'members': ['garet']}
with patch('salt.utils.files.fopen',
mock_open(read_data=raw)):
with patch('time.time',
MagicMock(return_value=1506121200)):
with patch('struct.unpack',
MagicMock(return_value=pack)):
fixing beacons group tests.
2018-05-20 04:35:37 +00:00
with patch('{0}.info'.format(groupadd),
new=MagicMock(return_value=mock_group_info)):
Consolidating duplicate code. Adding additional tests. Fixing some btmp vs wtmp typos.
2018-05-24 02:31:46 +00:00
config = [{'group': {'users': {'time_range': {'end': '5pm',
'start': '3pm'}}}}
Adding group support in the btmp and wtmp beacons.
2018-05-17 21:46:00 +00:00
]
ret = btmp.validate(config)
self.assertEqual(ret,
(True, 'Valid beacon configuration'))
_expected = [{'addr': 1505937373,
'exit_status': 0,
'inittab': '',
'hostname': '::1',
'PID': 29774,
'session': 0,
'user':
'garet',
'time': 0,
'line': 'ssh:notty',
'type': 6}]
ret = btmp.beacon(config)
self.assertEqual(ret, _expected)
Reference in New Issue Copy Permalink