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/cros/fake_system_state.h" 24 #include "update_engine/update_manager/next_update_check_policy_impl.h" 25 26 using base::Time; 27 using base::TimeDelta; 28 using chromeos_update_engine::ErrorCode; 29 using chromeos_update_engine::FakeSystemState; 30 using std::string; 31 using std::tuple; 32 using std::vector; 33 34 namespace chromeos_update_manager { 35 SetUp()36void UmPolicyTestBase::SetUp() { 37 loop_.SetAsCurrent(); 38 FakeSystemState::CreateInstance(); 39 fake_clock_ = FakeSystemState::Get()->fake_clock(); 40 SetUpDefaultClock(); 41 eval_ctx_.reset(new EvaluationContext(TimeDelta::FromSeconds(5))); 42 SetUpDefaultState(); 43 } 44 TearDown()45void UmPolicyTestBase::TearDown() { 46 EXPECT_FALSE(loop_.PendingTasks()); 47 } 48 49 // Sets the clock to fixed values. SetUpDefaultClock()50void UmPolicyTestBase::SetUpDefaultClock() { 51 fake_clock_->SetMonotonicTime(Time::FromInternalValue(12345678L)); 52 fake_clock_->SetWallclockTime(Time::FromInternalValue(12345678901234L)); 53 } 54 SetUpDefaultTimeProvider()55void UmPolicyTestBase::SetUpDefaultTimeProvider() { 56 Time current_time = FakeSystemState::Get()->clock()->GetWallclockTime(); 57 base::Time::Exploded exploded; 58 current_time.LocalExplode(&exploded); 59 fake_state_.time_provider()->var_curr_hour()->reset(new int(exploded.hour)); 60 fake_state_.time_provider()->var_curr_minute()->reset( 61 new int(exploded.minute)); 62 fake_state_.time_provider()->var_curr_date()->reset( 63 new Time(current_time.LocalMidnight())); 64 } 65 SetUpDefaultState()66void UmPolicyTestBase::SetUpDefaultState() { 67 fake_state_.updater_provider()->var_updater_started_time()->reset( 68 new Time(fake_clock_->GetWallclockTime())); 69 fake_state_.updater_provider()->var_last_checked_time()->reset( 70 new Time(fake_clock_->GetWallclockTime())); 71 fake_state_.updater_provider()->var_consecutive_failed_update_checks()->reset( 72 new unsigned int(0)); // NOLINT(readability/casting) 73 fake_state_.updater_provider()->var_server_dictated_poll_interval()->reset( 74 new unsigned int(0)); // NOLINT(readability/casting) 75 fake_state_.updater_provider()->var_forced_update_requested()->reset( 76 new UpdateRequestStatus{UpdateRequestStatus::kNone}); 77 78 // Chosen by fair dice roll. Guaranteed to be random. 79 fake_state_.random_provider()->var_seed()->reset(new uint64_t(4)); 80 } 81 82 // Returns a default UpdateState structure: GetDefaultUpdateState(TimeDelta first_seen_period)83UpdateState UmPolicyTestBase::GetDefaultUpdateState( 84 TimeDelta first_seen_period) { 85 Time first_seen_time = 86 FakeSystemState::Get()->clock()->GetWallclockTime() - first_seen_period; 87 UpdateState update_state = UpdateState(); 88 89 // This is a non-interactive check returning a delta payload, seen for the 90 // first time (|first_seen_period| ago). Clearly, there were no failed 91 // attempts so far. 92 update_state.interactive = false; 93 update_state.is_delta_payload = false; 94 update_state.first_seen = first_seen_time; 95 update_state.num_checks = 1; 96 update_state.num_failures = 0; 97 update_state.failures_last_updated = Time(); // Needs to be zero. 98 // There's a single HTTP download URL with a maximum of 10 retries. 99 update_state.download_urls = vector<string>{"http://fake/url/"}; 100 update_state.download_errors_max = 10; 101 // Download was never attempted. 102 update_state.last_download_url_idx = -1; 103 update_state.last_download_url_num_errors = 0; 104 // There were no download errors. 105 update_state.download_errors = vector<tuple<int, ErrorCode, Time>>(); 106 // P2P is not disabled by Omaha. 107 update_state.p2p_downloading_disabled = false; 108 update_state.p2p_sharing_disabled = false; 109 // P2P was not attempted. 110 update_state.p2p_num_attempts = 0; 111 update_state.p2p_first_attempted = Time(); 112 // No active backoff period, backoff is not disabled by Omaha. 113 update_state.backoff_expiry = Time(); 114 update_state.is_backoff_disabled = false; 115 // There is no active scattering wait period (max 7 days allowed) nor check 116 // threshold (none allowed). 117 update_state.scatter_wait_period = TimeDelta(); 118 update_state.scatter_check_threshold = 0; 119 update_state.scatter_wait_period_max = TimeDelta::FromDays(7); 120 update_state.scatter_check_threshold_min = 0; 121 update_state.scatter_check_threshold_max = 0; 122 123 return update_state; 124 } 125 126 } // namespace chromeos_update_manager 127