def get_pod_pool_size_map(self, key):
"""Return the mapping for a size calculation based on key."""
pools = {}
for pool in self.list_pools():
output = self.run(['pool-info', pool]).strip()
if output is None:
# Skip if cannot get more information.
continue
pools[pool] = convert_size_to_bytes(
self.get_key_value(output, "Capacity"))
return pools
def get_pod_available_local_storage(self):
"""Gets the available local storage for the pod."""
pools = self.list_pools()
local_storage = 0
for pool in pools:
output = self.run(['pool-info', pool]).strip()
if output is None:
maaslog.error("Failed to get available pod local storage")
return None
local_storage += convert_size_to_bytes(
self.get_key_value(output, "Available"))
# Local storage in bytes.
return local_storage
def test_convert_size_to_bytes(self):
self.assertEqual(self.expected, convert_size_to_bytes(self.value))