1 files changed, 58 insertions, 0 deletions
diff --git a/policy/tests/resilient_policy_util_test.cc b/policy/tests/resilient_policy_util_test.cc
new file mode 100644
index 0000000..0963b08
--- /dev/null
+++ b/policy/tests/resilient_policy_util_test.cc
@@ -0,0 +1,58 @@
+// Copyright 2017 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 <gtest/gtest.h>
+
+#include <map>
+#include <string>
+
+#include <base/files/file_path.h>
+#include <base/files/file_util.h>
+#include <base/files/scoped_temp_dir.h>
+
+#include "policy/resilient_policy_util.h"
+
+namespace {
+
+const char kDefaultResilientPolicyFilePath[] = "policy";
+
+void CreateFile(const base::FilePath& file_path) {
+  base::File file(file_path, base::File::FLAG_CREATE | base::File::FLAG_WRITE);
+}
+
+}  // namespace
+
+namespace policy {
+
+// Test that the policy files from the folder are identified correctly.
+TEST(DevicePolicyUtilTest, GetSortedResilientPolicyFilePaths) {
+  // Create the temporary directory.
+  base::ScopedTempDir temp_dir;
+  ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
+
+  base::FilePath file0(temp_dir.GetPath().Append("policy"));
+  base::FilePath file1(temp_dir.GetPath().Append("policy.12"));
+  base::FilePath file2(temp_dir.GetPath().Append("policy.2"));
+  base::FilePath file3(temp_dir.GetPath().Append("policy.30"));
+  base::FilePath invalid(temp_dir.GetPath().Append("policy_4"));
+
+  CreateFile(file0);
+  CreateFile(file1);
+  CreateFile(file2);
+  CreateFile(file3);
+
+  const base::FilePath test_file_path(
+      temp_dir.GetPath().Append(kDefaultResilientPolicyFilePath));
+  std::map<int, base::FilePath> sorted_file_paths =
+      GetSortedResilientPolicyFilePaths(test_file_path);
+
+  EXPECT_EQ(4, sorted_file_paths.size());
+  EXPECT_EQ(file0.value(), sorted_file_paths[0].value());
+  EXPECT_EQ(file1.value(), sorted_file_paths[12].value());
+  EXPECT_EQ(file2.value(), sorted_file_paths[2].value());
+  EXPECT_EQ(file3.value(), sorted_file_paths[30].value());
+  EXPECT_EQ("", sorted_file_paths[4].value());
+}
+
+}  // namespace policy