74 lines
3.0 KiB
Python
74 lines
3.0 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 devops.helpers import helpers
|
|
|
|
from fuel_ccp_tests import logger
|
|
|
|
LOG = logger.logger
|
|
LOG.addHandler(logger.console)
|
|
|
|
|
|
# TODO: replace check with deployment status request
|
|
def check_pods_status(k8sclient, timeout=4500, namespace='ccp'):
|
|
LOG.info("Check pods status")
|
|
|
|
def is_pod_running(cluster, pod_name, namespace=namespace):
|
|
def temporary_status():
|
|
pod = cluster.pods.get(name=pod_name, namespace=namespace)
|
|
if pod.status.phase in ('Running', 'Succeeded'):
|
|
return True
|
|
elif pod.status.phase == "Failed":
|
|
if pod.status.reason in ('NodeSelectorMismatching',
|
|
'MatchNodeSelector'):
|
|
return True
|
|
return False
|
|
return temporary_status
|
|
pod_names = [pod.name for pod in k8sclient.pods.list(namespace=namespace)]
|
|
LOG.info('Pods names {}'.format(pod_names))
|
|
for pod_name in pod_names:
|
|
predicate = is_pod_running(k8sclient, pod_name)
|
|
helpers.wait(predicate, timeout=timeout,
|
|
timeout_msg='Timeout waiting, pod {0}'
|
|
'is not in the \"Running\", \"Succeeded\" state'.format(
|
|
pod_name))
|
|
|
|
|
|
# TODO: replace check with deployment status request
|
|
def check_jobs_status(k8sclient, timeout=4500, namespace='ccp'):
|
|
LOG.info("Check jobs status")
|
|
|
|
def is_job_successful(cluster, job_name, namespace=namespace):
|
|
return lambda: (cluster.jobs.get(
|
|
name=job_name, namespace=namespace)).status.succeeded == 1
|
|
job_names = [job.name for job in k8sclient.jobs.list(namespace=namespace)]
|
|
for job_name in job_names:
|
|
predicate = is_job_successful(k8sclient, job_name)
|
|
helpers.wait(predicate, timeout=timeout,
|
|
timeout_msg='Timeout waiting job {0} '
|
|
'status is not successful'.format(job_name))
|
|
|
|
|
|
def check_pod_status_by_name(name, k8sclient, namespace='ccp', count=None):
|
|
pod_names = [pod.name for pod in
|
|
k8sclient.pods.list(namespace=namespace) if name in pod.name]
|
|
if count:
|
|
fail_msg = 'Unexpected count of pods {0}. Current {1}'.format(
|
|
pod_names, len(pod_names))
|
|
assert len(pod_names) == count, fail_msg
|
|
for pod in pod_names:
|
|
|
|
assert k8sclient.pods.get(
|
|
name=pod, namespace=namespace).status.phase in (
|
|
'Running', 'Succeeded')
|