mirror of
https://github.com/valitydev/salt.git
synced 2024-11-07 17:09:03 +00:00
90dcf8287c
Fix problem with TmpStringIO Class on Windows. The module uses this class twice in normal operation. In Windows there's an additional run before the real runs where it is opened with mode='w'. This causes the data to be wiped out. So, this sets it to only save the value in the instance to data if it is not empty. Use the windows path to the hosts file in Windows
230 lines
8.6 KiB
Python
230 lines
8.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
'''
|
|
:codeauthor: :email:`Jayesh Kariya <jayeshk@saltstack.com>`
|
|
'''
|
|
|
|
# Import Python libs
|
|
from __future__ import absolute_import
|
|
|
|
# Import Salt Testing Libs
|
|
from tests.support.mixins import LoaderModuleMockMixin
|
|
from tests.support.unit import TestCase
|
|
from tests.support.mock import (
|
|
MagicMock,
|
|
mock_open,
|
|
patch,
|
|
)
|
|
# Import Salt Libs
|
|
import salt.modules.hosts as hosts
|
|
import salt.utils
|
|
from salt.ext.six.moves import StringIO
|
|
|
|
|
|
class HostsTestCase(TestCase, LoaderModuleMockMixin):
|
|
'''
|
|
TestCase for salt.modules.hosts
|
|
'''
|
|
def setup_loader_modules(self):
|
|
return {hosts: {}}
|
|
|
|
# 'list_hosts' function tests: 1
|
|
|
|
def test_list_hosts(self):
|
|
'''
|
|
Tests return the hosts found in the hosts file
|
|
'''
|
|
with patch('salt.modules.hosts._list_hosts',
|
|
MagicMock(return_value={'10.10.10.10': ['Salt1', 'Salt2']})):
|
|
self.assertDictEqual({'10.10.10.10': ['Salt1', 'Salt2']},
|
|
hosts.list_hosts())
|
|
|
|
# 'get_ip' function tests: 3
|
|
|
|
def test_get_ip(self):
|
|
'''
|
|
Tests return ip associated with the named host
|
|
'''
|
|
with patch('salt.modules.hosts._list_hosts',
|
|
MagicMock(return_value={'10.10.10.10': ['Salt1', 'Salt2']})):
|
|
self.assertEqual('10.10.10.10', hosts.get_ip('Salt1'))
|
|
|
|
self.assertEqual('', hosts.get_ip('Salt3'))
|
|
|
|
def test_get_ip_none(self):
|
|
'''
|
|
Tests return ip associated with the named host
|
|
'''
|
|
with patch('salt.modules.hosts._list_hosts', MagicMock(return_value='')):
|
|
self.assertEqual('', hosts.get_ip('Salt1'))
|
|
|
|
# 'get_alias' function tests: 2
|
|
|
|
def test_get_alias(self):
|
|
'''
|
|
Tests return the list of aliases associated with an ip
|
|
'''
|
|
with patch('salt.modules.hosts._list_hosts',
|
|
MagicMock(return_value={'10.10.10.10': ['Salt1', 'Salt2']})):
|
|
self.assertListEqual(['Salt1', 'Salt2'], hosts.get_alias('10.10.10.10'))
|
|
|
|
def test_get_alias_none(self):
|
|
'''
|
|
Tests return the list of aliases associated with an ip
|
|
'''
|
|
with patch('salt.modules.hosts._list_hosts',
|
|
MagicMock(return_value={'10.10.10.10': ['Salt1', 'Salt2']})):
|
|
self.assertListEqual([], hosts.get_alias('10.10.10.11'))
|
|
|
|
# 'has_pair' function tests: 1
|
|
|
|
def test_has_pair(self):
|
|
'''
|
|
Tests return True / False if the alias is set
|
|
'''
|
|
with patch('salt.modules.hosts._list_hosts',
|
|
MagicMock(return_value={'10.10.10.10': ['Salt1', 'Salt2']})):
|
|
self.assertTrue(hosts.has_pair('10.10.10.10', 'Salt1'))
|
|
|
|
self.assertFalse(hosts.has_pair('10.10.10.10', 'Salt3'))
|
|
|
|
# 'set_host' function tests: 3
|
|
|
|
def test_set_host(self):
|
|
'''
|
|
Tests true if the alias is set
|
|
'''
|
|
hosts_file = '/etc/hosts'
|
|
if salt.utils.is_windows():
|
|
hosts_file = r'C:\Windows\System32\Drivers\etc\hosts'
|
|
|
|
with patch('salt.modules.hosts.__get_hosts_filename',
|
|
MagicMock(return_value=hosts_file)), \
|
|
patch('os.path.isfile', MagicMock(return_value=False)), \
|
|
patch.dict(hosts.__salt__,
|
|
{'config.option': MagicMock(return_value=None)}):
|
|
self.assertFalse(hosts.set_host('10.10.10.10', 'Salt1'))
|
|
|
|
def test_set_host_true(self):
|
|
'''
|
|
Tests true if the alias is set
|
|
'''
|
|
with patch('salt.modules.hosts.__get_hosts_filename',
|
|
MagicMock(return_value='/etc/hosts')), \
|
|
patch('os.path.isfile', MagicMock(return_value=True)), \
|
|
patch('salt.utils.fopen', mock_open()):
|
|
mock_opt = MagicMock(return_value=None)
|
|
with patch.dict(hosts.__salt__, {'config.option': mock_opt}):
|
|
self.assertTrue(hosts.set_host('10.10.10.10', 'Salt1'))
|
|
|
|
def test_set_host_true_remove(self):
|
|
'''
|
|
Test if an empty hosts value removes existing entries
|
|
'''
|
|
with patch('salt.modules.hosts.__get_hosts_filename',
|
|
MagicMock(return_value='/etc/hosts')), \
|
|
patch('os.path.isfile', MagicMock(return_value=True)):
|
|
data = ['\n'.join((
|
|
'1.1.1.1 foo.foofoo foo',
|
|
'2.2.2.2 bar.barbar bar',
|
|
'3.3.3.3 asdf.asdfadsf asdf',
|
|
'1.1.1.1 foofoo.foofoo foofoo',
|
|
))]
|
|
|
|
class TmpStringIO(StringIO):
|
|
def __init__(self, fn, mode='r'):
|
|
initial_value = data[0]
|
|
if 'w' in mode:
|
|
initial_value = ''
|
|
StringIO.__init__(self, initial_value)
|
|
|
|
def __enter__(self):
|
|
return self
|
|
|
|
def __exit__(self, exc_type, exc_value, traceback):
|
|
self.close()
|
|
|
|
def close(self):
|
|
# Don't save unless there's something there. In Windows
|
|
# the class gets initialized the first time with mode = w
|
|
# which sets the initial value to ''. When the class closes
|
|
# it clears out data and causes the test to fail.
|
|
# I don't know why it get's initialized with a mode of 'w'
|
|
# For the purposes of this test data shouldn't be empty
|
|
# This is a problem with this class and not with the hosts
|
|
# module
|
|
if self.getvalue():
|
|
data[0] = self.getvalue()
|
|
StringIO.close(self)
|
|
|
|
expected = '\n'.join((
|
|
'2.2.2.2 bar.barbar bar',
|
|
'3.3.3.3 asdf.asdfadsf asdf',
|
|
)) + '\n'
|
|
|
|
with patch('salt.utils.fopen', TmpStringIO):
|
|
mock_opt = MagicMock(return_value=None)
|
|
with patch.dict(hosts.__salt__, {'config.option': mock_opt}):
|
|
self.assertTrue(hosts.set_host('1.1.1.1', ' '))
|
|
|
|
self.assertEqual(data[0], expected)
|
|
|
|
# 'rm_host' function tests: 2
|
|
|
|
def test_rm_host(self):
|
|
'''
|
|
Tests if specified host entry gets removed from the hosts file
|
|
'''
|
|
with patch('salt.utils.fopen', mock_open()), \
|
|
patch('salt.modules.hosts.__get_hosts_filename',
|
|
MagicMock(return_value='/etc/hosts')), \
|
|
patch('salt.modules.hosts.has_pair',
|
|
MagicMock(return_value=True)), \
|
|
patch('os.path.isfile', MagicMock(return_value=True)):
|
|
mock_opt = MagicMock(return_value=None)
|
|
with patch.dict(hosts.__salt__, {'config.option': mock_opt}):
|
|
self.assertTrue(hosts.rm_host('10.10.10.10', 'Salt1'))
|
|
|
|
def test_rm_host_false(self):
|
|
'''
|
|
Tests if specified host entry gets removed from the hosts file
|
|
'''
|
|
with patch('salt.modules.hosts.has_pair', MagicMock(return_value=False)):
|
|
self.assertTrue(hosts.rm_host('10.10.10.10', 'Salt1'))
|
|
|
|
# 'add_host' function tests: 3
|
|
|
|
def test_add_host(self):
|
|
'''
|
|
Tests if specified host entry gets added from the hosts file
|
|
'''
|
|
hosts_file = '/etc/hosts'
|
|
if salt.utils.is_windows():
|
|
hosts_file = r'C:\Windows\System32\Drivers\etc\hosts'
|
|
|
|
with patch('salt.utils.fopen', mock_open()), \
|
|
patch('salt.modules.hosts.__get_hosts_filename',
|
|
MagicMock(return_value=hosts_file)):
|
|
mock_opt = MagicMock(return_value=None)
|
|
with patch.dict(hosts.__salt__, {'config.option': mock_opt}):
|
|
self.assertTrue(hosts.add_host('10.10.10.10', 'Salt1'))
|
|
|
|
def test_add_host_no_file(self):
|
|
'''
|
|
Tests if specified host entry gets added from the hosts file
|
|
'''
|
|
with patch('salt.utils.fopen', mock_open()), \
|
|
patch('os.path.isfile', MagicMock(return_value=False)):
|
|
mock_opt = MagicMock(return_value=None)
|
|
with patch.dict(hosts.__salt__, {'config.option': mock_opt}):
|
|
self.assertFalse(hosts.add_host('10.10.10.10', 'Salt1'))
|
|
|
|
def test_add_host_create_entry(self):
|
|
'''
|
|
Tests if specified host entry gets added from the hosts file
|
|
'''
|
|
with patch('salt.utils.fopen', mock_open()), \
|
|
patch('os.path.isfile', MagicMock(return_value=True)):
|
|
mock_opt = MagicMock(return_value=None)
|
|
with patch.dict(hosts.__salt__, {'config.option': mock_opt}):
|
|
self.assertTrue(hosts.add_host('10.10.10.10', 'Salt1'))
|