AU: Add unit tests for utils IsSymlink and SetProcessPriority.

utils_unittest.cc

 // Copyright (c) 2010 The Chromium OS 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 <map>
 #include <string>
@@ skipping 111 matching lines @@
   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, IsSymlinkTest) {
+  string temp_dir;
+  EXPECT_TRUE(utils::MakeTempDirectory("/tmp/symlink-test.XXXXXX", &temp_dir));
+  string temp_file = temp_dir + "temp-file";
+  EXPECT_TRUE(utils::WriteFile(temp_file.c_str(), "", 0));
+  string temp_symlink = temp_dir + "temp-symlink";
+  EXPECT_EQ(0, symlink(temp_file.c_str(), temp_symlink.c_str()));
+  EXPECT_FALSE(utils::IsSymlink(temp_dir.c_str()));
+  EXPECT_FALSE(utils::IsSymlink(temp_file.c_str()));
+  EXPECT_TRUE(utils::IsSymlink(temp_symlink.c_str()));
+  EXPECT_FALSE(utils::IsSymlink("/non/existent/path"));
+  EXPECT_TRUE(utils::RecursiveUnlinkDir(temp_dir));
+}
+
 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"));
 }
 
 TEST(UtilsTest, RootDeviceTest) {
   EXPECT_EQ("/dev/sda", utils::RootDevice("/dev/sda3"));
@@ skipping 15 matching lines @@
 TEST(UtilsTest, IsRemovableDeviceTest) {
   EXPECT_FALSE(utils::IsRemovableDevice(""));
   EXPECT_FALSE(utils::IsRemovableDevice("/dev/non-existent-device"));
 }
 
 TEST(UtilsTest, PartitionNumberTest) {
   EXPECT_EQ("3", utils::PartitionNumber("/dev/sda3"));
   EXPECT_EQ("3", utils::PartitionNumber("/dev/mmc0p3"));
 }
 
+
+TEST(UtilsTest, RunAsRootSetProcessPriorityTest) {
+  // getpriority may return -1 on error so the getpriority logic needs to be
+  // enhanced if any of the pre-defined priority constants are changed to -1.
+  ASSERT_NE(-1, utils::kProcessPriorityLow);
+  ASSERT_NE(-1, utils::kProcessPriorityNormal);
+  ASSERT_NE(-1, utils::kProcessPriorityHigh);
+  EXPECT_EQ(utils::kProcessPriorityNormal, getpriority(PRIO_PROCESS, 0));
+  EXPECT_TRUE(utils::SetProcessPriority(utils::kProcessPriorityHigh));
+  EXPECT_EQ(utils::kProcessPriorityHigh, getpriority(PRIO_PROCESS, 0));
+  EXPECT_TRUE(utils::SetProcessPriority(utils::kProcessPriorityLow));
+  EXPECT_EQ(utils::kProcessPriorityLow, getpriority(PRIO_PROCESS, 0));
+  EXPECT_TRUE(utils::SetProcessPriority(utils::kProcessPriorityNormal));
+  EXPECT_EQ(utils::kProcessPriorityNormal, getpriority(PRIO_PROCESS, 0));
+}
+
 TEST(UtilsTest, ComparePrioritiesTest) {
   EXPECT_LT(utils::ComparePriorities(utils::kProcessPriorityLow,
                                      utils::kProcessPriorityNormal), 0);
   EXPECT_GT(utils::ComparePriorities(utils::kProcessPriorityNormal,
                                      utils::kProcessPriorityLow), 0);
   EXPECT_EQ(utils::ComparePriorities(utils::kProcessPriorityNormal,
                                      utils::kProcessPriorityNormal), 0);
   EXPECT_GT(utils::ComparePriorities(utils::kProcessPriorityHigh,
                                      utils::kProcessPriorityNormal), 0);
 }
@@ skipping 42 matching lines @@
       img.c_str())));
   EXPECT_EQ(20 * 1024 * 1024, utils::FileSize(img));
   int block_count = 0;
   int block_size = 0;
   EXPECT_TRUE(utils::GetFilesystemSize(img, &block_count, &block_size));
   EXPECT_EQ(4096, block_size);
   EXPECT_EQ(10 * 1024 * 1024 / 4096, block_count);
 }
 
 }  // namespace chromeos_update_engine