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 #include "chrome/browser/android/offline_pages/prerendering_offliner.h" | 5 #include "chrome/browser/android/offline_pages/prerendering_offliner.h" |
6 | 6 |
7 #include <memory> | 7 #include <memory> |
8 | 8 |
9 #include "base/bind.h" | 9 #include "base/bind.h" |
10 #include "base/run_loop.h" | 10 #include "base/run_loop.h" |
(...skipping 24 matching lines...) Expand all Loading... |
35 // Mock Loader for testing the Offliner calls. | 35 // Mock Loader for testing the Offliner calls. |
36 class MockPrerenderingLoader : public PrerenderingLoader { | 36 class MockPrerenderingLoader : public PrerenderingLoader { |
37 public: | 37 public: |
38 explicit MockPrerenderingLoader(content::BrowserContext* browser_context) | 38 explicit MockPrerenderingLoader(content::BrowserContext* browser_context) |
39 : PrerenderingLoader(browser_context), | 39 : PrerenderingLoader(browser_context), |
40 can_prerender_(true), | 40 can_prerender_(true), |
41 mock_loading_(false), | 41 mock_loading_(false), |
42 mock_loaded_(false) {} | 42 mock_loaded_(false) {} |
43 ~MockPrerenderingLoader() override {} | 43 ~MockPrerenderingLoader() override {} |
44 | 44 |
45 bool LoadPage(const GURL& url, const LoadPageCallback& callback) override { | 45 bool LoadPage(const GURL& url, |
| 46 const LoadPageCallback& load_done_callback, |
| 47 const ProgressCallback& progress_callback) override { |
46 mock_loading_ = can_prerender_; | 48 mock_loading_ = can_prerender_; |
47 load_page_callback_ = callback; | 49 load_page_callback_ = load_done_callback; |
| 50 progress_callback_ = progress_callback; |
48 return mock_loading_; | 51 return mock_loading_; |
49 } | 52 } |
50 | 53 |
51 void StopLoading() override { | 54 void StopLoading() override { |
52 mock_loading_ = false; | 55 mock_loading_ = false; |
53 mock_loaded_ = false; | 56 mock_loaded_ = false; |
54 } | 57 } |
55 | 58 |
56 bool IsIdle() override { return !mock_loading_ && !mock_loaded_; } | 59 bool IsIdle() override { return !mock_loading_ && !mock_loaded_; } |
57 bool IsLoaded() override { return mock_loaded_; } | 60 bool IsLoaded() override { return mock_loaded_; } |
(...skipping 30 matching lines...) Expand all Loading... |
88 } | 91 } |
89 | 92 |
90 void DisablePrerendering() { can_prerender_ = false; } | 93 void DisablePrerendering() { can_prerender_ = false; } |
91 const LoadPageCallback& load_page_callback() { return load_page_callback_; } | 94 const LoadPageCallback& load_page_callback() { return load_page_callback_; } |
92 | 95 |
93 private: | 96 private: |
94 bool can_prerender_; | 97 bool can_prerender_; |
95 bool mock_loading_; | 98 bool mock_loading_; |
96 bool mock_loaded_; | 99 bool mock_loaded_; |
97 LoadPageCallback load_page_callback_; | 100 LoadPageCallback load_page_callback_; |
| 101 ProgressCallback progress_callback_; |
98 | 102 |
99 DISALLOW_COPY_AND_ASSIGN(MockPrerenderingLoader); | 103 DISALLOW_COPY_AND_ASSIGN(MockPrerenderingLoader); |
100 }; | 104 }; |
101 | 105 |
102 // Mock OfflinePageModel for testing the SavePage calls. | 106 // Mock OfflinePageModel for testing the SavePage calls. |
103 class MockOfflinePageModel : public StubOfflinePageModel { | 107 class MockOfflinePageModel : public StubOfflinePageModel { |
104 public: | 108 public: |
105 MockOfflinePageModel() : mock_saving_(false) {} | 109 MockOfflinePageModel() : mock_saving_(false) {} |
106 ~MockOfflinePageModel() override {} | 110 ~MockOfflinePageModel() override {} |
107 | 111 |
(...skipping 285 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
393 EXPECT_FALSE(loader()->IsIdle()); | 397 EXPECT_FALSE(loader()->IsIdle()); |
394 | 398 |
395 offliner()->SetApplicationStateForTesting( | 399 offliner()->SetApplicationStateForTesting( |
396 base::android::APPLICATION_STATE_HAS_RUNNING_ACTIVITIES); | 400 base::android::APPLICATION_STATE_HAS_RUNNING_ACTIVITIES); |
397 | 401 |
398 // Loader still loading since not low-end device. | 402 // Loader still loading since not low-end device. |
399 EXPECT_FALSE(loader()->IsIdle()); | 403 EXPECT_FALSE(loader()->IsIdle()); |
400 } | 404 } |
401 | 405 |
402 } // namespace offline_pages | 406 } // namespace offline_pages |
OLD | NEW |