Home | History | Annotate | Download | only in update_manager 
      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 
     25 using base::Time;
     26 using base::TimeDelta;
     27 using chromeos_update_engine::ErrorCode;
     28 using std::string;
     29 using std::tuple;
     30 using std::vector;
     31 
     32 namespace chromeos_update_manager {
     33 
     34 void UmPolicyTestBase::SetUp() {
     35   loop_.SetAsCurrent();
     36   SetUpDefaultClock();
     37   eval_ctx_ = new EvaluationContext(&fake_clock_, TimeDelta::FromSeconds(5));
     38   SetUpDefaultState();
     39 }
     40 
     41 void UmPolicyTestBase::TearDown() {
     42   EXPECT_FALSE(loop_.PendingTasks());
     43 }
     44 
     45 // Sets the clock to fixed values.
     46 void UmPolicyTestBase::SetUpDefaultClock() {
     47   fake_clock_.SetMonotonicTime(Time::FromInternalValue(12345678L));
     48   fake_clock_.SetWallclockTime(Time::FromInternalValue(12345678901234L));
     49 }
     50 
     51 void 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:
     68 UpdateState 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