| 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 "chrome/browser/android/offline_pages/prerendering_loader.h" | 10 #include "chrome/browser/android/offline_pages/prerendering_loader.h" |
| (...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 51 | 51 |
| 52 void SetUp() override; | 52 void SetUp() override; |
| 53 | 53 |
| 54 PrerenderingOffliner* offliner() const { return offliner_.get(); } | 54 PrerenderingOffliner* offliner() const { return offliner_.get(); } |
| 55 Offliner::CompletionCallback const callback() { | 55 Offliner::CompletionCallback const callback() { |
| 56 return base::Bind(&PrerenderingOfflinerTest::OnCompletion, | 56 return base::Bind(&PrerenderingOfflinerTest::OnCompletion, |
| 57 base::Unretained(this)); | 57 base::Unretained(this)); |
| 58 } | 58 } |
| 59 | 59 |
| 60 bool loading() const { return loader_->mock_loading(); } | 60 bool loading() const { return loader_->mock_loading(); } |
| 61 Offliner::CompletionStatus completion_status() { | 61 Offliner::RequestStatus completion_status() { return completion_status_; } |
| 62 return completion_status_; | |
| 63 } | |
| 64 | 62 |
| 65 private: | 63 private: |
| 66 void OnCompletion(const SavePageRequest& request, | 64 void OnCompletion(const SavePageRequest& request, |
| 67 Offliner::CompletionStatus status); | 65 Offliner::RequestStatus status); |
| 68 | 66 |
| 69 std::unique_ptr<PrerenderingOffliner> offliner_; | 67 std::unique_ptr<PrerenderingOffliner> offliner_; |
| 70 // Not owned. | 68 // Not owned. |
| 71 MockPrerenderingLoader* loader_; | 69 MockPrerenderingLoader* loader_; |
| 72 Offliner::CompletionStatus completion_status_; | 70 Offliner::RequestStatus completion_status_; |
| 73 | 71 |
| 74 DISALLOW_COPY_AND_ASSIGN(PrerenderingOfflinerTest); | 72 DISALLOW_COPY_AND_ASSIGN(PrerenderingOfflinerTest); |
| 75 }; | 73 }; |
| 76 | 74 |
| 77 PrerenderingOfflinerTest::PrerenderingOfflinerTest() {} | 75 PrerenderingOfflinerTest::PrerenderingOfflinerTest() {} |
| 78 | 76 |
| 79 PrerenderingOfflinerTest::~PrerenderingOfflinerTest() {} | 77 PrerenderingOfflinerTest::~PrerenderingOfflinerTest() {} |
| 80 | 78 |
| 81 void PrerenderingOfflinerTest::SetUp() { | 79 void PrerenderingOfflinerTest::SetUp() { |
| 82 offliner_.reset(new PrerenderingOffliner(nullptr, nullptr, nullptr)); | 80 offliner_.reset(new PrerenderingOffliner(nullptr, nullptr, nullptr)); |
| 83 std::unique_ptr<MockPrerenderingLoader> mock_loader( | 81 std::unique_ptr<MockPrerenderingLoader> mock_loader( |
| 84 new MockPrerenderingLoader(nullptr)); | 82 new MockPrerenderingLoader(nullptr)); |
| 85 loader_ = mock_loader.get(); | 83 loader_ = mock_loader.get(); |
| 86 offliner_->SetLoaderForTesting(std::move(mock_loader)); | 84 offliner_->SetLoaderForTesting(std::move(mock_loader)); |
| 87 } | 85 } |
| 88 | 86 |
| 89 void PrerenderingOfflinerTest::OnCompletion(const SavePageRequest& request, | 87 void PrerenderingOfflinerTest::OnCompletion(const SavePageRequest& request, |
| 90 Offliner::CompletionStatus status) { | 88 Offliner::RequestStatus status) { |
| 91 completion_status_ = status; | 89 completion_status_ = status; |
| 92 } | 90 } |
| 93 | 91 |
| 94 // Tests initiate loading. | 92 // Tests initiate loading. |
| 95 TEST_F(PrerenderingOfflinerTest, LoadAndSaveBadUrl) { | 93 TEST_F(PrerenderingOfflinerTest, LoadAndSaveBadUrl) { |
| 96 base::Time creation_time = base::Time::Now(); | 94 base::Time creation_time = base::Time::Now(); |
| 97 SavePageRequest request(kRequestId, kFileUrl, kClientId, creation_time); | 95 SavePageRequest request(kRequestId, kFileUrl, kClientId, creation_time); |
| 98 EXPECT_FALSE(offliner()->LoadAndSave(request, callback())); | 96 EXPECT_FALSE(offliner()->LoadAndSave(request, callback())); |
| 99 EXPECT_FALSE(loading()); | 97 EXPECT_FALSE(loading()); |
| 100 } | 98 } |
| (...skipping 11 matching lines...) Expand all Loading... |
| 112 base::Time creation_time = base::Time::Now(); | 110 base::Time creation_time = base::Time::Now(); |
| 113 SavePageRequest request(kRequestId, kHttpUrl, kClientId, creation_time); | 111 SavePageRequest request(kRequestId, kHttpUrl, kClientId, creation_time); |
| 114 EXPECT_TRUE(offliner()->LoadAndSave(request, callback())); | 112 EXPECT_TRUE(offliner()->LoadAndSave(request, callback())); |
| 115 EXPECT_TRUE(loading()); | 113 EXPECT_TRUE(loading()); |
| 116 | 114 |
| 117 offliner()->Cancel(); | 115 offliner()->Cancel(); |
| 118 EXPECT_FALSE(loading()); | 116 EXPECT_FALSE(loading()); |
| 119 } | 117 } |
| 120 | 118 |
| 121 } // namespace offline_pages | 119 } // namespace offline_pages |
| OLD | NEW |