1d7da740b2
Add a new module that contain generic wrapper for file and iterator, which are used to wrap image to upload and the request body iterator in upload and download cases repectively, to show and advance a pretty progress bar when this laters are consumed, The progress bar is triggered by adding a --progress command line argument to commands: image-create, image-download or image-update. Change-Id: I2ba42fd0c58f4fa087adb568ec3f08246cae3759 bug fix: LP#1112309 blueprint: progressbar-when-uploading
109 lines
3.6 KiB
Python
109 lines
3.6 KiB
Python
# Copyright 2012 OpenStack LLC.
|
|
# 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 sys
|
|
import StringIO
|
|
|
|
import testtools
|
|
|
|
from glanceclient.common import utils
|
|
|
|
|
|
class TestUtils(testtools.TestCase):
|
|
|
|
def test_make_size_human_readable(self):
|
|
self.assertEqual("106B", utils.make_size_human_readable(106))
|
|
self.assertEqual("1000kB", utils.make_size_human_readable(1024000))
|
|
self.assertEqual("1MB", utils.make_size_human_readable(1048576))
|
|
self.assertEqual("1.4GB", utils.make_size_human_readable(1476395008))
|
|
self.assertEqual("9.3MB", utils.make_size_human_readable(9761280))
|
|
|
|
def test_get_new_file_size(self):
|
|
size = 98304
|
|
file_obj = StringIO.StringIO('X' * size)
|
|
try:
|
|
self.assertEqual(utils.get_file_size(file_obj), size)
|
|
# Check that get_file_size didn't change original file position.
|
|
self.assertEqual(file_obj.tell(), 0)
|
|
finally:
|
|
file_obj.close()
|
|
|
|
def test_get_consumed_file_size(self):
|
|
size, consumed = 98304, 304
|
|
file_obj = StringIO.StringIO('X' * size)
|
|
file_obj.seek(consumed)
|
|
try:
|
|
self.assertEqual(utils.get_file_size(file_obj), size)
|
|
# Check that get_file_size didn't change original file position.
|
|
self.assertEqual(file_obj.tell(), consumed)
|
|
finally:
|
|
file_obj.close()
|
|
|
|
def test_prettytable(self):
|
|
class Struct:
|
|
def __init__(self, **entries):
|
|
self.__dict__.update(entries)
|
|
|
|
# test that the prettytable output is wellformatted (left-aligned)
|
|
columns = ['ID', 'Name']
|
|
val = ['Name1', 'another', 'veeeery long']
|
|
images = [Struct(**{'id': i ** 16, 'name': val[i]})
|
|
for i in range(len(val))]
|
|
|
|
saved_stdout = sys.stdout
|
|
try:
|
|
sys.stdout = output_list = StringIO.StringIO()
|
|
utils.print_list(images, columns)
|
|
|
|
sys.stdout = output_dict = StringIO.StringIO()
|
|
utils.print_dict({'K': 'k', 'Key': 'Value'})
|
|
|
|
finally:
|
|
sys.stdout = saved_stdout
|
|
|
|
self.assertEqual(output_list.getvalue(), '''\
|
|
+-------+--------------+
|
|
| ID | Name |
|
|
+-------+--------------+
|
|
| | Name1 |
|
|
| 1 | another |
|
|
| 65536 | veeeery long |
|
|
+-------+--------------+
|
|
''')
|
|
|
|
self.assertEqual(output_dict.getvalue(), '''\
|
|
+----------+-------+
|
|
| Property | Value |
|
|
+----------+-------+
|
|
| K | k |
|
|
| Key | Value |
|
|
+----------+-------+
|
|
''')
|
|
|
|
def test_exception_to_str(self):
|
|
class FakeException(Exception):
|
|
def __str__(self):
|
|
raise UnicodeError()
|
|
|
|
ret = utils.exception_to_str(Exception('error message'))
|
|
self.assertEqual(ret, 'error message')
|
|
|
|
ret = utils.exception_to_str(Exception('\xa5 error message'))
|
|
self.assertEqual(ret, ' error message')
|
|
|
|
ret = utils.exception_to_str(FakeException('\xa5 error message'))
|
|
self.assertEqual(ret, "Caught '%(exception)s' exception." %
|
|
{'exception': 'FakeException'})
|