Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 <initguid.h> // NOLINT. | 7 #include <initguid.h> // NOLINT. |
| 8 | 8 |
| 9 #include "base/file_path.h" | 9 #include "base/file_path.h" |
| 10 #include "base/file_util.h" | 10 #include "base/file_util.h" |
| (...skipping 123 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 134 EXPECT_STREQ(kTestSessionName, controller.session_name()); | 134 EXPECT_STREQ(kTestSessionName, controller.session_name()); |
| 135 | 135 |
| 136 EXPECT_HRESULT_SUCCEEDED(controller.Stop(NULL)); | 136 EXPECT_HRESULT_SUCCEEDED(controller.Stop(NULL)); |
| 137 EXPECT_EQ(NULL, controller.session()); | 137 EXPECT_EQ(NULL, controller.session()); |
| 138 EXPECT_STREQ(L"", controller.session_name()); | 138 EXPECT_STREQ(L"", controller.session_name()); |
| 139 } | 139 } |
| 140 | 140 |
| 141 TEST(EtwTraceControllerTest, StartFileSession) { | 141 TEST(EtwTraceControllerTest, StartFileSession) { |
| 142 FilePath temp; | 142 FilePath temp; |
| 143 | 143 |
| 144 ASSERT_HRESULT_SUCCEEDED(file_util::CreateTemporaryFile(&temp)); | 144 ASSERT_HRESULT_SUCCEEDED(file_util::CreateTemporaryFile(&temp)); |
|
Sigurður Ásgeirsson
2011/12/16 13:59:45
can this be a ScopedTempDir instead?
Lei Zhang
2011/12/16 20:00:22
Sure, in patch set 3.
| |
| 145 | 145 |
| 146 EtwTraceController controller; | 146 EtwTraceController controller; |
| 147 HRESULT hr = controller.StartFileSession(kTestSessionName, | 147 HRESULT hr = controller.StartFileSession(kTestSessionName, |
| 148 temp.value().c_str()); | 148 temp.value().c_str()); |
| 149 if (hr == E_ACCESSDENIED) { | 149 if (hr == E_ACCESSDENIED) { |
| 150 VLOG(1) << "You must be an administrator to run this test on Vista"; | 150 VLOG(1) << "You must be an administrator to run this test on Vista"; |
| 151 file_util::Delete(temp, false); | |
| 151 return; | 152 return; |
| 152 } | 153 } |
| 153 | 154 |
| 154 EXPECT_TRUE(NULL != controller.session()); | 155 EXPECT_TRUE(NULL != controller.session()); |
| 155 EXPECT_STREQ(kTestSessionName, controller.session_name()); | 156 EXPECT_STREQ(kTestSessionName, controller.session_name()); |
| 156 | 157 |
| 157 EXPECT_HRESULT_SUCCEEDED(controller.Stop(NULL)); | 158 EXPECT_HRESULT_SUCCEEDED(controller.Stop(NULL)); |
| 158 EXPECT_EQ(NULL, controller.session()); | 159 EXPECT_EQ(NULL, controller.session()); |
| 159 EXPECT_STREQ(L"", controller.session_name()); | 160 EXPECT_STREQ(L"", controller.session_name()); |
| 161 file_util::Delete(temp, false); | |
| 160 } | 162 } |
| 161 | 163 |
| 162 TEST(EtwTraceControllerTest, EnableDisable) { | 164 TEST(EtwTraceControllerTest, EnableDisable) { |
| 163 TestingProvider provider(kTestProvider); | 165 TestingProvider provider(kTestProvider); |
| 164 | 166 |
| 165 EXPECT_EQ(ERROR_SUCCESS, provider.Register()); | 167 EXPECT_EQ(ERROR_SUCCESS, provider.Register()); |
| 166 EXPECT_EQ(NULL, provider.session_handle()); | 168 EXPECT_EQ(NULL, provider.session_handle()); |
| 167 | 169 |
| 168 EtwTraceController controller; | 170 EtwTraceController controller; |
| 169 HRESULT hr = controller.StartRealtimeSession(kTestSessionName, 100 * 1024); | 171 HRESULT hr = controller.StartRealtimeSession(kTestSessionName, 100 * 1024); |
| (...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 201 EXPECT_EQ(kTestProviderFlags, provider.enable_flags()); | 203 EXPECT_EQ(kTestProviderFlags, provider.enable_flags()); |
| 202 | 204 |
| 203 EXPECT_HRESULT_SUCCEEDED(controller.Stop(NULL)); | 205 EXPECT_HRESULT_SUCCEEDED(controller.Stop(NULL)); |
| 204 | 206 |
| 205 provider.WaitForCallback(); | 207 provider.WaitForCallback(); |
| 206 | 208 |
| 207 // Session should have wound down. | 209 // Session should have wound down. |
| 208 EXPECT_EQ(0, provider.enable_level()); | 210 EXPECT_EQ(0, provider.enable_level()); |
| 209 EXPECT_EQ(0, provider.enable_flags()); | 211 EXPECT_EQ(0, provider.enable_flags()); |
| 210 } | 212 } |
| OLD | NEW |