import os
from . import base
from cfn import ssh, owner_ssh, generate_stack_from_input
from mock import patch, MagicMock

class TestCfn(base.BaseCase):
    def setUp(self):
        os.environ['LOGNAME'] = 'my_user'

    def tearDown(self):
        del os.environ['LOGNAME']

    @patch('cfn.local')
    @patch('buildercore.core.active_stack_names')
    @patch('buildercore.core.find_ec2_instances')
    def test_ssh_task(self, find_ec2_instances, active_stack_names, local):
        self._dummy_instance_is_active(find_ec2_instances, active_stack_names)
        ssh('dummy1--prod')
        local.assert_called_with('ssh elife@54.54.54.54')

    @patch('cfn.local')
    @patch('buildercore.core.active_stack_names')
    @patch('buildercore.core.find_ec2_instances')
    def test_owner_ssh_task(self, find_ec2_instances, active_stack_names, local):
        self._dummy_instance_is_active(find_ec2_instances, active_stack_names)
        owner_ssh('dummy1--prod')
        (args, _) = local.call_args
        self.assertRegex(args[0], 'ssh ubuntu@54.54.54.54 -i .+/.cfn/keypairs/dummy1--prod.pem')

    # all non-interactive cases
    def test_generate_stack_from_input(self):
        self.assertEqual(generate_stack_from_input('dummy1', 'prod'), 'dummy1--prod')
        self.assertEqual(generate_stack_from_input('dummy2', 'alt-config1'), 'dummy2--alt-config1')
        self.assertEqual(generate_stack_from_input('dummy2', 'end2end', alt_config='alt-config1'), 'dummy2--end2end')

    def _dummy_instance_is_active(self, find_ec2_instances, active_stack_names):
        active_stack_names.return_value = ['dummy1--prod']
        instance = MagicMock()
        instance.public_ip_address = '54.54.54.54'
        instance.tags = [{'Key': 'Name', 'Value': 'dummy1--test--1'}]
        find_ec2_instances.return_value = [
            instance
        ]

import os
from . import base
from cfn import ssh, owner_ssh, generate_stack_from_input
from mock import patch, MagicMock

class TestCfn(base.BaseCase):
    def setUp(self):
        os.environ['LOGNAME'] = 'my_user'

    def tearDown(self):
        del os.environ['LOGNAME']

    @patch('cfn.local')
    @patch('buildercore.core.active_stack_names')
    @patch('buildercore.core.find_ec2_instances')
    def test_ssh_task(self, find_ec2_instances, active_stack_names, local):
        self._dummy_instance_is_active(find_ec2_instances, active_stack_names)
        ssh('dummy1--prod')
        local.assert_called_with('ssh elife@54.54.54.54 -i ~/.ssh/id_rsa')

    @patch('cfn.local')
    @patch('buildercore.core.active_stack_names')
    @patch('buildercore.core.find_ec2_instances')
    def test_owner_ssh_task(self, find_ec2_instances, active_stack_names, local):
        self._dummy_instance_is_active(find_ec2_instances, active_stack_names)
        owner_ssh('dummy1--prod')
        (args, _) = local.call_args
        self.assertRegex(args[0], 'ssh ubuntu@54.54.54.54 -i .+/.cfn/keypairs/dummy1--prod.pem')

    # all non-interactive cases
    def test_generate_stack_from_input(self):
        self.assertEqual(generate_stack_from_input('dummy1', 'prod'), 'dummy1--prod')
        self.assertEqual(generate_stack_from_input('dummy2', 'alt-config1'), 'dummy2--alt-config1')
        self.assertEqual(generate_stack_from_input('dummy2', 'end2end', alt_config='alt-config1'), 'dummy2--end2end')

    def _dummy_instance_is_active(self, find_ec2_instances, active_stack_names):
        active_stack_names.return_value = ['dummy1--prod']
        instance = MagicMock()
        instance.public_ip_address = '54.54.54.54'
        instance.tags = [{'Key': 'Name', 'Value': 'dummy1--test--1'}]
        find_ec2_instances.return_value = [
            instance
        ]

Update unit test for ssh task