Case-insensitive string class

installer/src/installer-lib/test/process_test.cpp

 #include <gtest/gtest.h>
 #include "../process.h"
 #include <functional>
 
 // Turn off warnings for string copies
 #pragma warning( disable : 4996 )
 
 //-------------------------------------------------------
 // Comparison objects
 //-------------------------------------------------------
 
 const wchar_t exact_exe_name[] = L"installer-ca-tests.exe" ;
+const std::wstring exact_exe_string( exact_exe_name ) ;
+const wstring_ci exact_exe_string_ci( exact_exe_name ) ;
+
 const wchar_t mixedcase_exe_name[] = L"Installer-CA-Tests.exe" ;
-const wchar_t * multiple_exe_names[] = { mixedcase_exe_name, L"non-matching-name" } ;
+const wstring_ci mixedcase_exe_string_ci( mixedcase_exe_name ) ;
+
+const wchar_t unknown_name[] = L"non-matching-name" ;
+const wchar_t * multiple_exe_names[] = { mixedcase_exe_name, unknown_name } ;
 
 /**
  * Compare to our own process name, case-sensitive, no length limit
  */
 struct our_process_by_name
   : std::unary_function< PROCESSENTRY32W, bool >
 {
   bool operator()( const PROCESSENTRY32W & process )
   {
-    return 0 == wcscmp( process.szExeFile, exact_exe_name ) ;
+    return std::wstring( process.szExeFile ) == exact_exe_string ;
   } ;
 };
 
 /**
  * Compare to our own process name, case-insensitive, no length limit
  */
 struct our_process_by_name_CI
   : std::unary_function< PROCESSENTRY32W, bool >
 {
   bool operator()( const PROCESSENTRY32W & process ) 
   {
-    return 0 == wcscmpi( process.szExeFile, mixedcase_exe_name ) ;
-  } ;
-} ;
-
-/**
- * Compare to our own process name, case-insensitive, length-limited
- */
-struct our_process_by_name_CI_N
-  : std::unary_function< PROCESSENTRY32W, bool >
-{
-  bool operator()( const PROCESSENTRY32W & process ) 
-  {
-    return 0 == wcsncmpi( process.szExeFile, mixedcase_exe_name, sizeof( mixedcase_exe_name ) / sizeof( wchar_t ) ) ;
+    return wstring_ci( process.szExeFile ) == mixedcase_exe_string_ci ;
   } ;
 } ;
 
 
+struct our_process_by_name_subclassed
+  : public process_by_any_exe_name_CI
+{
+  our_process_by_name_subclassed( file_name_set( multiple_exe_names ) ) ;
+} ;
+
+/**
+ * Filter by process name. Comparison is case-insensitive.
+ */
+class process_by_name_CI
+  : public std::unary_function< PROCESSENTRY32W, bool >
+{
+  const wstring_ci _name ;
+public:
+  bool operator()( const PROCESSENTRY32W & process )
+  {
+    return _name == wstring_ci( process.szExeFile ) ;
+  }
+
+  process_by_name_CI( const wchar_t * name )
+    : _name( name )
+  {}
+} ;
+
 //-------------------------------------------------------
 // TESTS, no snapshots
 //-------------------------------------------------------
 PROCESSENTRY32 process_with_name( const wchar_t * s )
 {
   PROCESSENTRY32W p ;
   wcsncpy( p.szExeFile, s, MAX_PATH ) ;
   return p ;
 }
 
 PROCESSENTRY32 process_empty = process_with_name( L"" ) ;
 PROCESSENTRY32 process_exact = process_with_name( exact_exe_name ) ;
 PROCESSENTRY32 process_mixedcase = process_with_name( mixedcase_exe_name ) ;
 PROCESSENTRY32 process_explorer = process_with_name( L"explorer.exe" ) ;
 PROCESSENTRY32 process_absent = process_with_name( L"no_such_name" ) ;
 
-exe_name_set multiple_name_set( multiple_exe_names, 2 ) ;
+file_name_set multiple_name_set( multiple_exe_names ) ;
 process_by_any_exe_name_CI find_in_set( multiple_name_set ) ;
 
 TEST( exe_name_set, validate_setup )
 {
   ASSERT_EQ( 2u, multiple_name_set.size() ) ;
-  ASSERT_TRUE( multiple_name_set.find( mixedcase_exe_name ) != multiple_name_set.end() ) ;
-  ASSERT_TRUE( multiple_name_set.find( exact_exe_name ) != multiple_name_set.end() ) ;
+  ASSERT_TRUE( multiple_name_set.find( exact_exe_string_ci ) != multiple_name_set.end() ) ;
+  ASSERT_TRUE( multiple_name_set.find( mixedcase_exe_string_ci ) != multiple_name_set.end() ) ;
   ASSERT_TRUE( multiple_name_set.find( L"" ) == multiple_name_set.end() ) ;
   ASSERT_TRUE( multiple_name_set.find( L"not-in-list" ) == multiple_name_set.end() ) ;
 }
 
 TEST( process_by_any_exe_name_CI, empty )
 {
-  const wchar_t * elements[ 1 ] = { 0 } ;   // cheating a bit
-  exe_name_set s( elements, 0 ) ;
+  file_name_set s ;
   process_by_any_exe_name_CI x( s ) ;
 
   ASSERT_FALSE( x( process_empty ) ) ;
   ASSERT_FALSE( x( process_exact ) ) ;
   ASSERT_FALSE( x( process_mixedcase ) ) ;
   ASSERT_FALSE( x( process_explorer ) ) ;
   ASSERT_FALSE( x( process_absent ) ) ;
 }
 
-TEST( process_by_any_exe_name_CI, single_element )
+TEST( process_by_any_exe_name_CI, single_element_known )
 {
   const wchar_t * elements[ 1 ] = { exact_exe_name } ;
-  exe_name_set s( elements, 1 ) ;
+  file_name_set s( elements ) ;
   process_by_any_exe_name_CI x( s ) ;
 
   ASSERT_FALSE( x( process_empty ) ) ;
   ASSERT_TRUE( x( process_exact ) ) ;
   ASSERT_TRUE( x( process_mixedcase ) ) ;
   ASSERT_FALSE( x( process_explorer ) ) ;
   ASSERT_FALSE( x( process_absent ) ) ;
 }
 
+TEST( process_by_any_exe_name_CI, single_element_unknown )
+{
+  const wchar_t * elements[ 1 ] = { unknown_name } ;
+  file_name_set s( elements ) ;
+  process_by_any_exe_name_CI x( s ) ;
+
+  ASSERT_FALSE( x( process_empty ) ) ;
+  ASSERT_FALSE( x( process_exact ) ) ;
+  ASSERT_FALSE( x( process_mixedcase ) ) ;
+  ASSERT_FALSE( x( process_explorer ) ) ;
+  ASSERT_FALSE( x( process_absent ) ) ;
+}
+
 TEST( process_by_any_exe_name_CI, two_elements )
 {
-  exe_name_set s( multiple_exe_names, 2 ) ;
+  file_name_set s( multiple_exe_names ) ;
   process_by_any_exe_name_CI x( s ) ;
 
   ASSERT_FALSE( find_in_set( process_empty ) ) ;
   ASSERT_TRUE( find_in_set( process_exact ) ) ;
   ASSERT_TRUE( find_in_set( process_mixedcase ) ) ;
   ASSERT_FALSE( find_in_set( process_explorer ) ) ;
   ASSERT_FALSE( find_in_set( process_absent ) ) ;
 }
 
 //-------------------------------------------------------
@@ skipping 37 matching lines @@
 {
   std::vector< PROCESSENTRY32W > v ;
   initialize_process_list( v, our_process_by_name(), copy_all() ) ;
   unsigned int size( v.size() );
   EXPECT_EQ( 1u, size );    // Please, don't run multiple test executables simultaneously
   ASSERT_GE( 1u, size );
 }
 
 /**
  * The only process we are really guaranteed to have is this test process itself.
- * This test uses a filter function class with a special, fixed name comparison.
+ * This test uses the generic filter function.
  */
-TEST( Process_List_Test, find_our_process_CI_N_special )
+TEST( Process_List_Test, find_our_process_CI_generic )
 {
   std::vector< PROCESSENTRY32W > v ;
-  initialize_process_list( v, our_process_by_name_CI_N(), copy_all() ) ;
+  initialize_process_list( v, process_by_name_CI( mixedcase_exe_name ), copy_all() ) ;
   unsigned int size( v.size() );
   EXPECT_EQ( 1u, size );    // Please, don't run multiple test executables simultaneously
   ASSERT_GE( 1u, size );
 }
 
 /**
  * The only process we are really guaranteed to have is this test process itself.
- * This test uses the generic filter function.
+ * This test uses same one used in Process_Closer
  */
-TEST( Process_List_Test, find_our_process_CI_N_generic )
+TEST( Process_List_Test, find_our_process_CI_as_used )
 {
   std::vector< PROCESSENTRY32W > v ;
-  initialize_process_list( v, process_by_name_CI( mixedcase_exe_name ), copy_all() ) ;
+  initialize_process_list( v, process_by_any_exe_name_CI( file_name_set( multiple_exe_names ) ), copy_all() ) ;
   unsigned int size( v.size() );
   EXPECT_EQ( 1u, size );    // Please, don't run multiple test executables simultaneously
   ASSERT_GE( 1u, size );
 }
 
 /**
  * Locate the PID of our process.
  */
 TEST( Process_List_Test, find_our_PID )
 {
@@ skipping 35 matching lines @@
 }
 
 TEST( pid_set, find_our_process_in_set )
 {
   std::set< DWORD > s ;
   initialize_process_set( s, find_in_set, copy_PID() ) ;
   size_t size( s.size() ) ;
   EXPECT_EQ( size, 1u );
   ASSERT_GE( size, 1u );
 }