stacklight-integration-tests/stacklight_tests/kafka/api.py
Artem b0d0ba18f8 Fix for Kafka plugin on CI
Change-Id: Id053c5f8e3038ebe8692b1306ac794d0cd663d44
2016-09-02 11:24:23 +03:00

66 lines
2.4 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.
from fuelweb_test import logger
from proboscis import asserts
from stacklight_tests import base_test
from stacklight_tests.kafka import plugin_settings
class KafkaPluginApi(base_test.PluginApi):
def __init__(self):
super(KafkaPluginApi, self).__init__()
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 check_plugin_online(self):
nodes = []
for node in self.helpers.get_all_ready_nodes():
if "kafka" in node["roles"]:
nodes.append({"name": node["name"], "ip": node["ip"]})
for port in [
{"name": "Zookeeper", "number": 2181},
{"name": "Kafka", "number": 9092},
{"name": "Kafka(JMX)", "number": 9990}
]:
for node in nodes:
result = self.checkers.check_port(node["ip"], port["number"])
logger.info("Check that {} (port {}) on node {} (IP: {}) is up"
.format(port['name'], port['number'],
node['name'], node['ip']))
msg = "Port {} ({}) is unreachable on node {}".format(
port['name'], port['number'], node['name'])
asserts.assert_true(result, 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)