chromium/chrome/browser/ash/bluetooth/bluetooth_log_controller_unittest.cc

// 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 "chrome/browser/ash/bluetooth/bluetooth_log_controller.h"

#include "base/test/task_environment.h"
#include "chromeos/ash/components/dbus/upstart/fake_upstart_client.h"
#include "chromeos/ash/components/dbus/upstart/upstart_client.h"
#include "components/account_id/account_id.h"
#include "components/user_manager/fake_user_manager.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace ash {

class BluetoothLogControllerTest : public testing::Test {
 public:
  BluetoothLogControllerTest() = default;
  ~BluetoothLogControllerTest() override = default;

  void SetUp() override { UpstartClient::InitializeFake(); }

  void TearDown() override { UpstartClient::Shutdown(); }

  user_manager::FakeUserManager& user_manager() { return user_manager_; }
  BluetoothLogController& controller() { return controller_; }

 private:
  base::test::TaskEnvironment task_environment_;
  user_manager::FakeUserManager user_manager_;
  BluetoothLogController controller_{&user_manager_};
};

TEST_F(BluetoothLogControllerTest, GoogleInternalUser) {
  auto* upstart_client = FakeUpstartClient::Get();
  upstart_client->StartRecordingUpstartOperations();

  auto* user =
      user_manager().AddUser(AccountId::FromUserEmail("[email protected]"));
  // TODO(b/278643115): use UserManager::UserLoggedIn() to notify observer.
  controller().OnUserLoggedIn(*user);

  auto upstart_operations =
      upstart_client->GetRecordedUpstartOperationsForJob("bluetoothlog");
  ASSERT_EQ(1u, upstart_operations.size());
  EXPECT_EQ(FakeUpstartClient::UpstartOperationType::START,
            upstart_operations[0].type);
}

TEST_F(BluetoothLogControllerTest, NonGoolgeInternalUser) {
  auto* upstart_client = FakeUpstartClient::Get();
  upstart_client->StartRecordingUpstartOperations();

  auto* user =
      user_manager().AddUser(AccountId::FromUserEmail("[email protected]"));
  // TODO(b/278643115): use UserManager::UserLoggedIn() to notify observer.
  controller().OnUserLoggedIn(*user);

  auto upstart_operations =
      upstart_client->GetRecordedUpstartOperationsForJob("bluetoothlog");
  EXPECT_EQ(0u, upstart_operations.size());
}

TEST_F(BluetoothLogControllerTest, NonRegularUser) {
  auto* upstart_client = FakeUpstartClient::Get();
  upstart_client->StartRecordingUpstartOperations();

  auto* user = user_manager().AddKioskAppUser(
      AccountId::FromUserEmail("[email protected]"));
  // TODO(b/278643115): use UserManager::UserLoggedIn() to notify observer.
  controller().OnUserLoggedIn(*user);

  auto upstart_operations =
      upstart_client->GetRecordedUpstartOperationsForJob("bluetoothlog");
  EXPECT_EQ(0u, upstart_operations.size());
}

}  // namespace ash