Index: base/win/event_trace_consumer_unittest.cc |
diff --git a/base/win/event_trace_consumer_unittest.cc b/base/win/event_trace_consumer_unittest.cc |
deleted file mode 100644 |
index ecbf238bea48c88429e22930957e10bdf82280ab..0000000000000000000000000000000000000000 |
--- a/base/win/event_trace_consumer_unittest.cc |
+++ /dev/null |
@@ -1,366 +0,0 @@ |
-// 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. |
-// |
-// Unit tests for event trace consumer base class. |
-#include "base/win/event_trace_consumer.h" |
- |
-#include <list> |
- |
-#include <objbase.h> |
- |
-#include "base/basictypes.h" |
-#include "base/files/file_path.h" |
-#include "base/files/file_util.h" |
-#include "base/files/scoped_temp_dir.h" |
-#include "base/logging.h" |
-#include "base/process/process_handle.h" |
-#include "base/strings/stringprintf.h" |
-#include "base/win/event_trace_controller.h" |
-#include "base/win/event_trace_provider.h" |
-#include "base/win/scoped_handle.h" |
-#include "testing/gtest/include/gtest/gtest.h" |
- |
-#include <initguid.h> // NOLINT - has to be last |
- |
-namespace base { |
-namespace win { |
- |
-namespace { |
- |
-typedef std::list<EVENT_TRACE> EventQueue; |
- |
-class TestConsumer: public EtwTraceConsumerBase<TestConsumer> { |
- public: |
- TestConsumer() { |
- sank_event_.Set(::CreateEvent(NULL, TRUE, FALSE, NULL)); |
- ClearQueue(); |
- } |
- |
- ~TestConsumer() { |
- ClearQueue(); |
- sank_event_.Close(); |
- } |
- |
- void ClearQueue() { |
- for (EventQueue::const_iterator it(events_.begin()), end(events_.end()); |
- it != end; ++it) { |
- delete[] reinterpret_cast<char*>(it->MofData); |
- } |
- |
- events_.clear(); |
- } |
- |
- static void EnqueueEvent(EVENT_TRACE* event) { |
- events_.push_back(*event); |
- EVENT_TRACE& back = events_.back(); |
- |
- if (event->MofData != NULL && event->MofLength != 0) { |
- back.MofData = new char[event->MofLength]; |
- memcpy(back.MofData, event->MofData, event->MofLength); |
- } |
- } |
- |
- static void ProcessEvent(EVENT_TRACE* event) { |
- EnqueueEvent(event); |
- ::SetEvent(sank_event_.Get()); |
- } |
- |
- static ScopedHandle sank_event_; |
- static EventQueue events_; |
- |
- private: |
- DISALLOW_COPY_AND_ASSIGN(TestConsumer); |
-}; |
- |
-ScopedHandle TestConsumer::sank_event_; |
-EventQueue TestConsumer::events_; |
- |
-class EtwTraceConsumerBaseTest: public testing::Test { |
- public: |
- EtwTraceConsumerBaseTest() |
- : session_name_(StringPrintf(L"TestSession-%d", GetCurrentProcId())) { |
- } |
- |
- void SetUp() override { |
- // Cleanup any potentially dangling sessions. |
- EtwTraceProperties ignore; |
- EtwTraceController::Stop(session_name_.c_str(), &ignore); |
- |
- // Allocate a new GUID for each provider test. |
- ASSERT_HRESULT_SUCCEEDED(::CoCreateGuid(&test_provider_)); |
- } |
- |
- void TearDown() override { |
- // Cleanup any potentially dangling sessions. |
- EtwTraceProperties ignore; |
- EtwTraceController::Stop(session_name_.c_str(), &ignore); |
- } |
- |
- protected: |
- GUID test_provider_; |
- std::wstring session_name_; |
-}; |
- |
-} // namespace |
- |
-TEST_F(EtwTraceConsumerBaseTest, Initialize) { |
- TestConsumer consumer_; |
-} |
- |
-TEST_F(EtwTraceConsumerBaseTest, OpenRealtimeSucceedsWhenNoSession) { |
- TestConsumer consumer_; |
- ASSERT_HRESULT_SUCCEEDED( |
- consumer_.OpenRealtimeSession(session_name_.c_str())); |
-} |
- |
-TEST_F(EtwTraceConsumerBaseTest, ConsumerImmediateFailureWhenNoSession) { |
- TestConsumer consumer_; |
- ASSERT_HRESULT_SUCCEEDED( |
- consumer_.OpenRealtimeSession(session_name_.c_str())); |
- ASSERT_HRESULT_FAILED(consumer_.Consume()); |
-} |
- |
-namespace { |
- |
-class EtwTraceConsumerRealtimeTest: public EtwTraceConsumerBaseTest { |
- public: |
- void SetUp() override { |
- EtwTraceConsumerBaseTest::SetUp(); |
- ASSERT_HRESULT_SUCCEEDED( |
- consumer_.OpenRealtimeSession(session_name_.c_str())); |
- } |
- |
- void TearDown() override { |
- consumer_.Close(); |
- EtwTraceConsumerBaseTest::TearDown(); |
- } |
- |
- DWORD ConsumerThread() { |
- ::SetEvent(consumer_ready_.Get()); |
- return consumer_.Consume(); |
- } |
- |
- static DWORD WINAPI ConsumerThreadMainProc(void* arg) { |
- return reinterpret_cast<EtwTraceConsumerRealtimeTest*>(arg)-> |
- ConsumerThread(); |
- } |
- |
- HRESULT StartConsumerThread() { |
- consumer_ready_.Set(::CreateEvent(NULL, TRUE, FALSE, NULL)); |
- EXPECT_TRUE(consumer_ready_.IsValid()); |
- consumer_thread_.Set(::CreateThread(NULL, 0, ConsumerThreadMainProc, this, |
- 0, NULL)); |
- if (consumer_thread_.Get() == NULL) |
- return HRESULT_FROM_WIN32(::GetLastError()); |
- |
- HANDLE events[] = { consumer_ready_.Get(), consumer_thread_.Get() }; |
- DWORD result = ::WaitForMultipleObjects(arraysize(events), events, |
- FALSE, INFINITE); |
- switch (result) { |
- case WAIT_OBJECT_0: |
- // The event was set, the consumer_ is ready. |
- return S_OK; |
- case WAIT_OBJECT_0 + 1: { |
- // The thread finished. This may race with the event, so check |
- // explicitly for the event here, before concluding there's trouble. |
- if (::WaitForSingleObject(consumer_ready_.Get(), 0) == WAIT_OBJECT_0) |
- return S_OK; |
- DWORD exit_code = 0; |
- if (::GetExitCodeThread(consumer_thread_.Get(), &exit_code)) |
- return exit_code; |
- return HRESULT_FROM_WIN32(::GetLastError()); |
- } |
- default: |
- return E_UNEXPECTED; |
- } |
- } |
- |
- // Waits for consumer_ thread to exit, and returns its exit code. |
- HRESULT JoinConsumerThread() { |
- if (::WaitForSingleObject(consumer_thread_.Get(), INFINITE) != |
- WAIT_OBJECT_0) { |
- return HRESULT_FROM_WIN32(::GetLastError()); |
- } |
- |
- DWORD exit_code = 0; |
- if (::GetExitCodeThread(consumer_thread_.Get(), &exit_code)) |
- return exit_code; |
- |
- return HRESULT_FROM_WIN32(::GetLastError()); |
- } |
- |
- TestConsumer consumer_; |
- ScopedHandle consumer_ready_; |
- ScopedHandle consumer_thread_; |
-}; |
- |
-} // namespace |
- |
-TEST_F(EtwTraceConsumerRealtimeTest, ConsumerReturnsWhenSessionClosed) { |
- EtwTraceController controller; |
- if (controller.StartRealtimeSession(session_name_.c_str(), 100 * 1024) == |
- E_ACCESSDENIED) { |
- VLOG(1) << "You must be an administrator to run this test on Vista"; |
- return; |
- } |
- |
- // Start the consumer_. |
- ASSERT_HRESULT_SUCCEEDED(StartConsumerThread()); |
- |
- // Wait around for the consumer_ thread a bit. |
- ASSERT_EQ(WAIT_TIMEOUT, ::WaitForSingleObject(consumer_thread_.Get(), 50)); |
- ASSERT_HRESULT_SUCCEEDED(controller.Stop(NULL)); |
- |
- // The consumer_ returns success on session stop. |
- ASSERT_HRESULT_SUCCEEDED(JoinConsumerThread()); |
-} |
- |
-namespace { |
- |
-// {57E47923-A549-476f-86CA-503D57F59E62} |
-DEFINE_GUID( |
- kTestEventType, |
- 0x57e47923, 0xa549, 0x476f, 0x86, 0xca, 0x50, 0x3d, 0x57, 0xf5, 0x9e, 0x62); |
- |
-} // namespace |
- |
-TEST_F(EtwTraceConsumerRealtimeTest, ConsumeEvent) { |
- EtwTraceController controller; |
- if (controller.StartRealtimeSession(session_name_.c_str(), 100 * 1024) == |
- E_ACCESSDENIED) { |
- VLOG(1) << "You must be an administrator to run this test on Vista"; |
- return; |
- } |
- |
- ASSERT_HRESULT_SUCCEEDED(controller.EnableProvider( |
- test_provider_, TRACE_LEVEL_VERBOSE, 0xFFFFFFFF)); |
- |
- EtwTraceProvider provider(test_provider_); |
- ASSERT_EQ(ERROR_SUCCESS, provider.Register()); |
- |
- // Start the consumer_. |
- ASSERT_HRESULT_SUCCEEDED(StartConsumerThread()); |
- ASSERT_EQ(0, TestConsumer::events_.size()); |
- |
- EtwMofEvent<1> event(kTestEventType, 1, TRACE_LEVEL_ERROR); |
- EXPECT_EQ(ERROR_SUCCESS, provider.Log(&event.header)); |
- EXPECT_EQ(WAIT_OBJECT_0, |
- ::WaitForSingleObject(TestConsumer::sank_event_.Get(), INFINITE)); |
- ASSERT_HRESULT_SUCCEEDED(controller.Stop(NULL)); |
- ASSERT_HRESULT_SUCCEEDED(JoinConsumerThread()); |
- ASSERT_NE(0u, TestConsumer::events_.size()); |
-} |
- |
-namespace { |
- |
-// We run events through a file session to assert that |
-// the content comes through. |
-class EtwTraceConsumerDataTest: public EtwTraceConsumerBaseTest { |
- public: |
- EtwTraceConsumerDataTest() { |
- } |
- |
- void SetUp() override { |
- EtwTraceConsumerBaseTest::SetUp(); |
- |
- EtwTraceProperties prop; |
- EtwTraceController::Stop(session_name_.c_str(), &prop); |
- |
- // Create a temp dir for this test. |
- ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); |
- // Construct a temp file name in our dir. |
- temp_file_ = temp_dir_.path().Append(L"test.etl"); |
- } |
- |
- void TearDown() override { |
- EXPECT_TRUE(base::DeleteFile(temp_file_, false)); |
- |
- EtwTraceConsumerBaseTest::TearDown(); |
- } |
- |
- HRESULT LogEventToTempSession(PEVENT_TRACE_HEADER header) { |
- EtwTraceController controller; |
- |
- // Set up a file session. |
- HRESULT hr = controller.StartFileSession(session_name_.c_str(), |
- temp_file_.value().c_str()); |
- if (FAILED(hr)) |
- return hr; |
- |
- // Enable our provider. |
- EXPECT_HRESULT_SUCCEEDED(controller.EnableProvider( |
- test_provider_, TRACE_LEVEL_VERBOSE, 0xFFFFFFFF)); |
- |
- EtwTraceProvider provider(test_provider_); |
- // Then register our provider, means we get a session handle immediately. |
- EXPECT_EQ(ERROR_SUCCESS, provider.Register()); |
- // Trace the event, it goes to the temp file. |
- EXPECT_EQ(ERROR_SUCCESS, provider.Log(header)); |
- EXPECT_HRESULT_SUCCEEDED(controller.DisableProvider(test_provider_)); |
- EXPECT_HRESULT_SUCCEEDED(provider.Unregister()); |
- EXPECT_HRESULT_SUCCEEDED(controller.Flush(NULL)); |
- EXPECT_HRESULT_SUCCEEDED(controller.Stop(NULL)); |
- |
- return S_OK; |
- } |
- |
- HRESULT ConsumeEventFromTempSession() { |
- // Now consume the event(s). |
- TestConsumer consumer_; |
- HRESULT hr = consumer_.OpenFileSession(temp_file_.value().c_str()); |
- if (SUCCEEDED(hr)) |
- hr = consumer_.Consume(); |
- consumer_.Close(); |
- // And nab the result. |
- events_.swap(TestConsumer::events_); |
- return hr; |
- } |
- |
- HRESULT RoundTripEvent(PEVENT_TRACE_HEADER header, PEVENT_TRACE* trace) { |
- base::DeleteFile(temp_file_, false); |
- |
- HRESULT hr = LogEventToTempSession(header); |
- if (SUCCEEDED(hr)) |
- hr = ConsumeEventFromTempSession(); |
- |
- if (FAILED(hr)) |
- return hr; |
- |
- // We should now have the event in the queue. |
- if (events_.empty()) |
- return E_FAIL; |
- |
- *trace = &events_.back(); |
- return S_OK; |
- } |
- |
- EventQueue events_; |
- ScopedTempDir temp_dir_; |
- FilePath temp_file_; |
-}; |
- |
-} // namespace |
- |
- |
-TEST_F(EtwTraceConsumerDataTest, RoundTrip) { |
- EtwMofEvent<1> event(kTestEventType, 1, TRACE_LEVEL_ERROR); |
- |
- static const char kData[] = "This is but test data"; |
- event.fields[0].DataPtr = reinterpret_cast<ULONG64>(kData); |
- event.fields[0].Length = sizeof(kData); |
- |
- PEVENT_TRACE trace = NULL; |
- HRESULT hr = RoundTripEvent(&event.header, &trace); |
- if (hr == E_ACCESSDENIED) { |
- VLOG(1) << "You must be an administrator to run this test on Vista"; |
- return; |
- } |
- ASSERT_HRESULT_SUCCEEDED(hr) << "RoundTripEvent failed"; |
- ASSERT_TRUE(trace != NULL); |
- ASSERT_EQ(sizeof(kData), trace->MofLength); |
- ASSERT_STREQ(kData, reinterpret_cast<const char*>(trace->MofData)); |
-} |
- |
-} // namespace win |
-} // namespace base |