Use oslo_utils.units where appropriate

Use units from oslo_utils where appropriate instead of using magic
numbers.

Changes from patch Ib08b8d8843b72966e2cf87f741b1cc0eea0672e5 are also
included.

Co-Authored-By: Gorka Eguileor <geguileo@redhat.com>
Co-Authored-By: James Page <james.page@canonical.com>
Change-Id: I082ea91cb06e49659495cae9072b263eccda76a1

9 files changed, 36 insertions, 32 deletions

diff --git a/glance_store/_drivers/cinder.py b/glance_store/_drivers/cinder.py
index f91072c..ab9e284 100644
--- a/glance_store/_drivers/cinder.py
+++ b/glance_store/_drivers/cinder.py
@@ -18,6 +18,7 @@ from cinderclient import exceptions as cinder_exception
 from cinderclient import service_catalog
 from cinderclient.v2 import client as cinderclient
 from oslo_config import cfg
+from oslo_utils import units
 
 from glance_store import capabilities
 from glance_store.common import utils
@@ -173,7 +174,7 @@ class Store(glance_store.driver.Store):
             volume = get_cinderclient(self.conf,
                                       context).volumes.get(loc.volume_id)
             # GB unit convert to byte
-            return volume.size * (1024 ** 3)
+            return volume.size * units.Gi
         except cinder_exception.NotFound as e:
             reason = _("Failed to get image size due to "
                        "volume can not be found: %s") % self.volume_id
diff --git a/glance_store/_drivers/rbd.py b/glance_store/_drivers/rbd.py
index 09d2468..8d64e34 100644
--- a/glance_store/_drivers/rbd.py
+++ b/glance_store/_drivers/rbd.py
@@ -24,6 +24,7 @@ import math
 import urllib
 
 from oslo_config import cfg
+from oslo_utils import units
 
 from glance_store import capabilities
 from glance_store.common import utils
@@ -193,7 +194,7 @@ class Store(driver.Store):
         """
         try:
             chunk = self.conf.glance_store.rbd_store_chunk_size
-            self.chunk_size = chunk * (1024 ** 2)
+            self.chunk_size = chunk * units.Mi
             self.READ_CHUNKSIZE = self.chunk_size
             self.WRITE_CHUNKSIZE = self.READ_CHUNKSIZE
 
@@ -369,7 +370,7 @@ class Store(driver.Store):
                                 length = offset + chunk_length
                                 bytes_written += chunk_length
                                 LOG.debug(_("resizing image to %s KiB") %
-                                          (length / 1024))
+                                          (length / units.Ki))
                                 image.resize(length)
                             LOG.debug(_("writing chunk at offset %s") %
                                       (offset))
diff --git a/glance_store/_drivers/swift/store.py b/glance_store/_drivers/swift/store.py
index 30b69f3..4ec42f9 100644
--- a/glance_store/_drivers/swift/store.py
+++ b/glance_store/_drivers/swift/store.py
@@ -22,6 +22,7 @@ import math
 
 from oslo.config import cfg
 from oslo.utils import excutils
+from oslo_utils import units
 import six
 import six.moves.urllib.parse as urlparse
 import swiftclient
@@ -43,9 +44,9 @@ LOG = logging.getLogger(__name__)
 _LI = i18n._LI
 
 DEFAULT_CONTAINER = 'glance'
-DEFAULT_LARGE_OBJECT_SIZE = 5 * 1024  # 5GB
+DEFAULT_LARGE_OBJECT_SIZE = 5 * units.Ki  # 5GB
 DEFAULT_LARGE_OBJECT_CHUNK_SIZE = 200  # 200M
-ONE_MB = 1000 * 1024
+ONE_MB = units.k * units.Ki  # Here we used the mixed meaning of MB
 
 _SWIFT_OPTS = [
     cfg.StrOpt('swift_store_auth_version', default='2',
diff --git a/glance_store/driver.py b/glance_store/driver.py
index c7cf3a5..cf98d1c 100644
--- a/glance_store/driver.py
+++ b/glance_store/driver.py
@@ -20,6 +20,7 @@ import logging
 
 from oslo.config import cfg
 from oslo.utils import importutils
+from oslo_utils import units
 
 from glance_store import capabilities
 from glance_store.common import utils
@@ -33,7 +34,7 @@ LOG = logging.getLogger(__name__)
 class Store(capabilities.StoreCapability):
 
     OPTIONS = None
-    READ_CHUNKSIZE = 16 * (1024 * 1024)  # 16M
+    READ_CHUNKSIZE = 16 * units.Mi  # 16M
     WRITE_CHUNKSIZE = READ_CHUNKSIZE
 
     def __init__(self, conf):
diff --git a/tests/unit/test_cinder_store.py b/tests/unit/test_cinder_store.py
index 0972664..c526360 100644
--- a/tests/unit/test_cinder_store.py
+++ b/tests/unit/test_cinder_store.py
@@ -15,6 +15,8 @@
 
 import mock
 
+from oslo_utils import units
+
 import glance_store
 from glance_store._drivers import cinder
 from glance_store import exceptions
@@ -70,7 +72,7 @@ class TestCinderStore(base.StoreBaseTest,
             loc = location.get_location_from_uri(uri, conf=self.conf)
             image_size = self.store.get_size(loc, context=fake_context)
             self.assertEqual(image_size,
-                             fake_volumes.values()[0].size * (1024 ** 3))
+                             fake_volumes.values()[0].size * units.Gi)
 
     def test_cinder_delete_raise_error(self):
         uri = 'cinder://12345678-9012-3455-6789-012345678901'
diff --git a/tests/unit/test_filesystem_store.py b/tests/unit/test_filesystem_store.py
index ac72e89..27993e4 100644
--- a/tests/unit/test_filesystem_store.py
+++ b/tests/unit/test_filesystem_store.py
@@ -39,9 +39,6 @@ from glance_store.tests import base
 from tests.unit import test_store_capabilities
 
 
-KB = 1024
-
-
 class TestStore(base.StoreBaseTest,
                 test_store_capabilities.TestStoreCapabilitiesChecking):
 
@@ -155,9 +152,9 @@ class TestStore(base.StoreBaseTest,
 
     def test_add(self):
         """Test that we can add an image via the filesystem backend."""
-        ChunkedFile.CHUNKSIZE = 1024
+        ChunkedFile.CHUNKSIZE = units.Ki
         expected_image_id = str(uuid.uuid4())
-        expected_file_size = 5 * KB  # 5K
+        expected_file_size = 5 * units.Ki  # 5K
         expected_file_contents = "*" * expected_file_size
         expected_checksum = hashlib.md5(expected_file_contents).hexdigest()
         expected_location = "file://%s/%s" % (self.test_dir,
@@ -225,9 +222,9 @@ class TestStore(base.StoreBaseTest,
         Tests that adding an image with an existing identifier
         raises an appropriate exception
         """
-        ChunkedFile.CHUNKSIZE = 1024
+        ChunkedFile.CHUNKSIZE = units.Ki
         image_id = str(uuid.uuid4())
-        file_size = 5 * KB  # 5K
+        file_size = 5 * units.Ki  # 5K
         file_contents = "*" * file_size
         image_file = StringIO.StringIO(file_contents)
 
@@ -240,9 +237,9 @@ class TestStore(base.StoreBaseTest,
                           image_id, image_file, 0)
 
     def _do_test_add_write_failure(self, errno, exception):
-        ChunkedFile.CHUNKSIZE = 1024
+        ChunkedFile.CHUNKSIZE = units.Ki
         image_id = str(uuid.uuid4())
-        file_size = 5 * KB  # 5K
+        file_size = 5 * units.Ki  # 5K
         file_contents = "*" * file_size
         path = os.path.join(self.test_dir, image_id)
         image_file = StringIO.StringIO(file_contents)
@@ -291,9 +288,9 @@ class TestStore(base.StoreBaseTest,
         Tests the partial image file is cleaned up after a read
         failure.
         """
-        ChunkedFile.CHUNKSIZE = 1024
+        ChunkedFile.CHUNKSIZE = units.Ki
         image_id = str(uuid.uuid4())
-        file_size = 5 * KB  # 5K
+        file_size = 5 * units.Ki  # 5K
         file_contents = "*" * file_size
         path = os.path.join(self.test_dir, image_id)
         image_file = StringIO.StringIO(file_contents)
@@ -315,7 +312,7 @@ class TestStore(base.StoreBaseTest,
         """
         # First add an image
         image_id = str(uuid.uuid4())
-        file_size = 5 * KB  # 5K
+        file_size = 5 * units.Ki  # 5K
         file_contents = "*" * file_size
         image_file = StringIO.StringIO(file_contents)
 
@@ -468,7 +465,7 @@ class TestStore(base.StoreBaseTest,
         self.store.configure()
 
         """Test that we can add an image via the filesystem backend"""
-        ChunkedFile.CHUNKSIZE = 1024
+        ChunkedFile.CHUNKSIZE = units.Ki
         expected_image_id = str(uuid.uuid4())
         expected_file_size = 5 * units.Ki  # 5K
         expected_file_contents = "*" * expected_file_size
@@ -520,7 +517,7 @@ class TestStore(base.StoreBaseTest,
         with mock.patch.object(self.store, '_get_capacity_info') as capacity:
             capacity.return_value = 0
 
-            ChunkedFile.CHUNKSIZE = 1024
+            ChunkedFile.CHUNKSIZE = units.Ki
             expected_image_id = str(uuid.uuid4())
             expected_file_size = 5 * units.Ki  # 5K
             expected_file_contents = "*" * expected_file_size
@@ -573,7 +570,7 @@ class TestStore(base.StoreBaseTest,
 
         self.store.configure_add()
 
-        Store.WRITE_CHUNKSIZE = 1024
+        Store.WRITE_CHUNKSIZE = units.Ki
         expected_image_id = str(uuid.uuid4())
         expected_file_size = 5 * units.Ki  # 5K
         expected_file_contents = "*" * expected_file_size
@@ -614,7 +611,7 @@ class TestStore(base.StoreBaseTest,
 
         self.store.configure_add()
 
-        Store.WRITE_CHUNKSIZE = 1024
+        Store.WRITE_CHUNKSIZE = units.Ki
         expected_image_id = str(uuid.uuid4())
         expected_file_size = 5 * units.Ki  # 5K
         expected_file_contents = "*" * expected_file_size
diff --git a/tests/unit/test_rbd_store.py b/tests/unit/test_rbd_store.py
index 917b7ff..ede804a 100644
--- a/tests/unit/test_rbd_store.py
+++ b/tests/unit/test_rbd_store.py
@@ -16,6 +16,7 @@
 import StringIO
 
 import mock
+from oslo_utils import units
 
 from glance_store._drivers import rbd as rbd_store
 from glance_store import exceptions
@@ -163,12 +164,12 @@ class TestStore(base.StoreBaseTest,
         self.location = rbd_store.StoreLocation(self.store_specs,
                                                 self.conf)
         # Provide enough data to get more than one chunk iteration.
-        self.data_len = 3 * 1024
+        self.data_len = 3 * units.Ki
         self.data_iter = StringIO.StringIO('*' * self.data_len)
 
     def test_add_w_image_size_zero(self):
         """Assert that correct size is returned even though 0 was provided."""
-        self.store.chunk_size = 1024
+        self.store.chunk_size = units.Ki
         with mock.patch.object(rbd_store.rbd.Image, 'resize') as resize:
             with mock.patch.object(rbd_store.rbd.Image, 'write') as write:
                 ret = self.store.add('fake_image_id', self.data_iter, 0)
diff --git a/tests/unit/test_s3_store.py b/tests/unit/test_s3_store.py
index 39c0ba5..9345ac1 100644
--- a/tests/unit/test_s3_store.py
+++ b/tests/unit/test_s3_store.py
@@ -56,7 +56,7 @@ class FakeKey(object):
         self.data = None
         self.size = 0
         self.etag = None
-        self.BufferSize = 1024
+        self.BufferSize = units.Ki
 
     def close(self):
         pass
diff --git a/tests/unit/test_swift_store.py b/tests/unit/test_swift_store.py
index 8a541a4..4141d84 100644
--- a/tests/unit/test_swift_store.py
+++ b/tests/unit/test_swift_store.py
@@ -662,8 +662,8 @@ class SwiftTests(object):
         orig_max_size = self.store.large_object_size
         orig_temp_size = self.store.large_object_chunk_size
         try:
-            self.store.large_object_size = 1024
-            self.store.large_object_chunk_size = 1024
+            self.store.large_object_size = units.Ki
+            self.store.large_object_chunk_size = units.Ki
             loc, size, checksum, _ = self.store.add(expected_image_id,
                                                     image_swift,
                                                     expected_swift_size)
@@ -719,9 +719,9 @@ class SwiftTests(object):
         global MAX_SWIFT_OBJECT_SIZE
         orig_max_swift_object_size = MAX_SWIFT_OBJECT_SIZE
         try:
-            MAX_SWIFT_OBJECT_SIZE = 1024
-            self.store.large_object_size = 1024
-            self.store.large_object_chunk_size = 1024
+            MAX_SWIFT_OBJECT_SIZE = units.Ki
+            self.store.large_object_size = units.Ki
+            self.store.large_object_chunk_size = units.Ki
             loc, size, checksum, _ = self.store.add(expected_image_id,
                                                     image_swift, 0)
         finally:
@@ -1446,7 +1446,7 @@ class TestChunkReader(base.StoreBaseTest):
             bytes_read += len(chunk)
             if not chunk:
                 break
-        self.assertEqual(1024, bytes_read)
+        self.assertEqual(units.Ki, bytes_read)
         data_file.close()