def test_check_elasticsearch_masters(pods, expect_error):
test_pods = list(pods)
check = Elasticsearch(None, task_vars_config_base)
check.execute_module = lambda cmd, args: {
'result': test_pods.pop(0)['_test_master_name_str']
}
errors = check._check_elasticsearch_masters(pods_by_name(pods))
assert_error(''.join(errors), expect_error)
def test_check_elasticsearch_cluster_health(pods, health_data, expect_error):
test_health_data = list(health_data)
check = Elasticsearch(None, task_vars_config_base)
check.execute_module = lambda cmd, args: {
'result': json.dumps(test_health_data.pop(0))
}
errors = check._check_es_cluster_health(pods_by_name(pods))
assert_error(''.join(errors), expect_error)
def test_check_elasticsearch():
assert 'No logging Elasticsearch pods' in Elasticsearch(
).check_elasticsearch([])
# canned oc responses to match so all the checks pass
def _exec_oc(ns, cmd, args):
if '_cat/master' in cmd:
return 'name logging-es'
elif '/_nodes' in cmd:
return json.dumps(es_node_list)
elif '_cluster/health' in cmd:
return '{"status": "green"}'
elif ' df ' in cmd:
return 'IUse% Use%\n 3% 4%\n'
else:
raise Exception(cmd)
check = Elasticsearch(None, {})
check.exec_oc = _exec_oc
assert not check.check_elasticsearch([plain_es_pod])
def canned_elasticsearch(task_vars=None, exec_oc=None):
"""Create an Elasticsearch check object with stubbed exec_oc method"""
check = Elasticsearch(None, task_vars or {})
if exec_oc:
check.exec_oc = exec_oc
return check
def test_check_running_es_pods():
pods, errors = Elasticsearch().running_elasticsearch_pods(
[plain_es_pod, unready_es_pod])
assert plain_es_pod in pods
assert_error_in_list('PodNotRunning', errors)
def canned_elasticsearch(task_vars=None, exec_oc=None):
"""Create an Elasticsearch check object with stubbed exec_oc method"""
check = Elasticsearch(None, task_vars or {})
if exec_oc:
check.exec_oc = exec_oc
return check
def canned_elasticsearch(task_vars=None, exec_oc=None):
"""Create an Elasticsearch check object with canned exec_oc method"""
check = Elasticsearch("dummy", task_vars or {}) # fails if a module is actually invoked
if exec_oc:
check._exec_oc = exec_oc
return check
def test_check_elasticsearch_diskspace(disk_data, expect_error):
check = Elasticsearch(None, task_vars_config_base)
check.execute_module = lambda cmd, args: {'result': disk_data}
errors = check._check_elasticsearch_diskspace(pods_by_name([plain_es_pod]))
assert_error(''.join(errors), expect_error)
def test_check_elasticsearch_node_list(pods, node_list, expect_error):
check = Elasticsearch(None, task_vars_config_base)
check.execute_module = lambda cmd, args: {'result': json.dumps(node_list)}
errors = check._check_elasticsearch_node_list(pods_by_name(pods))
assert_error(''.join(errors), expect_error)
def canned_elasticsearch(exec_oc=None):
"""Create an Elasticsearch check object with canned exec_oc method"""
check = Elasticsearch("dummy") # fails if a module is actually invoked
if exec_oc:
check._exec_oc = exec_oc
return check