| OLD | NEW |
| (Empty) |
| 1 #!/usr/bin/env python | |
| 2 # | |
| 3 # Copyright 2010 Google Inc. | |
| 4 # | |
| 5 # Permission is hereby granted, free of charge, to any person obtaining a | |
| 6 # copy of this software and associated documentation files (the | |
| 7 # "Software"), to deal in the Software without restriction, including | |
| 8 # without limitation the rights to use, copy, modify, merge, publish, dis- | |
| 9 # tribute, sublicense, and/or sell copies of the Software, and to permit | |
| 10 # persons to whom the Software is furnished to do so, subject to the fol- | |
| 11 # lowing conditions: | |
| 12 # | |
| 13 # The above copyright notice and this permission notice shall be included | |
| 14 # in all copies or substantial portions of the Software. | |
| 15 # | |
| 16 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS | |
| 17 # OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABIL- | |
| 18 # ITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT | |
| 19 # SHALL THE AUTHOR BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, | |
| 20 # WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | |
| 21 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS | |
| 22 # IN THE SOFTWARE. | |
| 23 | |
| 24 """Utility class for gslib unit tests""" | |
| 25 | |
| 26 import sys | |
| 27 # Put local libs at front of path so tests will run latest lib code rather | |
| 28 # than whatever code is found on user's PYTHONPATH. | |
| 29 sys.path.insert(0, '.') | |
| 30 sys.path.insert(0, 'boto') | |
| 31 import boto | |
| 32 from gslib.project_id import ProjectIdHandler | |
| 33 from tests.s3 import mock_storage_service | |
| 34 import wildcard_iterator | |
| 35 | |
| 36 proj_id_handler = ProjectIdHandler() | |
| 37 | |
| 38 | |
| 39 def test_wildcard_iterator(uri_or_str, | |
| 40 result_type=wildcard_iterator.ResultType.URIS, | |
| 41 debug=0): | |
| 42 """ | |
| 43 Convenience method for instantiating a testing | |
| 44 instance of WildCardIterator, without having to specify | |
| 45 all the params of that class (like | |
| 46 bucket_storage_uri_class=mock_storage_service.MockBucketStorageUri). | |
| 47 Also naming the factory method this way makes it clearer in the test | |
| 48 code that WildcardIterator needs to be set up for testing. | |
| 49 | |
| 50 Args, Returns, and Raises are same as for | |
| 51 wildcard_iterator.wildcard_iterator(), except there's no | |
| 52 bucket_storage_uri_class arg. | |
| 53 """ | |
| 54 return wildcard_iterator.wildcard_iterator( | |
| 55 uri_or_str, proj_id_handler, result_type, | |
| 56 mock_storage_service.MockBucketStorageUri, headers={}, debug=debug) | |
| 57 | |
| 58 | |
| 59 def test_storage_uri(uri_str, default_scheme='file', debug=0, validate=True): | |
| 60 """ | |
| 61 Convenience method for instantiating a testing | |
| 62 instance of StorageUri, without having to specify | |
| 63 bucket_storage_uri_class=mock_storage_service.MockBucketStorageUri. | |
| 64 Also naming the factory method this way makes it clearer in the test | |
| 65 code that StorageUri needs to be set up for testing. | |
| 66 | |
| 67 Args, Returns, and Raises are same as for | |
| 68 boto.storage_uri(), except there's no bucket_storage_uri_class arg. | |
| 69 """ | |
| 70 | |
| 71 return boto.storage_uri(uri_str, default_scheme, debug, validate, | |
| 72 mock_storage_service.MockBucketStorageUri) | |
| OLD | NEW |