fbpreprocessor/fake_manager.cc

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

#include "fbpreprocessor/fake_manager.h"

#include <memory>
#include <utility>

#include <base/check.h>
#include <base/files/file_path.h>
#include <base/files/file_util.h>
#include <base/files/scoped_temp_dir.h>
#include <chromeos/dbus/fbpreprocessor/dbus-constants.h>
#include <dbus/bus.h>
#include <metrics/fake_metrics_library.h>

#include "fbpreprocessor/fake_platform_features_client.h"
#include "fbpreprocessor/fake_session_state_manager.h"
#include "fbpreprocessor/metrics.h"
#include "fbpreprocessor/output_manager.h"

namespace {
constexpr int kTestDefaultExpirationSeconds = 1800;
}  // namespace

namespace fbpreprocessor {

FakeManager::FakeManager()
    : default_file_expiration_in_secs_(kTestDefaultExpirationSeconds) {
  auto uma_lib = std::make_unique<FakeMetricsLibrary>();
  uma_lib_ = uma_lib.get();
  metrics().SetLibraryForTesting(std::move(uma_lib));
  platform_features_ = std::make_unique<FakePlatformFeaturesClient>();
  session_state_manager_ = std::make_unique<FakeSessionStateManager>();
  output_manager_ = std::make_unique<OutputManager>(this);
}

void FakeManager::Start(dbus::Bus* bus) {
  SetupFakeDaemonStore();
  output_manager_->set_base_dir_for_test(GetRootDir());
}

SessionStateManagerInterface* FakeManager::session_state_manager() const {
  return session_state_manager_.get();
}

void FakeManager::SimulateUserLogin() {
  session_state_manager_->SimulateLogin();
}

void FakeManager::SimulateUserLogout() {
  session_state_manager_->SimulateLogout();
}

void FakeManager::SetupFakeDaemonStore() {
  CHECK(root_dir_.CreateUniqueTempDir());
  CHECK(base::CreateDirectory(
      root_dir_.GetPath().Append(kTestUserHash).Append(kInputDirectory)));
  CHECK(base::CreateDirectory(
      root_dir_.GetPath().Append(kTestUserHash).Append(kProcessedDirectory)));
}

}  // namespace fbpreprocessor