#!/usr/bin/env # Copyright (c) 2012-2013 Mitch Garnaat http://garnaat.org/ # Copyright 2012-2014 Amazon.com, Inc. or its affiliates. All Rights Reserved. # # Licensed under the Apache License, Version 2.0 (the "License"). You # may not use this file except in compliance with the License. A copy of # the License is located at # # http://aws.amazon.com/apache2.0/ # # or in the "license" file accompanying this file. This file is # distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF # ANY KIND, either express or implied. See the License for the specific # language governing permissions and limitations under the License. from tests import unittest, BaseEnvVar import os import mock import tempfile import shutil import botocore.exceptions from botocore.configloader import raw_config_parse, load_config, \ multi_file_load_config from botocore.compat import six def path(filename): directory = os.path.join(os.path.dirname(__file__), 'cfg') if isinstance(filename, six.binary_type): directory = six.b(directory) return os.path.join(directory, filename) class TestConfigLoader(BaseEnvVar): def setUp(self): self.tempdir = tempfile.mkdtemp() def tearDown(self): shutil.rmtree(self.tempdir) def create_config_file(self, filename): contents = ( '[default]\n' 'aws_access_key_id = foo\n' 'aws_secret_access_key = bar\n\n' '[profile "personal"]\n' 'aws_access_key_id = fie\n' 'aws_secret_access_key = baz\n' 'aws_security_token = fiebaz\n' ) directory = self.tempdir if isinstance(filename, six.binary_type): directory = six.b(directory) full_path = os.path.join(directory, filename) with open(full_path, 'w') as f: f.write(contents) return full_path def test_config_not_found(self): with self.assertRaises(botocore.exceptions.ConfigNotFound): loaded_config = raw_config_parse(path('aws_config_notfound')) def test_config_parse_error(self): filename = path('aws_config_bad') with self.assertRaises(botocore.exceptions.ConfigParseError): raw_config_parse(filename) def test_config_parse_error_filesystem_encoding_none(self): filename = path('aws_config_bad') with mock.patch('sys.getfilesystemencoding') as encoding: encoding.return_value = None with self.assertRaises(botocore.exceptions.ConfigParseError): raw_config_parse(filename) def test_config(self): loaded_config = raw_config_parse(path('aws_config')) self.assertIn('default', loaded_config) self.assertIn('profile "personal"', loaded_config) def test_profile_map_conversion(self): loaded_config = load_config(path('aws_config')) self.assertIn('profiles', loaded_config) self.assertEqual(sorted(loaded_config['profiles'].keys()), ['default', 'personal']) def test_bad_profiles_are_ignored(self): filename = path('aws_bad_profile') loaded_config = load_config(filename) self.assertEqual(len(loaded_config['profiles']), 3) profiles = loaded_config['profiles'] self.assertIn('my profile', profiles) self.assertIn('personal1', profiles) self.assertIn('default', profiles) def test_nested_hierarchy_parsing(self): filename = path('aws_config_nested') loaded_config = load_config(filename) config = loaded_config['profiles']['default'] self.assertEqual(config['aws_access_key_id'], 'foo') self.assertEqual(config['region'], 'us-west-2') self.assertEqual(config['s3']['signature_version'], 's3v4') self.assertEqual(config['cloudwatch']['signature_version'], 'v4') def test_nested_hierarchy_with_no_subsection_parsing(self): filename = path('aws_config_nested') raw_config = raw_config_parse(filename, False)['default'] self.assertEqual(raw_config['aws_access_key_id'], 'foo') self.assertEqual(raw_config['region'], 'us-west-2') # Specifying False for pase_subsections in raw_config_parse # will make sure that indented sections such as singature_version # will not be treated as another subsection but rather # its literal value. self.assertEqual( raw_config['cloudwatch'], '\nsignature_version = v4') self.assertEqual( raw_config['s3'], '\nsignature_version = s3v4' '\naddressing_style = path' ) def test_nested_bad_config(self): filename = path('aws_config_nested_bad') with self.assertRaises(botocore.exceptions.ConfigParseError): loaded_config = load_config(filename) def test_nested_bad_config_filesystem_encoding_none(self): filename = path('aws_config_nested_bad') with mock.patch('sys.getfilesystemencoding') as encoding: encoding.return_value = None with self.assertRaises(botocore.exceptions.ConfigParseError): loaded_config = load_config(filename) def test_multi_file_load(self): filenames = [path('aws_config_other'), path('aws_config'), path('aws_third_config'), path('aws_config_notfound')] loaded_config = multi_file_load_config(*filenames) config = loaded_config['profiles']['default'] self.assertEqual(config['aws_access_key_id'], 'other_foo') self.assertEqual(config['aws_secret_access_key'], 'other_bar') second_config = loaded_config['profiles']['personal'] self.assertEqual(second_config['aws_access_key_id'], 'fie') self.assertEqual(second_config['aws_secret_access_key'], 'baz') self.assertEqual(second_config['aws_security_token'], 'fiebaz') third_config = loaded_config['profiles']['third'] self.assertEqual(third_config['aws_access_key_id'], 'third_fie') self.assertEqual(third_config['aws_secret_access_key'], 'third_baz') self.assertEqual(third_config['aws_security_token'], 'third_fiebaz') def test_unicode_bytes_path_not_found(self): with self.assertRaises(botocore.exceptions.ConfigNotFound): with mock.patch('sys.getfilesystemencoding') as encoding: encoding.return_value = 'utf-8' load_config(path(b'\xe2\x9c\x93')) def test_unicode_bytes_path_not_found_filesystem_encoding_none(self): with mock.patch('sys.getfilesystemencoding') as encoding: encoding.return_value = None with self.assertRaises(botocore.exceptions.ConfigNotFound): load_config(path(b'\xe2\x9c\x93')) def test_unicode_bytes_path(self): filename = self.create_config_file(b'aws_config_unicode\xe2\x9c\x93') with mock.patch('sys.getfilesystemencoding') as encoding: encoding.return_value = 'utf-8' loaded_config = load_config(filename) self.assertIn('default', loaded_config['profiles']) self.assertIn('personal', loaded_config['profiles']) if __name__ == "__main__": unittest.main()