| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 // | 4 // |
| 5 // Unit tests for event trace controller. | 5 // Unit tests for event trace controller. |
| 6 | 6 |
| 7 #include <objbase.h> | 7 #include <objbase.h> |
| 8 #include <initguid.h> | 8 #include <initguid.h> |
| 9 | 9 |
| 10 #include "base/files/file_path.h" | 10 #include "base/files/file_path.h" |
| (...skipping 44 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 55 | 55 |
| 56 TEST(EtwTracePropertiesTest, Initialization) { | 56 TEST(EtwTracePropertiesTest, Initialization) { |
| 57 EtwTraceProperties prop; | 57 EtwTraceProperties prop; |
| 58 | 58 |
| 59 EVENT_TRACE_PROPERTIES* p = prop.get(); | 59 EVENT_TRACE_PROPERTIES* p = prop.get(); |
| 60 EXPECT_NE(0u, p->Wnode.BufferSize); | 60 EXPECT_NE(0u, p->Wnode.BufferSize); |
| 61 EXPECT_EQ(0u, p->Wnode.ProviderId); | 61 EXPECT_EQ(0u, p->Wnode.ProviderId); |
| 62 EXPECT_EQ(0u, p->Wnode.HistoricalContext); | 62 EXPECT_EQ(0u, p->Wnode.HistoricalContext); |
| 63 | 63 |
| 64 EXPECT_TRUE(kGuidNull == p->Wnode.Guid); | 64 EXPECT_TRUE(kGuidNull == p->Wnode.Guid); |
| 65 EXPECT_EQ(0, p->Wnode.ClientContext); | 65 EXPECT_EQ(0u, p->Wnode.ClientContext); |
| 66 EXPECT_EQ(WNODE_FLAG_TRACED_GUID, p->Wnode.Flags); | 66 EXPECT_EQ(static_cast<ULONG>(WNODE_FLAG_TRACED_GUID), p->Wnode.Flags); |
| 67 | 67 |
| 68 EXPECT_EQ(0, p->BufferSize); | 68 EXPECT_EQ(0u, p->BufferSize); |
| 69 EXPECT_EQ(0, p->MinimumBuffers); | 69 EXPECT_EQ(0u, p->MinimumBuffers); |
| 70 EXPECT_EQ(0, p->MaximumBuffers); | 70 EXPECT_EQ(0u, p->MaximumBuffers); |
| 71 EXPECT_EQ(0, p->MaximumFileSize); | 71 EXPECT_EQ(0u, p->MaximumFileSize); |
| 72 EXPECT_EQ(0, p->LogFileMode); | 72 EXPECT_EQ(0u, p->LogFileMode); |
| 73 EXPECT_EQ(0, p->FlushTimer); | 73 EXPECT_EQ(0u, p->FlushTimer); |
| 74 EXPECT_EQ(0, p->EnableFlags); | 74 EXPECT_EQ(0u, p->EnableFlags); |
| 75 EXPECT_EQ(0, p->AgeLimit); | 75 EXPECT_EQ(0, p->AgeLimit); |
| 76 | 76 |
| 77 EXPECT_EQ(0, p->NumberOfBuffers); | 77 EXPECT_EQ(0u, p->NumberOfBuffers); |
| 78 EXPECT_EQ(0, p->FreeBuffers); | 78 EXPECT_EQ(0u, p->FreeBuffers); |
| 79 EXPECT_EQ(0, p->EventsLost); | 79 EXPECT_EQ(0u, p->EventsLost); |
| 80 EXPECT_EQ(0, p->BuffersWritten); | 80 EXPECT_EQ(0u, p->BuffersWritten); |
| 81 EXPECT_EQ(0, p->LogBuffersLost); | 81 EXPECT_EQ(0u, p->LogBuffersLost); |
| 82 EXPECT_EQ(0, p->RealTimeBuffersLost); | 82 EXPECT_EQ(0u, p->RealTimeBuffersLost); |
| 83 EXPECT_EQ(0, p->LoggerThreadId); | 83 EXPECT_EQ(0u, p->LoggerThreadId); |
| 84 EXPECT_NE(0u, p->LogFileNameOffset); | 84 EXPECT_NE(0u, p->LogFileNameOffset); |
| 85 EXPECT_NE(0u, p->LoggerNameOffset); | 85 EXPECT_NE(0u, p->LoggerNameOffset); |
| 86 } | 86 } |
| 87 | 87 |
| 88 TEST(EtwTracePropertiesTest, Strings) { | 88 TEST(EtwTracePropertiesTest, Strings) { |
| 89 EtwTraceProperties prop; | 89 EtwTraceProperties prop; |
| 90 | 90 |
| 91 ASSERT_STREQ(L"", prop.GetLoggerFileName()); | 91 ASSERT_STREQ(L"", prop.GetLoggerFileName()); |
| 92 ASSERT_STREQ(L"", prop.GetLoggerName()); | 92 ASSERT_STREQ(L"", prop.GetLoggerName()); |
| 93 | 93 |
| (...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 126 protected: | 126 protected: |
| 127 GUID test_provider_; | 127 GUID test_provider_; |
| 128 std::wstring session_name_; | 128 std::wstring session_name_; |
| 129 }; | 129 }; |
| 130 | 130 |
| 131 } // namespace | 131 } // namespace |
| 132 | 132 |
| 133 TEST_F(EtwTraceControllerTest, Initialize) { | 133 TEST_F(EtwTraceControllerTest, Initialize) { |
| 134 EtwTraceController controller; | 134 EtwTraceController controller; |
| 135 | 135 |
| 136 EXPECT_EQ(NULL, controller.session()); | 136 EXPECT_EQ(0u, controller.session()); |
| 137 EXPECT_STREQ(L"", controller.session_name()); | 137 EXPECT_STREQ(L"", controller.session_name()); |
| 138 } | 138 } |
| 139 | 139 |
| 140 | 140 |
| 141 TEST_F(EtwTraceControllerTest, StartRealTimeSession) { | 141 TEST_F(EtwTraceControllerTest, StartRealTimeSession) { |
| 142 EtwTraceController controller; | 142 EtwTraceController controller; |
| 143 | 143 |
| 144 HRESULT hr = controller.StartRealtimeSession(session_name_.c_str(), | 144 HRESULT hr = controller.StartRealtimeSession(session_name_.c_str(), |
| 145 100 * 1024); | 145 100 * 1024); |
| 146 if (hr == E_ACCESSDENIED) { | 146 if (hr == E_ACCESSDENIED) { |
| 147 VLOG(1) << "You must be an administrator to run this test on Vista"; | 147 VLOG(1) << "You must be an administrator to run this test on Vista"; |
| 148 return; | 148 return; |
| 149 } | 149 } |
| 150 | 150 |
| 151 EXPECT_TRUE(NULL != controller.session()); | 151 EXPECT_NE(0u, controller.session()); |
| 152 EXPECT_STREQ(session_name_.c_str(), controller.session_name()); | 152 EXPECT_STREQ(session_name_.c_str(), controller.session_name()); |
| 153 | 153 |
| 154 EXPECT_HRESULT_SUCCEEDED(controller.Stop(NULL)); | 154 EXPECT_HRESULT_SUCCEEDED(controller.Stop(NULL)); |
| 155 EXPECT_EQ(NULL, controller.session()); | 155 EXPECT_EQ(0u, controller.session()); |
| 156 EXPECT_STREQ(L"", controller.session_name()); | 156 EXPECT_STREQ(L"", controller.session_name()); |
| 157 } | 157 } |
| 158 | 158 |
| 159 TEST_F(EtwTraceControllerTest, StartFileSession) { | 159 TEST_F(EtwTraceControllerTest, StartFileSession) { |
| 160 ScopedTempDir temp_dir; | 160 ScopedTempDir temp_dir; |
| 161 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); | 161 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); |
| 162 FilePath temp; | 162 FilePath temp; |
| 163 ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir.path(), &temp)); | 163 ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir.path(), &temp)); |
| 164 | 164 |
| 165 EtwTraceController controller; | 165 EtwTraceController controller; |
| 166 HRESULT hr = controller.StartFileSession(session_name_.c_str(), | 166 HRESULT hr = controller.StartFileSession(session_name_.c_str(), |
| 167 temp.value().c_str()); | 167 temp.value().c_str()); |
| 168 if (hr == E_ACCESSDENIED) { | 168 if (hr == E_ACCESSDENIED) { |
| 169 VLOG(1) << "You must be an administrator to run this test on Vista"; | 169 VLOG(1) << "You must be an administrator to run this test on Vista"; |
| 170 base::DeleteFile(temp, false); | 170 base::DeleteFile(temp, false); |
| 171 return; | 171 return; |
| 172 } | 172 } |
| 173 | 173 |
| 174 EXPECT_TRUE(NULL != controller.session()); | 174 EXPECT_NE(0u, controller.session()); |
| 175 EXPECT_STREQ(session_name_.c_str(), controller.session_name()); | 175 EXPECT_STREQ(session_name_.c_str(), controller.session_name()); |
| 176 | 176 |
| 177 EXPECT_HRESULT_SUCCEEDED(controller.Stop(NULL)); | 177 EXPECT_HRESULT_SUCCEEDED(controller.Stop(NULL)); |
| 178 EXPECT_EQ(NULL, controller.session()); | 178 EXPECT_EQ(0u, controller.session()); |
| 179 EXPECT_STREQ(L"", controller.session_name()); | 179 EXPECT_STREQ(L"", controller.session_name()); |
| 180 base::DeleteFile(temp, false); | 180 base::DeleteFile(temp, false); |
| 181 } | 181 } |
| 182 | 182 |
| 183 // This test is flaky for unclear reasons. See bugs 525297 and 534184 | 183 // This test is flaky for unclear reasons. See bugs 525297 and 534184 |
| 184 TEST_F(EtwTraceControllerTest, DISABLED_EnableDisable) { | 184 TEST_F(EtwTraceControllerTest, DISABLED_EnableDisable) { |
| 185 TestingProvider provider(test_provider_); | 185 TestingProvider provider(test_provider_); |
| 186 | 186 |
| 187 EXPECT_EQ(ERROR_SUCCESS, provider.Register()); | 187 EXPECT_EQ(static_cast<DWORD>(ERROR_SUCCESS), provider.Register()); |
| 188 EXPECT_EQ(NULL, provider.session_handle()); | 188 EXPECT_EQ(0u, provider.session_handle()); |
| 189 | 189 |
| 190 EtwTraceController controller; | 190 EtwTraceController controller; |
| 191 HRESULT hr = controller.StartRealtimeSession(session_name_.c_str(), | 191 HRESULT hr = controller.StartRealtimeSession(session_name_.c_str(), |
| 192 100 * 1024); | 192 100 * 1024); |
| 193 if (hr == E_ACCESSDENIED) { | 193 if (hr == E_ACCESSDENIED) { |
| 194 VLOG(1) << "You must be an administrator to run this test on Vista"; | 194 VLOG(1) << "You must be an administrator to run this test on Vista"; |
| 195 return; | 195 return; |
| 196 } | 196 } |
| 197 | 197 |
| 198 EXPECT_HRESULT_SUCCEEDED(controller.EnableProvider(test_provider_, | 198 EXPECT_HRESULT_SUCCEEDED(controller.EnableProvider(test_provider_, |
| 199 TRACE_LEVEL_VERBOSE, kTestProviderFlags)); | 199 TRACE_LEVEL_VERBOSE, kTestProviderFlags)); |
| 200 | 200 |
| 201 provider.WaitForCallback(); | 201 provider.WaitForCallback(); |
| 202 | 202 |
| 203 EXPECT_EQ(TRACE_LEVEL_VERBOSE, provider.enable_level()); | 203 EXPECT_EQ(TRACE_LEVEL_VERBOSE, provider.enable_level()); |
| 204 EXPECT_EQ(kTestProviderFlags, provider.enable_flags()); | 204 EXPECT_EQ(kTestProviderFlags, provider.enable_flags()); |
| 205 | 205 |
| 206 EXPECT_HRESULT_SUCCEEDED(controller.DisableProvider(test_provider_)); | 206 EXPECT_HRESULT_SUCCEEDED(controller.DisableProvider(test_provider_)); |
| 207 | 207 |
| 208 provider.WaitForCallback(); | 208 provider.WaitForCallback(); |
| 209 | 209 |
| 210 EXPECT_EQ(0, provider.enable_level()); | 210 EXPECT_EQ(0, provider.enable_level()); |
| 211 EXPECT_EQ(0, provider.enable_flags()); | 211 EXPECT_EQ(0u, provider.enable_flags()); |
| 212 | 212 |
| 213 EXPECT_EQ(ERROR_SUCCESS, provider.Unregister()); | 213 EXPECT_EQ(static_cast<DWORD>(ERROR_SUCCESS), provider.Unregister()); |
| 214 | 214 |
| 215 // Enable the provider again, before registering. | 215 // Enable the provider again, before registering. |
| 216 EXPECT_HRESULT_SUCCEEDED(controller.EnableProvider(test_provider_, | 216 EXPECT_HRESULT_SUCCEEDED(controller.EnableProvider(test_provider_, |
| 217 TRACE_LEVEL_VERBOSE, kTestProviderFlags)); | 217 TRACE_LEVEL_VERBOSE, kTestProviderFlags)); |
| 218 | 218 |
| 219 // Register the provider again, the settings above | 219 // Register the provider again, the settings above |
| 220 // should take immediate effect. | 220 // should take immediate effect. |
| 221 EXPECT_EQ(ERROR_SUCCESS, provider.Register()); | 221 EXPECT_EQ(static_cast<DWORD>(ERROR_SUCCESS), provider.Register()); |
| 222 | 222 |
| 223 EXPECT_EQ(TRACE_LEVEL_VERBOSE, provider.enable_level()); | 223 EXPECT_EQ(TRACE_LEVEL_VERBOSE, provider.enable_level()); |
| 224 EXPECT_EQ(kTestProviderFlags, provider.enable_flags()); | 224 EXPECT_EQ(kTestProviderFlags, provider.enable_flags()); |
| 225 | 225 |
| 226 // Consume the callback event of the previous controller.EnableProvider(). | 226 // Consume the callback event of the previous controller.EnableProvider(). |
| 227 provider.WaitForCallback(); | 227 provider.WaitForCallback(); |
| 228 | 228 |
| 229 EXPECT_HRESULT_SUCCEEDED(controller.Stop(NULL)); | 229 EXPECT_HRESULT_SUCCEEDED(controller.Stop(NULL)); |
| 230 | 230 |
| 231 // Windows 7 does not call the callback when Stop() is called so we | 231 // Windows 7 does not call the callback when Stop() is called so we |
| 232 // can't wait, and enable_level and enable_flags are not zeroed. | 232 // can't wait, and enable_level and enable_flags are not zeroed. |
| 233 if (base::win::GetVersion() >= VERSION_WIN8) { | 233 if (base::win::GetVersion() >= VERSION_WIN8) { |
| 234 provider.WaitForCallback(); | 234 provider.WaitForCallback(); |
| 235 | 235 |
| 236 // Session should have wound down. | 236 // Session should have wound down. |
| 237 EXPECT_EQ(0, provider.enable_level()); | 237 EXPECT_EQ(0, provider.enable_level()); |
| 238 EXPECT_EQ(0, provider.enable_flags()); | 238 EXPECT_EQ(0u, provider.enable_flags()); |
| 239 } | 239 } |
| 240 } | 240 } |
| 241 | 241 |
| 242 } // namespace win | 242 } // namespace win |
| 243 } // namespace base | 243 } // namespace base |
| OLD | NEW |