Chromium Code Reviews| Index: remoting/base/breakpad_win_unittest.cc |
| diff --git a/remoting/base/breakpad_win_unittest.cc b/remoting/base/breakpad_win_unittest.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..d0b59a0ddfe6ceb7735ddf956404d805d05421eb |
| --- /dev/null |
| +++ b/remoting/base/breakpad_win_unittest.cc |
| @@ -0,0 +1,136 @@ |
| +// Copyright (c) 2012 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 <stdio.h> |
| + |
| +#include "base/compiler_specific.h" |
| +#include "base/environment.h" |
| +#include "base/logging.h" |
| +#include "base/memory/scoped_ptr.h" |
| +#include "base/string16.h" |
| +#include "base/stringprintf.h" |
| +#include "base/utf_string_conversions.h" |
| +#include "breakpad/src/client/windows/crash_generation/client_info.h" |
| +#include "breakpad/src/client/windows/crash_generation/crash_generation_server.h" |
| +#include "testing/gmock/include/gmock/gmock.h" |
| +#include "testing/gtest/include/gtest/gtest.h" |
| + |
| +namespace remoting { |
|
Peter Kasting
2012/06/14 20:16:27
Nit: Does the anonymous namespace need to be insid
alexeypa (please no reviews)
2012/06/15 18:45:49
Done.
|
| + |
| +namespace { |
| + |
| +// The name of the environment variable used to pass the crash server pipe name |
| +// to the crashing child process. |
| +const char kPipeVariableName[] = "REMOTING_BREAKPAD_WIN_DEATH_TEST_PIPE_NAME"; |
| + |
| +// The prefix string used to generate a unique crash server pipe name. |
| +// The name has to be unique as multiple test instances can be running |
| +// simultaneously. |
| +const wchar_t kPipeNamePrefix[] = L"\\\\.\\pipe\\"; |
| + |
| +class MockCrashServerCallbacks { |
| + public: |
| + MockCrashServerCallbacks() {} |
| + virtual ~MockCrashServerCallbacks() {} |
|
Peter Kasting
2012/06/14 20:16:27
Nit: Even in tests, try to avoid inlining function
alexeypa (please no reviews)
2012/06/15 18:45:49
Done. I removed inlining of the constructor as wel
|
| + |
| + // |google_breakpad::CrashGenerationServer| invokes callbacks from artitrary |
| + // thread pool threads. |OnClientDumpRequested| is the only one that happened |
| + // to be called in synchronous manner. While it is still called on |
| + // a thread pool thread, the crashing process will wait until the server |
| + // signals an event after |OnClientDumpRequested| completes (or until 15 |
| + // seconds timeout expires). |
| + MOCK_METHOD0(OnClientDumpRequested, void()); |
| + |
| + static void OnClientDumpRequestCallback( |
| + void* context, |
| + const google_breakpad::ClientInfo* client_info, |
| + const string16* file_path) { |
| + reinterpret_cast<MockCrashServerCallbacks*>(context)-> |
| + OnClientDumpRequested(); |
| + } |
| +}; |
| + |
| +} // namespace |
| + |
| +void InitializeCrashReportingForTest(const wchar_t*); |
| + |
| +class BreakpadWinDeathTest : public testing::Test { |
| + public: |
| + BreakpadWinDeathTest() {} |
| + virtual void SetUp() OVERRIDE { |
| + scoped_ptr<base::Environment> environment(base::Environment::Create()); |
| + std::string pipe_name; |
| + if (environment->GetVar(kPipeVariableName, &pipe_name)) { |
| + // This is a child process. Initialize crash dump reporting to the crash |
| + // dump server. |
| + pipe_name_ = UTF8ToUTF16(pipe_name); |
|
Peter Kasting
2012/06/14 20:16:27
Use UTF8ToWide() here (see below).
alexeypa (please no reviews)
2012/06/15 18:45:49
Done.
|
| + ::remoting::InitializeCrashReportingForTest(pipe_name_.c_str()); |
| + } else { |
| + // This is the parent process. Generate a unique pipe name and setup |
| + // a dummy crash dump server. |
| + UUID guid = {0}; |
| + RPC_STATUS status = ::UuidCreate(&guid); |
|
Peter Kasting
2012/06/14 20:16:27
Nit: Is the :: really necessary?
alexeypa (please no reviews)
2012/06/15 18:45:49
Done.
|
| + EXPECT_TRUE(status == RPC_S_OK || status == RPC_S_UUID_LOCAL_ONLY); |
| + |
| + pipe_name_ = kPipeNamePrefix + |
|
Peter Kasting
2012/06/14 20:16:27
Nit: If using StringPrintf() anyway, add a "%s" to
alexeypa (please no reviews)
2012/06/15 18:45:49
Done.
|
| + base::StringPrintf( |
| + L"%08x-%04x-%04x-%02x%02x-%02x%02x%02x%02x%02x%02x", |
| + guid.Data1, guid.Data2, guid.Data3, |
| + guid.Data4[0], guid.Data4[1], guid.Data4[2], |
| + guid.Data4[3], guid.Data4[4], guid.Data4[5], |
| + guid.Data4[6], guid.Data4[7]); |
| + bool result = environment->SetVar(kPipeVariableName, |
| + UTF16ToUTF8(pipe_name_)); |
|
Peter Kasting
2012/06/14 20:16:27
Use WideToUTF8 here (see below).
alexeypa (please no reviews)
2012/06/15 18:45:49
Done.
|
| + EXPECT_TRUE(result); |
|
Peter Kasting
2012/06/14 20:16:27
Nit: Unlike with DCHECK(), it's OK to directly wra
alexeypa (please no reviews)
2012/06/15 18:45:49
Done.
|
| + |
| + // Setup a dummy crash dump server. |
| + callbacks_.reset(new MockCrashServerCallbacks()); |
| + crash_server_.reset( |
| + new google_breakpad::CrashGenerationServer( |
| + pipe_name_, NULL, |
|
Peter Kasting
2012/06/14 20:16:27
Nit: Strange wrapping here. More common choices a
alexeypa (please no reviews)
2012/06/15 18:45:49
Done.
|
| + NULL, 0, |
| + MockCrashServerCallbacks::OnClientDumpRequestCallback, |
| + callbacks_.get(), |
| + NULL, 0, |
| + false, NULL)); |
| + |
| + result = crash_server_->Start(); |
| + ASSERT_TRUE(result); |
| + } |
| + } |
| + |
| + protected: |
| + scoped_ptr<google_breakpad::CrashGenerationServer> crash_server_; |
| + scoped_ptr<MockCrashServerCallbacks> callbacks_; |
| + string16 pipe_name_; |
|
Peter Kasting
2012/06/14 20:16:27
This should be a wstring.
alexeypa (please no reviews)
2012/06/15 18:45:49
Done.
|
| +}; |
| + |
| +TEST_F(BreakpadWinDeathTest, TestAccessViolation) { |
| + if (callbacks_.get()) { |
| + EXPECT_CALL(*callbacks_, OnClientDumpRequested()); |
| + } |
| + |
| + // Generate access violation exception. |
| + ASSERT_DEATH(*reinterpret_cast<int*>(0) = 1, ""); |
|
Peter Kasting
2012/06/14 20:16:27
Nit: 0 -> NULL?
alexeypa (please no reviews)
2012/06/15 18:45:49
Done.
|
| +} |
| + |
| +TEST_F(BreakpadWinDeathTest, TestInvalidParameter) { |
| + if (callbacks_.get()) { |
| + EXPECT_CALL(*callbacks_, OnClientDumpRequested()); |
| + } |
| + |
| + // Cause the invalid parameter callback to be called. |
| + ASSERT_EXIT(printf(NULL), ::testing::ExitedWithCode(0), ""); |
| +} |
| + |
| +TEST_F(BreakpadWinDeathTest, TestDebugbreak) { |
| + if (callbacks_.get()) { |
| + EXPECT_CALL(*callbacks_, OnClientDumpRequested()); |
| + } |
| + |
| + // See if __debugbreak() is intercepted. |
| + ASSERT_DEATH(__debugbreak(), ""); |
| +} |
| + |
| +} // namespace remoting |