Index: base/event_trace_controller_unittest.cc |
diff --git a/base/event_trace_controller_unittest.cc b/base/event_trace_controller_unittest.cc |
deleted file mode 100644 |
index 6ce027b5188c7154e553a6f6b0b107a99a00c012..0000000000000000000000000000000000000000 |
--- a/base/event_trace_controller_unittest.cc |
+++ /dev/null |
@@ -1,239 +0,0 @@ |
-// Copyright 2010 Google Inc. |
-// |
-// Licensed under the Apache License, Version 2.0 (the "License"); |
-// you may not use this file except in compliance with the License. |
-// You may obtain a copy of the License at |
-// |
-// http://www.apache.org/licenses/LICENSE-2.0 |
-// |
-// Unless required by applicable law or agreed to in writing, software |
-// distributed under the License is distributed on an "AS IS" BASIS, |
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
-// See the License for the specific language governing permissions and |
-// limitations under the License. |
-// ======================================================================== |
-// |
-// Unit tests for event trace controller. |
-#include "omaha/base/event_trace_controller.h" |
-#include <atlsync.h> |
-#include "omaha/base/app_util.h" |
-#include "omaha/base/event_trace_provider.h" |
-#include "omaha/testing/unit_test.h" |
-#include <initguid.h> // NOLINT - must be last. |
- |
-namespace { |
- |
-using omaha::EtwTraceController; |
-using omaha::EtwTraceProvider; |
-using omaha::EtwTraceProperties; |
- |
-const wchar_t kTestSessionName[] = L"TestLogSession"; |
- |
-// {0D236A42-CD18-4e3d-9975-DCEEA2106E05} |
-DEFINE_GUID(kTestProvider, |
- 0xd236a42, 0xcd18, 0x4e3d, 0x99, 0x75, 0xdc, 0xee, 0xa2, 0x10, 0x6e, 0x5); |
- |
-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_.Create(NULL, TRUE, FALSE, NULL); |
- } |
- |
- void WaitForCallback() { |
- ::WaitForSingleObject(callback_event_, INFINITE); |
- callback_event_.Reset(); |
- } |
- |
- private: |
- virtual void OnEventsEnabled() { |
- callback_event_.Set(); |
- } |
- virtual void OnEventsDisabled() { |
- callback_event_.Set(); |
- } |
- |
- CEvent callback_event_; |
- |
- DISALLOW_COPY_AND_ASSIGN(TestingProvider); |
-}; |
- |
-// These fixtures make sure we clean up dangling trace sessions |
-// prior to all tests, to make the tests stable against crashes |
-// and failures. |
-class EtwTracePropertiesTest : public testing::Test { |
- public: |
- virtual void SetUp() { |
- CloseTestTraceSession(); |
- } |
- |
- virtual void TearDown() { |
- CloseTestTraceSession(); |
- } |
- |
- private: |
- void CloseTestTraceSession() { |
- // Clean up potential leftover sessions from previous unsuccessful runs. |
- EtwTraceProperties prop; |
- EtwTraceController::Stop(kTestSessionName, &prop); |
- } |
-}; |
- |
-class EtwTraceControllerTest : public EtwTracePropertiesTest { |
-}; |
- |
-} // namespace |
- |
-namespace omaha { |
- |
-TEST_F(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_F(EtwTracePropertiesTest, Strings) { |
- EtwTraceProperties prop; |
- |
- EXPECT_STREQ(L"", prop.GetLoggerFileName()); |
- EXPECT_STREQ(L"", prop.GetLoggerName()); |
- |
- std::wstring name(1023, L'A'); |
- EXPECT_HRESULT_SUCCEEDED(prop.SetLoggerFileName(name.c_str())); |
- EXPECT_HRESULT_SUCCEEDED(prop.SetLoggerName(name.c_str())); |
- EXPECT_STREQ(name.c_str(), prop.GetLoggerFileName()); |
- EXPECT_STREQ(name.c_str(), prop.GetLoggerName()); |
- |
- std::wstring name2(1024, L'A'); |
- EXPECT_HRESULT_FAILED(prop.SetLoggerFileName(name2.c_str())); |
- EXPECT_HRESULT_FAILED(prop.SetLoggerName(name2.c_str())); |
-} |
- |
-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(kTestSessionName, 100 * 1024); |
- if (hr == E_ACCESSDENIED) { |
- SUCCEED() << "You must be an administrator to run this test on Vista"; |
- return; |
- } |
- |
- EXPECT_TRUE(NULL != controller.session()); |
- EXPECT_STREQ(kTestSessionName, controller.session_name()); |
- |
- EXPECT_HRESULT_SUCCEEDED(controller.Stop(NULL)); |
- EXPECT_EQ(NULL, controller.session()); |
- EXPECT_STREQ(L"", controller.session_name()); |
-} |
- |
-TEST_F(EtwTraceControllerTest, StartFileSession) { |
- CString temp; |
- EXPECT_TRUE(::GetTempFileName(app_util::GetTempDir(), _T("tmp"), 0, |
- CStrBuf(temp, MAX_PATH))); |
- |
- EtwTraceController controller; |
- HRESULT hr = controller.StartFileSession(kTestSessionName, temp); |
- if (hr == E_ACCESSDENIED) { |
- SUCCEED() << "You must be an administrator to run this test on Vista"; |
- return; |
- } |
- |
- EXPECT_TRUE(NULL != controller.session()); |
- EXPECT_STREQ(kTestSessionName, controller.session_name()); |
- |
- EXPECT_HRESULT_SUCCEEDED(controller.Stop(NULL)); |
- EXPECT_EQ(NULL, controller.session()); |
- EXPECT_STREQ(L"", controller.session_name()); |
- |
- EXPECT_TRUE(::DeleteFile(temp)); |
-} |
- |
-TEST_F(EtwTraceControllerTest, EnableDisable) { |
- TestingProvider provider(kTestProvider); |
- |
- EXPECT_EQ(ERROR_SUCCESS, provider.Register()); |
- EXPECT_EQ(NULL, provider.session_handle()); |
- |
- EtwTraceController controller; |
- HRESULT hr = controller.StartRealtimeSession(kTestSessionName, 100 * 1024); |
- if (hr == E_ACCESSDENIED) { |
- SUCCEED() << "You must be an administrator to run this test on Vista"; |
- return; |
- } |
- |
- EXPECT_HRESULT_SUCCEEDED(controller.EnableProvider(kTestProvider, |
- 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(kTestProvider)); |
- |
- 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(kTestProvider, |
- 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 omaha |