def test_create_body_valid(self): client = Mock() client.indices.get_settings.return_value = testvars.settings_one client.cluster.state.return_value = testvars.clu_state_one client.indices.stats.return_value = testvars.stats_one ilo = curator.IndexList(client) ao = curator.Allocation(ilo, key='key', value='value') self.assertEqual('index.routing.allocation.require.key=value', ao.body)
def test_do_action(self): client = Mock() client.indices.get_settings.return_value = testvars.settings_one client.cluster.state.return_value = testvars.clu_state_one client.indices.stats.return_value = testvars.stats_one client.indices.put_settings.return_value = None ilo = curator.IndexList(client) ao = curator.Allocation(ilo, key='key', value='value') self.assertIsNone(ao.do_action())
def test_init(self): client = Mock() client.indices.get_settings.return_value = testvars.settings_one client.cluster.state.return_value = testvars.clu_state_one client.indices.stats.return_value = testvars.stats_one ilo = curator.IndexList(client) ao = curator.Allocation(ilo, key='key', value='value') self.assertEqual(ilo, ao.index_list) self.assertEqual(client, ao.client)
def test_do_action_raise_on_put_settings(self): client = Mock() client.indices.get_settings.return_value = testvars.settings_one client.cluster.state.return_value = testvars.clu_state_one client.indices.stats.return_value = testvars.stats_one client.indices.put_settings.return_value = None client.indices.put_settings.side_effect = testvars.fake_fail ilo = curator.IndexList(client) ao = curator.Allocation(ilo, key='key', value='value') self.assertRaises(Exception, ao.do_action)
def test_do_action_wait(self): client = Mock() client.info.return_value = {'version': {'number': '2.4.1'}} client.indices.get_settings.return_value = testvars.settings_one client.cluster.state.return_value = testvars.clu_state_one client.indices.stats.return_value = testvars.stats_one client.indices.put_settings.return_value = None client.cluster.health.return_value = None ilo = curator.IndexList(client) ao = curator.Allocation(ilo, key='key', value='value', wait_for_completion=True) self.assertIsNone(ao.do_action())