#    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 proboscis import test

from fuelweb_test.helpers.decorators import log_snapshot_after_test

from stacklight_tests.elasticsearch_kibana import api


@test(groups=["plugins"])
class TestNodesElasticsearshPlugin(api.ElasticsearchPluginApi):
    """Class for system tests for Elasticsearch-Kibana plugin."""

    @test(depends_on_groups=['deploy_ha_elasticsearch_kibana_plugin'],
          groups=["check_scaling_elasticsearch_kibana", "scaling",
                  "elasticsearch_kibana", "system",
                  "add_remove_controller_elasticsearch_kibana_plugin"])
    @log_snapshot_after_test
    def add_remove_controller_elasticsearch_kibana_plugin(self):
        """Verify that the number of controllers can scale up and down

        Scenario:
            1. Revert snapshot with 9 deployed nodes in HA configuration
            2. Remove one controller node and update the cluster
            3. Check that plugin is working
            4. Run OSTF
            5. Add one controller node (return previous state) and
               update the cluster
            6. Check that plugin is working
            7. Run OSTF

        Duration 120m
        """
        self.env.revert_snapshot("deploy_ha_elasticsearch_kibana_plugin")

        target_node = {'slave-03': ['controller']}

        # NOTE(vgusev): We set "check_services=False" and
        # "should_fail=1" parameters in deploy_cluster_wait and run_ostf
        # methods because after removing one node
        # nova has been keeping it in service list

        # Remove controller
        self.helpers.remove_node_from_cluster(target_node)

        self.check_plugin_online()

        self.helpers.run_ostf(should_fail=1)

        # Add controller
        self.helpers.add_node_to_cluster(target_node)

        self.check_plugin_online()

        self.helpers.run_ostf(should_fail=1)

        self.env.make_snapshot(
            "add_remove_controller_elasticsearch_kibana_plugin")

    @test(depends_on_groups=['deploy_ha_elasticsearch_kibana_plugin'],
          groups=["check_scaling_elasticsearch_kibana", "scaling",
                  "elasticsearch_kibana", "system",
                  "add_remove_compute_elasticsearch_kibana_plugin"])
    @log_snapshot_after_test
    def add_remove_compute_elasticsearch_kibana_plugin(self):
        """Verify that the number of computes can scale up and down

        Scenario:
            1. Revert snapshot with 9 deployed nodes in HA configuration
            2. Remove one compute node and update the cluster
            3. Check that plugin is working
            4. Run OSTF
            5. Add one compute node (return previous state) and
               update the cluster
            6. Check that plugin is working
            7. Run OSTF

        Duration 120m
        """
        self.env.revert_snapshot("deploy_ha_elasticsearch_kibana_plugin")

        target_node = {'slave-04': ['compute', 'cinder']}

        # NOTE(vgusev): We set "check_services=False" and
        # "should_fail=1" parameters in deploy_cluster_wait and run_ostf
        # methods because after removing one node
        # nova has been keeping it in service list

        # Remove compute
        self.helpers.remove_node_from_cluster(target_node)

        self.check_plugin_online()

        self.helpers.run_ostf(should_fail=1)

        # Add compute
        self.helpers.add_node_to_cluster(target_node)

        self.check_plugin_online()

        self.helpers.run_ostf(should_fail=1)

        self.env.make_snapshot(
            "add_remove_compute_elasticsearch_kibana_plugin")

    @test(depends_on_groups=['deploy_ha_elasticsearch_kibana_plugin'],
          groups=["check_scaling_elasticsearch_kibana", "scaling",
                  "elasticsearch_kibana", "system",
                  "add_remove_node_with_elasticsearch_kibana_plugin"])
    @log_snapshot_after_test
    def add_remove_node_with_elasticsearch_kibana_plugin(self):
        """Verify that the number of Elasticsearch-Kibana nodes
        can scale up and down

        Scenario:
            1. Revert snapshot with 9 deployed nodes in HA configuration
            2. Remove one Elasticsearch-Kibana node and update the cluster
            3. Check that plugin is working
            4. Run OSTF
            5. Add one Elasticsearch-Kibana node (return previous state) and
               update the cluster
            6. Check that plugin is working
            7. Run OSTF

        Duration 120m
        """
        self.env.revert_snapshot("deploy_ha_elasticsearch_kibana_plugin")

        self.check_elasticsearch_nodes_count(3)

        target_node = {'slave-07': self.settings.role_name}

        # Remove Elasticsearch-Kibana node
        self.helpers.remove_node_from_cluster(target_node)

        self.check_plugin_online()

        self.check_elasticsearch_nodes_count(2)

        self.fuel_web.run_ostf()

        # Add Elasticsearch-Kibana node
        self.helpers.add_node_to_cluster(target_node)

        self.check_plugin_online()

        self.check_elasticsearch_nodes_count(3)

        self.helpers.run_ostf()

        self.env.make_snapshot(
            "add_remove_node_with_elasticsearch_kibana_plugin")