platform_PartitionCheck: Allow 4GB root partition size

The current partition check only expects to see 2GB root partition
size, but there are now systems with 4GB root partition so check
for this before failing.

BUG=b:37746057
TEST=pass platform_PartitionCheck on Eve

Change-Id: Ia2d539319def58fd406c4d9397df0909cfc069d7
Signed-off-by: Duncan Laurie <dlaurie@google.com>
Reviewed-on: https://chromium-review.googlesource.com/515383
Reviewed-by: Todd Broch <tbroch@chromium.org>
diff --git a/client/site_tests/platform_PartitionCheck/platform_PartitionCheck.py b/client/site_tests/platform_PartitionCheck/platform_PartitionCheck.py
index 469f5a5..1a5cded 100644
--- a/client/site_tests/platform_PartitionCheck/platform_PartitionCheck.py
+++ b/client/site_tests/platform_PartitionCheck/platform_PartitionCheck.py
@@ -8,7 +8,8 @@
 from autotest_lib.client.bin import test, utils
 from autotest_lib.client.common_lib import error
 
-ROOTFS_SIZE = 2 * 1024 * 1024 * 1024
+ROOTFS_SIZE_2G = 2 * 1024 * 1024 * 1024
+ROOTFS_SIZE_4G = 4 * 1024 * 1024 * 1024
 
 class platform_PartitionCheck(test.test):
     """
@@ -64,7 +65,7 @@
         for p in partitions:
             pblocks = self.get_partition_size(device, p)
             psize = pblocks * block_size
-            if psize != ROOTFS_SIZE:
+            if psize != ROOTFS_SIZE_2G and psize != ROOTFS_SIZE_4G:
                 errmsg = ('%s is %d bytes, expected %d' %
                           (p, psize, ROOTFS_SIZE))
                 logging.warning(errmsg)