diff --git a/glance/common/utils.py b/glance/common/utils.py index feb2c81565..860b9fb600 100644 --- a/glance/common/utils.py +++ b/glance/common/utils.py @@ -724,13 +724,12 @@ def get_store_weight(location): if not store_id: return 0 try: - store = glance_store.get_store_from_store_identifier(store_id) + return glance_store.get_store_weight(store_id) except glance_store.exceptions.UnknownScheme: msg = (_LW("Unable to find store '%s', returning " "default weight '0'") % store_id) LOG.warning(msg) return 0 - return store.weight if store is not None else 0 sorted_locations = sorted(locations, key=get_store_weight, reverse=True) LOG.debug(('Sorted locations: %s'), sorted_locations) diff --git a/glance/tests/unit/common/test_utils.py b/glance/tests/unit/common/test_utils.py index 817ea05bb4..f77fa502c8 100644 --- a/glance/tests/unit/common/test_utils.py +++ b/glance/tests/unit/common/test_utils.py @@ -189,7 +189,7 @@ def test_sort_image_locations_multistore_disabled(self): 'url': 'rbd://cccccccc/images/id', 'metadata': {'store': 'rbd3'} }] - mp = "glance.common.utils.glance_store.get_store_from_store_identifier" + mp = "glance.common.utils.glance_store.get_store_weight" with mock.patch(mp) as mock_get_store: utils.sort_image_locations(locations) @@ -215,10 +215,9 @@ def test_sort_image_locations(self): 'url': 'rbd://cccccccc/images/id', 'metadata': {'store': 'rbd3'} }] - mp = "glance.common.utils.glance_store.get_store_from_store_identifier" + mp = "glance.common.utils.glance_store.get_store_weight" with mock.patch(mp) as mock_get_store: - mock_store = mock_get_store.return_value - mock_store.weight = 100 + mock_get_store.return_value = 100 utils.sort_image_locations(locations) # Since 3 stores are configured, internal method will be called 3 times @@ -243,7 +242,7 @@ def test_sort_image_locations_without_metadata(self): 'url': 'rbd://cccccccc/images/id', 'metadata': {} }] - mp = "glance.common.utils.glance_store.get_store_from_store_identifier" + mp = "glance.common.utils.glance_store.get_store_weight" with mock.patch(mp) as mock_get_store: utils.sort_image_locations(locations) @@ -270,10 +269,9 @@ def test_sort_image_locations_with_partial_metadata(self): 'url': 'rbd://cccccccc/images/id', 'metadata': {} }] - mp = "glance.common.utils.glance_store.get_store_from_store_identifier" + mp = "glance.common.utils.glance_store.get_store_weight" with mock.patch(mp) as mock_get_store: - mock_store = mock_get_store.return_value - mock_store.weight = 100 + mock_get_store.return_value = 100 utils.sort_image_locations(locations) # Since 3 stores are configured, but only one location has @@ -300,7 +298,7 @@ def test_sort_image_locations_unknownscheme(self): 'url': 'rbd://cccccccc/images/id', 'metadata': {'store': 'rbd3'} }] - mp = "glance.common.utils.glance_store.get_store_from_store_identifier" + mp = "glance.common.utils.glance_store.get_store_weight" with mock.patch(mp) as mock_get_store: mock_get_store.side_effect = store.UnknownScheme() sorted_locations = utils.sort_image_locations(locations)