#
#    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 heat.common import environment_util as env_util
from heat.tests import common


class TestEnvironmentUtil(common.HeatTestCase):

    def test_empty_merge_strategies(self):
        merge_strategies = {}
        param_strategy = env_util.get_param_merge_strategy(merge_strategies,
                                                           'param1')
        self.assertEqual(env_util.OVERWRITE, param_strategy)

    def test_default_merge_strategy(self):
        merge_strategies = {'default': 'deep_merge'}
        param_strategy = env_util.get_param_merge_strategy(merge_strategies,
                                                           'param1')
        self.assertEqual(env_util.DEEP_MERGE, param_strategy)

    def test_param_sepcific_merge_strategy(self):
        merge_strategies = {'default': 'merge',
                            'param1': 'deep_merge'}
        param_strategy = env_util.get_param_merge_strategy(merge_strategies,
                                                           'param1')
        self.assertEqual(env_util.DEEP_MERGE, param_strategy)

    def test_wrong_param_strategy(self):
        merge_strategies = {'default': 'merge',
                            'param1': 'unknown'}
        param_strategy = env_util.get_param_merge_strategy(merge_strategies,
                                                           'param1')
        self.assertEqual(env_util.MERGE, param_strategy)

    def test_merge_startegies_none(self):
        merge_strategies = None
        param_strategy = env_util.get_param_merge_strategy(merge_strategies,
                                                           'param1')
        self.assertEqual(env_util.OVERWRITE, param_strategy)


class TestMergeEnvironments(common.HeatTestCase):

    def test_merge_environments(self):
        # Setup
        params = {'parameters': {
            'p0': 'CORRECT',
            'p1': 'INCORRECT',
            'p2': 'INCORRECT'}
        }
        env_1 = '''
        {'parameters' : {
            'p1': 'CORRECT',
            'p2': 'INCORRECT-ENV-1',
        }}'''
        env_2 = '''
        {'parameters': {
            'p2': 'CORRECT'
        }}'''

        files = {'env_1': env_1, 'env_2': env_2}
        environment_files = ['env_1', 'env_2']

        # Test
        env_util.merge_environments(environment_files, files, params)

        # Verify
        expected = {'parameters': {
            'p0': 'CORRECT',
            'p1': 'CORRECT',
            'p2': 'CORRECT',
        }}
        self.assertEqual(expected, params)

    def test_merge_environments_no_env_files(self):
        params = {'parameters': {'p0': 'CORRECT'}}
        env_1 = '''
        {'parameters' : {
            'p0': 'INCORRECT',
        }}'''

        files = {'env_1': env_1}

        # Test - Should ignore env_1 in files
        env_util.merge_environments(None, files, params)

        # Verify
        expected = {'parameters': {'p0': 'CORRECT'}}
        self.assertEqual(expected, params)