cros_build_lib: Return a list from GetImageDiskPartitionInfo
BUG=None
TEST=Updated unit tests all pass
Change-Id: I25e3f78ebba00f692eb8981f4f7d9fff832feb8e
Reviewed-on: https://chromium-review.googlesource.com/1359008
Commit-Ready: LaMont Jones <lamontjones@chromium.org>
Tested-by: LaMont Jones <lamontjones@chromium.org>
Reviewed-by: Mike Frysinger <vapier@chromium.org>
diff --git a/scripts/test_image_unittest.py b/scripts/test_image_unittest.py
index 1859dfc..c1d7f20 100644
--- a/scripts/test_image_unittest.py
+++ b/scripts/test_image_unittest.py
@@ -30,11 +30,11 @@
self.image_file = os.path.join(self.tempdir,
constants.BASE_IMAGE_NAME + '.bin')
osutils.WriteFile(self.image_file, '')
- fake_partitions = {
- 1: cros_build_lib.PartitionInfo(1, 0, 0, 0, 'fs', 'STATE', 'flag'),
- 2: cros_build_lib.PartitionInfo(2, 0, 0, 0, 'fs', 'KERN-A', 'flag'),
- 3: cros_build_lib.PartitionInfo(3, 0, 0, 0, 'fs', 'ROOT-A', 'flag'),
- }
+ fake_partitions = (
+ cros_build_lib.PartitionInfo(1, 0, 0, 0, 'fs', 'STATE', 'flag'),
+ cros_build_lib.PartitionInfo(2, 0, 0, 0, 'fs', 'KERN-A', 'flag'),
+ cros_build_lib.PartitionInfo(3, 0, 0, 0, 'fs', 'ROOT-A', 'flag'),
+ )
self.PatchObject(cros_build_lib, 'GetImageDiskPartitionInfo',
autospec=True, return_value=fake_partitions)
self.PatchObject(osutils.MountImageContext, '_Mount', autospec=True)