| OLD | NEW |
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/shell_integration.h" | 5 #include "chrome/browser/shell_integration.h" |
| 6 | 6 |
| 7 #include <map> | 7 #include <map> |
| 8 | 8 |
| 9 #include "base/file_path.h" | 9 #include "base/file_path.h" |
| 10 #include "base/file_util.h" | 10 #include "base/file_util.h" |
| 11 #include "base/message_loop.h" | 11 #include "base/message_loop.h" |
| 12 #include "base/scoped_temp_dir.h" | 12 #include "base/scoped_temp_dir.h" |
| 13 #include "base/stl_util-inl.h" | 13 #include "base/stl_util-inl.h" |
| 14 #include "base/string_util.h" | 14 #include "base/string_util.h" |
| 15 #include "chrome/browser/chrome_thread.h" | 15 #include "chrome/browser/chrome_thread.h" |
| 16 #include "chrome/common/chrome_constants.h" | 16 #include "chrome/common/chrome_constants.h" |
| 17 #include "chrome/common/chrome_paths_internal.h" | 17 #include "chrome/common/chrome_paths_internal.h" |
| 18 #include "googleurl/src/gurl.h" | 18 #include "googleurl/src/gurl.h" |
| 19 #include "testing/gtest/include/gtest/gtest.h" | 19 #include "testing/gtest/include/gtest/gtest.h" |
| 20 | 20 |
| 21 #if defined(OS_WIN) | 21 #if defined(OS_WIN) |
| 22 #include "chrome/installer/util/browser_distribution.h" | 22 #include "chrome/installer/util/browser_distribution.h" |
| 23 #elif defined(OS_LINUX) | 23 #elif defined(OS_LINUX) |
| 24 #include "base/env_var.h" | 24 #include "base/environment.h" |
| 25 #endif // defined(OS_LINUX) | 25 #endif // defined(OS_LINUX) |
| 26 | 26 |
| 27 #define FPL FILE_PATH_LITERAL | 27 #define FPL FILE_PATH_LITERAL |
| 28 | 28 |
| 29 #if defined(OS_LINUX) | 29 #if defined(OS_LINUX) |
| 30 namespace { | 30 namespace { |
| 31 | 31 |
| 32 // Provides mock environment variables values based on a stored map. | 32 // Provides mock environment variables values based on a stored map. |
| 33 class MockEnvVarGetter : public base::EnvVarGetter { | 33 class MockEnvironment : public base::Environment { |
| 34 public: | 34 public: |
| 35 MockEnvVarGetter() {} | 35 MockEnvironment() {} |
| 36 | 36 |
| 37 void Set(const std::string& name, const std::string& value) { | 37 void Set(const std::string& name, const std::string& value) { |
| 38 variables_[name] = value; | 38 variables_[name] = value; |
| 39 } | 39 } |
| 40 | 40 |
| 41 virtual bool GetEnv(const char* variable_name, std::string* result) { | 41 virtual bool GetEnv(const char* variable_name, std::string* result) { |
| 42 if (ContainsKey(variables_, variable_name)) { | 42 if (ContainsKey(variables_, variable_name)) { |
| 43 *result = variables_[variable_name]; | 43 *result = variables_[variable_name]; |
| 44 return true; | 44 return true; |
| 45 } | 45 } |
| 46 | 46 |
| 47 return false; | 47 return false; |
| 48 } | 48 } |
| 49 | 49 |
| 50 virtual bool SetEnv(const char* variable_name, const std::string& new_value) { | 50 virtual bool SetEnv(const char* variable_name, const std::string& new_value) { |
| 51 ADD_FAILURE(); | 51 ADD_FAILURE(); |
| 52 return false; | 52 return false; |
| 53 } | 53 } |
| 54 | 54 |
| 55 virtual bool UnSetEnv(const char* variable_name) { | 55 virtual bool UnSetEnv(const char* variable_name) { |
| 56 ADD_FAILURE(); | 56 ADD_FAILURE(); |
| 57 return false; | 57 return false; |
| 58 } | 58 } |
| 59 | 59 |
| 60 private: | 60 private: |
| 61 std::map<std::string, std::string> variables_; | 61 std::map<std::string, std::string> variables_; |
| 62 | 62 |
| 63 DISALLOW_COPY_AND_ASSIGN(MockEnvVarGetter); | 63 DISALLOW_COPY_AND_ASSIGN(MockEnvironment); |
| 64 }; | 64 }; |
| 65 | 65 |
| 66 } // namespace | 66 } // namespace |
| 67 | 67 |
| 68 TEST(ShellIntegrationTest, GetDesktopShortcutTemplate) { | 68 TEST(ShellIntegrationTest, GetDesktopShortcutTemplate) { |
| 69 #if defined(GOOGLE_CHROME_BUILD) | 69 #if defined(GOOGLE_CHROME_BUILD) |
| 70 const char kTemplateFilename[] = "google-chrome.desktop"; | 70 const char kTemplateFilename[] = "google-chrome.desktop"; |
| 71 #else // CHROMIUM_BUILD | 71 #else // CHROMIUM_BUILD |
| 72 const char kTemplateFilename[] = "chromium-browser.desktop"; | 72 const char kTemplateFilename[] = "chromium-browser.desktop"; |
| 73 #endif | 73 #endif |
| 74 | 74 |
| 75 const char kTestData1[] = "a magical testing string"; | 75 const char kTestData1[] = "a magical testing string"; |
| 76 const char kTestData2[] = "a different testing string"; | 76 const char kTestData2[] = "a different testing string"; |
| 77 | 77 |
| 78 MessageLoop message_loop; | 78 MessageLoop message_loop; |
| 79 ChromeThread file_thread(ChromeThread::FILE, &message_loop); | 79 ChromeThread file_thread(ChromeThread::FILE, &message_loop); |
| 80 | 80 |
| 81 { | 81 { |
| 82 ScopedTempDir temp_dir; | 82 ScopedTempDir temp_dir; |
| 83 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); | 83 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); |
| 84 | 84 |
| 85 MockEnvVarGetter env_getter; | 85 MockEnvironment env; |
| 86 env_getter.Set("XDG_DATA_HOME", temp_dir.path().value()); | 86 env.Set("XDG_DATA_HOME", temp_dir.path().value()); |
| 87 ASSERT_TRUE(file_util::WriteFile( | 87 ASSERT_TRUE(file_util::WriteFile( |
| 88 temp_dir.path().AppendASCII(kTemplateFilename), | 88 temp_dir.path().AppendASCII(kTemplateFilename), |
| 89 kTestData1, strlen(kTestData1))); | 89 kTestData1, strlen(kTestData1))); |
| 90 std::string contents; | 90 std::string contents; |
| 91 ASSERT_TRUE(ShellIntegration::GetDesktopShortcutTemplate(&env_getter, | 91 ASSERT_TRUE(ShellIntegration::GetDesktopShortcutTemplate(&env, |
| 92 &contents)); | 92 &contents)); |
| 93 EXPECT_EQ(kTestData1, contents); | 93 EXPECT_EQ(kTestData1, contents); |
| 94 } | 94 } |
| 95 | 95 |
| 96 { | 96 { |
| 97 ScopedTempDir temp_dir; | 97 ScopedTempDir temp_dir; |
| 98 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); | 98 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); |
| 99 | 99 |
| 100 MockEnvVarGetter env_getter; | 100 MockEnvironment env; |
| 101 env_getter.Set("XDG_DATA_DIRS", temp_dir.path().value()); | 101 env.Set("XDG_DATA_DIRS", temp_dir.path().value()); |
| 102 ASSERT_TRUE(file_util::CreateDirectory( | 102 ASSERT_TRUE(file_util::CreateDirectory( |
| 103 temp_dir.path().AppendASCII("applications"))); | 103 temp_dir.path().AppendASCII("applications"))); |
| 104 ASSERT_TRUE(file_util::WriteFile( | 104 ASSERT_TRUE(file_util::WriteFile( |
| 105 temp_dir.path().AppendASCII("applications") | 105 temp_dir.path().AppendASCII("applications") |
| 106 .AppendASCII(kTemplateFilename), | 106 .AppendASCII(kTemplateFilename), |
| 107 kTestData2, strlen(kTestData2))); | 107 kTestData2, strlen(kTestData2))); |
| 108 std::string contents; | 108 std::string contents; |
| 109 ASSERT_TRUE(ShellIntegration::GetDesktopShortcutTemplate(&env_getter, | 109 ASSERT_TRUE(ShellIntegration::GetDesktopShortcutTemplate(&env, |
| 110 &contents)); | 110 &contents)); |
| 111 EXPECT_EQ(kTestData2, contents); | 111 EXPECT_EQ(kTestData2, contents); |
| 112 } | 112 } |
| 113 | 113 |
| 114 { | 114 { |
| 115 ScopedTempDir temp_dir; | 115 ScopedTempDir temp_dir; |
| 116 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); | 116 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); |
| 117 | 117 |
| 118 MockEnvVarGetter env_getter; | 118 MockEnvironment env; |
| 119 env_getter.Set("XDG_DATA_DIRS", temp_dir.path().value() + ":" + | 119 env.Set("XDG_DATA_DIRS", temp_dir.path().value() + ":" + |
| 120 temp_dir.path().AppendASCII("applications").value()); | 120 temp_dir.path().AppendASCII("applications").value()); |
| 121 ASSERT_TRUE(file_util::CreateDirectory( | 121 ASSERT_TRUE(file_util::CreateDirectory( |
| 122 temp_dir.path().AppendASCII("applications"))); | 122 temp_dir.path().AppendASCII("applications"))); |
| 123 ASSERT_TRUE(file_util::WriteFile( | 123 ASSERT_TRUE(file_util::WriteFile( |
| 124 temp_dir.path().AppendASCII(kTemplateFilename), | 124 temp_dir.path().AppendASCII(kTemplateFilename), |
| 125 kTestData1, strlen(kTestData1))); | 125 kTestData1, strlen(kTestData1))); |
| 126 ASSERT_TRUE(file_util::WriteFile( | 126 ASSERT_TRUE(file_util::WriteFile( |
| 127 temp_dir.path().AppendASCII("applications") | 127 temp_dir.path().AppendASCII("applications") |
| 128 .AppendASCII(kTemplateFilename), | 128 .AppendASCII(kTemplateFilename), |
| 129 kTestData2, strlen(kTestData2))); | 129 kTestData2, strlen(kTestData2))); |
| 130 std::string contents; | 130 std::string contents; |
| 131 ASSERT_TRUE(ShellIntegration::GetDesktopShortcutTemplate(&env_getter, | 131 ASSERT_TRUE(ShellIntegration::GetDesktopShortcutTemplate(&env, |
| 132 &contents)); | 132 &contents)); |
| 133 EXPECT_EQ(kTestData1, contents); | 133 EXPECT_EQ(kTestData1, contents); |
| 134 } | 134 } |
| 135 } | 135 } |
| 136 | 136 |
| 137 TEST(ShellIntegrationTest, GetDesktopShortcutFilename) { | 137 TEST(ShellIntegrationTest, GetDesktopShortcutFilename) { |
| 138 const struct { | 138 const struct { |
| 139 const FilePath::CharType* path; | 139 const FilePath::CharType* path; |
| 140 const char* url; | 140 const char* url; |
| 141 } test_cases[] = { | 141 } test_cases[] = { |
| (...skipping 167 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 309 // Non-default profile path should get chrome::kBrowserAppID joined with | 309 // Non-default profile path should get chrome::kBrowserAppID joined with |
| 310 // profile info. | 310 // profile info. |
| 311 FilePath profile_path(FILE_PATH_LITERAL("root")); | 311 FilePath profile_path(FILE_PATH_LITERAL("root")); |
| 312 profile_path = profile_path.Append(FILE_PATH_LITERAL("udd")); | 312 profile_path = profile_path.Append(FILE_PATH_LITERAL("udd")); |
| 313 profile_path = profile_path.Append(FILE_PATH_LITERAL("User Data - Test")); | 313 profile_path = profile_path.Append(FILE_PATH_LITERAL("User Data - Test")); |
| 314 EXPECT_EQ(BrowserDistribution::GetDistribution()->GetBrowserAppId() + | 314 EXPECT_EQ(BrowserDistribution::GetDistribution()->GetBrowserAppId() + |
| 315 L".udd.UserDataTest", | 315 L".udd.UserDataTest", |
| 316 ShellIntegration::GetChromiumAppId(profile_path)); | 316 ShellIntegration::GetChromiumAppId(profile_path)); |
| 317 } | 317 } |
| 318 #endif | 318 #endif |
| OLD | NEW |