def test_index_template_present_check_definition_alias_not_empty(self): ''' Test to manage a elasticsearch index template. with check_definition set and alias is not empty ''' name = 'foo' index_template = {name: {"test2": "key", "aliases": {}, "mappings": {}, "settings": {}}} expected = {'name': name, 'result': True, 'comment': 'Successfully updated index template foo', 'changes': {'new': {'aliases': {'alias1': {}}}, 'old': {'aliases': {}}}} mock_exists = MagicMock(side_effect=[True]) mock_create = MagicMock(side_effect=[True]) mock_get = MagicMock(side_effect=[index_template]) with patch.dict(elasticsearch.__salt__, {'elasticsearch.index_template_get': mock_get, 'elasticsearch.index_template_create': mock_create, 'elasticsearch.index_template_exists': mock_exists}): ret = elasticsearch.index_template_present(name, {"test2": "key", "aliases": {'alias1': {}}}, check_definition=True) self.assertDictEqual(expected, ret)
def test_index_template_present_check_definition_alias_not_empty(): """ Test to manage a elasticsearch index template. with check_definition set and alias is not empty """ name = "foo" index_template = { name: { "test2": "key", "aliases": {}, "mappings": {}, "settings": {} } } expected = { "name": name, "result": True, "comment": "Successfully updated index template foo", "changes": { "new": { "aliases": { "alias1": {} } }, "old": { "aliases": {} } }, } mock_exists = MagicMock(side_effect=[True]) mock_create = MagicMock(side_effect=[True]) mock_get = MagicMock(side_effect=[index_template]) with patch.dict( elasticsearch.__salt__, { "elasticsearch.index_template_get": mock_get, "elasticsearch.index_template_create": mock_create, "elasticsearch.index_template_exists": mock_exists, }, ): ret = elasticsearch.index_template_present(name, { "test2": "key", "aliases": { "alias1": {} } }, check_definition=True) assert expected == ret
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)
def test_index_template_present(): """ 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, }, ): assert elasticsearch.index_template_present(name, {"test2": "key"}) == ret ret.update({ "comment": "Successfully created index template foo", "changes": { "new": { "test": "key" } }, }) assert elasticsearch.index_template_present(name, {"test2": "key"}) == ret ret.update({ "comment": "Cannot create index template foo, False", "result": False, "changes": {}, }) assert 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" } }, }) assert elasticsearch.index_template_present( name, {"test2": "key"}) == ret ret.update({"comment": "", "result": False, "changes": {}}) assert elasticsearch.index_template_present(name, {}) == ret ret.update({"comment": "", "result": False, "changes": {}}) assert elasticsearch.index_template_present(name, {}) == ret ret.update({"comment": "", "result": False, "changes": {}}) assert elasticsearch.index_template_present(name, {}) == ret