AU: Update Downloader to support our image formats.

src/platform/update_engine/utils_unittest.cc

 // Copyright (c) 2009 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include <sys/stat.h>
 #include <sys/types.h>
 #include <errno.h>
 #include <string>
 #include <vector>
 #include <gtest/gtest.h>
@@ skipping 70 matching lines @@
   EXPECT_FALSE(utils::StringHasPrefix("", "x"));
   EXPECT_FALSE(utils::StringHasPrefix("abcdefgh", "bc"));
   EXPECT_FALSE(utils::StringHasPrefix("abcdefgh", "gh"));
 }
 
 TEST(UtilsTest, BootDeviceTest) {
   // Pretty lame test...
   EXPECT_FALSE(utils::BootDevice().empty());
 }
 
+TEST(UtilsTest, BootKernelDeviceTest) {
+  EXPECT_EQ("", utils::BootKernelDevice("foo"));
+  EXPECT_EQ("", utils::BootKernelDevice("/dev/sda0"));
+  EXPECT_EQ("", utils::BootKernelDevice("/dev/sda1"));
+  EXPECT_EQ("", utils::BootKernelDevice("/dev/sda2"));
+  EXPECT_EQ("/dev/sda2", utils::BootKernelDevice("/dev/sda3"));
+  EXPECT_EQ("", utils::BootKernelDevice("/dev/sda4"));
+  EXPECT_EQ("/dev/sda4", utils::BootKernelDevice("/dev/sda5"));
+  EXPECT_EQ("", utils::BootKernelDevice("/dev/sda6"));
+  EXPECT_EQ("/dev/sda6", utils::BootKernelDevice("/dev/sda7"));
+  EXPECT_EQ("", utils::BootKernelDevice("/dev/sda8"));
+  EXPECT_EQ("", utils::BootKernelDevice("/dev/sda9"));
+}
+
 TEST(UtilsTest, RecursiveUnlinkDirTest) {
   EXPECT_EQ(0, mkdir("RecursiveUnlinkDirTest-a", 0755));
   EXPECT_EQ(0, mkdir("RecursiveUnlinkDirTest-b", 0755));
   EXPECT_EQ(0, symlink("../RecursiveUnlinkDirTest-a",
                        "RecursiveUnlinkDirTest-b/link"));
   EXPECT_EQ(0, system("echo hi > RecursiveUnlinkDirTest-b/file"));
   EXPECT_EQ(0, mkdir("RecursiveUnlinkDirTest-b/dir", 0755));
   EXPECT_EQ(0, system("echo ok > RecursiveUnlinkDirTest-b/dir/subfile"));
   EXPECT_TRUE(utils::RecursiveUnlinkDir("RecursiveUnlinkDirTest-b"));
   EXPECT_TRUE(utils::FileExists("RecursiveUnlinkDirTest-a"));
   EXPECT_EQ(0, system("rm -rf RecursiveUnlinkDirTest-a"));
   EXPECT_FALSE(utils::FileExists("RecursiveUnlinkDirTest-b"));
   EXPECT_TRUE(utils::RecursiveUnlinkDir("/something/that/doesnt/exist"));
 }
 
 TEST(UtilsTest, TempFilenameTest) {
   const string original = "/foo.XXXXXX";
   const string result = utils::TempFilename(original);
   EXPECT_EQ(original.size(), result.size());
   EXPECT_TRUE(utils::StringHasPrefix(result, "/foo."));
   EXPECT_FALSE(utils::StringHasSuffix(result, "XXXXXX"));
 }
 
 }  // namespace chromeos_update_engine