Refactor multiprocess test code to allow multiple child processes to be launched.

snapshot/win/pe_image_annotations_reader_test.cc

 // Copyright 2015 The Crashpad Authors. All rights reserved.
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
 // You may obtain a copy of the License at
 //
 //     http://www.apache.org/licenses/LICENSE-2.0
 //
 // Unless required by applicable law or agreed to in writing, software
 // distributed under the License is distributed on an "AS IS" BASIS,
@@ skipping 18 matching lines @@
 #include "snapshot/win/pe_image_reader.h"
 #include "snapshot/win/process_reader_win.h"
 #include "test/win/win_multiprocess.h"
 #include "util/file/file_io.h"
 #include "util/win/process_info.h"
 
 namespace crashpad {
 namespace test {
 namespace {
 
+enum TestType {
+  // Don't crash, just test the CrashpadInfo interface.
+  kDontCrash = 0,
+
+  // The child process should crash by __debugbreak().
+  kCrashDebugBreak,
+};
+
+template <TestType Type>
 class TestPEImageAnnotationsReader final : public WinMultiprocess {
  public:
-  enum TestType {
-    // Don't crash, just test the CrashpadInfo interface.
-    kDontCrash = 0,
-
-    // The child process should crash by __debugbreak().
-    kCrashDebugBreak,
-  };
-
-  explicit TestPEImageAnnotationsReader(TestType test_type)
-      : WinMultiprocess(), test_type_(test_type) {}
-
+  TestPEImageAnnotationsReader() {}
   ~TestPEImageAnnotationsReader() {}
 
  private:
   // WinMultiprocess:
 
   void WinMultiprocessParent() override {
     ProcessReaderWin process_reader;
     ASSERT_TRUE(process_reader.Initialize(ChildProcess()));
 
     // Wait for the child process to indicate that it's done setting up its
@@ skipping 18 matching lines @@
                                         module_annotations_simple_map.end());
     }
 
     EXPECT_GE(all_annotations_simple_map.size(), 5u);
     EXPECT_EQ("crash", all_annotations_simple_map["#TEST# pad"]);
     EXPECT_EQ("value", all_annotations_simple_map["#TEST# key"]);
     EXPECT_EQ("y", all_annotations_simple_map["#TEST# x"]);
     EXPECT_EQ("shorter", all_annotations_simple_map["#TEST# longer"]);
     EXPECT_EQ("", all_annotations_simple_map["#TEST# empty_value"]);
 
-    if (test_type_ == kCrashDebugBreak)
+    if (Type == kCrashDebugBreak)
       SetExpectedChildExitCode(STATUS_BREAKPOINT);
 
     // Tell the child process to continue.
     CheckedWriteFile(WritePipeHandle(), &c, sizeof(c));
   }
 
   void WinMultiprocessChild() override {
     CrashpadInfo* crashpad_info = CrashpadInfo::GetCrashpadInfo();
 
     // This is "leaked" to crashpad_info.
     SimpleStringDictionary* simple_annotations = new SimpleStringDictionary();
     simple_annotations->SetKeyValue("#TEST# pad", "break");
     simple_annotations->SetKeyValue("#TEST# key", "value");
     simple_annotations->SetKeyValue("#TEST# pad", "crash");
     simple_annotations->SetKeyValue("#TEST# x", "y");
     simple_annotations->SetKeyValue("#TEST# longer", "shorter");
     simple_annotations->SetKeyValue("#TEST# empty_value", "");
 
     crashpad_info->set_simple_annotations(simple_annotations);
 
     // Tell the parent that the environment has been set up.
     char c = '\0';
     CheckedWriteFile(WritePipeHandle(), &c, sizeof(c));
 
     // Wait for the parent to indicate that it's safe to continue/crash.
     CheckedReadFile(ReadPipeHandle(), &c, sizeof(c));
 
-    switch (test_type_) {
+    switch (Type) {
       case kDontCrash:
         break;
 
       case kCrashDebugBreak:
         __debugbreak();
         break;
     }
   }
 
-  TestType test_type_;
-
   DISALLOW_COPY_AND_ASSIGN(TestPEImageAnnotationsReader);
 };
 
 TEST(PEImageAnnotationsReader, DontCrash) {
-  TestPEImageAnnotationsReader test_pe_image_annotations_reader(
-      TestPEImageAnnotationsReader::kDontCrash);
-  test_pe_image_annotations_reader.Run();
+  WinMultiprocess::Run<TestPEImageAnnotationsReader<kDontCrash>>();
 }
 
 TEST(PEImageAnnotationsReader, CrashDebugBreak) {
-  TestPEImageAnnotationsReader test_pe_image_annotations_reader(
-      TestPEImageAnnotationsReader::kCrashDebugBreak);
-  test_pe_image_annotations_reader.Run();
+  WinMultiprocess::Run<TestPEImageAnnotationsReader<kCrashDebugBreak>>();
 }
 
 }  // namespace
 }  // namespace test
 }  // namespace crashpad