OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 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 #include "base/trace_event/memory_dump_manager.h" | 5 #include "base/trace_event/memory_dump_manager.h" |
6 | 6 |
7 #include "base/trace_event/memory_dump_provider.h" | 7 #include "base/trace_event/memory_dump_provider.h" |
8 #include "base/trace_event/process_memory_dump.h" | 8 #include "base/trace_event/process_memory_dump.h" |
9 #include "testing/gmock/include/gmock/gmock.h" | 9 #include "testing/gmock/include/gmock/gmock.h" |
10 #include "testing/gtest/include/gtest/gtest.h" | 10 #include "testing/gtest/include/gtest/gtest.h" |
11 | 11 |
12 using testing::_; | 12 using testing::_; |
13 | 13 |
14 namespace base { | 14 namespace base { |
15 namespace trace_event { | 15 namespace trace_event { |
16 | 16 |
17 class MemoryDumpManagerTest : public testing::Test { | 17 class MemoryDumpManagerTest : public testing::Test { |
18 public: | 18 public: |
19 void SetUp() override { | 19 void SetUp() override { |
| 20 mdm_.reset(new MemoryDumpManager()); |
| 21 MemoryDumpManager::SetInstanceForTesting(mdm_.get()); |
| 22 ASSERT_EQ(mdm_, MemoryDumpManager::GetInstance()); |
20 MemoryDumpManager::GetInstance()->Initialize(); | 23 MemoryDumpManager::GetInstance()->Initialize(); |
21 mdm_ = MemoryDumpManager::GetInstance(); | |
22 } | 24 } |
23 | 25 |
24 void TearDown() override { | 26 void TearDown() override { |
25 MemoryDumpManager::DeleteForTesting(); | 27 MemoryDumpManager::SetInstanceForTesting(nullptr); |
| 28 mdm_.reset(); |
26 TraceLog::DeleteForTesting(); | 29 TraceLog::DeleteForTesting(); |
27 mdm_ = NULL; | |
28 } | 30 } |
29 | 31 |
30 protected: | 32 protected: |
31 const char* const kTraceCategory = MemoryDumpManager::kTraceCategory; | 33 const char* const kTraceCategory = MemoryDumpManager::kTraceCategory; |
32 | 34 |
33 void EnableTracing(const char* category) { | 35 void EnableTracing(const char* category) { |
34 TraceLog::GetInstance()->SetEnabled( | 36 TraceLog::GetInstance()->SetEnabled( |
35 CategoryFilter(category), TraceLog::RECORDING_MODE, TraceOptions()); | 37 CategoryFilter(category), TraceLog::RECORDING_MODE, TraceOptions()); |
36 } | 38 } |
37 | 39 |
38 void DisableTracing() { TraceLog::GetInstance()->SetDisabled(); } | 40 void DisableTracing() { TraceLog::GetInstance()->SetDisabled(); } |
39 | 41 |
40 MemoryDumpManager* mdm_; | 42 scoped_ptr<MemoryDumpManager> mdm_; |
41 | 43 |
42 private: | 44 private: |
43 // We want our singleton torn down after each test. | 45 // We want our singleton torn down after each test. |
44 ShadowingAtExitManager at_exit_manager_; | 46 ShadowingAtExitManager at_exit_manager_; |
45 }; | 47 }; |
46 | 48 |
47 class MockDumpProvider : public MemoryDumpProvider { | 49 class MockDumpProvider : public MemoryDumpProvider { |
48 public: | 50 public: |
49 MOCK_METHOD1(DumpInto, void(ProcessMemoryDump* pmd)); | 51 MOCK_METHOD1(DumpInto, void(ProcessMemoryDump* pmd)); |
50 }; | 52 }; |
(...skipping 65 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
116 mdm_->RegisterDumpProvider(&mdp1); | 118 mdm_->RegisterDumpProvider(&mdp1); |
117 EnableTracing(kTraceCategory); | 119 EnableTracing(kTraceCategory); |
118 EXPECT_CALL(mdp1, DumpInto(_)).Times(1); | 120 EXPECT_CALL(mdp1, DumpInto(_)).Times(1); |
119 EXPECT_CALL(mdp2, DumpInto(_)).Times(1); | 121 EXPECT_CALL(mdp2, DumpInto(_)).Times(1); |
120 mdm_->RequestDumpPoint(DumpPointType::EXPLICITLY_TRIGGERED); | 122 mdm_->RequestDumpPoint(DumpPointType::EXPLICITLY_TRIGGERED); |
121 DisableTracing(); | 123 DisableTracing(); |
122 } | 124 } |
123 | 125 |
124 } // namespace trace_Event | 126 } // namespace trace_Event |
125 } // namespace base | 127 } // namespace base |
OLD | NEW |