components/offline_pages/background/mark_attempt_started_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_started_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_started_task.cc ('k') | components/offline_pages/background/offliner.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_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 AddItemToStore(RequestQueueStore* store);

31

32 void AddRequestDone(ItemActionStatus status);

33

34 void ChangeRequestsStateCallback(

35 std::unique_ptr<UpdateRequestsResult> result);

36

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

38

39 private:

40 std::unique_ptr<UpdateRequestsResult> result_;

41 scoped_refptr<base::TestSimpleTaskRunner> task_runner_;

42 base::ThreadTaskRunnerHandle task_runner_handle_;

43 };

44

45 MarkAttemptStartedTaskTest::MarkAttemptStartedTaskTest()

46 : task_runner_(new base::TestSimpleTaskRunner),

47 task_runner_handle_(task_runner_) {}

48

49 MarkAttemptStartedTaskTest::~MarkAttemptStartedTaskTest() {}

50

51 void MarkAttemptStartedTaskTest::PumpLoop() {

52 task_runner_->RunUntilIdle();

53 }

54

55 void MarkAttemptStartedTaskTest::AddItemToStore(RequestQueueStore* store) {

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

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

58 true);

59 store->AddRequest(request_1,

60 base::Bind(&MarkAttemptStartedTaskTest::AddRequestDone,

61 base::Unretained(this)));

62 PumpLoop();

63 }

64

65 void MarkAttemptStartedTaskTest::AddRequestDone(ItemActionStatus status) {

66 ASSERT_EQ(ItemActionStatus::SUCCESS, status);

67 }

68

69 void MarkAttemptStartedTaskTest::ChangeRequestsStateCallback(

70 std::unique_ptr<UpdateRequestsResult> result) {

71 result_ = std::move(result);

72 }

73

74 TEST_F(MarkAttemptStartedTaskTest, MarkAttemptStartedWhenStoreEmpty) {

75 RequestQueueInMemoryStore store;

76 MarkAttemptStartedTask task(

77 &store, kRequestId1,

78 base::Bind(&MarkAttemptStartedTaskTest::ChangeRequestsStateCallback,

79 base::Unretained(this)));

80 task.Run();

81 PumpLoop();

82 ASSERT_TRUE(last_result());

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

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

85 EXPECT_EQ(ItemActionStatus::NOT_FOUND,

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

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

88 }

89

90 TEST_F(MarkAttemptStartedTaskTest, MarkAttemptStartedWhenExists) {

91 RequestQueueInMemoryStore store;

92 AddItemToStore(&store);

93

94 MarkAttemptStartedTask task(

95 &store, kRequestId1,

96 base::Bind(&MarkAttemptStartedTaskTest::ChangeRequestsStateCallback,

97 base::Unretained(this)));

98

99 // Current time for verification.

100 base::Time before_time = base::Time::Now();

101 task.Run();

102 PumpLoop();

103 ASSERT_TRUE(last_result());

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

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

106 EXPECT_EQ(ItemActionStatus::SUCCESS,

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

108 EXPECT_EQ(1UL, last_result()->updated_items.size());

109 EXPECT_LE(before_time,

110 last_result()->updated_items.at(0).last_attempt_time());

111 EXPECT_GE(base::Time::Now(),

112 last_result()->updated_items.at(0).last_attempt_time());

113 EXPECT_EQ(1, last_result()->updated_items.at(0).started_attempt_count());

114 EXPECT_EQ(SavePageRequest::RequestState::OFFLINING,

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

116 }

117

118 TEST_F(MarkAttemptStartedTaskTest, MarkAttemptStartedWhenItemMissing) {

119 RequestQueueInMemoryStore store;

120 AddItemToStore(&store);

121 MarkAttemptStartedTask task(

122 &store, kRequestId2,

123 base::Bind(&MarkAttemptStartedTaskTest::ChangeRequestsStateCallback,

124 base::Unretained(this)));

125 task.Run();

126 PumpLoop();

127 ASSERT_TRUE(last_result());

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

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

130 EXPECT_EQ(ItemActionStatus::NOT_FOUND,

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

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

133 }

134

135 } // namespace offline_pages

OLD	NEW