| 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 <stddef.h> | 5 #include <stddef.h> |
| 6 | 6 |
| 7 #include <iterator> | 7 #include <iterator> |
| 8 | 8 |
| 9 #include "base/memory/ref_counted.h" | 9 #include "base/memory/ref_counted.h" |
| 10 #include "base/trace_event/heap_profiler_allocation_context.h" | 10 #include "base/trace_event/heap_profiler_allocation_context.h" |
| 11 #include "base/trace_event/heap_profiler_allocation_context_tracker.h" | 11 #include "base/trace_event/heap_profiler_allocation_context_tracker.h" |
| 12 #include "base/trace_event/trace_event.h" | 12 #include "base/trace_event/trace_event.h" |
| 13 #include "testing/gtest/include/gtest/gtest.h" | 13 #include "testing/gtest/include/gtest/gtest.h" |
| 14 | 14 |
| 15 namespace base { | 15 namespace base { |
| 16 namespace trace_event { | 16 namespace trace_event { |
| 17 | 17 |
| 18 // Define all strings once, because the pseudo stack requires pointer equality, | 18 // Define all strings once, because the pseudo stack requires pointer equality, |
| 19 // and string interning is unreliable. | 19 // and string interning is unreliable. |
| 20 const char kCupcake[] = "Cupcake"; | 20 const char kCupcake[] = "Cupcake"; |
| 21 const char kDonut[] = "Donut"; | 21 const char kDonut[] = "Donut"; |
| 22 const char kEclair[] = "Eclair"; | 22 const char kEclair[] = "Eclair"; |
| 23 const char kFroyo[] = "Froyo"; | 23 const char kFroyo[] = "Froyo"; |
| 24 const char kGingerbread[] = "Gingerbread"; | 24 const char kGingerbread[] = "Gingerbread"; |
| 25 | 25 |
| 26 // Asserts that the fixed-size array |expected_backtrace| matches the backtrace | 26 // Asserts that the fixed-size array |expected_backtrace| matches the backtrace |
| 27 // in |AllocationContextTracker::GetContextSnapshot|. | 27 // in |AllocationContextTracker::GetContextSnapshot|. |
| 28 template <size_t N> | 28 template <size_t N> |
| 29 void AssertBacktraceEquals(const StackFrame(&expected_backtrace)[N]) { | 29 void AssertBacktraceEquals(const StackFrame(&expected_backtrace)[N]) { |
| 30 AllocationContext ctx = | 30 AllocationContext ctx = AllocationContextTracker::GetContextSnapshot(); |
| 31 AllocationContextTracker::GetInstanceForCurrentThread() | |
| 32 ->GetContextSnapshot(); | |
| 33 | 31 |
| 34 auto actual = std::begin(ctx.backtrace.frames); | 32 auto actual = std::begin(ctx.backtrace.frames); |
| 35 auto actual_bottom = std::end(ctx.backtrace.frames); | 33 auto actual_bottom = std::end(ctx.backtrace.frames); |
| 36 auto expected = std::begin(expected_backtrace); | 34 auto expected = std::begin(expected_backtrace); |
| 37 auto expected_bottom = std::end(expected_backtrace); | 35 auto expected_bottom = std::end(expected_backtrace); |
| 38 | 36 |
| 39 // Note that this requires the pointers to be equal, this is not doing a deep | 37 // Note that this requires the pointers to be equal, this is not doing a deep |
| 40 // string comparison. | 38 // string comparison. |
| 41 for (; actual != actual_bottom && expected != expected_bottom; | 39 for (; actual != actual_bottom && expected != expected_bottom; |
| 42 actual++, expected++) | 40 actual++, expected++) |
| 43 ASSERT_EQ(*expected, *actual); | 41 ASSERT_EQ(*expected, *actual); |
| 44 | 42 |
| 45 // Ensure that the height of the stacks is the same. | 43 // Ensure that the height of the stacks is the same. |
| 46 ASSERT_EQ(actual, actual_bottom); | 44 ASSERT_EQ(actual, actual_bottom); |
| 47 ASSERT_EQ(expected, expected_bottom); | 45 ASSERT_EQ(expected, expected_bottom); |
| 48 } | 46 } |
| 49 | 47 |
| 50 void AssertBacktraceEmpty() { | 48 void AssertBacktraceEmpty() { |
| 51 AllocationContext ctx = | 49 AllocationContext ctx = AllocationContextTracker::GetContextSnapshot(); |
| 52 AllocationContextTracker::GetInstanceForCurrentThread() | |
| 53 ->GetContextSnapshot(); | |
| 54 | 50 |
| 55 for (StackFrame frame : ctx.backtrace.frames) | 51 for (StackFrame frame : ctx.backtrace.frames) |
| 56 ASSERT_EQ(nullptr, frame); | 52 ASSERT_EQ(nullptr, frame); |
| 57 } | 53 } |
| 58 | 54 |
| 59 class AllocationContextTrackerTest : public testing::Test { | 55 class AllocationContextTrackerTest : public testing::Test { |
| 60 public: | 56 public: |
| 61 void SetUp() override { | 57 void SetUp() override { |
| 62 TraceConfig config(""); | 58 TraceConfig config(""); |
| 63 TraceLog::GetInstance()->SetEnabled(config, TraceLog::RECORDING_MODE); | 59 TraceLog::GetInstance()->SetEnabled(config, TraceLog::RECORDING_MODE); |
| (...skipping 140 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 204 TRACE_EVENT0("Testing", kCupcake); | 200 TRACE_EVENT0("Testing", kCupcake); |
| 205 TRACE_EVENT0("Testing", kCupcake); | 201 TRACE_EVENT0("Testing", kCupcake); |
| 206 | 202 |
| 207 TRACE_EVENT0("Testing", kCupcake); | 203 TRACE_EVENT0("Testing", kCupcake); |
| 208 TRACE_EVENT0("Testing", kDonut); | 204 TRACE_EVENT0("Testing", kDonut); |
| 209 TRACE_EVENT0("Testing", kEclair); | 205 TRACE_EVENT0("Testing", kEclair); |
| 210 TRACE_EVENT0("Testing", kFroyo); | 206 TRACE_EVENT0("Testing", kFroyo); |
| 211 | 207 |
| 212 { | 208 { |
| 213 TRACE_EVENT0("Testing", kGingerbread); | 209 TRACE_EVENT0("Testing", kGingerbread); |
| 214 AllocationContext ctx = | 210 AllocationContext ctx = AllocationContextTracker::GetContextSnapshot(); |
| 215 AllocationContextTracker::GetInstanceForCurrentThread() | |
| 216 ->GetContextSnapshot(); | |
| 217 | 211 |
| 218 // The pseudo stack relies on pointer equality, not deep string comparisons. | 212 // The pseudo stack relies on pointer equality, not deep string comparisons. |
| 219 ASSERT_EQ(kCupcake, ctx.backtrace.frames[0]); | 213 ASSERT_EQ(kCupcake, ctx.backtrace.frames[0]); |
| 220 ASSERT_EQ(kFroyo, ctx.backtrace.frames[11]); | 214 ASSERT_EQ(kFroyo, ctx.backtrace.frames[11]); |
| 221 } | 215 } |
| 222 | 216 |
| 223 { | 217 { |
| 224 AllocationContext ctx = | 218 AllocationContext ctx = AllocationContextTracker::GetContextSnapshot(); |
| 225 AllocationContextTracker::GetInstanceForCurrentThread() | |
| 226 ->GetContextSnapshot(); | |
| 227 ASSERT_EQ(kCupcake, ctx.backtrace.frames[0]); | 219 ASSERT_EQ(kCupcake, ctx.backtrace.frames[0]); |
| 228 ASSERT_EQ(kFroyo, ctx.backtrace.frames[11]); | 220 ASSERT_EQ(kFroyo, ctx.backtrace.frames[11]); |
| 229 } | 221 } |
| 230 } | 222 } |
| 231 | 223 |
| 232 } // namespace trace_event | 224 } // namespace trace_event |
| 233 } // namespace base | 225 } // namespace base |
| OLD | NEW |