Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(562)

Unified Diff: base/win/event_trace_consumer_unittest.cc

Issue 1446363003: Deleted OS_WIN and all Windows specific files from base. (Closed) Base URL: https://github.com/domokit/mojo.git@base_tests
Patch Set: Created 5 years, 1 month ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « base/win/event_trace_consumer.h ('k') | base/win/event_trace_controller.h » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
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
« no previous file with comments | « base/win/event_trace_consumer.h ('k') | base/win/event_trace_controller.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698