OLD | NEW |
| (Empty) |
1 // Copyright 2017 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "base/trace_event/memory_dump_scheduler.h" | |
6 | |
7 #include <memory> | |
8 | |
9 #include "base/single_thread_task_runner.h" | |
10 #include "testing/gtest/include/gtest/gtest.h" | |
11 | |
12 namespace base { | |
13 namespace trace_event { | |
14 | |
15 class MemoryDumpSchedulerPollingTest : public testing::Test { | |
16 public: | |
17 static const uint32_t kMinPollsToDump = 5; | |
18 | |
19 MemoryDumpSchedulerPollingTest() | |
20 : testing::Test(), | |
21 num_samples_tracked_( | |
22 MemoryDumpScheduler::PollingTriggerState::kMaxNumMemorySamples) {} | |
23 | |
24 void SetUp() override { | |
25 MemoryDumpScheduler::SetPollingIntervalForTesting(1); | |
26 uint32_t kMinPollsToDump = 5; | |
27 mds_.reset(new MemoryDumpScheduler(nullptr, nullptr)); | |
28 mds_->AddTrigger(MemoryDumpType::PEAK_MEMORY_USAGE, | |
29 MemoryDumpLevelOfDetail::LIGHT, kMinPollsToDump); | |
30 mds_->polling_state_.ResetTotals(); | |
31 mds_->polling_state_.current_state = | |
32 MemoryDumpScheduler::PollingTriggerState::ENABLED; | |
33 } | |
34 | |
35 void TearDown() override { | |
36 mds_->polling_state_.current_state = | |
37 MemoryDumpScheduler::PollingTriggerState::DISABLED; | |
38 mds_.reset(); | |
39 } | |
40 | |
41 protected: | |
42 bool ShouldTriggerDump(uint64_t total) { | |
43 return mds_->ShouldTriggerDump(total); | |
44 } | |
45 | |
46 uint32_t num_samples_tracked_; | |
47 std::unique_ptr<MemoryDumpScheduler> mds_; | |
48 }; | |
49 | |
50 TEST_F(MemoryDumpSchedulerPollingTest, PeakDetection) { | |
51 for (uint32_t i = 0; i < num_samples_tracked_ * 6; ++i) { | |
52 // Memory is increased in steps and dumps must be triggered at every step. | |
53 uint64_t total = (2 + (i / (2 * num_samples_tracked_))) * 1024 * 1204; | |
54 bool did_trigger = ShouldTriggerDump(total); | |
55 // Dumps must be triggered only at specific iterations. | |
56 bool should_have_triggered = i == 0; | |
57 should_have_triggered |= | |
58 (i > num_samples_tracked_) && (i % (2 * num_samples_tracked_) == 1); | |
59 if (should_have_triggered) { | |
60 ASSERT_TRUE(did_trigger) << "Dump wasn't triggered at " << i; | |
61 } else { | |
62 ASSERT_FALSE(did_trigger) << "Unexpected dump at " << i; | |
63 } | |
64 } | |
65 } | |
66 | |
67 TEST_F(MemoryDumpSchedulerPollingTest, SlowGrowthDetection) { | |
68 for (uint32_t i = 0; i < 15; ++i) { | |
69 // Record 1GiB of increase in each call. Dumps are triggered with 1% w.r.t | |
70 // system's total memory. | |
71 uint64_t total = static_cast<uint64_t>(i + 1) * 1024 * 1024 * 1024; | |
72 bool did_trigger = ShouldTriggerDump(total); | |
73 bool should_have_triggered = i % kMinPollsToDump == 0; | |
74 if (should_have_triggered) { | |
75 ASSERT_TRUE(did_trigger) << "Dump wasn't triggered at " << i; | |
76 } else { | |
77 ASSERT_FALSE(did_trigger) << "Unexpected dump at " << i; | |
78 } | |
79 } | |
80 } | |
81 | |
82 TEST_F(MemoryDumpSchedulerPollingTest, NotifyDumpTriggered) { | |
83 for (uint32_t i = 0; i < num_samples_tracked_ * 6; ++i) { | |
84 uint64_t total = (2 + (i / (2 * num_samples_tracked_))) * 1024 * 1204; | |
85 if (i % num_samples_tracked_ == 0) | |
86 mds_->NotifyDumpTriggered(); | |
87 bool did_trigger = ShouldTriggerDump(total); | |
88 // Dumps should never be triggered since NotifyDumpTriggered() is called | |
89 // frequently. | |
90 ASSERT_FALSE(did_trigger && i); | |
91 } | |
92 } | |
93 | |
94 } // namespace trace_event | |
95 } // namespace base | |
OLD | NEW |