import mock
import unittest
import yaml

# Required Oslo configuration setup
from armada.conf import default
default.register_opts()

from armada.handlers.armada import Armada

class ArmadaTestCase(unittest.TestCase):
    test_yaml = """
        armada:
          release_prefix: armada
          charts:
              - description: this is a test
                sequenced: False
                chart_group:
                  - chart:
                      name: test_chart_1
                      release_name: test_chart_1
                      namespace: test
                      values: {}
                      source:
                        type: git
                        location: git://github.com/dummy/armada
                        subpath: chart_1
                        reference: master
                      dependencies: []
                      timeout: 50

                  - chart:
                      name: test_chart_2
                      release_name: test_chart_2
                      namespace: test
                      values: {}
                      source:
                        type: local
                        location: /tmp/dummy/armada
                        subpath: chart_2
                        reference: null
                      dependencies: []
                      timeout: 5
    """

    @mock.patch('armada.handlers.armada.git')
    @mock.patch('armada.handlers.armada.lint')
    @mock.patch('armada.handlers.armada.Tiller')
    def test_pre_flight_ops(self, mock_tiller, mock_lint, mock_git):
        '''Test pre-flight checks and operations'''
        armada = Armada('')
        armada.tiller = mock_tiller
        armada.config = yaml.load(self.test_yaml)

        CHART_SOURCES = [('git://github.com/dummy/armada', 'chart_1'),
                         ('/tmp/dummy/armada', 'chart_2')]

        # mock methods called by pre_flight_ops()
        mock_tiller.tiller_status.return_value = True
        mock_lint.valid_manifest.return_value = True
        mock_git.git_clone.return_value = CHART_SOURCES[0][0]

        armada.pre_flight_ops()

        mock_git.git_clone.assert_called_once_with(CHART_SOURCES[0][0],
                                                   'master')
        for group in armada.config.get('armada').get('charts'):
            for counter, chart in enumerate(group.get('chart_group')):
                self.assertEqual(chart.get('chart').get('source_dir')[0],
                                 CHART_SOURCES[counter][0])
                self.assertEqual(chart.get('chart').get('source_dir')[1],
                                 CHART_SOURCES[counter][1])

    @mock.patch('armada.handlers.armada.git')
    @mock.patch('armada.handlers.armada.lint')
    @mock.patch('armada.handlers.armada.Tiller')
    def test_post_flight_ops(self, mock_tiller, mock_lint, mock_git):
        '''Test post-flight operations'''
        armada = Armada('')
        armada.tiller = mock_tiller
        armada.config = yaml.load(self.test_yaml)

        CHART_SOURCES = [('git://github.com/dummy/armada', 'chart_1'),
                         ('/tmp/dummy/armada', 'chart_2')]

        # mock methods called by pre_flight_ops()
        mock_tiller.tiller_status.return_value = True
        mock_lint.valid_manifest.return_value = True
        mock_git.git_clone.return_value = CHART_SOURCES[0][0]
        armada.pre_flight_ops()

        armada.post_flight_ops()

        for group in yaml.load(self.test_yaml).get('armada').get('charts'):
            for counter, chart in enumerate(group.get('chart_group')):
                if chart.get('chart').get('source').get('type') == 'git':
                    mock_git.source_cleanup \
                            .assert_called_with(CHART_SOURCES[counter][0])

    @mock.patch.object(Armada, 'post_flight_ops')
    @mock.patch.object(Armada, 'pre_flight_ops')
    @mock.patch('armada.handlers.armada.ChartBuilder')
    @mock.patch('armada.handlers.armada.Tiller')
    def test_install(self, mock_tiller, mock_chartbuilder,
                     mock_pre_flight, mock_post_flight):
        '''Test install functionality from the sync() method'''

        # instantiate Armada and Tiller objects
        armada = Armada('',
                        wait=True,
                        timeout=1000)
        armada.tiller = mock_tiller
        armada.config = yaml.load(self.test_yaml)

        charts = armada.config['armada']['charts'][0]['chart_group']
        chart_1 = charts[0]['chart']
        chart_2 = charts[1]['chart']

        # mock irrelevant methods called by armada.sync()
        mock_tiller.list_charts.return_value = []
        mock_chartbuilder.get_source_path.return_value = None
        mock_chartbuilder.get_helm_chart.return_value = None

        armada.sync()

        # check params that should be passed to tiller.install_release()
        method_calls = [mock.call(mock_chartbuilder().get_helm_chart(),
                                  armada.dry_run, chart_1['release_name'],
                                  chart_1['namespace'],
                                  armada.config['armada']['release_prefix'],
                                  values=yaml.safe_dump(chart_1['values']),
                                  wait=armada.wait,
                                  timeout=1000),
                        mock.call(mock_chartbuilder().get_helm_chart(),
                                  armada.dry_run, chart_2['release_name'],
                                  chart_2['namespace'],
                                  armada.config['armada']['release_prefix'],
                                  values=yaml.safe_dump(chart_2['values']),
                                  wait=armada.wait,
                                  timeout=1000)]
        mock_tiller.install_release.assert_has_calls(method_calls)

    @unittest.skip('skipping update')
    def test_upgrade(self):
        '''Test upgrade functionality from the sync() method'''
        # TODO