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