#    Copyright 2015 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 system_test.helpers.utils import case_factory
from proboscis import factory

from system_test.tests.strength import strength_base
from system_test.helpers.decorators import make_snapshot_if_step_fail
from system_test.helpers.decorators import deferred_decorator
from system_test.helpers.decorators import action


class StrengthDestroyFirstController(strength_base.StrengthBaseActions):
    """Destroy two controllers and check pacemaker status is correct

    Scenario:
        1. Create Environment
        2. Add nodes to Environment
        3. Run network checker
        4. Deploy Environment
        5. Run network checker
        6. Run OSTF
        7. Make or use existing snapshot of ready Environment
        8. Destroy first controller
        9. Check pacemaker status
        10. Wait offline status in nailgun
        11. Run OSTF

    """

    base_group = ['system_test',
                  'system_test.failover',
                  'system_test.failover.destroy_controllers',
                  'system_test.failover.destroy_controllers.first']

    actions_order = [
        'setup_master',
        'config_release',
        'make_slaves',
        'revert_slaves',
        'create_env',
        'add_nodes',
        'network_check',
        'deploy_cluster',
        'network_check',
        'health_check',
        'save_load_environment',
        'destroy_first_controller',
        'check_pacemaker_status',
        'wait_offline_nodes',
        'check_ha_service_ready',
        'check_os_services_ready',
        'wait_galera_cluster',
        'health_check',
    ]

    @deferred_decorator([make_snapshot_if_step_fail])
    @action
    def destroy_first_controller(self):
        """Destroy first controller"""
        self._destroy_controller('slave-01')


class StrengthDestroySecondController(strength_base.StrengthBaseActions):
    """Destroy two controllers and check pacemaker status is correct

    Scenario:
        1. Create Environment
        2. Add nodes to Environment
        3. Run network checker
        4. Deploy Environment
        5. Run network checker
        6. Run OSTF
        7. Make or use existing snapshot of ready Environment
        8. Destroy second controller
        9. Check pacemaker status
        10. Wait offline status in nailgun
        11. Run OSTF

    """

    base_group = ['system_test',
                  'system_test.failover',
                  'system_test.failover.destroy_controllers',
                  'system_test.failover.destroy_controllers.second']

    actions_order = [
        'setup_master',
        'config_release',
        'make_slaves',
        'revert_slaves',
        'create_env',
        'add_nodes',
        'network_check',
        'deploy_cluster',
        'network_check',
        'health_check',
        'save_load_environment',
        'destroy_second_controller',
        'check_pacemaker_status',
        'wait_offline_nodes',
        'check_ha_service_ready',
        'check_os_services_ready',
        'wait_galera_cluster',
        'health_check',
    ]

    @deferred_decorator([make_snapshot_if_step_fail])
    @action
    def destroy_second_controller(self):
        """Destroy second controller"""
        self._destroy_controller('slave-02')


@factory
def cases():
    return (case_factory(StrengthDestroyFirstController) +
            case_factory(StrengthDestroySecondController))