| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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/installer/mini_installer/configuration.h" | 5 #include "chrome/installer/mini_installer/configuration.h" |
| 6 | 6 |
| 7 #include <stdlib.h> | 7 #include <stdlib.h> |
| 8 | 8 |
| 9 #include "base/basictypes.h" | 9 #include "base/basictypes.h" |
| 10 #include "chrome/installer/mini_installer/appid.h" | 10 #include "chrome/installer/mini_installer/appid.h" |
| 11 #include "testing/gtest/include/gtest/gtest.h" | 11 #include "testing/gtest/include/gtest/gtest.h" |
| 12 | 12 |
| 13 using mini_installer::Configuration; | 13 using mini_installer::Configuration; |
| 14 | 14 |
| 15 class TestConfiguration : public Configuration { | 15 class TestConfiguration : public Configuration { |
| 16 public: | 16 public: |
| 17 explicit TestConfiguration(const wchar_t* command_line) : Configuration() { | 17 explicit TestConfiguration(const wchar_t* command_line) : Configuration() { |
| 18 Initialize(command_line); | 18 Initialize(command_line); |
| 19 } | 19 } |
| 20 explicit TestConfiguration(const wchar_t* command_line, | |
| 21 LONG ret, const wchar_t* value) | |
| 22 : Configuration(), | |
| 23 open_registry_key_result_(true), | |
| 24 read_registry_value_result_(ret), | |
| 25 read_registry_value_(value) { | |
| 26 Initialize(command_line); | |
| 27 } | |
| 28 void SetRegistryResults(bool openkey, LONG ret, const wchar_t* value) { | |
| 29 } | |
| 30 private: | 20 private: |
| 31 bool open_registry_key_result_; | |
| 32 LONG read_registry_value_result_; | |
| 33 const wchar_t* read_registry_value_ = L""; | |
| 34 | |
| 35 void Initialize(const wchar_t* command_line) { | 21 void Initialize(const wchar_t* command_line) { |
| 36 Clear(); | 22 Clear(); |
| 37 ASSERT_TRUE(ParseCommandLine(command_line)); | 23 ASSERT_TRUE(ParseCommandLine(command_line)); |
| 38 } | 24 } |
| 39 bool ReadClientStateRegistryValue( | |
| 40 const HKEY root_key, const wchar_t* app_guid, | |
| 41 LONG* retval, ValueString& value) override { | |
| 42 *retval = read_registry_value_result_; | |
| 43 value.assign(read_registry_value_); | |
| 44 return open_registry_key_result_; | |
| 45 } | |
| 46 }; | 25 }; |
| 47 | 26 |
| 48 // Test that the operation type is CLEANUP iff --cleanup is on the cmdline. | 27 // Test that the operation type is CLEANUP iff --cleanup is on the cmdline. |
| 49 TEST(MiniInstallerConfigurationTest, Operation) { | 28 TEST(MiniInstallerConfigurationTest, Operation) { |
| 50 EXPECT_EQ(Configuration::INSTALL_PRODUCT, | 29 EXPECT_EQ(Configuration::INSTALL_PRODUCT, |
| 51 TestConfiguration(L"spam.exe").operation()); | 30 TestConfiguration(L"spam.exe").operation()); |
| 52 EXPECT_EQ(Configuration::INSTALL_PRODUCT, | 31 EXPECT_EQ(Configuration::INSTALL_PRODUCT, |
| 53 TestConfiguration(L"spam.exe --clean").operation()); | 32 TestConfiguration(L"spam.exe --clean").operation()); |
| 54 EXPECT_EQ(Configuration::INSTALL_PRODUCT, | 33 EXPECT_EQ(Configuration::INSTALL_PRODUCT, |
| 55 TestConfiguration(L"spam.exe --cleanupthis").operation()); | 34 TestConfiguration(L"spam.exe --cleanupthis").operation()); |
| (...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 87 TestConfiguration(kCommandLines[i]).command_line()); | 66 TestConfiguration(kCommandLines[i]).command_line()); |
| 88 } | 67 } |
| 89 } | 68 } |
| 90 | 69 |
| 91 TEST(MiniInstallerConfigurationTest, ChromeAppGuid) { | 70 TEST(MiniInstallerConfigurationTest, ChromeAppGuid) { |
| 92 EXPECT_TRUE(std::wstring(google_update::kAppGuid) == | 71 EXPECT_TRUE(std::wstring(google_update::kAppGuid) == |
| 93 TestConfiguration(L"spam.exe").chrome_app_guid()); | 72 TestConfiguration(L"spam.exe").chrome_app_guid()); |
| 94 EXPECT_TRUE(std::wstring(google_update::kAppGuid) == | 73 EXPECT_TRUE(std::wstring(google_update::kAppGuid) == |
| 95 TestConfiguration(L"spam.exe --chrome").chrome_app_guid()); | 74 TestConfiguration(L"spam.exe --chrome").chrome_app_guid()); |
| 96 EXPECT_TRUE(std::wstring(google_update::kAppGuid) == | 75 EXPECT_TRUE(std::wstring(google_update::kAppGuid) == |
| 76 TestConfiguration(L"spam.exe --multi-install --chrome") |
| 77 .chrome_app_guid()); |
| 78 EXPECT_TRUE(std::wstring(google_update::kAppGuid) == |
| 97 TestConfiguration(L"spam.exe --chrome-frame").chrome_app_guid()); | 79 TestConfiguration(L"spam.exe --chrome-frame").chrome_app_guid()); |
| 98 EXPECT_TRUE(std::wstring(google_update::kSxSAppGuid) == | 80 EXPECT_TRUE(std::wstring(google_update::kSxSAppGuid) == |
| 99 TestConfiguration(L"spam.exe --chrome-sxs").chrome_app_guid()); | 81 TestConfiguration(L"spam.exe --chrome-sxs").chrome_app_guid()); |
| 100 EXPECT_TRUE(std::wstring(google_update::kMultiInstallAppGuid) == | |
| 101 TestConfiguration(L"spam.exe --multi-install --chrome") | |
| 102 .chrome_app_guid()); | |
| 103 EXPECT_TRUE(std::wstring(google_update::kMultiInstallAppGuid) == | |
| 104 TestConfiguration(L"spam.exe --multi-install --chrome", | |
| 105 ERROR_INVALID_FUNCTION, L"") | |
| 106 .chrome_app_guid()); | |
| 107 EXPECT_TRUE(std::wstring(google_update::kAppGuid) == | |
| 108 TestConfiguration(L"spam.exe --multi-install --chrome", | |
| 109 ERROR_FILE_NOT_FOUND, L"") | |
| 110 .chrome_app_guid()); | |
| 111 EXPECT_TRUE(std::wstring(google_update::kAppGuid) == | |
| 112 TestConfiguration(L"spam.exe --multi-install --chrome", | |
| 113 ERROR_SUCCESS, L"foo-bar") | |
| 114 .chrome_app_guid()); | |
| 115 EXPECT_TRUE(std::wstring(google_update::kMultiInstallAppGuid) == | |
| 116 TestConfiguration(L"spam.exe --multi-install --chrome", | |
| 117 ERROR_SUCCESS, L"foo-multi") | |
| 118 .chrome_app_guid()); | |
| 119 } | 82 } |
| 120 | 83 |
| 121 TEST(MiniInstallerConfigurationTest, HasChrome) { | 84 TEST(MiniInstallerConfigurationTest, HasChrome) { |
| 122 EXPECT_TRUE(TestConfiguration(L"spam.exe").has_chrome()); | 85 EXPECT_TRUE(TestConfiguration(L"spam.exe").has_chrome()); |
| 123 EXPECT_TRUE(TestConfiguration(L"spam.exe --chrome").has_chrome()); | 86 EXPECT_TRUE(TestConfiguration(L"spam.exe --chrome").has_chrome()); |
| 124 EXPECT_TRUE(TestConfiguration(L"spam.exe --multi-install --chrome") | 87 EXPECT_TRUE(TestConfiguration(L"spam.exe --multi-install --chrome") |
| 125 .has_chrome()); | 88 .has_chrome()); |
| 126 EXPECT_FALSE(TestConfiguration(L"spam.exe --chrome-frame").has_chrome()); | 89 EXPECT_FALSE(TestConfiguration(L"spam.exe --chrome-frame").has_chrome()); |
| 127 EXPECT_FALSE(TestConfiguration(L"spam.exe --multi-install").has_chrome()); | 90 EXPECT_FALSE(TestConfiguration(L"spam.exe --multi-install").has_chrome()); |
| 128 } | 91 } |
| (...skipping 21 matching lines...) Expand all Loading... |
| 150 .is_multi_install()); | 113 .is_multi_install()); |
| 151 EXPECT_TRUE(TestConfiguration(L"spam.exe --multi-install") | 114 EXPECT_TRUE(TestConfiguration(L"spam.exe --multi-install") |
| 152 .is_multi_install()); | 115 .is_multi_install()); |
| 153 } | 116 } |
| 154 | 117 |
| 155 TEST(MiniInstallerConfigurationTest, IsSystemLevel) { | 118 TEST(MiniInstallerConfigurationTest, IsSystemLevel) { |
| 156 EXPECT_FALSE(TestConfiguration(L"spam.exe").is_system_level()); | 119 EXPECT_FALSE(TestConfiguration(L"spam.exe").is_system_level()); |
| 157 EXPECT_FALSE(TestConfiguration(L"spam.exe --chrome").is_system_level()); | 120 EXPECT_FALSE(TestConfiguration(L"spam.exe --chrome").is_system_level()); |
| 158 EXPECT_TRUE(TestConfiguration(L"spam.exe --system-level").is_system_level()); | 121 EXPECT_TRUE(TestConfiguration(L"spam.exe --system-level").is_system_level()); |
| 159 } | 122 } |
| OLD | NEW |