chrome/browser/ash/crosapi/device_settings_ash_unittest.cc

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

#include "chrome/browser/ash/crosapi/device_settings_ash.h"

#include "base/run_loop.h"
#include "base/test/task_environment.h"
#include "base/test/test_future.h"
#include "chrome/browser/ash/policy/core/device_policy_builder.h"
#include "chrome/browser/ash/settings/device_settings_service.h"
#include "chromeos/ash/components/dbus/session_manager/fake_session_manager_client.h"
#include "components/ownership/mock_owner_key_util.h"
#include "components/policy/proto/device_management_backend.pb.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace crosapi {
namespace {

class DeviceSettingsAshTest : public ::testing::Test {
 protected:
  void SetUp() override {
    ::ash::DeviceSettingsService::Initialize();
    scoped_refptr<ownership::MockOwnerKeyUtil> owner_key_util(
        new ownership::MockOwnerKeyUtil());
    owner_key_util->SetPublicKeyFromPrivateKey(*device_policy_.GetSigningKey());
    ::ash::DeviceSettingsService::Get()->SetSessionManager(
        &session_manager_client_, owner_key_util);
  }

  void TearDown() override {
    if (::ash::DeviceSettingsService::IsInitialized()) {
      ::ash::DeviceSettingsService::Get()->UnsetSessionManager();
      ::ash::DeviceSettingsService::Shutdown();
    }
  }

  void SetDeviceAssociationState(
      ::enterprise_management::PolicyData::AssociationState association_state) {
    device_policy_.policy_data().set_state(association_state);
    device_policy_.Build();
    session_manager_client_.set_device_policy(device_policy_.GetBlob());

    auto* const device_settings_service = ::ash::DeviceSettingsService::Get();
    base::RunLoop run_loop;
    device_settings_service->Store(device_policy_.GetCopy(),
                                   run_loop.QuitClosure());
    run_loop.Run();
    ASSERT_EQ(device_settings_service->status(),
              ::ash::DeviceSettingsService::STORE_SUCCESS);
  }

  base::test::TaskEnvironment task_environment_;
  ::ash::FakeSessionManagerClient session_manager_client_;
  ::policy::DevicePolicyBuilder device_policy_;
};

TEST_F(DeviceSettingsAshTest, ShouldReturnFalseWhenServiceUninitialized) {
  // Uninitialize device settings service since it is initialized during setup.
  ::ash::DeviceSettingsService::Shutdown();

  DeviceSettingsAsh device_settings_service;
  base::test::TestFuture<bool> test_future;
  device_settings_service.IsDeviceDeprovisioned(test_future.GetCallback());
  EXPECT_FALSE(test_future.Get());
}

TEST_F(DeviceSettingsAshTest, ShouldReturnTrueWhenDeviceDeprovisioned) {
  DeviceSettingsAsh device_settings_service;
  base::test::TestFuture<bool> test_future;
  SetDeviceAssociationState(::enterprise_management::PolicyData::DEPROVISIONED);
  device_settings_service.IsDeviceDeprovisioned(test_future.GetCallback());
  EXPECT_TRUE(test_future.Get());
}

TEST_F(DeviceSettingsAshTest, ShouldReturnFalseIfDeviceNotDeprovisioned) {
  DeviceSettingsAsh device_settings_service;
  base::test::TestFuture<bool> test_future;
  SetDeviceAssociationState(::enterprise_management::PolicyData::ACTIVE);
  device_settings_service.IsDeviceDeprovisioned(test_future.GetCallback());
  EXPECT_FALSE(test_future.Get());
}

}  // namespace
}  // namespace crosapi