| // 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 |