OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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/util/shell_util.h" | 5 #include "chrome/installer/util/shell_util.h" |
6 | 6 |
7 #include <vector> | 7 #include <vector> |
8 | 8 |
9 #include "base/base_paths.h" | 9 #include "base/base_paths.h" |
10 #include "base/base_paths_win.h" | 10 #include "base/base_paths_win.h" |
(...skipping 807 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
818 base::win::RegKey key; | 818 base::win::RegKey key; |
819 ASSERT_EQ( | 819 ASSERT_EQ( |
820 ERROR_SUCCESS, | 820 ERROR_SUCCESS, |
821 key.Create( | 821 key.Create( |
822 HKEY_CURRENT_USER, L"Software\\Classes\\.test2", KEY_ALL_ACCESS)); | 822 HKEY_CURRENT_USER, L"Software\\Classes\\.test2", KEY_ALL_ACCESS)); |
823 EXPECT_EQ(ERROR_SUCCESS, key.WriteValue(L"", L"SomeOtherApp")); | 823 EXPECT_EQ(ERROR_SUCCESS, key.WriteValue(L"", L"SomeOtherApp")); |
824 } | 824 } |
825 | 825 |
826 static base::CommandLine OpenCommand() { | 826 static base::CommandLine OpenCommand() { |
827 base::FilePath open_command_path(kTestOpenCommand); | 827 base::FilePath open_command_path(kTestOpenCommand); |
828 return base::CommandLine(open_command_path); | 828 base::CommandLine open_command(open_command_path); |
| 829 // The "%1" should automatically be quoted. |
| 830 open_command.AppendArg("%1"); |
| 831 return open_command; |
829 } | 832 } |
830 | 833 |
831 static std::set<base::string16> FileExtensions() { | 834 static std::set<base::string16> FileExtensions() { |
832 std::set<base::string16> file_extensions; | 835 std::set<base::string16> file_extensions; |
833 for (size_t i = 0; i < arraysize(kTestFileExtensions); ++i) | 836 for (size_t i = 0; i < arraysize(kTestFileExtensions); ++i) |
834 file_extensions.insert(kTestFileExtensions[i]); | 837 file_extensions.insert(kTestFileExtensions[i]); |
835 return file_extensions; | 838 return file_extensions; |
836 } | 839 } |
837 | 840 |
838 private: | 841 private: |
(...skipping 22 matching lines...) Expand all Loading... |
861 key.Open(HKEY_CURRENT_USER, | 864 key.Open(HKEY_CURRENT_USER, |
862 L"Software\\Classes\\TestApp\\DefaultIcon", | 865 L"Software\\Classes\\TestApp\\DefaultIcon", |
863 KEY_READ)); | 866 KEY_READ)); |
864 EXPECT_EQ(ERROR_SUCCESS, key.ReadValue(L"", &value)); | 867 EXPECT_EQ(ERROR_SUCCESS, key.ReadValue(L"", &value)); |
865 EXPECT_EQ(L"D:\\test.ico,0", value); | 868 EXPECT_EQ(L"D:\\test.ico,0", value); |
866 ASSERT_EQ(ERROR_SUCCESS, | 869 ASSERT_EQ(ERROR_SUCCESS, |
867 key.Open(HKEY_CURRENT_USER, | 870 key.Open(HKEY_CURRENT_USER, |
868 L"Software\\Classes\\TestApp\\shell\\open\\command", | 871 L"Software\\Classes\\TestApp\\shell\\open\\command", |
869 KEY_READ)); | 872 KEY_READ)); |
870 EXPECT_EQ(ERROR_SUCCESS, key.ReadValue(L"", &value)); | 873 EXPECT_EQ(ERROR_SUCCESS, key.ReadValue(L"", &value)); |
871 EXPECT_EQ(L"\"C:\\test.exe\"", value); | 874 EXPECT_EQ(L"\"C:\\test.exe\" \"%1\"", value); |
872 | 875 |
873 // .test1 should be default-associated with our test app. | 876 // .test1 should be default-associated with our test app. |
874 ASSERT_EQ( | 877 ASSERT_EQ( |
875 ERROR_SUCCESS, | 878 ERROR_SUCCESS, |
876 key.Open(HKEY_CURRENT_USER, L"Software\\Classes\\.test1", KEY_READ)); | 879 key.Open(HKEY_CURRENT_USER, L"Software\\Classes\\.test1", KEY_READ)); |
877 EXPECT_EQ(ERROR_SUCCESS, key.ReadValue(L"", &value)); | 880 EXPECT_EQ(ERROR_SUCCESS, key.ReadValue(L"", &value)); |
878 EXPECT_EQ(L"TestApp", value); | 881 EXPECT_EQ(L"TestApp", value); |
879 ASSERT_EQ(ERROR_SUCCESS, | 882 ASSERT_EQ(ERROR_SUCCESS, |
880 key.Open(HKEY_CURRENT_USER, | 883 key.Open(HKEY_CURRENT_USER, |
881 L"Software\\Classes\\.test1\\OpenWithProgids", | 884 L"Software\\Classes\\.test1\\OpenWithProgids", |
(...skipping 118 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1000 | 1003 |
1001 const base::string16 expected[] = { L"", L"MY", L"MZXQ", L"MZXW6", L"MZXW6YQ", | 1004 const base::string16 expected[] = { L"", L"MY", L"MZXQ", L"MZXW6", L"MZXW6YQ", |
1002 L"MZXW6YTB", L"MZXW6YTBOI"}; | 1005 L"MZXW6YTB", L"MZXW6YTBOI"}; |
1003 | 1006 |
1004 // Run the tests, with one more letter in the input every pass. | 1007 // Run the tests, with one more letter in the input every pass. |
1005 for (int i = 0; i < arraysize(expected); ++i) { | 1008 for (int i = 0; i < arraysize(expected); ++i) { |
1006 ASSERT_EQ(expected[i], | 1009 ASSERT_EQ(expected[i], |
1007 ShellUtil::ByteArrayToBase32(test_array, i)); | 1010 ShellUtil::ByteArrayToBase32(test_array, i)); |
1008 } | 1011 } |
1009 } | 1012 } |
OLD | NEW |