// Copyright 2020 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/arc/arc_demo_mode_delegate_impl.h"

#include <memory>
#include <optional>

#include "base/check_deref.h"
#include "chrome/browser/ash/login/demo_mode/demo_mode_test_helper.h"
#include "chrome/browser/ash/login/demo_mode/demo_session.h"
#include "chrome/test/base/testing_browser_process.h"
#include "chromeos/ash/components/install_attributes/stub_install_attributes.h"
#include "components/session_manager/test/test_user_session_manager.h"
#include "content/public/test/browser_task_environment.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace arc {
namespace {

class ArcDemoModeDelegateImplTest : public testing::Test {
 public:
  ArcDemoModeDelegateImplTest() = default;
  ArcDemoModeDelegateImplTest(const ArcDemoModeDelegateImplTest&) = delete;
  ArcDemoModeDelegateImplTest& operator=(const ArcDemoModeDelegateImplTest&) =
      delete;
  ~ArcDemoModeDelegateImplTest() override = default;

 protected:
  void SetUp() override {
    stub_install_attributes_ =
        std::make_unique<ash::ScopedStubInstallAttributes>(
            ash::StubInstallAttributes::CreateDemoMode());
    test_user_session_manager_ =
        std::make_unique<ash::test::TestUserSessionManager>(
            TestingBrowserProcess::GetGlobal()->local_state());

    demo_helper_.emplace();
    delegate_.emplace();
  }

  void TearDown() override {
    delegate_.reset();
    demo_helper_.reset();
    test_user_session_manager_.reset();
    stub_install_attributes_.reset();
  }

  ash::StubInstallAttributes& stub_install_attributes() {
    return CHECK_DEREF(stub_install_attributes_->Get());
  }
  ash::DemoModeTestHelper& demo_helper() { return *demo_helper_; }
  ArcDemoModeDelegateImpl& delegate() { return *delegate_; }

 private:
  content::BrowserTaskEnvironment browser_task_environment_;
  std::unique_ptr<ash::ScopedStubInstallAttributes> stub_install_attributes_;
  std::unique_ptr<ash::test::TestUserSessionManager> test_user_session_manager_;
  std::optional<ash::DemoModeTestHelper> demo_helper_;
  std::optional<ArcDemoModeDelegateImpl> delegate_;
};

// Test that EnsureResourcesLoaded returns immediately if demo mode is
// not enabled.
TEST_F(ArcDemoModeDelegateImplTest, EnsureResourcesLoaded_NotEnabled) {
  ash::DemoSession::SetDemoConfigForTesting(
      ash::DemoSession::DemoModeConfig::kNone);
  stub_install_attributes().SetConsumerOwned();

  bool was_called = false;
  base::OnceClosure callback =
      base::BindOnce([](bool* was_called) { *was_called = true; }, &was_called);
  delegate().EnsureResourcesLoaded(std::move(callback));
  EXPECT_TRUE(was_called);
}

// Test that EnsureResourcesLoaded returns after resources are loaded if
// demo mode is enabled.
TEST_F(ArcDemoModeDelegateImplTest, EnsureResourcesLoaded_Enabled) {
  demo_helper().InitializeSessionWithPendingComponent();

  bool was_called = false;
  base::OnceClosure callback =
      base::BindOnce([](bool* was_called) { *was_called = true; }, &was_called);
  delegate().EnsureResourcesLoaded(std::move(callback));
  EXPECT_FALSE(was_called);

  demo_helper().FinishLoadingComponent();
  EXPECT_TRUE(was_called);
}

// Test that GetDemoAppsPath returns empty path if demo mode is not enabled.
TEST_F(ArcDemoModeDelegateImplTest, GetDemoAppsPath_NotEnabled) {
  ash::DemoSession::SetDemoConfigForTesting(
      ash::DemoSession::DemoModeConfig::kNone);
  stub_install_attributes().SetConsumerOwned();

  base::FilePath demo_session_apps_path = delegate().GetDemoAppsPath();
  EXPECT_TRUE(demo_session_apps_path.empty());
}

// Test that GetDemoAppsPath returns the correct path if demo mode is enabled.
TEST_F(ArcDemoModeDelegateImplTest, GetDemoAppsPath_Enabled) {
  demo_helper().InitializeSession();

  base::FilePath demo_session_apps_path = delegate().GetDemoAppsPath();
  EXPECT_FALSE(demo_session_apps_path.empty());
  EXPECT_EQ(
      demo_helper().GetDemoResourcesPath().Append("android_demo_apps.squash"),
      demo_session_apps_path);
}

}  // namespace
}  // namespace arc