update_engine/update_manager/policy_test_utils.cc

// Copyright 2017 The ChromiumOS Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "update_engine/update_manager/policy_test_utils.h"

#include <memory>
#include <utility>

#include "update_engine/cros/fake_system_state.h"
#include "update_engine/update_manager/next_update_check_policy_impl.h"

using base::Time;
using chromeos_update_engine::ErrorCode;
using chromeos_update_engine::FakeSystemState;

namespace chromeos_update_manager {

void UmPolicyTestBase::SetUp() {
  loop_.SetAsCurrent();
  FakeSystemState::CreateInstance();
  fake_clock_ = FakeSystemState::Get()->fake_clock();
  SetUpDefaultClock();
  eval_ctx_.reset(new EvaluationContext(base::Seconds(5)));
  SetUpDefaultState();

  evaluator_ = std::make_unique<PolicyEvaluator>(
      &fake_state_, std::make_unique<EvaluationContext>(base::Seconds(5)),
      std::move(policy_2_), policy_data_);
}

void UmPolicyTestBase::TearDown() {
  EXPECT_FALSE(loop_.PendingTasks());
}

// Sets the clock to fixed values.
void UmPolicyTestBase::SetUpDefaultClock() {
  fake_clock_->SetMonotonicTime(Time::FromInternalValue(12345678L));
  fake_clock_->SetWallclockTime(Time::FromInternalValue(12345678901234L));
}

void UmPolicyTestBase::SetUpDefaultTimeProvider() {
  Time current_time = FakeSystemState::Get()->clock()->GetWallclockTime();
  base::Time::Exploded exploded;
  current_time.LocalExplode(&exploded);
  fake_state_.time_provider()->var_curr_hour()->reset(new int(exploded.hour));
  fake_state_.time_provider()->var_curr_minute()->reset(
      new int(exploded.minute));
  fake_state_.time_provider()->var_curr_date()->reset(
      new Time(current_time.LocalMidnight()));
}

void UmPolicyTestBase::SetUpDefaultState() {
  fake_state_.updater_provider()->var_updater_started_time()->reset(
      new Time(fake_clock_->GetWallclockTime()));
  fake_state_.updater_provider()->var_last_checked_time()->reset(
      new Time(fake_clock_->GetWallclockTime()));
  fake_state_.updater_provider()->var_consecutive_failed_update_checks()->reset(
      new unsigned int(0));  // NOLINT(readability/casting)
  fake_state_.updater_provider()->var_server_dictated_poll_interval()->reset(
      new unsigned int(0));  // NOLINT(readability/casting)
  fake_state_.updater_provider()->var_forced_update_requested()->reset(
      new UpdateRequestStatus{UpdateRequestStatus::kNone});

  // Chosen by fair dice roll.  Guaranteed to be random.
  fake_state_.random_provider()->var_seed()->reset(new uint64_t(4));
}

}  // namespace chromeos_update_manager