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()36 void 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()45 void UmPolicyTestBase::TearDown() {
46   EXPECT_FALSE(loop_.PendingTasks());
47 }
48 
49 // Sets the clock to fixed values.
SetUpDefaultClock()50 void UmPolicyTestBase::SetUpDefaultClock() {
51   fake_clock_->SetMonotonicTime(Time::FromInternalValue(12345678L));
52   fake_clock_->SetWallclockTime(Time::FromInternalValue(12345678901234L));
53 }
54 
SetUpDefaultTimeProvider()55 void 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()66 void 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)83 UpdateState 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