mirror of
https://github.com/valitydev/salt.git
synced 2024-11-09 01:36:48 +00:00
126 lines
4.9 KiB
Python
126 lines
4.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
'''
|
|
:codeauthor: :email:`Nicole Thomas <nicole@saltstack.com>`
|
|
'''
|
|
|
|
# Import python libs
|
|
from __future__ import absolute_import
|
|
import logging
|
|
|
|
# Import Salt Testing Libs
|
|
from salttesting import TestCase, skipIf
|
|
from salttesting.helpers import ensure_in_syspath
|
|
from salttesting.mock import (
|
|
MagicMock,
|
|
patch,
|
|
mock_open,
|
|
call,
|
|
NO_MOCK,
|
|
NO_MOCK_REASON
|
|
)
|
|
|
|
ensure_in_syspath('../../')
|
|
|
|
# Import Salt Libs
|
|
from salt.modules import dnsutil
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
mock_hosts_file = '##\n' \
|
|
'# Host Database\n' \
|
|
'#\n' \
|
|
'# localhost is used to configure the loopback interface\n' \
|
|
'# when the system is booting. Do not change this entry.\n' \
|
|
'##\n' \
|
|
'127.0.0.1 localhost\n' \
|
|
'255.255.255.255 broadcasthost\n' \
|
|
'::1 localhost\n' \
|
|
'fe80::1%lo0 localhost'
|
|
|
|
mock_hosts_file_rtn = {'::1': ['localhost'], '255.255.255.255': ['broadcasthost'],
|
|
'127.0.0.1': ['localhost'], 'fe80::1%lo0': ['localhost']}
|
|
|
|
mock_soa_zone = '$TTL 3D\n' \
|
|
'@ IN SOA land-5.com. root.land-5.com. (\n' \
|
|
'199609203 ; Serial\n' \
|
|
'28800 ; Refresh\n' \
|
|
'7200 ; Retry\n' \
|
|
'604800 ; Expire\n' \
|
|
'86400) ; Minimum TTL\n' \
|
|
'NS land-5.com.\n\n' \
|
|
'1 PTR localhost.'
|
|
|
|
if NO_MOCK is False:
|
|
mock_calls_list = [call.read(), call.write('##\n'),
|
|
call.write('# Host Database\n'),
|
|
call.write('#\n'),
|
|
call.write('# localhost is used to configure the loopback interface\n'),
|
|
call.write('# when the system is booting. Do not change this entry.\n'),
|
|
call.write('##\n'),
|
|
call.write('127.0.0.1 localhost'),
|
|
call.write('\n'),
|
|
call.write('255.255.255.255 broadcasthost'),
|
|
call.write('\n'),
|
|
call.write('::1 localhost'),
|
|
call.write('\n'),
|
|
call.write('fe80::1%lo0 localhost'),
|
|
call.write('\n'),
|
|
call.close()]
|
|
|
|
|
|
@skipIf(NO_MOCK, NO_MOCK_REASON)
|
|
class DNSUtilTestCase(TestCase):
|
|
def test_parse_hosts(self):
|
|
with patch('salt.utils.fopen', mock_open(read_data=mock_hosts_file)):
|
|
self.assertEqual(dnsutil.parse_hosts(), {'::1': ['localhost'],
|
|
'255.255.255.255': ['broadcasthost'],
|
|
'127.0.0.1': ['localhost'],
|
|
'fe80::1%lo0': ['localhost']})
|
|
|
|
@patch('salt.modules.dnsutil.parse_hosts', MagicMock(return_value=mock_hosts_file_rtn))
|
|
def test_hosts_append(self):
|
|
with patch('salt.utils.fopen', mock_open(read_data=mock_hosts_file)) as m_open:
|
|
dnsutil.hosts_append('/etc/hosts', '127.0.0.1', 'ad1.yuk.co,ad2.yuk.co')
|
|
helper_open = m_open()
|
|
helper_open.write.assert_called_once_with('\n127.0.0.1 ad1.yuk.co ad2.yuk.co')
|
|
|
|
def test_hosts_remove(self):
|
|
to_remove = 'ad1.yuk.co'
|
|
new_mock_file = mock_hosts_file + '\n127.0.0.1 ' + to_remove + '\n'
|
|
with patch('salt.utils.fopen', mock_open(read_data=new_mock_file)) as m_open:
|
|
dnsutil.hosts_remove('/etc/hosts', to_remove)
|
|
helper_open = m_open()
|
|
calls_list = helper_open.method_calls
|
|
self.assertEqual(calls_list, mock_calls_list)
|
|
|
|
@skipIf(True, 'Waiting on bug report fixes')
|
|
def test_parse_zone(self):
|
|
with patch('salt.utils.fopen', mock_open(read_data=mock_soa_zone)):
|
|
log.debug(mock_soa_zone)
|
|
log.debug(dnsutil.parse_zone('/var/lib/named/example.com.zone'))
|
|
|
|
def test_to_seconds_hour(self):
|
|
self.assertEqual(dnsutil._to_seconds('4H'), 14400,
|
|
msg='Did not detect valid hours as invalid')
|
|
|
|
def test_to_seconds_day(self):
|
|
self.assertEqual(dnsutil._to_seconds('1D'), 86400,
|
|
msg='Did not detect valid day as invalid')
|
|
|
|
def test_to_seconds_week(self):
|
|
self.assertEqual(dnsutil._to_seconds('2W'), 604800,
|
|
msg='Did not set time greater than one week to one week')
|
|
|
|
def test_to_seconds_empty(self):
|
|
self.assertEqual(dnsutil._to_seconds(''), 604800,
|
|
msg='Did not set empty time to one week')
|
|
|
|
def test_to_seconds_large(self):
|
|
self.assertEqual(dnsutil._to_seconds('604801'), 604800,
|
|
msg='Did not set time greater than one week to one week')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
from integration import run_tests
|
|
run_tests(DNSUtilTestCase, needs_daemon=False)
|