OLD | NEW |
| (Empty) |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "chrome/browser/safe_browsing/path_sanitizer.h" | |
6 | |
7 #include <vector> | |
8 | |
9 #include "base/logging.h" | |
10 #include "base/path_service.h" | |
11 #include "testing/gtest/include/gtest/gtest.h" | |
12 | |
13 namespace { | |
14 | |
15 // Returns the root directory with a trailing separator. Works on all platforms. | |
16 base::FilePath GetRootDirectory() { | |
17 base::FilePath dir_temp; | |
18 if (!PathService::Get(base::DIR_TEMP, &dir_temp)) | |
19 NOTREACHED(); | |
20 | |
21 std::vector<base::FilePath::StringType> components; | |
22 dir_temp.GetComponents(&components); | |
23 | |
24 return base::FilePath(components[0]).AsEndingWithSeparator(); | |
25 } | |
26 | |
27 } // namespace | |
28 | |
29 TEST(SafeBrowsingPathSanitizerTest, HomeDirectoryIsNotEmpty) { | |
30 safe_browsing::PathSanitizer path_sanitizer; | |
31 | |
32 ASSERT_FALSE(path_sanitizer.GetHomeDirectory().empty()); | |
33 } | |
34 | |
35 TEST(SafeBrowsingPathSanitizerTest, DontStripHomeDirectoryTest) { | |
36 // Test with path not in home directory. | |
37 base::FilePath path = | |
38 GetRootDirectory().Append(FILE_PATH_LITERAL("not_in_home_directory.ext")); | |
39 base::FilePath path_expected = path; | |
40 | |
41 safe_browsing::PathSanitizer path_sanitizer; | |
42 path_sanitizer.StripHomeDirectory(&path); | |
43 | |
44 ASSERT_EQ(path.value(), path_expected.value()); | |
45 } | |
46 | |
47 TEST(SafeBrowsingPathSanitizerTest, DoStripHomeDirectoryTest) { | |
48 // Test with path in home directory. | |
49 safe_browsing::PathSanitizer path_sanitizer; | |
50 | |
51 base::FilePath path = path_sanitizer.GetHomeDirectory().Append( | |
52 FILE_PATH_LITERAL("in_home_directory.ext")); | |
53 base::FilePath path_expected = base::FilePath(FILE_PATH_LITERAL("~")).Append( | |
54 FILE_PATH_LITERAL("in_home_directory.ext")); | |
55 | |
56 path_sanitizer.StripHomeDirectory(&path); | |
57 | |
58 ASSERT_EQ(path.value(), path_expected.value()); | |
59 } | |
OLD | NEW |