blob: 330f62e9836d7d48fa8e4cff339d97237105d63f [file] [log] [blame]
#!/usr/bin/env python3
# -*- encoding: utf-8 -*-
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements. See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership. The ASF licenses this file
# to you 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
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.
import logging
import subprocess
from ..common import status
import test_template
class TestScaleUp(test_template.TestTemplate):
expected_container_count = 1
expected_instance_count = 3
def get_expected_container_count(self):
return self.expected_container_count
def get_expected_min_instance_count(self):
return self.expected_instance_count
def execute_test_case(self):
scale_up(self.params['cliPath'], self.params['cluster'], self.params['topologyName'])
self.expected_container_count += 1
self.expected_instance_count += 1
def pre_check_results(self, physical_plan_json):
instances = physical_plan_json['instances']
instance_count = len(instances)
if instance_count != self.expected_instance_count:
raise status.TestFailure("Found %s instances but expected %s: %s" %
(instance_count, self.expected_instance_count, instances))
def scale_up(heron_cli_path, test_cluster, topology_name):
splitcmd = [
heron_cli_path, 'update', '--verbose', test_cluster, topology_name,
]"Increasing number of component instances: %s", splitcmd)
if != 0:
raise status.TestFailure("Unable to update topology %s" % topology_name)"Increased number of component instances")