| FazBrowse GitHub Viewer | Trending | | Home |
| Tools: [Download Repo ZIP] [Original HTTPS Page] |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -232,6 +232,8 @@ def upload_file(self, filename, key=None): | |||
| 232 | 232 | to the root of the bucket | |
| 233 | 233 | with the same name as on your local file system. | |
| 234 | 234 | """ | |
| 235 | + if key is None: | ||
| 236 | + key = os.path.basename(filename) | ||
| 235 | 237 | key = self.new_key(key) | |
| 236 | 238 | return key.set_contents_from_filename(filename) | |
| 237 | 239 | ||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -248,8 +248,22 @@ def test_delete_keys_miss(self): | |||
| 248 | 248 | self.assertEqual(kw[1]['method'], 'DELETE') | |
| 249 | 249 | self.assertEqual(kw[1]['path'], '/b/%s/o/%s' % (NAME, NONESUCH)) | |
| 250 | 250 | ||
| 251 | - # See: https://github.com/GoogleCloudPlatform/gcloud-python/issues/137 | ||
| 252 | - #def test_upload_file_default_key(self): | ||
| 251 | + def test_upload_file_default_key(self): | ||
| 252 | + from gcloud.test_credentials import _Monkey | ||
| 253 | + from gcloud.storage import bucket as MUT | ||
| 254 | + BASENAME = 'file.ext' | ||
| 255 | + FILENAME = '/path/to/%s' % BASENAME | ||
| 256 | + _uploaded = [] | ||
| 257 | + class _Key(object): | ||
| 258 | + def __init__(self, bucket, name): | ||
| 259 | + self._bucket = bucket | ||
| 260 | + self._name = name | ||
| 261 | + def set_contents_from_filename(self, filename): | ||
| 262 | + _uploaded.append((self._bucket, self._name, filename)) | ||
| 263 | + bucket = self._makeOne() | ||
| 264 | + with _Monkey(MUT, Key=_Key): | ||
| 265 | + bucket.upload_file(FILENAME) | ||
| 266 | + self.assertEqual(_uploaded, [(bucket, BASENAME, FILENAME)]) | ||
| 253 | 267 | ||
| 254 | 268 | def test_upload_file_explicit_key(self): | |
| 255 | 269 | from gcloud.test_credentials import _Monkey | |
| Back | FazBrowse Home | New Git URL |
0 commit comments