Index: base/win/event_trace_controller_unittest.cc |
diff --git a/base/win/event_trace_controller_unittest.cc b/base/win/event_trace_controller_unittest.cc |
deleted file mode 100644 |
index a2cd81cf7097d1427540cbe475b685fdaa6ba877..0000000000000000000000000000000000000000 |
--- a/base/win/event_trace_controller_unittest.cc |
+++ /dev/null |
@@ -1,234 +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 controller. |
- |
-#include <objbase.h> |
-#include <initguid.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/sys_info.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" |
- |
-namespace base { |
-namespace win { |
- |
-namespace { |
- |
-DEFINE_GUID(kGuidNull, |
- 0x0000000, 0x0000, 0x0000, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0); |
- |
-const ULONG kTestProviderFlags = 0xCAFEBABE; |
- |
-class TestingProvider: public EtwTraceProvider { |
- public: |
- explicit TestingProvider(const GUID& provider_name) |
- : EtwTraceProvider(provider_name) { |
- callback_event_.Set(::CreateEvent(NULL, TRUE, FALSE, NULL)); |
- } |
- |
- void WaitForCallback() { |
- ::WaitForSingleObject(callback_event_.Get(), INFINITE); |
- ::ResetEvent(callback_event_.Get()); |
- } |
- |
- private: |
- void OnEventsEnabled() override { ::SetEvent(callback_event_.Get()); } |
- void PostEventsDisabled() override { ::SetEvent(callback_event_.Get()); } |
- |
- ScopedHandle callback_event_; |
- |
- DISALLOW_COPY_AND_ASSIGN(TestingProvider); |
-}; |
- |
-} // namespace |
- |
-TEST(EtwTracePropertiesTest, Initialization) { |
- EtwTraceProperties prop; |
- |
- EVENT_TRACE_PROPERTIES* p = prop.get(); |
- EXPECT_NE(0u, p->Wnode.BufferSize); |
- EXPECT_EQ(0u, p->Wnode.ProviderId); |
- EXPECT_EQ(0u, p->Wnode.HistoricalContext); |
- |
- EXPECT_TRUE(kGuidNull == p->Wnode.Guid); |
- EXPECT_EQ(0, p->Wnode.ClientContext); |
- EXPECT_EQ(WNODE_FLAG_TRACED_GUID, p->Wnode.Flags); |
- |
- EXPECT_EQ(0, p->BufferSize); |
- EXPECT_EQ(0, p->MinimumBuffers); |
- EXPECT_EQ(0, p->MaximumBuffers); |
- EXPECT_EQ(0, p->MaximumFileSize); |
- EXPECT_EQ(0, p->LogFileMode); |
- EXPECT_EQ(0, p->FlushTimer); |
- EXPECT_EQ(0, p->EnableFlags); |
- EXPECT_EQ(0, p->AgeLimit); |
- |
- EXPECT_EQ(0, p->NumberOfBuffers); |
- EXPECT_EQ(0, p->FreeBuffers); |
- EXPECT_EQ(0, p->EventsLost); |
- EXPECT_EQ(0, p->BuffersWritten); |
- EXPECT_EQ(0, p->LogBuffersLost); |
- EXPECT_EQ(0, p->RealTimeBuffersLost); |
- EXPECT_EQ(0, p->LoggerThreadId); |
- EXPECT_NE(0u, p->LogFileNameOffset); |
- EXPECT_NE(0u, p->LoggerNameOffset); |
-} |
- |
-TEST(EtwTracePropertiesTest, Strings) { |
- EtwTraceProperties prop; |
- |
- ASSERT_STREQ(L"", prop.GetLoggerFileName()); |
- ASSERT_STREQ(L"", prop.GetLoggerName()); |
- |
- std::wstring name(1023, L'A'); |
- ASSERT_HRESULT_SUCCEEDED(prop.SetLoggerFileName(name.c_str())); |
- ASSERT_HRESULT_SUCCEEDED(prop.SetLoggerName(name.c_str())); |
- ASSERT_STREQ(name.c_str(), prop.GetLoggerFileName()); |
- ASSERT_STREQ(name.c_str(), prop.GetLoggerName()); |
- |
- std::wstring name2(1024, L'A'); |
- ASSERT_HRESULT_FAILED(prop.SetLoggerFileName(name2.c_str())); |
- ASSERT_HRESULT_FAILED(prop.SetLoggerName(name2.c_str())); |
-} |
- |
-namespace { |
- |
-class EtwTraceControllerTest : public testing::Test { |
- public: |
- EtwTraceControllerTest() |
- : session_name_(StringPrintf(L"TestSession-%d", GetCurrentProcId())) { |
- } |
- |
- void SetUp() override { |
- EtwTraceProperties ignore; |
- EtwTraceController::Stop(session_name_.c_str(), &ignore); |
- |
- // Allocate a new provider name GUID for each test. |
- ASSERT_HRESULT_SUCCEEDED(::CoCreateGuid(&test_provider_)); |
- } |
- |
- void TearDown() override { |
- EtwTraceProperties prop; |
- EtwTraceController::Stop(session_name_.c_str(), &prop); |
- } |
- |
- protected: |
- GUID test_provider_; |
- std::wstring session_name_; |
-}; |
- |
-} // namespace |
- |
-TEST_F(EtwTraceControllerTest, Initialize) { |
- EtwTraceController controller; |
- |
- EXPECT_EQ(NULL, controller.session()); |
- EXPECT_STREQ(L"", controller.session_name()); |
-} |
- |
- |
-TEST_F(EtwTraceControllerTest, StartRealTimeSession) { |
- EtwTraceController controller; |
- |
- HRESULT hr = controller.StartRealtimeSession(session_name_.c_str(), |
- 100 * 1024); |
- if (hr == E_ACCESSDENIED) { |
- VLOG(1) << "You must be an administrator to run this test on Vista"; |
- return; |
- } |
- |
- EXPECT_TRUE(NULL != controller.session()); |
- EXPECT_STREQ(session_name_.c_str(), controller.session_name()); |
- |
- EXPECT_HRESULT_SUCCEEDED(controller.Stop(NULL)); |
- EXPECT_EQ(NULL, controller.session()); |
- EXPECT_STREQ(L"", controller.session_name()); |
-} |
- |
-TEST_F(EtwTraceControllerTest, StartFileSession) { |
- ScopedTempDir temp_dir; |
- ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); |
- FilePath temp; |
- ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir.path(), &temp)); |
- |
- EtwTraceController controller; |
- HRESULT hr = controller.StartFileSession(session_name_.c_str(), |
- temp.value().c_str()); |
- if (hr == E_ACCESSDENIED) { |
- VLOG(1) << "You must be an administrator to run this test on Vista"; |
- base::DeleteFile(temp, false); |
- return; |
- } |
- |
- EXPECT_TRUE(NULL != controller.session()); |
- EXPECT_STREQ(session_name_.c_str(), controller.session_name()); |
- |
- EXPECT_HRESULT_SUCCEEDED(controller.Stop(NULL)); |
- EXPECT_EQ(NULL, controller.session()); |
- EXPECT_STREQ(L"", controller.session_name()); |
- base::DeleteFile(temp, false); |
-} |
- |
-TEST_F(EtwTraceControllerTest, EnableDisable) { |
- TestingProvider provider(test_provider_); |
- |
- EXPECT_EQ(ERROR_SUCCESS, provider.Register()); |
- EXPECT_EQ(NULL, provider.session_handle()); |
- |
- EtwTraceController controller; |
- HRESULT hr = controller.StartRealtimeSession(session_name_.c_str(), |
- 100 * 1024); |
- if (hr == E_ACCESSDENIED) { |
- VLOG(1) << "You must be an administrator to run this test on Vista"; |
- return; |
- } |
- |
- EXPECT_HRESULT_SUCCEEDED(controller.EnableProvider(test_provider_, |
- TRACE_LEVEL_VERBOSE, kTestProviderFlags)); |
- |
- provider.WaitForCallback(); |
- |
- EXPECT_EQ(TRACE_LEVEL_VERBOSE, provider.enable_level()); |
- EXPECT_EQ(kTestProviderFlags, provider.enable_flags()); |
- |
- EXPECT_HRESULT_SUCCEEDED(controller.DisableProvider(test_provider_)); |
- |
- provider.WaitForCallback(); |
- |
- EXPECT_EQ(0, provider.enable_level()); |
- EXPECT_EQ(0, provider.enable_flags()); |
- |
- EXPECT_EQ(ERROR_SUCCESS, provider.Unregister()); |
- |
- // Enable the provider again, before registering. |
- EXPECT_HRESULT_SUCCEEDED(controller.EnableProvider(test_provider_, |
- TRACE_LEVEL_VERBOSE, kTestProviderFlags)); |
- |
- // Register the provider again, the settings above |
- // should take immediate effect. |
- EXPECT_EQ(ERROR_SUCCESS, provider.Register()); |
- |
- EXPECT_EQ(TRACE_LEVEL_VERBOSE, provider.enable_level()); |
- EXPECT_EQ(kTestProviderFlags, provider.enable_flags()); |
- |
- EXPECT_HRESULT_SUCCEEDED(controller.Stop(NULL)); |
- |
- provider.WaitForCallback(); |
- |
- // Session should have wound down. |
- EXPECT_EQ(0, provider.enable_level()); |
- EXPECT_EQ(0, provider.enable_flags()); |
-} |
- |
-} // namespace win |
-} // namespace base |