mirror of
https://github.com/valitydev/salt.git
synced 2024-11-07 17:09:03 +00:00
47 lines
1.3 KiB
Python
47 lines
1.3 KiB
Python
# -*- 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)
|