mirror of
https://github.com/valitydev/salt.git
synced 2024-11-09 01:36:48 +00:00
445 lines
22 KiB
Python
445 lines
22 KiB
Python
# -*- coding: utf-8 -*-
|
|
'''
|
|
:codeauthor: :email:`Lukas Raska <lukas@raska.me>`
|
|
'''
|
|
# Import Python libs
|
|
from __future__ import absolute_import, print_function, unicode_literals
|
|
|
|
# Import Salt Testing Libs
|
|
from tests.support.mixins import LoaderModuleMockMixin
|
|
from tests.support.unit import skipIf, TestCase
|
|
from tests.support.mock import (
|
|
NO_MOCK,
|
|
NO_MOCK_REASON,
|
|
MagicMock,
|
|
patch)
|
|
|
|
from salt.exceptions import CommandExecutionError
|
|
|
|
# Import Salt Libs
|
|
import salt.utils.dictdiffer as dictdiffer
|
|
from salt.states import elasticsearch
|
|
|
|
|
|
@skipIf(NO_MOCK, NO_MOCK_REASON)
|
|
class ElasticsearchTestCase(TestCase, LoaderModuleMockMixin):
|
|
'''
|
|
Test cases for salt.states.elasticsearch
|
|
'''
|
|
def setup_loader_modules(self):
|
|
return {
|
|
elasticsearch: {
|
|
'__opts__': {'test': False},
|
|
'__utils__': {'dictdiffer.deep_diff': dictdiffer.deep_diff}
|
|
}
|
|
}
|
|
|
|
# 'index_absent' function tests: 1
|
|
|
|
def test_index_absent(self):
|
|
'''
|
|
Test to manage a elasticsearch index.
|
|
'''
|
|
name = 'foo'
|
|
|
|
ret = {'name': name,
|
|
'result': True,
|
|
'comment': 'Index foo is already absent',
|
|
'changes': {}}
|
|
|
|
mock_get = MagicMock(side_effect=[None, {name: {"test": "key"}}, {name: {}}, {name: {"test": "key"}}, CommandExecutionError, {name: {"test": "key"}}])
|
|
mock_delete = MagicMock(side_effect=[True, False, CommandExecutionError])
|
|
|
|
with patch.dict(elasticsearch.__salt__, {'elasticsearch.index_get': mock_get,
|
|
'elasticsearch.index_delete': mock_delete}):
|
|
self.assertDictEqual(elasticsearch.index_absent(name), ret)
|
|
|
|
ret.update({'comment': 'Successfully removed index foo', 'changes': {"old": {"test": "key"}}})
|
|
self.assertDictEqual(elasticsearch.index_absent(name), ret)
|
|
|
|
ret.update({'comment': 'Failed to remove index foo for unknown reasons', 'result': False, 'changes': {}})
|
|
self.assertDictEqual(elasticsearch.index_absent(name), ret)
|
|
|
|
with patch.dict(elasticsearch.__opts__, {'test': True}):
|
|
ret.update({'comment': "Index foo will be removed", 'result': None, 'changes': {"old": {"test": "key"}}})
|
|
self.assertDictEqual(elasticsearch.index_absent(name), ret)
|
|
|
|
ret.update({'comment': '', 'result': False, 'changes': {}})
|
|
self.assertDictEqual(elasticsearch.index_absent(name), ret)
|
|
|
|
ret.update({'comment': '', 'result': False, 'changes': {}})
|
|
self.assertDictEqual(elasticsearch.index_absent(name), ret)
|
|
|
|
# 'index_present' function tests: 1
|
|
|
|
def test_index_present(self):
|
|
'''
|
|
Test to manage a elasticsearch index.
|
|
'''
|
|
name = 'foo'
|
|
|
|
ret = {'name': name,
|
|
'result': True,
|
|
'comment': 'Index foo is already present',
|
|
'changes': {}}
|
|
|
|
mock_exists = MagicMock(side_effect=[True, False, False, False, CommandExecutionError, False, False])
|
|
mock_get = MagicMock(side_effect=[{name: {"test": "key"}}, CommandExecutionError])
|
|
mock_create = MagicMock(side_effect=[True, False, CommandExecutionError, True])
|
|
|
|
with patch.dict(elasticsearch.__salt__, {'elasticsearch.index_get': mock_get,
|
|
'elasticsearch.index_exists': mock_exists,
|
|
'elasticsearch.index_create': mock_create}):
|
|
self.assertDictEqual(elasticsearch.index_present(name), ret)
|
|
|
|
ret.update({'comment': 'Successfully created index foo', 'changes': {"new": {"test": "key"}}})
|
|
self.assertDictEqual(elasticsearch.index_present(name), ret)
|
|
|
|
ret.update({'comment': 'Cannot create index foo, False', 'result': False, 'changes': {}})
|
|
self.assertDictEqual(elasticsearch.index_present(name), ret)
|
|
|
|
with patch.dict(elasticsearch.__opts__, {'test': True}):
|
|
ret.update({'comment': "Index foo does not exist and will be created", 'result': None, 'changes': {"new": {"test2": "key"}}})
|
|
self.assertDictEqual(elasticsearch.index_present(name, {"test2": "key"}), ret)
|
|
|
|
ret.update({'comment': '', 'result': False, 'changes': {}})
|
|
self.assertDictEqual(elasticsearch.index_absent(name), ret)
|
|
|
|
ret.update({'comment': '', 'result': False, 'changes': {}})
|
|
self.assertDictEqual(elasticsearch.index_absent(name), ret)
|
|
|
|
ret.update({'comment': '', 'result': False, 'changes': {}})
|
|
self.assertDictEqual(elasticsearch.index_absent(name), ret)
|
|
|
|
# 'alias_absent' function tests: 1
|
|
|
|
def test_alias_absent(self):
|
|
'''
|
|
Test to manage a elasticsearch alias.
|
|
'''
|
|
name = 'foo'
|
|
index = 'bar'
|
|
|
|
alias = {index: {"aliases": {name: {"test": "key"}}}}
|
|
|
|
ret = {'name': name,
|
|
'result': True,
|
|
'comment': 'Alias foo for index bar is already absent',
|
|
'changes': {}}
|
|
|
|
mock_get = MagicMock(side_effect=[None, {"foo2": {}}, alias, alias, alias, CommandExecutionError, alias])
|
|
mock_delete = MagicMock(side_effect=[True, False, CommandExecutionError])
|
|
|
|
with patch.dict(elasticsearch.__salt__, {'elasticsearch.alias_get': mock_get,
|
|
'elasticsearch.alias_delete': mock_delete}):
|
|
self.assertDictEqual(elasticsearch.alias_absent(name, index), ret)
|
|
self.assertDictEqual(elasticsearch.alias_absent(name, index), ret)
|
|
|
|
ret.update({'comment': 'Successfully removed alias foo for index bar', 'changes': {"old": {"test": "key"}}})
|
|
self.assertDictEqual(elasticsearch.alias_absent(name, index), ret)
|
|
|
|
ret.update({'comment': 'Failed to remove alias foo for index bar for unknown reasons', 'result': False, 'changes': {}})
|
|
self.assertDictEqual(elasticsearch.alias_absent(name, index), ret)
|
|
|
|
with patch.dict(elasticsearch.__opts__, {'test': True}):
|
|
ret.update({'comment': "Alias foo for index bar will be removed", 'result': None, 'changes': {"old": {"test": "key"}}})
|
|
self.assertDictEqual(elasticsearch.alias_absent(name, index), ret)
|
|
|
|
ret.update({'comment': '', 'result': False, 'changes': {}})
|
|
self.assertDictEqual(elasticsearch.alias_absent(name, index), ret)
|
|
|
|
ret.update({'comment': '', 'result': False, 'changes': {}})
|
|
self.assertDictEqual(elasticsearch.alias_absent(name, index), ret)
|
|
|
|
# 'alias_present' function tests: 1
|
|
|
|
def test_alias_present(self):
|
|
'''
|
|
Test to manage a elasticsearch alias.
|
|
'''
|
|
name = 'foo'
|
|
index = 'bar'
|
|
|
|
alias = {index: {"aliases": {name: {"test": "key"}}}}
|
|
|
|
ret = {'name': name,
|
|
'result': True,
|
|
'comment': 'Alias foo for index bar is already present',
|
|
'changes': {}}
|
|
|
|
mock_get = MagicMock(side_effect=[alias, alias, None, None, None, alias, CommandExecutionError, None])
|
|
mock_create = MagicMock(side_effect=[True, True, False, CommandExecutionError])
|
|
|
|
with patch.dict(elasticsearch.__salt__, {'elasticsearch.alias_get': mock_get,
|
|
'elasticsearch.alias_create': mock_create}):
|
|
self.assertDictEqual(elasticsearch.alias_present(name, index, {"test": "key"}), ret)
|
|
|
|
ret.update({'comment': "Successfully replaced alias foo for index bar", 'changes': {'old': {"test": "key"}, 'new': {"test2": "key"}}})
|
|
self.assertDictEqual(elasticsearch.alias_present(name, index, {"test2": "key"}), ret)
|
|
|
|
ret.update({'comment': "Successfully created alias foo for index bar", 'changes': {'new': {"test2": "key"}}})
|
|
self.assertDictEqual(elasticsearch.alias_present(name, index, {"test2": "key"}), ret)
|
|
|
|
ret.update({'comment': 'Cannot create alias foo for index bar, False', 'result': False})
|
|
self.assertDictEqual(elasticsearch.alias_present(name, index, {"test2": "key"}), ret)
|
|
|
|
with patch.dict(elasticsearch.__opts__, {'test': True}):
|
|
ret.update({'comment': "Alias foo for index bar does not exist and will be created", 'result': None, 'changes': {'new': {"test2": "key"}}})
|
|
self.assertDictEqual(elasticsearch.alias_present(name, index, {"test2": "key"}), ret)
|
|
|
|
ret.update({'comment': "Alias foo for index bar exists with wrong configuration and will be overriden", 'result': None, 'changes': {'old': {"test": "key"}, 'new': {"test2": "key"}}})
|
|
self.assertDictEqual(elasticsearch.alias_present(name, index, {"test2": "key"}), ret)
|
|
|
|
ret.update({'comment': '', 'result': False, 'changes': {}})
|
|
self.assertDictEqual(elasticsearch.alias_present(name, index), ret)
|
|
|
|
ret.update({'comment': '', 'result': False, 'changes': {}})
|
|
self.assertDictEqual(elasticsearch.alias_present(name, index), ret)
|
|
|
|
# 'index_template_absent' function tests: 1
|
|
|
|
def test_index_template_absent(self):
|
|
'''
|
|
Test to manage a elasticsearch index template.
|
|
'''
|
|
name = 'foo'
|
|
|
|
index_template = {name: {"test": "key"}}
|
|
|
|
ret = {'name': name,
|
|
'result': True,
|
|
'comment': 'Index template foo is already absent',
|
|
'changes': {}}
|
|
|
|
mock_get = MagicMock(side_effect=[None, {"bar": {}}, index_template, index_template, index_template, CommandExecutionError, index_template])
|
|
mock_delete = MagicMock(side_effect=[True, False, CommandExecutionError])
|
|
|
|
with patch.dict(elasticsearch.__salt__, {'elasticsearch.index_template_get': mock_get,
|
|
'elasticsearch.index_template_delete': mock_delete}):
|
|
self.assertDictEqual(elasticsearch.index_template_absent(name), ret)
|
|
self.assertDictEqual(elasticsearch.index_template_absent(name), ret)
|
|
|
|
ret.update({'comment': 'Successfully removed index template foo', 'changes': {"old": {"test": "key"}}})
|
|
self.assertDictEqual(elasticsearch.index_template_absent(name), ret)
|
|
|
|
ret.update({'comment': 'Failed to remove index template foo for unknown reasons', 'result': False, 'changes': {}})
|
|
self.assertDictEqual(elasticsearch.index_template_absent(name), ret)
|
|
|
|
with patch.dict(elasticsearch.__opts__, {'test': True}):
|
|
ret.update({'comment': "Index template foo will be removed", 'result': None, 'changes': {"old": {"test": "key"}}})
|
|
self.assertDictEqual(elasticsearch.index_template_absent(name), ret)
|
|
|
|
ret.update({'comment': '', 'result': False, 'changes': {}})
|
|
self.assertDictEqual(elasticsearch.index_template_absent(name), ret)
|
|
|
|
ret.update({'comment': '', 'result': False, 'changes': {}})
|
|
self.assertDictEqual(elasticsearch.index_template_absent(name), ret)
|
|
|
|
# 'index_template_present' function tests: 1
|
|
|
|
def test_index_template_present(self):
|
|
'''
|
|
Test to manage a elasticsearch index template.
|
|
'''
|
|
name = 'foo'
|
|
|
|
index_template = {name: {"test": "key"}}
|
|
|
|
ret = {'name': name,
|
|
'result': True,
|
|
'comment': 'Index template foo is already present',
|
|
'changes': {}}
|
|
|
|
mock_exists = MagicMock(side_effect=[True, False, False, False, CommandExecutionError, False, False])
|
|
mock_create = MagicMock(side_effect=[True, False, CommandExecutionError, True])
|
|
mock_get = MagicMock(side_effect=[index_template, CommandExecutionError])
|
|
|
|
with patch.dict(elasticsearch.__salt__, {'elasticsearch.index_template_get': mock_get,
|
|
'elasticsearch.index_template_create': mock_create,
|
|
'elasticsearch.index_template_exists': mock_exists}):
|
|
self.assertDictEqual(elasticsearch.index_template_present(name, {"test2": "key"}), ret)
|
|
|
|
ret.update({'comment': "Successfully created index template foo", 'changes': {'new': {"test": "key"}}})
|
|
self.assertDictEqual(elasticsearch.index_template_present(name, {"test2": "key"}), ret)
|
|
|
|
ret.update({'comment': 'Cannot create index template foo, False', 'result': False, 'changes': {}})
|
|
self.assertDictEqual(elasticsearch.index_template_present(name, {"test2": "key"}), ret)
|
|
|
|
with patch.dict(elasticsearch.__opts__, {'test': True}):
|
|
ret.update({'comment': "Index template foo does not exist and will be created", 'result': None, 'changes': {'new': {"test2": "key"}}})
|
|
self.assertDictEqual(elasticsearch.index_template_present(name, {"test2": "key"}), ret)
|
|
|
|
ret.update({'comment': '', 'result': False, 'changes': {}})
|
|
self.assertDictEqual(elasticsearch.index_template_present(name, {}), ret)
|
|
|
|
ret.update({'comment': '', 'result': False, 'changes': {}})
|
|
self.assertDictEqual(elasticsearch.index_template_present(name, {}), ret)
|
|
|
|
ret.update({'comment': '', 'result': False, 'changes': {}})
|
|
self.assertDictEqual(elasticsearch.index_template_present(name, {}), ret)
|
|
|
|
# 'pipeline_absent' function tests: 1
|
|
|
|
def test_pipeline_absent(self):
|
|
'''
|
|
Test to manage a elasticsearch pipeline.
|
|
'''
|
|
name = 'foo'
|
|
|
|
pipeline = {name: {"test": "key"}}
|
|
|
|
ret = {'name': name,
|
|
'result': True,
|
|
'comment': 'Pipeline foo is already absent',
|
|
'changes': {}}
|
|
|
|
mock_get = MagicMock(side_effect=[None, {"foo2": {}}, pipeline, pipeline, pipeline, CommandExecutionError, pipeline])
|
|
mock_delete = MagicMock(side_effect=[True, False, CommandExecutionError])
|
|
|
|
with patch.dict(elasticsearch.__salt__, {'elasticsearch.pipeline_get': mock_get,
|
|
'elasticsearch.pipeline_delete': mock_delete}):
|
|
self.assertDictEqual(elasticsearch.pipeline_absent(name), ret)
|
|
self.assertDictEqual(elasticsearch.pipeline_absent(name), ret)
|
|
|
|
ret.update({'comment': 'Successfully removed pipeline foo', 'changes': {"old": {"test": "key"}}})
|
|
self.assertDictEqual(elasticsearch.pipeline_absent(name), ret)
|
|
|
|
ret.update({'comment': 'Failed to remove pipeline foo for unknown reasons', 'result': False, 'changes': {}})
|
|
self.assertDictEqual(elasticsearch.pipeline_absent(name), ret)
|
|
|
|
with patch.dict(elasticsearch.__opts__, {'test': True}):
|
|
ret.update({'comment': "Pipeline foo will be removed", 'result': None, 'changes': {"old": {"test": "key"}}})
|
|
self.assertDictEqual(elasticsearch.pipeline_absent(name), ret)
|
|
|
|
ret.update({'comment': '', 'result': False, 'changes': {}})
|
|
self.assertDictEqual(elasticsearch.pipeline_absent(name), ret)
|
|
|
|
ret.update({'comment': '', 'result': False, 'changes': {}})
|
|
self.assertDictEqual(elasticsearch.pipeline_absent(name), ret)
|
|
|
|
# 'pipeline_present' function tests: 1
|
|
|
|
def test_pipeline_present(self):
|
|
'''
|
|
Test to manage a elasticsearch pipeline.
|
|
'''
|
|
name = 'foo'
|
|
|
|
pipeline = {name: {"test": "key"}}
|
|
|
|
ret = {'name': name,
|
|
'result': True,
|
|
'comment': 'Pipeline foo is already present',
|
|
'changes': {}}
|
|
|
|
mock_get = MagicMock(side_effect=[pipeline, pipeline, None, None, None, pipeline, CommandExecutionError, None])
|
|
mock_create = MagicMock(side_effect=[True, True, False, CommandExecutionError])
|
|
|
|
with patch.dict(elasticsearch.__salt__, {'elasticsearch.pipeline_get': mock_get,
|
|
'elasticsearch.pipeline_create': mock_create}):
|
|
self.assertDictEqual(elasticsearch.pipeline_present(name, {"test": "key"}), ret)
|
|
|
|
ret.update({'comment': "Successfully replaced pipeline foo", 'changes': {'old': {"test": "key"}, 'new': {"test2": "key"}}})
|
|
self.assertDictEqual(elasticsearch.pipeline_present(name, {"test2": "key"}), ret)
|
|
|
|
ret.update({'comment': "Successfully created pipeline foo", 'changes': {'new': {"test2": "key"}}})
|
|
self.assertDictEqual(elasticsearch.pipeline_present(name, {"test2": "key"}), ret)
|
|
|
|
ret.update({'comment': 'Cannot create pipeline foo, False', 'result': False})
|
|
self.assertDictEqual(elasticsearch.pipeline_present(name, {"test2": "key"}), ret)
|
|
|
|
with patch.dict(elasticsearch.__opts__, {'test': True}):
|
|
ret.update({'comment': "Pipeline foo does not exist and will be created", 'result': None, 'changes': {'new': {"test2": "key"}}})
|
|
self.assertDictEqual(elasticsearch.pipeline_present(name, {"test2": "key"}), ret)
|
|
|
|
ret.update({'comment': "Pipeline foo exists with wrong configuration and will be overriden", 'result': None, 'changes': {'old': {"test": "key"}, 'new': {"test2": "key"}}})
|
|
self.assertDictEqual(elasticsearch.pipeline_present(name, {"test2": "key"}), ret)
|
|
|
|
ret.update({'comment': '', 'result': False, 'changes': {}})
|
|
self.assertDictEqual(elasticsearch.pipeline_present(name, {}), ret)
|
|
|
|
ret.update({'comment': '', 'result': False, 'changes': {}})
|
|
self.assertDictEqual(elasticsearch.pipeline_present(name, {}), ret)
|
|
|
|
# 'search_template_absent' function tests: 1
|
|
|
|
def test_search_template_absent(self):
|
|
'''
|
|
Test to manage a elasticsearch search template.
|
|
'''
|
|
name = 'foo'
|
|
|
|
template = {"template": '{"test": "key"}'}
|
|
|
|
ret = {'name': name,
|
|
'result': True,
|
|
'comment': 'Search template foo is already absent',
|
|
'changes': {}}
|
|
|
|
mock_get = MagicMock(side_effect=[None, template, template, template, CommandExecutionError, template])
|
|
mock_delete = MagicMock(side_effect=[True, False, CommandExecutionError])
|
|
|
|
with patch.dict(elasticsearch.__salt__, {'elasticsearch.search_template_get': mock_get,
|
|
'elasticsearch.search_template_delete': mock_delete}):
|
|
self.assertDictEqual(elasticsearch.search_template_absent(name), ret)
|
|
|
|
ret.update({'comment': 'Successfully removed search template foo', 'changes': {"old": {"test": "key"}}})
|
|
self.assertDictEqual(elasticsearch.search_template_absent(name), ret)
|
|
|
|
ret.update({'comment': 'Failed to remove search template foo for unknown reasons', 'result': False, 'changes': {}})
|
|
self.assertDictEqual(elasticsearch.search_template_absent(name), ret)
|
|
|
|
with patch.dict(elasticsearch.__opts__, {'test': True}):
|
|
ret.update({'comment': "Search template foo will be removed", 'result': None, 'changes': {"old": {"test": "key"}}})
|
|
self.assertDictEqual(elasticsearch.search_template_absent(name), ret)
|
|
|
|
ret.update({'comment': '', 'result': False, 'changes': {}})
|
|
self.assertDictEqual(elasticsearch.search_template_absent(name), ret)
|
|
|
|
ret.update({'comment': '', 'result': False, 'changes': {}})
|
|
self.assertDictEqual(elasticsearch.search_template_absent(name), ret)
|
|
|
|
# 'pipeline_present' function tests: 1
|
|
|
|
def test_search_template_present(self):
|
|
'''
|
|
Test to manage a elasticsearch search template.
|
|
'''
|
|
name = 'foo'
|
|
|
|
template = {"template": '{"test": "key"}'}
|
|
|
|
ret = {'name': name,
|
|
'result': True,
|
|
'comment': 'Search template foo is already present',
|
|
'changes': {}}
|
|
|
|
mock_get = MagicMock(side_effect=[template, template, None, None, None, template, CommandExecutionError, None])
|
|
mock_create = MagicMock(side_effect=[True, True, False, CommandExecutionError])
|
|
|
|
with patch.dict(elasticsearch.__salt__, {'elasticsearch.search_template_get': mock_get,
|
|
'elasticsearch.search_template_create': mock_create}):
|
|
self.assertDictEqual(elasticsearch.search_template_present(name, {"test": "key"}), ret)
|
|
|
|
ret.update({'comment': "Successfully replaced search template foo", 'changes': {'old': {"test": "key"}, 'new': {"test2": "key"}}})
|
|
self.assertDictEqual(elasticsearch.search_template_present(name, {"test2": "key"}), ret)
|
|
|
|
ret.update({'comment': "Successfully created search template foo", 'changes': {'new': {"test2": "key"}}})
|
|
self.assertDictEqual(elasticsearch.search_template_present(name, {"test2": "key"}), ret)
|
|
|
|
ret.update({'comment': 'Cannot create search template foo, False', 'result': False})
|
|
self.assertDictEqual(elasticsearch.search_template_present(name, {"test2": "key"}), ret)
|
|
|
|
with patch.dict(elasticsearch.__opts__, {'test': True}):
|
|
ret.update({'comment': "Search template foo does not exist and will be created", 'result': None, 'changes': {'new': {"test2": "key"}}})
|
|
self.assertDictEqual(elasticsearch.search_template_present(name, {"test2": "key"}), ret)
|
|
|
|
ret.update({'comment': "Search template foo exists with wrong configuration and will be overriden", 'result': None, 'changes': {'old': {"test": "key"}, 'new': {"test2": "key"}}})
|
|
self.assertDictEqual(elasticsearch.search_template_present(name, {"test2": "key"}), ret)
|
|
|
|
ret.update({'comment': '', 'result': False, 'changes': {}})
|
|
self.assertDictEqual(elasticsearch.search_template_present(name, {}), ret)
|
|
|
|
ret.update({'comment': '', 'result': False, 'changes': {}})
|
|
self.assertDictEqual(elasticsearch.search_template_present(name, {}), ret)
|