components/offline_pages/background/mark_attempt_aborted_task_unittest.cc - Issue 2489443002: Move all components/offline_pages/ files into component/offline_pages/core

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

Issue 2489443002: Move all components/offline_pages/ files into component/offline_pages/core (Closed)

Patch Set: update Created 4 years, 1 month ago

Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.

Jump to:

« components/BUILD.gn ('K') | « components/offline_pages/background/mark_attempt_aborted_task.cc ('k') | components/offline_pages/background/mark_attempt_completed_task.h » ('j') | components/offline_pages/core/DEPS » ('J')
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Hide Comments ('s')

OLD	NEW
	(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_aborted_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/mark_attempt_started_task.h"

13 #include "components/offline_pages/background/request_queue_in_memory_store.h"

14 #include "testing/gtest/include/gtest/gtest.h"

15

16 namespace offline_pages {

17 namespace {

18 const int64_t kRequestId1 = 42;

19 const int64_t kRequestId2 = 44;

20 const GURL kUrl1("http://example.com");

21 const ClientId kClientId1("download", "1234");

22 } // namespace

23

24 class MarkAttemptAbortedTaskTest : public testing::Test {

25 public:

26 MarkAttemptAbortedTaskTest();

27 ~MarkAttemptAbortedTaskTest() override;

28

29 void PumpLoop();

30

31 void AddItemToStore(RequestQueueStore* store);

32

33 void AddRequestDone(ItemActionStatus status);

34

35 void ChangeRequestsStateCallback(

36 std::unique_ptr<UpdateRequestsResult> result);

37

38 void ClearResults();

39

40 UpdateRequestsResult* last_result() const { return result_.get(); }

41

42 private:

43 std::unique_ptr<UpdateRequestsResult> result_;

44 scoped_refptr<base::TestSimpleTaskRunner> task_runner_;

45 base::ThreadTaskRunnerHandle task_runner_handle_;

46 };

47

48 MarkAttemptAbortedTaskTest::MarkAttemptAbortedTaskTest()

49 : task_runner_(new base::TestSimpleTaskRunner),

50 task_runner_handle_(task_runner_) {}

51

52 MarkAttemptAbortedTaskTest::~MarkAttemptAbortedTaskTest() {}

53

54 void MarkAttemptAbortedTaskTest::PumpLoop() {

55 task_runner_->RunUntilIdle();

56 }

57

58 void MarkAttemptAbortedTaskTest::AddItemToStore(RequestQueueStore* store) {

59 base::Time creation_time = base::Time::Now();

60 SavePageRequest request_1(kRequestId1, kUrl1, kClientId1, creation_time,

61 true);

62 store->AddRequest(request_1,

63 base::Bind(&MarkAttemptAbortedTaskTest::AddRequestDone,

64 base::Unretained(this)));

65 PumpLoop();

66 }

67

68 void MarkAttemptAbortedTaskTest::AddRequestDone(ItemActionStatus status) {

69 ASSERT_EQ(ItemActionStatus::SUCCESS, status);

70 }

71

72 void MarkAttemptAbortedTaskTest::ChangeRequestsStateCallback(

73 std::unique_ptr<UpdateRequestsResult> result) {

74 result_ = std::move(result);

75 }

76

77 void MarkAttemptAbortedTaskTest::ClearResults() {

78 result_.reset(nullptr);

79 }

80

81 TEST_F(MarkAttemptAbortedTaskTest, MarkAttemptAbortedWhenStoreEmpty) {

82 RequestQueueInMemoryStore store;

83 MarkAttemptAbortedTask task(

84 &store, kRequestId1,

85 base::Bind(&MarkAttemptAbortedTaskTest::ChangeRequestsStateCallback,

86 base::Unretained(this)));

87 task.Run();

88 PumpLoop();

89 ASSERT_TRUE(last_result());

90 EXPECT_EQ(1UL, last_result()->item_statuses.size());

91 EXPECT_EQ(kRequestId1, last_result()->item_statuses.at(0).first);

92 EXPECT_EQ(ItemActionStatus::NOT_FOUND,

93 last_result()->item_statuses.at(0).second);

94 EXPECT_EQ(0UL, last_result()->updated_items.size());

95 }

96

97 TEST_F(MarkAttemptAbortedTaskTest, MarkAttemptAbortedWhenExists) {

98 RequestQueueInMemoryStore store;

99 AddItemToStore(&store);

100

101 // First mark attempt started.

102 MarkAttemptStartedTask start_request_task(

103 &store, kRequestId1,

104 base::Bind(&MarkAttemptAbortedTaskTest::ChangeRequestsStateCallback,

105 base::Unretained(this)));

106 start_request_task.Run();

107 PumpLoop();

108 ClearResults();

109

110 MarkAttemptAbortedTask task(

111 &store, kRequestId1,

112 base::Bind(&MarkAttemptAbortedTaskTest::ChangeRequestsStateCallback,

113 base::Unretained(this)));

114

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_EQ(SavePageRequest::RequestState::AVAILABLE,

124 last_result()->updated_items.at(0).request_state());

125 }

126

127 TEST_F(MarkAttemptAbortedTaskTest, MarkAttemptAbortedWhenItemMissing) {

128 RequestQueueInMemoryStore store;

129 AddItemToStore(&store);

130 MarkAttemptAbortedTask task(

131 &store, kRequestId2,

132 base::Bind(&MarkAttemptAbortedTaskTest::ChangeRequestsStateCallback,

133 base::Unretained(this)));

134 task.Run();

135 PumpLoop();

136 ASSERT_TRUE(last_result());

137 EXPECT_EQ(1UL, last_result()->item_statuses.size());

138 EXPECT_EQ(kRequestId2, last_result()->item_statuses.at(0).first);

139 EXPECT_EQ(ItemActionStatus::NOT_FOUND,

140 last_result()->item_statuses.at(0).second);

141 EXPECT_EQ(0UL, last_result()->updated_items.size());

142 }

143

144 } // namespace offline_pages

OLD	NEW