valitydev/salt
14
0
Fork 0
mirror of https://github.com/valitydev/salt.git synced 2024-11-07 17:09:03 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity
6d861f9a74
salt/tests/unit/states/test_elasticsearch.py
Pedro Algarvio f8d7530a58
Proper mocking.
2017-04-01 14:22:59 +01:00

445 lines
22 KiB
Python
Raw Blame History

# -*- coding: utf-8 -*-
'''
:codeauthor: :email:`Lukas Raska <lukas@raska.me>`
'''
# Import Python libs
from __future__ import absolute_import
# 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
from salt.states import elasticsearch
from salt.utils import dictdiffer
@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)
Reference in New Issue View Git Blame Copy Permalink