Index: components/browser_watcher/postmortem_minidump_writer_win_unittest.cc |
diff --git a/components/browser_watcher/postmortem_minidump_writer_win_unittest.cc b/components/browser_watcher/postmortem_minidump_writer_win_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..c93e5cadabe3913255ba637286db2edbc8bb7850 |
--- /dev/null |
+++ b/components/browser_watcher/postmortem_minidump_writer_win_unittest.cc |
@@ -0,0 +1,71 @@ |
+// Copyright 2016 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#include "components/browser_watcher/postmortem_minidump_writer.h" |
+ |
+#include <windows.h> // NOLINT |
+#include <dbghelp.h> |
+ |
+#include "base/files/file_path.h" |
+#include "base/files/file_util.h" |
+#include "base/files/scoped_temp_dir.h" |
+#include "components/browser_watcher/stability_report.pb.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+ |
+namespace browser_watcher { |
+ |
+TEST(PostmortemMinidumpWriterWinTest, WriteDumpBasicTest) { |
+ // Create a stability report. |
+ // TODO(manzagop): flesh out the report once proto is more detailed. |
+ StabilityReport expected_report; |
+ ProcessState* process_state = expected_report.add_process_states(); |
+ CodeModule* module = process_state->add_modules(); |
+ module->set_base_address(1024); |
+ module->set_code_file("some_code_file.dll"); |
+ |
+ // Write to minidump. |
+ base::ScopedTempDir temp_dir; |
+ ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); |
+ base::FilePath minidump_path = temp_dir.path().AppendASCII("minidump.dmp"); |
+ { |
+ // Note: the inner scope to ensure the writer goes out of scope, which will |
+ // close the minidump file. |
+ PostmortemMinidumpWriter writer; |
+ ASSERT_TRUE(writer.WriteDump(minidump_path, expected_report)); |
+ } |
+ |
+ // Read back the minidump. |
+ base::ScopedFILE minidump_file; |
+ minidump_file.reset(base::OpenFile(minidump_path, "rb")); |
+ ASSERT_TRUE(minidump_file.get()); |
+ |
+ MINIDUMP_HEADER header = {}; |
+ ASSERT_EQ(1, fread(&header, sizeof(header), 1, minidump_file.get())); |
+ ASSERT_EQ(MINIDUMP_SIGNATURE, header.Signature); |
+ ASSERT_EQ(1, header.NumberOfStreams); |
+ RVA directory_rva = header.StreamDirectoryRva; |
+ |
+ MINIDUMP_DIRECTORY directory = {}; |
+ ASSERT_EQ(0, fseek(minidump_file.get(), directory_rva, SEEK_SET)); |
+ ASSERT_EQ(1, fread(&directory, sizeof(directory), 1, minidump_file.get())); |
+ ASSERT_EQ(0x4B6B0002, directory.StreamType); |
+ RVA report_rva = directory.Location.Rva; |
+ ULONG32 report_size_bytes = directory.Location.DataSize; |
+ |
+ std::string recovered_serialized_report; |
+ recovered_serialized_report.resize(report_size_bytes); |
+ ASSERT_EQ(0, fseek(minidump_file.get(), report_rva, SEEK_SET)); |
+ ASSERT_EQ(report_size_bytes, fread(&recovered_serialized_report.at(0), 1, |
+ report_size_bytes, minidump_file.get())); |
+ |
+ // Validate the recovered report. |
+ std::string expected_serialized_report; |
+ expected_report.SerializeToString(&expected_serialized_report); |
+ ASSERT_EQ(expected_serialized_report, recovered_serialized_report); |
+ |
+ StabilityReport recovered_report; |
+ ASSERT_TRUE(recovered_report.ParseFromString(recovered_serialized_report)); |
+} |
+ |
+} // namespace browser_watcher |