156 lines
6.6 KiB
Python
156 lines
6.6 KiB
Python
# Copyright 2016 Mirantis, Inc.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import elasticsearch
|
|
from fuelweb_test import logger
|
|
from proboscis import asserts
|
|
|
|
from stacklight_tests import base_test
|
|
from stacklight_tests.elasticsearch_kibana import plugin_settings
|
|
|
|
|
|
class ElasticsearchPluginApi(base_test.PluginApi):
|
|
def __init__(self):
|
|
super(ElasticsearchPluginApi, self).__init__()
|
|
self._es_client = None
|
|
|
|
@property
|
|
def es(self):
|
|
if self._es_client is None:
|
|
self._es_client = elasticsearch.Elasticsearch(
|
|
[{'host': self.get_plugin_vip(), 'port': 9200}])
|
|
return self._es_client
|
|
|
|
def get_plugin_settings(self):
|
|
return plugin_settings
|
|
|
|
def prepare_plugin(self):
|
|
self.helpers.prepare_plugin(self.settings.plugin_path)
|
|
|
|
def activate_plugin(self, options=None):
|
|
if options is None:
|
|
options = self.settings.default_options
|
|
self.helpers.activate_plugin(
|
|
self.settings.name, self.settings.version, options)
|
|
|
|
def get_plugin_vip(self):
|
|
return self.helpers.get_plugin_vip(self.settings.vip_name)
|
|
|
|
def get_elasticsearch_url(self, query=''):
|
|
return "http://{}:9200/{}".format(self.get_plugin_vip(), query)
|
|
|
|
def get_kibana_url(self):
|
|
return "http://{}:80/".format(self.get_plugin_vip())
|
|
|
|
def check_plugin_online(self):
|
|
logger.info("Check that Elasticsearch is ready")
|
|
msg = "Elasticsearch responded with {0}, expected {1}"
|
|
self.checkers.check_http_get_response(
|
|
self.get_elasticsearch_url(), msg=msg)
|
|
|
|
logger.info("Check that Kibana is running")
|
|
msg = "Kibana responded with {0}, expected {1}"
|
|
self.checkers.check_http_get_response(self.get_kibana_url(), msg=msg)
|
|
|
|
def check_elasticsearch_nodes_count(self, expected_count):
|
|
logger.debug("Get information about Elasticsearch nodes")
|
|
url = self.get_elasticsearch_url(query='_nodes')
|
|
response = self.checkers.check_http_get_response(url)
|
|
nodes_count = len(response.json()['nodes'])
|
|
|
|
logger.debug("Check that the number of nodes is equal to the expected")
|
|
msg = ("Expected count of elasticsearch nodes {}, "
|
|
"actual count {}".format(expected_count, nodes_count))
|
|
asserts.assert_equal(expected_count, nodes_count, msg)
|
|
|
|
def uninstall_plugin(self):
|
|
return self.helpers.uninstall_plugin(
|
|
self.settings.name, self.settings.version)
|
|
|
|
def check_uninstall_failure(self):
|
|
return self.helpers.check_plugin_cannot_be_uninstalled(
|
|
self.settings.name, self.settings.version)
|
|
|
|
def query_elasticsearch(self, index_type, time_range="now-1h",
|
|
query_filter="*", size=100):
|
|
all_indices = self.es.indices.get_aliases().keys()
|
|
indices = filter(lambda x: index_type in x, sorted(all_indices))
|
|
return self.es.search(index=indices, body={
|
|
"query": {"filtered": {
|
|
"query": {"bool": {"should": {"query_string": {
|
|
"query": query_filter}}}},
|
|
"filter": {"bool": {"must": {"range": {
|
|
"Timestamp": {"from": time_range}}}}}}},
|
|
"size": size})
|
|
|
|
def make_instance_actions(self):
|
|
net_name = self.fuel_web.get_cluster_predefined_networks_name(
|
|
self.helpers.cluster_id)['private_net']
|
|
os_conn = self.helpers.os_conn
|
|
flavors = os_conn.nova.flavors.list(sort_key="memory_mb")
|
|
logger.info("Launch an instance")
|
|
instance = os_conn.create_server_for_migration(
|
|
label=net_name, flavor=flavors[0])
|
|
logger.info("Update the instance")
|
|
os_conn.nova.servers.update(instance, name="test-server")
|
|
self.helpers.wait_for_resource_status(
|
|
os_conn.nova.servers, instance, "ACTIVE")
|
|
image = self.helpers.os_conn._get_cirros_image()
|
|
logger.info("Rebuild the instance")
|
|
os_conn.nova.servers.rebuild(
|
|
instance, image, name="rebuilded_instance")
|
|
self.helpers.wait_for_resource_status(
|
|
os_conn.nova.servers, instance, "ACTIVE")
|
|
logger.info("Resize the instance")
|
|
os_conn.nova.servers.resize(instance, flavors[1])
|
|
self.helpers.wait_for_resource_status(
|
|
os_conn.nova.servers, instance, "VERIFY_RESIZE")
|
|
logger.info("Confirm the resize")
|
|
os_conn.nova.servers.confirm_resize(instance)
|
|
self.helpers.wait_for_resource_status(
|
|
os_conn.nova.servers, instance, "ACTIVE")
|
|
logger.info("Resize the instance")
|
|
os_conn.nova.servers.resize(instance, flavors[2])
|
|
self.helpers.wait_for_resource_status(
|
|
os_conn.nova.servers, instance, "VERIFY_RESIZE")
|
|
logger.info("Revert the resize")
|
|
os_conn.nova.servers.revert_resize(instance)
|
|
self.helpers.wait_for_resource_status(
|
|
os_conn.nova.servers, instance, "ACTIVE")
|
|
logger.info("Stop the instance")
|
|
os_conn.nova.servers.stop(instance)
|
|
self.helpers.wait_for_resource_status(
|
|
os_conn.nova.servers, instance, "SHUTOFF")
|
|
logger.info("Start the instance")
|
|
os_conn.nova.servers.start(instance)
|
|
self.helpers.wait_for_resource_status(
|
|
os_conn.nova.servers, instance, "ACTIVE")
|
|
logger.info("Suspend the instance")
|
|
os_conn.nova.servers.suspend(instance)
|
|
self.helpers.wait_for_resource_status(
|
|
os_conn.nova.servers, instance, "SUSPENDED")
|
|
logger.info("Resume the instance")
|
|
os_conn.nova.servers.resume(instance)
|
|
self.helpers.wait_for_resource_status(
|
|
os_conn.nova.servers, instance, "ACTIVE")
|
|
logger.info("Create an instance snapshot")
|
|
snapshot = os_conn.nova.servers.create_image(instance, "test-image")
|
|
self.helpers.wait_for_resource_status(
|
|
os_conn.nova.images, snapshot, "ACTIVE")
|
|
logger.info("Delete the instance")
|
|
os_conn.nova.servers.delete(instance)
|
|
logger.info("Check that the instance was deleted")
|
|
os_conn.verify_srv_deleted(instance)
|
|
return instance.id
|