1//
2// Copyright (C) 2017 The Android Open Source Project
3//
4// Licensed under the Apache License, Version 2.0 (the "License");
5// you may not use this file except in compliance with the License.
6// You may obtain a copy of the License at
7//
8//      http://www.apache.org/licenses/LICENSE-2.0
9//
10// Unless required by applicable law or agreed to in writing, software
11// distributed under the License is distributed on an "AS IS" BASIS,
12// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13// See the License for the specific language governing permissions and
14// limitations under the License.
15//
16
17#include "update_engine/update_manager/policy_test_utils.h"
18
19#include <memory>
20#include <tuple>
21#include <vector>
22
23#include "update_engine/update_manager/next_update_check_policy_impl.h"
24
25using base::Time;
26using base::TimeDelta;
27using chromeos_update_engine::ErrorCode;
28using std::string;
29using std::tuple;
30using std::vector;
31
32namespace chromeos_update_manager {
33
34void UmPolicyTestBase::SetUp() {
35  loop_.SetAsCurrent();
36  SetUpDefaultClock();
37  eval_ctx_ = new EvaluationContext(&fake_clock_, TimeDelta::FromSeconds(5));
38  SetUpDefaultState();
39}
40
41void UmPolicyTestBase::TearDown() {
42  EXPECT_FALSE(loop_.PendingTasks());
43}
44
45// Sets the clock to fixed values.
46void UmPolicyTestBase::SetUpDefaultClock() {
47  fake_clock_.SetMonotonicTime(Time::FromInternalValue(12345678L));
48  fake_clock_.SetWallclockTime(Time::FromInternalValue(12345678901234L));
49}
50
51void UmPolicyTestBase::SetUpDefaultState() {
52  fake_state_.updater_provider()->var_updater_started_time()->reset(
53      new Time(fake_clock_.GetWallclockTime()));
54  fake_state_.updater_provider()->var_last_checked_time()->reset(
55      new Time(fake_clock_.GetWallclockTime()));
56  fake_state_.updater_provider()->var_consecutive_failed_update_checks()->reset(
57      new unsigned int(0));  // NOLINT(readability/casting)
58  fake_state_.updater_provider()->var_server_dictated_poll_interval()->reset(
59      new unsigned int(0));  // NOLINT(readability/casting)
60  fake_state_.updater_provider()->var_forced_update_requested()->reset(
61      new UpdateRequestStatus{UpdateRequestStatus::kNone});
62
63  // Chosen by fair dice roll.  Guaranteed to be random.
64  fake_state_.random_provider()->var_seed()->reset(new uint64_t(4));
65}
66
67// Returns a default UpdateState structure:
68UpdateState UmPolicyTestBase::GetDefaultUpdateState(
69    TimeDelta first_seen_period) {
70  Time first_seen_time = fake_clock_.GetWallclockTime() - first_seen_period;
71  UpdateState update_state = UpdateState();
72
73  // This is a non-interactive check returning a delta payload, seen for the
74  // first time (|first_seen_period| ago). Clearly, there were no failed
75  // attempts so far.
76  update_state.is_interactive = false;
77  update_state.is_delta_payload = false;
78  update_state.first_seen = first_seen_time;
79  update_state.num_checks = 1;
80  update_state.num_failures = 0;
81  update_state.failures_last_updated = Time();  // Needs to be zero.
82  // There's a single HTTP download URL with a maximum of 10 retries.
83  update_state.download_urls = vector<string>{"http://fake/url/"};
84  update_state.download_errors_max = 10;
85  // Download was never attempted.
86  update_state.last_download_url_idx = -1;
87  update_state.last_download_url_num_errors = 0;
88  // There were no download errors.
89  update_state.download_errors = vector<tuple<int, ErrorCode, Time>>();
90  // P2P is not disabled by Omaha.
91  update_state.p2p_downloading_disabled = false;
92  update_state.p2p_sharing_disabled = false;
93  // P2P was not attempted.
94  update_state.p2p_num_attempts = 0;
95  update_state.p2p_first_attempted = Time();
96  // No active backoff period, backoff is not disabled by Omaha.
97  update_state.backoff_expiry = Time();
98  update_state.is_backoff_disabled = false;
99  // There is no active scattering wait period (max 7 days allowed) nor check
100  // threshold (none allowed).
101  update_state.scatter_wait_period = TimeDelta();
102  update_state.scatter_check_threshold = 0;
103  update_state.scatter_wait_period_max = TimeDelta::FromDays(7);
104  update_state.scatter_check_threshold_min = 0;
105  update_state.scatter_check_threshold_max = 0;
106
107  return update_state;
108}
109
110}  // namespace chromeos_update_manager
111