from unittest.mock import patch

import src.actions.actions as actions
import unit_tests.test_utils


class TestActions(unit_tests.test_utils.CharmTestCase):

    def setUp(self):
        super(TestActions, self).setUp()
        self.patches = []
        self.patch_all()
        self.patch_object(actions, 'hookenv', name='mock_hookenv')

    def test_generate_cert_not_leader(self):
        """Test when not leader, action fails"""
        self.mock_hookenv.is_leader.return_value = False

        actions.generate_cert()

        # Action should fail
        self.mock_hookenv.action_fail.assert_called_with(
            'Please run action on lead unit'
        )
        self.mock_hookenv.action_set.assert_not_called()

    @patch.object(actions, 'vault_pki')
    def test_generate_cert(self, mock_vault_pki):
        self.mock_hookenv.is_leader.return_value = True
        self.mock_hookenv.action_get.return_value = {
            'sans': 'foobar 1.2.3.4',
            'common-name': 'bazbuz',
            'ttl': '5m',
            'max-ttl': '5y',
        }
        mock_vault_pki.generate_certificate.return_value = 'shiny-cert'

        actions.generate_cert()

        # Validate the request for the cert was called
        mock_vault_pki.generate_certificate.assert_called_with(
            cert_type='server', common_name='bazbuz',
            sans=['foobar', '1.2.3.4'], ttl='5m', max_ttl='5y',
        )
        self.mock_hookenv.action_set.assert_called_with({
            'output': 'shiny-cert',
        })

    @patch.object(actions, 'vault_pki')
    def test_generate_cert_vault_failure(self, mock_vault_pki):
        """Test failure interacting with vault_pki"""
        self.mock_hookenv.is_leader.return_value = True
        self.mock_hookenv.action_get.return_value = {
            'sans': 'foobar',
            'common-name': 'bazbuz',
            'ttl': '5m',
            'max-ttl': '5y',
        }
        mock_vault_pki.generate_certificate.side_effect = \
            actions.vault.VaultNotReady(1)

        actions.generate_cert()

        # Validate the request for the cert was called
        self.mock_hookenv.action_set.assert_not_called
        self.mock_hookenv.action_fail.assert_called_with(
            'Vault is not ready (1)'
        )