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