salt/tests/unit/cloud/clouds/ec2_test.py

# -*- coding: utf-8 -*-

# Import Python libs
from __future__ import absolute_import
import os
import tempfile

# Import Salt Libs
from salt.cloud.clouds import ec2
from salt.exceptions import SaltCloudSystemExit

# Import Salt Testing Libs
from salttesting import TestCase, skipIf
from salttesting.mock import NO_MOCK, NO_MOCK_REASON
from salttesting.helpers import ensure_in_syspath

ensure_in_syspath('../../../')


@skipIf(NO_MOCK, NO_MOCK_REASON)
class EC2TestCase(TestCase):
    '''
    Unit TestCase for salt.cloud.clouds.ec2 module.
    '''
    def test__validate_key_path_and_mode(self):
        with tempfile.NamedTemporaryFile() as f:
            key_file = f.name

            os.chmod(key_file, 0o644)
            self.assertRaises(SaltCloudSystemExit,
                              ec2._validate_key_path_and_mode,
                              key_file)
            os.chmod(key_file, 0o600)
            self.assertTrue(ec2._validate_key_path_and_mode(key_file))
            os.chmod(key_file, 0o400)
            self.assertTrue(ec2._validate_key_path_and_mode(key_file))

        # tmp file removed
        self.assertRaises(SaltCloudSystemExit,
                          ec2._validate_key_path_and_mode,
                          key_file)


if __name__ == '__main__':
    from integration import run_tests
    run_tests(EC2TestCase, needs_daemon=False)