OLD | NEW |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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 #import "ios/chrome/app/application_delegate/memory_warning_helper.h" | 5 #import "ios/chrome/app/application_delegate/memory_warning_helper.h" |
6 | 6 |
7 #include "base/mac/bind_objc_block.h" | 7 #include "base/mac/bind_objc_block.h" |
8 #import "base/mac/scoped_nsobject.h" | |
9 #include "base/memory/memory_pressure_listener.h" | 8 #include "base/memory/memory_pressure_listener.h" |
10 #include "base/message_loop/message_loop.h" | 9 #include "base/message_loop/message_loop.h" |
11 #include "base/run_loop.h" | 10 #include "base/run_loop.h" |
12 #include "base/threading/thread.h" | 11 #include "base/threading/thread.h" |
13 #import "ios/chrome/browser/metrics/previous_session_info.h" | 12 #import "ios/chrome/browser/metrics/previous_session_info.h" |
14 #include "testing/platform_test.h" | 13 #include "testing/platform_test.h" |
15 | 14 |
| 15 #if !defined(__has_feature) || !__has_feature(objc_arc) |
| 16 #error "This file requires ARC support." |
| 17 #endif |
| 18 |
16 using previous_session_info_constants:: | 19 using previous_session_info_constants:: |
17 kDidSeeMemoryWarningShortlyBeforeTerminating; | 20 kDidSeeMemoryWarningShortlyBeforeTerminating; |
18 | 21 |
19 class MemoryWarningHelperTest : public PlatformTest { | 22 class MemoryWarningHelperTest : public PlatformTest { |
20 protected: | 23 protected: |
21 MemoryWarningHelperTest() { | 24 MemoryWarningHelperTest() { |
22 // Set up |memory_pressure_listener_| to invoke |OnMemoryPressure| which | 25 // Set up |memory_pressure_listener_| to invoke |OnMemoryPressure| which |
23 // will store the memory pressure level sent to the callback in | 26 // will store the memory pressure level sent to the callback in |
24 // |memory_pressure_level_| so that tests can verify the level is correct. | 27 // |memory_pressure_level_| so that tests can verify the level is correct. |
25 memory_pressure_listener_.reset(new base::MemoryPressureListener(base::Bind( | 28 memory_pressure_listener_.reset(new base::MemoryPressureListener(base::Bind( |
26 &MemoryWarningHelperTest::OnMemoryPressure, base::Unretained(this)))); | 29 &MemoryWarningHelperTest::OnMemoryPressure, base::Unretained(this)))); |
27 memory_pressure_level_ = | 30 memory_pressure_level_ = |
28 base::MemoryPressureListener::MEMORY_PRESSURE_LEVEL_MODERATE; | 31 base::MemoryPressureListener::MEMORY_PRESSURE_LEVEL_MODERATE; |
29 } | 32 } |
30 | 33 |
31 base::MemoryPressureListener::MemoryPressureLevel GetMemoryPressureLevel() { | 34 base::MemoryPressureListener::MemoryPressureLevel GetMemoryPressureLevel() { |
32 return memory_pressure_level_; | 35 return memory_pressure_level_; |
33 } | 36 } |
34 | 37 |
35 MemoryWarningHelper* GetMemoryHelper() { | 38 MemoryWarningHelper* GetMemoryHelper() { |
36 if (!memory_helper_) { | 39 if (!memory_helper_) { |
37 memory_helper_.reset([[MemoryWarningHelper alloc] init]); | 40 memory_helper_ = [[MemoryWarningHelper alloc] init]; |
38 } | 41 } |
39 return memory_helper_; | 42 return memory_helper_; |
40 } | 43 } |
41 | 44 |
42 // Callback for |memory_pressure_listener_|. | 45 // Callback for |memory_pressure_listener_|. |
43 void OnMemoryPressure( | 46 void OnMemoryPressure( |
44 base::MemoryPressureListener::MemoryPressureLevel memory_pressure_level) { | 47 base::MemoryPressureListener::MemoryPressureLevel memory_pressure_level) { |
45 memory_pressure_level_ = memory_pressure_level; | 48 memory_pressure_level_ = memory_pressure_level; |
46 run_loop_.QuitWhenIdle(); | 49 run_loop_.QuitWhenIdle(); |
47 } | 50 } |
48 | 51 |
49 void RunMessageLoop() { run_loop_.Run(); } | 52 void RunMessageLoop() { run_loop_.Run(); } |
50 | 53 |
51 private: | 54 private: |
52 base::MessageLoop message_loop_; | 55 base::MessageLoop message_loop_; |
53 base::RunLoop run_loop_; | 56 base::RunLoop run_loop_; |
54 base::MemoryPressureListener::MemoryPressureLevel memory_pressure_level_; | 57 base::MemoryPressureListener::MemoryPressureLevel memory_pressure_level_; |
55 std::unique_ptr<base::MemoryPressureListener> memory_pressure_listener_; | 58 std::unique_ptr<base::MemoryPressureListener> memory_pressure_listener_; |
56 base::scoped_nsobject<MemoryWarningHelper> memory_helper_; | 59 MemoryWarningHelper* memory_helper_; |
57 | 60 |
58 DISALLOW_COPY_AND_ASSIGN(MemoryWarningHelperTest); | 61 DISALLOW_COPY_AND_ASSIGN(MemoryWarningHelperTest); |
59 }; | 62 }; |
60 | 63 |
61 // Invokes resetForegroundMemoryWarningCount and verifies the | 64 // Invokes resetForegroundMemoryWarningCount and verifies the |
62 // foregroundMemoryWarningCount is setted to 0. | 65 // foregroundMemoryWarningCount is setted to 0. |
63 TEST_F(MemoryWarningHelperTest, VerifyForegroundMemoryWarningCountReset) { | 66 TEST_F(MemoryWarningHelperTest, VerifyForegroundMemoryWarningCountReset) { |
64 // Setup. | 67 // Setup. |
65 [GetMemoryHelper() handleMemoryPressure]; | 68 [GetMemoryHelper() handleMemoryPressure]; |
66 ASSERT_TRUE(GetMemoryHelper().foregroundMemoryWarningCount != 0); | 69 ASSERT_TRUE(GetMemoryHelper().foregroundMemoryWarningCount != 0); |
(...skipping 29 matching lines...) Expand all Loading... |
96 // Action. | 99 // Action. |
97 [GetMemoryHelper() handleMemoryPressure]; | 100 [GetMemoryHelper() handleMemoryPressure]; |
98 | 101 |
99 // Tests. | 102 // Tests. |
100 EXPECT_TRUE([[NSUserDefaults standardUserDefaults] | 103 EXPECT_TRUE([[NSUserDefaults standardUserDefaults] |
101 boolForKey:kDidSeeMemoryWarningShortlyBeforeTerminating]); | 104 boolForKey:kDidSeeMemoryWarningShortlyBeforeTerminating]); |
102 EXPECT_FALSE(memoryWarningFlagBeforeAlert); | 105 EXPECT_FALSE(memoryWarningFlagBeforeAlert); |
103 EXPECT_EQ(foregroundMemoryWarningCountBeforeWarning + 1, | 106 EXPECT_EQ(foregroundMemoryWarningCountBeforeWarning + 1, |
104 GetMemoryHelper().foregroundMemoryWarningCount); | 107 GetMemoryHelper().foregroundMemoryWarningCount); |
105 } | 108 } |
OLD | NEW |