Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(312)

Issues Search
    My Issues | Starred     Open | Closed | All

Unified Diff: tools/telemetry/third_party/gsutilz/third_party/boto/tests/unit/provider/test_provider.py

Issue 1493973002: Remove telemetry/third_party/gsutilz (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@gsutil_changes
Patch Set: rebase Created 5 years ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « tools/telemetry/third_party/gsutilz/third_party/boto/tests/unit/provider/__init__.py ('k') | tools/telemetry/third_party/gsutilz/third_party/boto/tests/unit/rds/__init__.py » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: tools/telemetry/third_party/gsutilz/third_party/boto/tests/unit/provider/test_provider.py
diff --git a/tools/telemetry/third_party/gsutilz/third_party/boto/tests/unit/provider/test_provider.py b/tools/telemetry/third_party/gsutilz/third_party/boto/tests/unit/provider/test_provider.py
deleted file mode 100755
index 89092253e44a297d5a735c0a5e4538a866368952..0000000000000000000000000000000000000000
--- a/tools/telemetry/third_party/gsutilz/third_party/boto/tests/unit/provider/test_provider.py
+++ /dev/null
@@ -1,441 +0,0 @@
-#!/usr/bin/env python
-from datetime import datetime, timedelta
-
-from tests.compat import mock, unittest
-import os
-
-from boto import provider
-from boto.compat import expanduser
-
-
-INSTANCE_CONFIG = {
- 'allowall': {
- u'AccessKeyId': u'iam_access_key',
- u'Code': u'Success',
- u'Expiration': u'2012-09-01T03:57:34Z',
- u'LastUpdated': u'2012-08-31T21:43:40Z',
- u'SecretAccessKey': u'iam_secret_key',
- u'Token': u'iam_token',
- u'Type': u'AWS-HMAC'
- }
-}
-
-
-class TestProvider(unittest.TestCase):
- def setUp(self):
- self.environ = {}
- self.config = {}
- self.shared_config = {}
-
- self.metadata_patch = mock.patch('boto.utils.get_instance_metadata')
- self.config_patch = mock.patch('boto.provider.config.get',
- self.get_config)
- self.has_config_patch = mock.patch('boto.provider.config.has_option',
- self.has_config)
- self.config_object_patch = mock.patch.object(
- provider.Config, 'get', self.get_shared_config)
- self.has_config_object_patch = mock.patch.object(
- provider.Config, 'has_option', self.has_shared_config)
- self.environ_patch = mock.patch('os.environ', self.environ)
-
- self.get_instance_metadata = self.metadata_patch.start()
- self.get_instance_metadata.return_value = None
- self.config_patch.start()
- self.has_config_patch.start()
- self.config_object_patch.start()
- self.has_config_object_patch.start()
- self.environ_patch.start()
-
-
- def tearDown(self):
- self.metadata_patch.stop()
- self.config_patch.stop()
- self.has_config_patch.stop()
- self.config_object_patch.stop()
- self.has_config_object_patch.stop()
- self.environ_patch.stop()
-
- def has_config(self, section_name, key):
- try:
- self.config[section_name][key]
- return True
- except KeyError:
- return False
-
- def get_config(self, section_name, key):
- try:
- return self.config[section_name][key]
- except KeyError:
- return None
-
- def has_shared_config(self, section_name, key):
- try:
- self.shared_config[section_name][key]
- return True
- except KeyError:
- return False
-
- def get_shared_config(self, section_name, key):
- try:
- return self.shared_config[section_name][key]
- except KeyError:
- return None
-
- def test_passed_in_values_are_used(self):
- p = provider.Provider('aws', 'access_key', 'secret_key', 'security_token')
- self.assertEqual(p.access_key, 'access_key')
- self.assertEqual(p.secret_key, 'secret_key')
- self.assertEqual(p.security_token, 'security_token')
-
- def test_environment_variables_are_used(self):
- self.environ['AWS_ACCESS_KEY_ID'] = 'env_access_key'
- self.environ['AWS_SECRET_ACCESS_KEY'] = 'env_secret_key'
- p = provider.Provider('aws')
- self.assertEqual(p.access_key, 'env_access_key')
- self.assertEqual(p.secret_key, 'env_secret_key')
- self.assertIsNone(p.security_token)
-
- def test_environment_variable_aws_security_token(self):
- self.environ['AWS_ACCESS_KEY_ID'] = 'env_access_key'
- self.environ['AWS_SECRET_ACCESS_KEY'] = 'env_secret_key'
- self.environ['AWS_SECURITY_TOKEN'] = 'env_security_token'
- p = provider.Provider('aws')
- self.assertEqual(p.access_key, 'env_access_key')
- self.assertEqual(p.secret_key, 'env_secret_key')
- self.assertEqual(p.security_token, 'env_security_token')
-
- def test_no_credentials_provided(self):
- p = provider.Provider(
- 'aws',
- provider.NO_CREDENTIALS_PROVIDED,
- provider.NO_CREDENTIALS_PROVIDED,
- provider.NO_CREDENTIALS_PROVIDED
- )
- self.assertEqual(p.access_key, provider.NO_CREDENTIALS_PROVIDED)
- self.assertEqual(p.secret_key, provider.NO_CREDENTIALS_PROVIDED)
- self.assertEqual(p.security_token, provider.NO_CREDENTIALS_PROVIDED)
-
- def test_config_profile_values_are_used(self):
- self.config = {
- 'profile dev': {
- 'aws_access_key_id': 'dev_access_key',
- 'aws_secret_access_key': 'dev_secret_key',
- }, 'profile prod': {
- 'aws_access_key_id': 'prod_access_key',
- 'aws_secret_access_key': 'prod_secret_key',
- }, 'profile prod_withtoken': {
- 'aws_access_key_id': 'prod_access_key',
- 'aws_secret_access_key': 'prod_secret_key',
- 'aws_security_token': 'prod_token',
- }, 'Credentials': {
- 'aws_access_key_id': 'default_access_key',
- 'aws_secret_access_key': 'default_secret_key'
- }
- }
- p = provider.Provider('aws', profile_name='prod')
- self.assertEqual(p.access_key, 'prod_access_key')
- self.assertEqual(p.secret_key, 'prod_secret_key')
- p = provider.Provider('aws', profile_name='prod_withtoken')
- self.assertEqual(p.access_key, 'prod_access_key')
- self.assertEqual(p.secret_key, 'prod_secret_key')
- self.assertEqual(p.security_token, 'prod_token')
- q = provider.Provider('aws', profile_name='dev')
- self.assertEqual(q.access_key, 'dev_access_key')
- self.assertEqual(q.secret_key, 'dev_secret_key')
-
- def test_config_missing_profile(self):
- # None of these default profiles should be loaded!
- self.shared_config = {
- 'default': {
- 'aws_access_key_id': 'shared_access_key',
- 'aws_secret_access_key': 'shared_secret_key',
- }
- }
- self.config = {
- 'Credentials': {
- 'aws_access_key_id': 'default_access_key',
- 'aws_secret_access_key': 'default_secret_key'
- }
- }
- with self.assertRaises(provider.ProfileNotFoundError):
- provider.Provider('aws', profile_name='doesntexist')
-
- def test_config_values_are_used(self):
- self.config = {
- 'Credentials': {
- 'aws_access_key_id': 'cfg_access_key',
- 'aws_secret_access_key': 'cfg_secret_key',
- }
- }
- p = provider.Provider('aws')
- self.assertEqual(p.access_key, 'cfg_access_key')
- self.assertEqual(p.secret_key, 'cfg_secret_key')
- self.assertIsNone(p.security_token)
-
- def test_config_value_security_token_is_used(self):
- self.config = {
- 'Credentials': {
- 'aws_access_key_id': 'cfg_access_key',
- 'aws_secret_access_key': 'cfg_secret_key',
- 'aws_security_token': 'cfg_security_token',
- }
- }
- p = provider.Provider('aws')
- self.assertEqual(p.access_key, 'cfg_access_key')
- self.assertEqual(p.secret_key, 'cfg_secret_key')
- self.assertEqual(p.security_token, 'cfg_security_token')
-
- def test_keyring_is_used(self):
- self.config = {
- 'Credentials': {
- 'aws_access_key_id': 'cfg_access_key',
- 'keyring': 'test',
- }
- }
- import sys
- try:
- import keyring
- imported = True
- except ImportError:
- sys.modules['keyring'] = keyring = type(mock)('keyring', '')
- imported = False
-
- try:
- with mock.patch('keyring.get_password', create=True):
- keyring.get_password.side_effect = (
- lambda kr, login: kr+login+'pw')
- p = provider.Provider('aws')
- self.assertEqual(p.access_key, 'cfg_access_key')
- self.assertEqual(p.secret_key, 'testcfg_access_keypw')
- self.assertIsNone(p.security_token)
- finally:
- if not imported:
- del sys.modules['keyring']
-
- def test_passed_in_values_beat_env_vars(self):
- self.environ['AWS_ACCESS_KEY_ID'] = 'env_access_key'
- self.environ['AWS_SECRET_ACCESS_KEY'] = 'env_secret_key'
- self.environ['AWS_SECURITY_TOKEN'] = 'env_security_token'
- p = provider.Provider('aws', 'access_key', 'secret_key')
- self.assertEqual(p.access_key, 'access_key')
- self.assertEqual(p.secret_key, 'secret_key')
- self.assertEqual(p.security_token, None)
-
- def test_env_vars_beat_shared_creds_values(self):
- self.environ['AWS_ACCESS_KEY_ID'] = 'env_access_key'
- self.environ['AWS_SECRET_ACCESS_KEY'] = 'env_secret_key'
- self.shared_config = {
- 'default': {
- 'aws_access_key_id': 'shared_access_key',
- 'aws_secret_access_key': 'shared_secret_key',
- }
- }
- p = provider.Provider('aws')
- self.assertEqual(p.access_key, 'env_access_key')
- self.assertEqual(p.secret_key, 'env_secret_key')
- self.assertIsNone(p.security_token)
-
- def test_shared_creds_beat_config_values(self):
- self.shared_config = {
- 'default': {
- 'aws_access_key_id': 'shared_access_key',
- 'aws_secret_access_key': 'shared_secret_key',
- }
- }
- self.config = {
- 'Credentials': {
- 'aws_access_key_id': 'cfg_access_key',
- 'aws_secret_access_key': 'cfg_secret_key',
- }
- }
- p = provider.Provider('aws')
- self.assertEqual(p.access_key, 'shared_access_key')
- self.assertEqual(p.secret_key, 'shared_secret_key')
- self.assertIsNone(p.security_token)
-
- def test_shared_creds_profile_beats_defaults(self):
- self.shared_config = {
- 'default': {
- 'aws_access_key_id': 'shared_access_key',
- 'aws_secret_access_key': 'shared_secret_key',
- },
- 'foo': {
- 'aws_access_key_id': 'foo_access_key',
- 'aws_secret_access_key': 'foo_secret_key',
- }
- }
- p = provider.Provider('aws', profile_name='foo')
- self.assertEqual(p.access_key, 'foo_access_key')
- self.assertEqual(p.secret_key, 'foo_secret_key')
- self.assertIsNone(p.security_token)
-
- def test_env_profile_loads_profile(self):
- self.environ['AWS_PROFILE'] = 'foo'
- self.shared_config = {
- 'default': {
- 'aws_access_key_id': 'shared_access_key',
- 'aws_secret_access_key': 'shared_secret_key',
- },
- 'foo': {
- 'aws_access_key_id': 'shared_access_key_foo',
- 'aws_secret_access_key': 'shared_secret_key_foo',
- }
- }
- self.config = {
- 'profile foo': {
- 'aws_access_key_id': 'cfg_access_key_foo',
- 'aws_secret_access_key': 'cfg_secret_key_foo',
- },
- 'Credentials': {
- 'aws_access_key_id': 'cfg_access_key',
- 'aws_secret_access_key': 'cfg_secret_key',
- }
- }
- p = provider.Provider('aws')
- self.assertEqual(p.access_key, 'shared_access_key_foo')
- self.assertEqual(p.secret_key, 'shared_secret_key_foo')
- self.assertIsNone(p.security_token)
-
- self.shared_config = {}
- p = provider.Provider('aws')
- self.assertEqual(p.access_key, 'cfg_access_key_foo')
- self.assertEqual(p.secret_key, 'cfg_secret_key_foo')
- self.assertIsNone(p.security_token)
-
- def test_env_vars_security_token_beats_config_values(self):
- self.environ['AWS_ACCESS_KEY_ID'] = 'env_access_key'
- self.environ['AWS_SECRET_ACCESS_KEY'] = 'env_secret_key'
- self.environ['AWS_SECURITY_TOKEN'] = 'env_security_token'
- self.shared_config = {
- 'default': {
- 'aws_access_key_id': 'shared_access_key',
- 'aws_secret_access_key': 'shared_secret_key',
- 'aws_security_token': 'shared_security_token',
- }
- }
- self.config = {
- 'Credentials': {
- 'aws_access_key_id': 'cfg_access_key',
- 'aws_secret_access_key': 'cfg_secret_key',
- 'aws_security_token': 'cfg_security_token',
- }
- }
- p = provider.Provider('aws')
- self.assertEqual(p.access_key, 'env_access_key')
- self.assertEqual(p.secret_key, 'env_secret_key')
- self.assertEqual(p.security_token, 'env_security_token')
-
- self.environ.clear()
- p = provider.Provider('aws')
- self.assertEqual(p.security_token, 'shared_security_token')
-
- self.shared_config.clear()
- p = provider.Provider('aws')
- self.assertEqual(p.security_token, 'cfg_security_token')
-
- def test_metadata_server_credentials(self):
- self.get_instance_metadata.return_value = INSTANCE_CONFIG
- p = provider.Provider('aws')
- self.assertEqual(p.access_key, 'iam_access_key')
- self.assertEqual(p.secret_key, 'iam_secret_key')
- self.assertEqual(p.security_token, 'iam_token')
- self.assertEqual(
- self.get_instance_metadata.call_args[1]['data'],
- 'meta-data/iam/security-credentials/')
-
- def test_refresh_credentials(self):
- now = datetime.utcnow()
- first_expiration = (now + timedelta(seconds=10)).strftime(
- "%Y-%m-%dT%H:%M:%SZ")
- credentials = {
- u'AccessKeyId': u'first_access_key',
- u'Code': u'Success',
- u'Expiration': first_expiration,
- u'LastUpdated': u'2012-08-31T21:43:40Z',
- u'SecretAccessKey': u'first_secret_key',
- u'Token': u'first_token',
- u'Type': u'AWS-HMAC'
- }
- instance_config = {'allowall': credentials}
- self.get_instance_metadata.return_value = instance_config
- p = provider.Provider('aws')
- self.assertEqual(p.access_key, 'first_access_key')
- self.assertEqual(p.secret_key, 'first_secret_key')
- self.assertEqual(p.security_token, 'first_token')
- self.assertIsNotNone(p._credential_expiry_time)
-
- # Now set the expiration to something in the past.
- expired = now - timedelta(seconds=20)
- p._credential_expiry_time = expired
- credentials['AccessKeyId'] = 'second_access_key'
- credentials['SecretAccessKey'] = 'second_secret_key'
- credentials['Token'] = 'second_token'
- self.get_instance_metadata.return_value = instance_config
-
- # Now upon attribute access, the credentials should be updated.
- self.assertEqual(p.access_key, 'second_access_key')
- self.assertEqual(p.secret_key, 'second_secret_key')
- self.assertEqual(p.security_token, 'second_token')
-
- @mock.patch('boto.provider.config.getint')
- @mock.patch('boto.provider.config.getfloat')
- def test_metadata_config_params(self, config_float, config_int):
- config_int.return_value = 10
- config_float.return_value = 4.0
- self.get_instance_metadata.return_value = INSTANCE_CONFIG
- p = provider.Provider('aws')
- self.assertEqual(p.access_key, 'iam_access_key')
- self.assertEqual(p.secret_key, 'iam_secret_key')
- self.assertEqual(p.security_token, 'iam_token')
- self.get_instance_metadata.assert_called_with(
- timeout=4.0, num_retries=10,
- data='meta-data/iam/security-credentials/')
-
- def test_provider_google(self):
- self.environ['GS_ACCESS_KEY_ID'] = 'env_access_key'
- self.environ['GS_SECRET_ACCESS_KEY'] = 'env_secret_key'
- self.shared_config = {
- 'default': {
- 'gs_access_key_id': 'shared_access_key',
- 'gs_secret_access_key': 'shared_secret_key',
- }
- }
- self.config = {
- 'Credentials': {
- 'gs_access_key_id': 'cfg_access_key',
- 'gs_secret_access_key': 'cfg_secret_key',
- }
- }
- p = provider.Provider('google')
- self.assertEqual(p.access_key, 'env_access_key')
- self.assertEqual(p.secret_key, 'env_secret_key')
-
- self.environ.clear()
- p = provider.Provider('google')
- self.assertEqual(p.access_key, 'shared_access_key')
- self.assertEqual(p.secret_key, 'shared_secret_key')
-
- self.shared_config.clear()
- p = provider.Provider('google')
- self.assertEqual(p.access_key, 'cfg_access_key')
- self.assertEqual(p.secret_key, 'cfg_secret_key')
-
- @mock.patch('os.path.isfile', return_value=True)
- @mock.patch.object(provider.Config, 'load_from_path')
- def test_shared_config_loading(self, load_from_path, exists):
- provider.Provider('aws')
- path = os.path.join(expanduser('~'), '.aws', 'credentials')
- exists.assert_called_once_with(path)
- load_from_path.assert_called_once_with(path)
-
- exists.reset_mock()
- load_from_path.reset_mock()
-
- provider.Provider('google')
- path = os.path.join(expanduser('~'), '.google', 'credentials')
- exists.assert_called_once_with(path)
- load_from_path.assert_called_once_with(path)
-
-
-if __name__ == '__main__':
- unittest.main()
« no previous file with comments | « tools/telemetry/third_party/gsutilz/third_party/boto/tests/unit/provider/__init__.py ('k') | tools/telemetry/third_party/gsutilz/third_party/boto/tests/unit/rds/__init__.py » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698