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

#include "chromeos/components/magic_boost/test/fake_magic_boost_state.h"

#include "base/types/expected.h"
#include "chromeos/components/magic_boost/public/cpp/magic_boost_state.h"

namespace chromeos {
namespace test {

bool FakeMagicBoostState::ShouldIncludeOrcaInOptInSync() {
  return false;
}

bool FakeMagicBoostState::CanShowNoticeBannerForHMR() {
  return false;
}

int32_t FakeMagicBoostState::AsyncIncrementHMRConsentWindowDismissCount() {
  return 0;
}

void FakeMagicBoostState::AsyncWriteConsentStatus(
    chromeos::HMRConsentStatus consent_status) {
  UpdateHMRConsentStatus(consent_status);
}

void FakeMagicBoostState::AsyncWriteHMREnabled(bool enabled) {
  UpdateHMREnabled(enabled);
}

void FakeMagicBoostState::SetAvailability(bool available) {
  UpdateUserEligibleForGenAIFeatures(available);
}

void FakeMagicBoostState::SetMagicBoostEnabled(bool enabled) {
  UpdateMagicBoostEnabled(enabled);
}

base::expected<bool, chromeos::MagicBoostState::Error>
FakeMagicBoostState::IsUserEligibleForGenAIFeaturesExpected() const {
  // Availability needs to be set explicitly via `SetAvailability` for this
  // fake.
  return base::unexpected(chromeos::MagicBoostState::Error::kUninitialized);
}

}  // namespace test
}  // namespace chromeos