Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(602)

Side by Side Diff: components/offline_pages/background/mark_attempt_started_task_unittest.cc

Issue 2489443002: Move all components/offline_pages/ files into component/offline_pages/core (Closed)
Patch Set: rebase Created 4 years ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
(Empty)
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
3 // found in the LICENSE file.
4
5 #include "components/offline_pages/background/mark_attempt_started_task.h"
6
7 #include <memory>
8
9 #include "base/bind.h"
10 #include "base/test/test_simple_task_runner.h"
11 #include "base/threading/thread_task_runner_handle.h"
12 #include "components/offline_pages/background/request_queue_in_memory_store.h"
13 #include "testing/gtest/include/gtest/gtest.h"
14
15 namespace offline_pages {
16 namespace {
17 const int64_t kRequestId1 = 42;
18 const int64_t kRequestId2 = 44;
19 const GURL kUrl1("http://example.com");
20 const ClientId kClientId1("download", "1234");
21 } // namespace
22
23 class MarkAttemptStartedTaskTest : public testing::Test {
24 public:
25 MarkAttemptStartedTaskTest();
26 ~MarkAttemptStartedTaskTest() override;
27
28 void PumpLoop();
29
30 void InitializeStore(RequestQueueStore* store);
31 void AddItemToStore(RequestQueueStore* store);
32 void ChangeRequestsStateCallback(
33 std::unique_ptr<UpdateRequestsResult> result);
34
35 UpdateRequestsResult* last_result() const { return result_.get(); }
36
37 private:
38 void InitializeStoreDone(bool success);
39 void AddRequestDone(ItemActionStatus status);
40
41 std::unique_ptr<UpdateRequestsResult> result_;
42 scoped_refptr<base::TestSimpleTaskRunner> task_runner_;
43 base::ThreadTaskRunnerHandle task_runner_handle_;
44 };
45
46 MarkAttemptStartedTaskTest::MarkAttemptStartedTaskTest()
47 : task_runner_(new base::TestSimpleTaskRunner),
48 task_runner_handle_(task_runner_) {}
49
50 MarkAttemptStartedTaskTest::~MarkAttemptStartedTaskTest() {}
51
52 void MarkAttemptStartedTaskTest::PumpLoop() {
53 task_runner_->RunUntilIdle();
54 }
55
56 void MarkAttemptStartedTaskTest::InitializeStore(RequestQueueStore* store) {
57 store->Initialize(base::Bind(&MarkAttemptStartedTaskTest::InitializeStoreDone,
58 base::Unretained(this)));
59 PumpLoop();
60 }
61
62 void MarkAttemptStartedTaskTest::AddItemToStore(RequestQueueStore* store) {
63 base::Time creation_time = base::Time::Now();
64 SavePageRequest request_1(kRequestId1, kUrl1, kClientId1, creation_time,
65 true);
66 store->AddRequest(request_1,
67 base::Bind(&MarkAttemptStartedTaskTest::AddRequestDone,
68 base::Unretained(this)));
69 PumpLoop();
70 }
71
72 void MarkAttemptStartedTaskTest::ChangeRequestsStateCallback(
73 std::unique_ptr<UpdateRequestsResult> result) {
74 result_ = std::move(result);
75 }
76
77 void MarkAttemptStartedTaskTest::InitializeStoreDone(bool success) {
78 ASSERT_TRUE(success);
79 }
80
81 void MarkAttemptStartedTaskTest::AddRequestDone(ItemActionStatus status) {
82 ASSERT_EQ(ItemActionStatus::SUCCESS, status);
83 }
84
85 TEST_F(MarkAttemptStartedTaskTest, MarkAttemptStartedWhenStoreEmpty) {
86 RequestQueueInMemoryStore store;
87 InitializeStore(&store);
88
89 MarkAttemptStartedTask task(
90 &store, kRequestId1,
91 base::Bind(&MarkAttemptStartedTaskTest::ChangeRequestsStateCallback,
92 base::Unretained(this)));
93 task.Run();
94 PumpLoop();
95 ASSERT_TRUE(last_result());
96 EXPECT_EQ(1UL, last_result()->item_statuses.size());
97 EXPECT_EQ(kRequestId1, last_result()->item_statuses.at(0).first);
98 EXPECT_EQ(ItemActionStatus::NOT_FOUND,
99 last_result()->item_statuses.at(0).second);
100 EXPECT_EQ(0UL, last_result()->updated_items.size());
101 }
102
103 TEST_F(MarkAttemptStartedTaskTest, MarkAttemptStartedWhenExists) {
104 RequestQueueInMemoryStore store;
105 InitializeStore(&store);
106 AddItemToStore(&store);
107
108 MarkAttemptStartedTask task(
109 &store, kRequestId1,
110 base::Bind(&MarkAttemptStartedTaskTest::ChangeRequestsStateCallback,
111 base::Unretained(this)));
112
113 // Current time for verification.
114 base::Time before_time = base::Time::Now();
115 task.Run();
116 PumpLoop();
117 ASSERT_TRUE(last_result());
118 EXPECT_EQ(1UL, last_result()->item_statuses.size());
119 EXPECT_EQ(kRequestId1, last_result()->item_statuses.at(0).first);
120 EXPECT_EQ(ItemActionStatus::SUCCESS,
121 last_result()->item_statuses.at(0).second);
122 EXPECT_EQ(1UL, last_result()->updated_items.size());
123 EXPECT_LE(before_time,
124 last_result()->updated_items.at(0).last_attempt_time());
125 EXPECT_GE(base::Time::Now(),
126 last_result()->updated_items.at(0).last_attempt_time());
127 EXPECT_EQ(1, last_result()->updated_items.at(0).started_attempt_count());
128 EXPECT_EQ(SavePageRequest::RequestState::OFFLINING,
129 last_result()->updated_items.at(0).request_state());
130 }
131
132 TEST_F(MarkAttemptStartedTaskTest, MarkAttemptStartedWhenItemMissing) {
133 RequestQueueInMemoryStore store;
134 InitializeStore(&store);
135 AddItemToStore(&store);
136
137 MarkAttemptStartedTask task(
138 &store, kRequestId2,
139 base::Bind(&MarkAttemptStartedTaskTest::ChangeRequestsStateCallback,
140 base::Unretained(this)));
141 task.Run();
142 PumpLoop();
143 ASSERT_TRUE(last_result());
144 EXPECT_EQ(1UL, last_result()->item_statuses.size());
145 EXPECT_EQ(kRequestId2, last_result()->item_statuses.at(0).first);
146 EXPECT_EQ(ItemActionStatus::NOT_FOUND,
147 last_result()->item_statuses.at(0).second);
148 EXPECT_EQ(0UL, last_result()->updated_items.size());
149 }
150
151 } // namespace offline_pages
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698