Scripts to download files from google storage based on sha1 sums

third_party/gsutil/gslib/tests/util.py

Index: third_party/gsutil/gslib/tests/util.py
diff --git a/third_party/gsutil/gslib/tests/util.py b/third_party/gsutil/gslib/tests/util.py
new file mode 100644
index 0000000000000000000000000000000000000000..3f52b0967b4ff1faa3d86fa98fa0393b17bed424
--- /dev/null
+++ b/third_party/gsutil/gslib/tests/util.py
@@ -0,0 +1,79 @@
+# Copyright 2013 Google Inc. 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.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License 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.
+
+import posixpath
+import os.path
+import time
+import urllib
+import urlparse
+
+import unittest
+if not hasattr(unittest.TestCase, 'assertIsNone'):
+  # external dependency unittest2 required for Python <= 2.6
+  import unittest2 as unittest
+
+# Flags for running different types of tests.
+RUN_INTEGRATION_TESTS = True
+RUN_UNIT_TESTS = True
+
+# Whether the tests are running verbose or not.
+VERBOSE_OUTPUT = False
+
+
+def _NormalizeURI(uri):
+  """Normalizes the path component of a URI.
+
+  Examples:
+    gs://foo//bar -> gs://foo/bar
+    gs://foo/./bar -> gs://foo/bar
+  """
+  # Note: we have to do this dance of changing gs:// to file:// because on
+  # Windows, the urlparse function won't work with URL schemes that are not
+  # known. urlparse('gs://foo/bar') on Windows turns into:
+  #     scheme='gs', netloc='', path='//foo/bar'
+  # while on non-Windows platforms, it turns into:
+  #     scheme='gs', netloc='foo', path='/bar'
+  uri = uri.replace('gs://', 'file://')
+  parsed = list(urlparse.urlparse(uri))
+  parsed[2] = posixpath.normpath(parsed[2])
+  if parsed[2].startswith('//'):
+    # The normpath function doesn't change '//foo' -> '/foo' by design.
+    parsed[2] = parsed[2][1:]
+  unparsed = urlparse.urlunparse(parsed)
+  unparsed = unparsed.replace('file://', 'gs://')
+  return unparsed
+
+
+def ObjectToURI(obj, *suffixes):
+  """Returns the storage URI string for a given StorageUri or file object.
+
+  Args:
+    obj: The object to get the URI from. Can be a file object, a subclass of
+         boto.storage_uri.StorageURI, or a string. If a string, it is assumed to
+         be a local on-disk path.
+    suffixes: Suffixes to append. For example, ObjectToUri(bucketuri, 'foo')
+              would return the URI for a key name 'foo' inside the given bucket.
+  """
+  if isinstance(obj, file):
+    return 'file://%s' % os.path.abspath(os.path.join(obj.name, *suffixes))
+  if isinstance(obj, basestring):
+    return 'file://%s' % os.path.join(obj, *suffixes)
+  uri = obj.uri
+  if suffixes:
+    uri = _NormalizeURI('/'.join([uri] + list(suffixes)))
+
+  # Storage URIs shouldn't contain a trailing slash.
+  if uri.endswith('/'):
+    uri = uri[:-1]
+  return uri