| OLD | NEW |
| 1 #include <gtest/gtest.h> | 1 #include <gtest/gtest.h> |
| 2 #include "../process.h" | 2 #include "../process.h" |
| 3 #include <functional> | 3 #include <functional> |
| 4 | 4 |
| 5 // Turn off warnings for string copies | 5 // Turn off warnings for string copies |
| 6 #pragma warning( disable : 4996 ) | 6 #pragma warning( disable : 4996 ) |
| 7 | 7 |
| 8 //------------------------------------------------------- | 8 //------------------------------------------------------- |
| 9 // Comparison objects | 9 // Comparison objects |
| 10 //------------------------------------------------------- | 10 //------------------------------------------------------- |
| 11 | 11 |
| 12 const wchar_t exact_exe_name[] = L"installer-ca-tests.exe" ; | 12 const wchar_t exact_exe_name[] = L"installer-ca-tests.exe" ; |
| 13 const wchar_t mixedcase_exe_name[] = L"Installer-CA-Tests.exe" ; | 13 const wchar_t mixedcase_exe_name[] = L"Installer-CA-Tests.exe" ; |
| 14 const wchar_t * multiple_exe_names[] = { mixedcase_exe_name, L"non-matching-name
" } ; | 14 const wchar_t * multiple_exe_names[] = { mixedcase_exe_name, L"non-matching-name
" } ; |
| 15 const wchar_t * multiple_module_names[] = { L"kernel32.dll", L"non-matching-name
" } ; |
| 16 const wchar_t * non_existent_module_names[] = { L"non-matching-name" } ; |
| 15 | 17 |
| 16 /** | 18 /** |
| 17 * Compare to our own process name, case-sensitive, no length limit | 19 * Compare to our own process name, case-sensitive, no length limit |
| 18 */ | 20 */ |
| 19 struct our_process_by_name | 21 struct our_process_by_name |
| 20 : std::unary_function< PROCESSENTRY32W, bool > | 22 : std::unary_function< PROCESSENTRY32W, bool > |
| 21 { | 23 { |
| 22 bool operator()( const PROCESSENTRY32W & process ) | 24 bool operator()( const PROCESSENTRY32W & process ) |
| 23 { | 25 { |
| 24 return 0 == wcscmp( process.szExeFile, exact_exe_name ) ; | 26 return 0 == wcscmp( process.szExeFile, exact_exe_name ) ; |
| (...skipping 35 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 60 return p ; | 62 return p ; |
| 61 } | 63 } |
| 62 | 64 |
| 63 PROCESSENTRY32 process_empty = process_with_name( L"" ) ; | 65 PROCESSENTRY32 process_empty = process_with_name( L"" ) ; |
| 64 PROCESSENTRY32 process_exact = process_with_name( exact_exe_name ) ; | 66 PROCESSENTRY32 process_exact = process_with_name( exact_exe_name ) ; |
| 65 PROCESSENTRY32 process_mixedcase = process_with_name( mixedcase_exe_name ) ; | 67 PROCESSENTRY32 process_mixedcase = process_with_name( mixedcase_exe_name ) ; |
| 66 PROCESSENTRY32 process_explorer = process_with_name( L"explorer.exe" ) ; | 68 PROCESSENTRY32 process_explorer = process_with_name( L"explorer.exe" ) ; |
| 67 PROCESSENTRY32 process_absent = process_with_name( L"no_such_name" ) ; | 69 PROCESSENTRY32 process_absent = process_with_name( L"no_such_name" ) ; |
| 68 | 70 |
| 69 exe_name_set multiple_name_set( multiple_exe_names, 2 ) ; | 71 exe_name_set multiple_name_set( multiple_exe_names, 2 ) ; |
| 72 exe_name_set multiple_name_set_modules( multiple_module_names, 2 ) ; |
| 73 exe_name_set non_existent_name_set_modules( non_existent_module_names, 1 ) ; |
| 70 process_by_any_exe_name_CI find_in_set( multiple_name_set ) ; | 74 process_by_any_exe_name_CI find_in_set( multiple_name_set ) ; |
| 75 process_by_any_exe_name_CI_w_ABP find_in_set_w_kernel32( multiple_name_set, mult
iple_name_set_modules ) ; |
| 76 process_by_any_exe_name_CI_w_ABP find_in_set_w_non_existent( multiple_name_set,
non_existent_name_set_modules ) ; |
| 71 | 77 |
| 72 TEST( exe_name_set, validate_setup ) | 78 TEST( exe_name_set, validate_setup ) |
| 73 { | 79 { |
| 74 ASSERT_EQ( 2u, multiple_name_set.size() ) ; | 80 ASSERT_EQ( 2u, multiple_name_set.size() ) ; |
| 75 ASSERT_TRUE( multiple_name_set.find( mixedcase_exe_name ) != multiple_name_set
.end() ) ; | 81 ASSERT_TRUE( multiple_name_set.find( mixedcase_exe_name ) != multiple_name_set
.end() ) ; |
| 76 ASSERT_TRUE( multiple_name_set.find( exact_exe_name ) != multiple_name_set.end
() ) ; | 82 ASSERT_TRUE( multiple_name_set.find( exact_exe_name ) != multiple_name_set.end
() ) ; |
| 77 ASSERT_TRUE( multiple_name_set.find( L"" ) == multiple_name_set.end() ) ; | 83 ASSERT_TRUE( multiple_name_set.find( L"" ) == multiple_name_set.end() ) ; |
| 78 ASSERT_TRUE( multiple_name_set.find( L"not-in-list" ) == multiple_name_set.end
() ) ; | 84 ASSERT_TRUE( multiple_name_set.find( L"not-in-list" ) == multiple_name_set.end
() ) ; |
| 79 } | 85 } |
| 80 | 86 |
| (...skipping 150 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 231 } | 237 } |
| 232 | 238 |
| 233 TEST( pid_set, find_our_process_in_set ) | 239 TEST( pid_set, find_our_process_in_set ) |
| 234 { | 240 { |
| 235 std::set< DWORD > s ; | 241 std::set< DWORD > s ; |
| 236 initialize_process_set( s, find_in_set, copy_PID() ) ; | 242 initialize_process_set( s, find_in_set, copy_PID() ) ; |
| 237 size_t size( s.size() ) ; | 243 size_t size( s.size() ) ; |
| 238 EXPECT_EQ( size, 1u ); | 244 EXPECT_EQ( size, 1u ); |
| 239 ASSERT_GE( size, 1u ); | 245 ASSERT_GE( size, 1u ); |
| 240 } | 246 } |
| 247 |
| 248 TEST( pid_set, find_our_process_in_set_w_kernel32 ) |
| 249 { |
| 250 std::set< DWORD > s ; |
| 251 initialize_process_set( s, find_in_set_w_kernel32, copy_PID() ) ; |
| 252 size_t size( s.size() ) ; |
| 253 EXPECT_EQ( size, 1u ); |
| 254 ASSERT_GE( size, 1u ); |
| 255 } |
| 256 TEST( pid_set, find_our_process_in_set_w_non_existant ) |
| 257 { |
| 258 std::set< DWORD > s ; |
| 259 initialize_process_set( s, find_in_set_w_non_existent, copy_PID() ) ; |
| 260 size_t size( s.size() ) ; |
| 261 EXPECT_EQ( size, 0u ); |
| 262 ASSERT_GE( size, 0u ); |
| 263 } |
| OLD | NEW |