[catapult] - Copy Telemetry's gsutilz over to third_party.

third_party/gsutil/third_party/boto/tests/unit/utils/test_utils.py

+# Copyright (c) 2010 Robert Mela
+#
+# Permission is hereby granted, free of charge, to any person obtaining a
+# copy of this software and associated documentation files (the
+# "Software"), to deal in the Software without restriction, including
+# without limitation the rights to use, copy, modify, merge, publish, dis-
+# tribute, sublicense, and/or sell copies of the Software, and to permit
+# persons to whom the Software is furnished to do so, subject to the fol-
+# lowing conditions:
+#
+# The above copyright notice and this permission notice shall be included
+# in all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
+# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABIL-
+# ITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT
+# SHALL THE AUTHOR BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
+# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
+# IN THE SOFTWARE.
+#
+from tests.compat import mock, unittest
+
+import datetime
+import hashlib
+import hmac
+import locale
+import time
+
+import boto.utils
+from boto.utils import Password
+from boto.utils import pythonize_name
+from boto.utils import _build_instance_metadata_url
+from boto.utils import get_instance_userdata
+from boto.utils import retry_url
+from boto.utils import LazyLoadMetadata
+
+from boto.compat import json, _thread
+
+
+@unittest.skip("http://bugs.python.org/issue7980")
+class TestThreadImport(unittest.TestCase):
+    def test_strptime(self):
+        def f():
+            for m in range(1, 13):
+                for d in range(1,29):
+                    boto.utils.parse_ts('2013-01-01T00:00:00Z')
+
+        for _ in range(10):
+            _thread.start_new_thread(f, ())
+
+        time.sleep(3)
+
+
+class TestPassword(unittest.TestCase):
+    """Test basic password functionality"""
+
+    def clstest(self, cls):
+        """Insure that password.__eq__ hashes test value before compare."""
+        password = cls('foo')
+        self.assertNotEquals(password, 'foo')
+
+        password.set('foo')
+        hashed = str(password)
+        self.assertEquals(password, 'foo')
+        self.assertEquals(password.str, hashed)
+
+        password = cls(hashed)
+        self.assertNotEquals(password.str, 'foo')
+        self.assertEquals(password, 'foo')
+        self.assertEquals(password.str, hashed)
+
+    def test_aaa_version_1_9_default_behavior(self):
+        self.clstest(Password)
+
+    def test_custom_hashclass(self):
+        class SHA224Password(Password):
+            hashfunc = hashlib.sha224
+
+        password = SHA224Password()
+        password.set('foo')
+        self.assertEquals(hashlib.sha224(b'foo').hexdigest(), str(password))
+
+    def test_hmac(self):
+        def hmac_hashfunc(cls, msg):
+            if not isinstance(msg, bytes):
+                msg = msg.encode('utf-8')
+            return hmac.new(b'mysecretkey', msg)
+
+        class HMACPassword(Password):
+            hashfunc = hmac_hashfunc
+
+        self.clstest(HMACPassword)
+        password = HMACPassword()
+        password.set('foo')
+
+        self.assertEquals(str(password),
+                          hmac.new(b'mysecretkey', b'foo').hexdigest())
+
+    def test_constructor(self):
+        hmac_hashfunc = lambda msg: hmac.new(b'mysecretkey', msg)
+
+        password = Password(hashfunc=hmac_hashfunc)
+        password.set('foo')
+        self.assertEquals(password.str,
+                          hmac.new(b'mysecretkey', b'foo').hexdigest())
+
+
+class TestPythonizeName(unittest.TestCase):
+    def test_empty_string(self):
+        self.assertEqual(pythonize_name(''), '')
+
+    def test_all_lower_case(self):
+        self.assertEqual(pythonize_name('lowercase'), 'lowercase')
+
+    def test_all_upper_case(self):
+        self.assertEqual(pythonize_name('UPPERCASE'), 'uppercase')
+
+    def test_camel_case(self):
+        self.assertEqual(pythonize_name('OriginallyCamelCased'),
+                         'originally_camel_cased')
+
+    def test_already_pythonized(self):
+        self.assertEqual(pythonize_name('already_pythonized'),
+                         'already_pythonized')
+
+    def test_multiple_upper_cased_letters(self):
+        self.assertEqual(pythonize_name('HTTPRequest'), 'http_request')
+        self.assertEqual(pythonize_name('RequestForHTTP'), 'request_for_http')
+
+    def test_string_with_numbers(self):
+        self.assertEqual(pythonize_name('HTTPStatus200Ok'), 'http_status_200_ok')
+
+
+class TestBuildInstanceMetadataURL(unittest.TestCase):
+    def test_normal(self):
+        # This is the all-defaults case.
+        self.assertEqual(_build_instance_metadata_url(
+                'http://169.254.169.254',
+                'latest',
+                'meta-data/'
+            ),
+            'http://169.254.169.254/latest/meta-data/'
+        )
+
+    def test_custom_path(self):
+        self.assertEqual(_build_instance_metadata_url(
+                'http://169.254.169.254',
+                'latest',
+                'dynamic/'
+            ),
+            'http://169.254.169.254/latest/dynamic/'
+        )
+
+    def test_custom_version(self):
+        self.assertEqual(_build_instance_metadata_url(
+                'http://169.254.169.254',
+                '1.0',
+                'meta-data/'
+            ),
+            'http://169.254.169.254/1.0/meta-data/'
+        )
+
+    def test_custom_url(self):
+        self.assertEqual(_build_instance_metadata_url(
+                'http://10.0.1.5',
+                'latest',
+                'meta-data/'
+            ),
+            'http://10.0.1.5/latest/meta-data/'
+        )
+
+    def test_all_custom(self):
+        self.assertEqual(_build_instance_metadata_url(
+                'http://10.0.1.5',
+                '2013-03-22',
+                'user-data'
+            ),
+            'http://10.0.1.5/2013-03-22/user-data'
+        )
+
+class TestRetryURL(unittest.TestCase):
+    def setUp(self):
+        self.urlopen_patch = mock.patch('boto.compat.urllib.request.urlopen')
+        self.opener_patch = mock.patch('boto.compat.urllib.request.build_opener')
+        self.urlopen = self.urlopen_patch.start()
+        self.opener = self.opener_patch.start()
+
+    def tearDown(self):
+        self.urlopen_patch.stop()
+        self.opener_patch.stop()
+
+    def set_normal_response(self, response):
+        fake_response = mock.Mock()
+        fake_response.read.return_value = response
+        self.urlopen.return_value = fake_response
+
+    def set_no_proxy_allowed_response(self, response):
+        fake_response = mock.Mock()
+        fake_response.read.return_value = response
+        self.opener.return_value.open.return_value = fake_response
+
+    def test_retry_url_uses_proxy(self):
+        self.set_normal_response('normal response')
+        self.set_no_proxy_allowed_response('no proxy response')
+
+        response = retry_url('http://10.10.10.10/foo', num_retries=1)
+        self.assertEqual(response, 'no proxy response')
+
+    def test_retry_url_using_bytes_and_string_response(self):
+        test_value = 'normal response'
+        fake_response = mock.Mock()
+
+        # test using unicode
+        fake_response.read.return_value = test_value
+        self.opener.return_value.open.return_value = fake_response
+        response = retry_url('http://10.10.10.10/foo', num_retries=1)
+        self.assertEqual(response, test_value)
+
+        # test using bytes
+        fake_response.read.return_value = test_value.encode('utf-8')
+        self.opener.return_value.open.return_value = fake_response
+        response = retry_url('http://10.10.10.10/foo', num_retries=1)
+        self.assertEqual(response, test_value)
+
+class TestLazyLoadMetadata(unittest.TestCase):
+
+    def setUp(self):
+        self.retry_url_patch = mock.patch('boto.utils.retry_url')
+        boto.utils.retry_url = self.retry_url_patch.start()
+
+    def tearDown(self):
+        self.retry_url_patch.stop()
+
+    def set_normal_response(self, data):
+        # here "data" should be a list of return values in some order
+        fake_response = mock.Mock()
+        fake_response.side_effect = data
+        boto.utils.retry_url = fake_response
+
+    def test_meta_data_with_invalid_json_format_happened_once(self):
+        # here "key_data" will be stored in the "self._leaves"
+        # when the class "LazyLoadMetadata" initialized
+        key_data = "test"
+        invalid_data = '{"invalid_json_format" : true,}'
+        valid_data = '{ "%s" : {"valid_json_format": true}}' % key_data
+        url = "/".join(["http://169.254.169.254", key_data])
+        num_retries = 2
+
+        self.set_normal_response([key_data, invalid_data, valid_data])
+        response = LazyLoadMetadata(url, num_retries)
+        self.assertEqual(list(response.values())[0], json.loads(valid_data))
+
+    def test_meta_data_with_invalid_json_format_happened_twice(self):
+        key_data = "test"
+        invalid_data = '{"invalid_json_format" : true,}'
+        valid_data = '{ "%s" : {"valid_json_format": true}}' % key_data
+        url = "/".join(["http://169.254.169.254", key_data])
+        num_retries = 2
+
+        self.set_normal_response([key_data, invalid_data, invalid_data])
+        response = LazyLoadMetadata(url, num_retries)
+        with self.assertRaises(ValueError):
+            response.values()[0]
+
+    def test_user_data(self):
+        self.set_normal_response(['foo'])
+
+        userdata = get_instance_userdata()
+
+        self.assertEqual('foo', userdata)
+
+        boto.utils.retry_url.assert_called_with(
+            'http://169.254.169.254/latest/user-data',
+            retry_on_404=False,
+            num_retries=5, timeout=None)
+
+    def test_user_data_timeout(self):
+        self.set_normal_response(['foo'])
+
+        userdata = get_instance_userdata(timeout=1, num_retries=2)
+
+        self.assertEqual('foo', userdata)
+
+        boto.utils.retry_url.assert_called_with(
+            'http://169.254.169.254/latest/user-data',
+            retry_on_404=False,
+            num_retries=2, timeout=1)
+
+
+class TestStringToDatetimeParsing(unittest.TestCase):
+    """ Test string to datetime parsing """
+    def setUp(self):
+        self._saved = locale.setlocale(locale.LC_ALL)
+        try:
+            locale.setlocale(locale.LC_ALL, 'de_DE.UTF-8')
+        except locale.Error:
+            self.skipTest('Unsupported locale setting')
+
+    def tearDown(self):
+        locale.setlocale(locale.LC_ALL, self._saved)
+
+    def test_nonus_locale(self):
+        test_string = 'Thu, 15 May 2014 09:06:03 GMT'
+
+        # Default strptime shoudl fail
+        with self.assertRaises(ValueError):
+            datetime.datetime.strptime(test_string, boto.utils.RFC1123)
+
+        # Our parser should succeed
+        result = boto.utils.parse_ts(test_string)
+
+        self.assertEqual(2014, result.year)
+        self.assertEqual(5, result.month)
+        self.assertEqual(15, result.day)
+        self.assertEqual(9, result.hour)
+        self.assertEqual(6, result.minute)
+
+
+if __name__ == '__main__':
+    unittest.main()