AU: ApplyMap utility function

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>
 #include <vector>
 
 #include "gtest/gtest.h"
 #include "update_engine/utils.h"
 
+using std::map;
 using std::string;
 using std::vector;
 
 namespace chromeos_update_engine {
 
 class UtilsTest : public ::testing::Test { };
 
 TEST(UtilsTest, IsOfficialBuild) {
   // Pretty lame test...
   EXPECT_TRUE(utils::IsOfficialBuild());
@@ skipping 151 matching lines @@
     unsigned int range = kRanges[r];
     const int kValue = 50;
     for (int tries = 0; tries < 100; ++tries) {
       int value = utils::FuzzInt(kValue, range);
       EXPECT_GE(value, kValue - range / 2);
       EXPECT_LE(value, kValue + range - range / 2);
     }
   }
 }
 
+TEST(UtilsTest, ApplyMapTest) {
+  int initial_values[] = {1, 2, 3, 4, 6};
+  vector<int> collection(&initial_values[0],
+                         initial_values + arraysize(initial_values));
+  EXPECT_EQ(arraysize(initial_values), collection.size());
+  int expected_values[] = {1, 2, 5, 4, 8};
+  map<int, int> value_map;
+  value_map[3] = 5;
+  value_map[6] = 8;
+  value_map[5] = 10;
+
+  utils::ApplyMap(&collection, value_map);
+
+  size_t index = 0;
+  for (vector<int>::iterator it = collection.begin(), e = collection.end();
+       it != e; ++it) {
+    EXPECT_EQ(expected_values[index++], *it);
+  }
+}
+
 }  // namespace chromeos_update_engine