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

#include <memory>

#include "base/check_deref.h"
#include "base/command_line.h"
#include "base/memory/ptr_util.h"
#include "base/memory/raw_ptr.h"
#include "chrome/browser/apps/platform_apps/app_browsertest_util.h"
#include "chrome/browser/ash/app_mode/consumer_kiosk_test_helper.h"
#include "chrome/browser/ash/app_mode/kiosk_chrome_app_manager.h"
#include "chrome/browser/ash/ownership/fake_owner_settings_service.h"
#include "chrome/browser/ash/settings/scoped_cros_settings_test_helper.h"
#include "chrome/browser/ash/test/kiosk_app_logged_in_browser_test_mixin.h"
#include "chrome/browser/extensions/extension_apitest.h"
#include "chrome/browser/ui/browser_window/public/browser_window_interface.h"
#include "chromeos/ash/components/browser_context_helper/browser_context_helper.h"
#include "chromeos/components/kiosk/kiosk_test_utils.h"
#include "components/account_id/account_id.h"
#include "components/user_manager/user_manager.h"
#include "content/public/test/browser_test.h"
#include "extensions/common/switches.h"
#include "testing/gmock/include/gmock/gmock.h"

namespace {
const char kTestingAppId[] = "pjdjhejcdkeebjehnokfbfnjmgmgdjlc";
}  // namespace

namespace extensions {
namespace {

// This class contains chrome.bluetoothLowEnergy API tests.
class BluetoothLowEnergyApiTestChromeOs : public PlatformAppBrowserTest {
 public:
  ~BluetoothLowEnergyApiTestChromeOs() override = default;

  void SetUpOnMainThread() override {
    PlatformAppBrowserTest::SetUpOnMainThread();
    settings_helper_.ReplaceDeviceSettingsProviderWithStub();

    auto* browser_context =
        ash::BrowserContextHelper::Get()->GetBrowserContextByUser(
            user_manager::UserManager::Get()->GetActiveUser());
    owner_settings_service_ = settings_helper_.CreateOwnerSettingsService(
        Profile::FromBrowserContext(browser_context));
  }

  void TearDownOnMainThread() override {
    owner_settings_service_.reset();
    settings_helper_.RestoreRealDeviceSettingsProvider();
    PlatformAppBrowserTest::TearDownOnMainThread();
  }

 protected:
  ash::ScopedCrosSettingsTestHelper settings_helper_{false};
  std::unique_ptr<ash::FakeOwnerSettingsService> owner_settings_service_;
};

class BluetoothLowEnergyApiKioskTestChromeOs
    : public BluetoothLowEnergyApiTestChromeOs {
 public:
  BluetoothLowEnergyApiKioskTestChromeOs() { set_chromeos_user_ = false; }

 protected:
  Profile* GetProfile() {
    auto* user = user_manager::UserManager::Get()->GetActiveUser();
    return Profile::FromBrowserContext(
        ash::BrowserContextHelper::Get()->GetBrowserContextByUser(user));
  }

  // TODO(hidehiko): consider to move this into KioskAppLoggedInBrowserTestMixin
  // so the test can be exercised in the environment closer to the production.
  void SetAutoLaunchApp() {
    auto* manager = ash::KioskChromeAppManager::Get();

    AddConsumerKioskChromeAppForTesting(
        CHECK_DEREF(owner_settings_service_.get()), kTestingAppId);
    SetConsumerKioskAutoLaunchChromeAppForTesting(
        CHECK_DEREF(manager), CHECK_DEREF(owner_settings_service_.get()),
        kTestingAppId);
    manager->SetAppWasAutoLaunchedWithZeroDelay(kTestingAppId);
  }

 private:
  ash::KioskAppLoggedInBrowserTestMixin kiosk_mixin_{&mixin_host_,
                                                     "kiosk-account"};
};

IN_PROC_BROWSER_TEST_F(BluetoothLowEnergyApiTestChromeOs,
                       RegisterAdvertisement_Flag) {
  base::CommandLine::ForCurrentProcess()->AppendSwitch(
      switches::kEnableBLEAdvertising);
  ASSERT_TRUE(RunExtensionTest(
      "api_test/bluetooth_low_energy/register_advertisement_flag",
      {.launch_as_platform_app = true}))
      << message_;
}

IN_PROC_BROWSER_TEST_F(BluetoothLowEnergyApiTestChromeOs,
                       RegisterAdvertisement_NotKioskSession) {
  ASSERT_TRUE(RunExtensionTest(
      "api_test/bluetooth_low_energy/register_advertisement_no_kiosk_mode",
      {.launch_as_platform_app = true}))
      << message_;
}

IN_PROC_BROWSER_TEST_F(BluetoothLowEnergyApiKioskTestChromeOs,
                       RegisterAdvertisement_KioskSessionOnly) {
  ASSERT_TRUE(RunExtensionTest(
      "api_test/bluetooth_low_energy/"
      "register_advertisement_kiosk_session_only",
      {.launch_as_platform_app = true, .profile = GetProfile()}))
      << message_;
}

IN_PROC_BROWSER_TEST_F(BluetoothLowEnergyApiKioskTestChromeOs,
                       RegisterAdvertisement) {
  SetAutoLaunchApp();
  ASSERT_TRUE(RunExtensionTest(
      "api_test/bluetooth_low_energy/register_advertisement",
      {.launch_as_platform_app = true, .profile = GetProfile()}))
      << message_;
}

IN_PROC_BROWSER_TEST_F(BluetoothLowEnergyApiKioskTestChromeOs,
                       ResetAdvertising) {
  SetAutoLaunchApp();
  ASSERT_TRUE(RunExtensionTest(
      "api_test/bluetooth_low_energy/reset_all_advertisements",
      {.launch_as_platform_app = true, .profile = GetProfile()}))
      << message_;
}

IN_PROC_BROWSER_TEST_F(BluetoothLowEnergyApiKioskTestChromeOs,
                       SetAdvertisingInterval) {
  SetAutoLaunchApp();
  ASSERT_TRUE(RunExtensionTest(
      "api_test/bluetooth_low_energy/"
      "set_advertising_interval",
      {.launch_as_platform_app = true, .profile = GetProfile()}))
      << message_;
}

IN_PROC_BROWSER_TEST_F(BluetoothLowEnergyApiKioskTestChromeOs, CreateService) {
  SetAutoLaunchApp();
  ASSERT_TRUE(RunExtensionTest(
      "api_test/bluetooth_low_energy/"
      "create_service",
      {.launch_as_platform_app = true, .profile = GetProfile()}))
      << message_;
}

IN_PROC_BROWSER_TEST_F(BluetoothLowEnergyApiTestChromeOs, CreateService_Flag) {
  base::CommandLine::ForCurrentProcess()->AppendSwitch(
      switches::kEnableBLEAdvertising);
  ASSERT_TRUE(
      RunExtensionTest("api_test/bluetooth_low_energy/create_service_flag",
                       {.launch_as_platform_app = true}))
      << message_;
}

IN_PROC_BROWSER_TEST_F(BluetoothLowEnergyApiTestChromeOs,
                       CreateService_NotKioskSession) {
  ASSERT_TRUE(RunExtensionTest(
      "api_test/bluetooth_low_energy/create_service_no_kiosk_mode",
      {.launch_as_platform_app = true}))
      << message_;
}

IN_PROC_BROWSER_TEST_F(BluetoothLowEnergyApiKioskTestChromeOs,
                       CreateService_KioskSessionOnly) {
  ASSERT_TRUE(RunExtensionTest(
      "api_test/bluetooth_low_energy/"
      "create_service_kiosk_session_only",
      {.launch_as_platform_app = true, .profile = GetProfile()}))
      << message_;
}

IN_PROC_BROWSER_TEST_F(BluetoothLowEnergyApiKioskTestChromeOs,
                       CreateCharacteristic) {
  SetAutoLaunchApp();
  ASSERT_TRUE(RunExtensionTest(
      "api_test/bluetooth_low_energy/"
      "create_characteristic",
      {.launch_as_platform_app = true, .profile = GetProfile()}))
      << message_;
}

IN_PROC_BROWSER_TEST_F(BluetoothLowEnergyApiKioskTestChromeOs,
                       CreateDescriptor) {
  SetAutoLaunchApp();
  ASSERT_TRUE(RunExtensionTest(
      "api_test/bluetooth_low_energy/"
      "create_descriptor",
      {.launch_as_platform_app = true, .profile = GetProfile()}))
      << message_;
}

IN_PROC_BROWSER_TEST_F(BluetoothLowEnergyApiKioskTestChromeOs,
                       RegisterService) {
  SetAutoLaunchApp();
  ASSERT_TRUE(RunExtensionTest(
      "api_test/bluetooth_low_energy/"
      "register_service",
      {.launch_as_platform_app = true, .profile = GetProfile()}))
      << message_;
}

IN_PROC_BROWSER_TEST_F(BluetoothLowEnergyApiKioskTestChromeOs,
                       UnregisterService) {
  SetAutoLaunchApp();
  ASSERT_TRUE(RunExtensionTest(
      "api_test/bluetooth_low_energy/"
      "unregister_service",
      {.launch_as_platform_app = true, .profile = GetProfile()}))
      << message_;
}

IN_PROC_BROWSER_TEST_F(BluetoothLowEnergyApiKioskTestChromeOs, RemoveService) {
  SetAutoLaunchApp();
  ASSERT_TRUE(RunExtensionTest(
      "api_test/bluetooth_low_energy/"
      "remove_service",
      {.launch_as_platform_app = true, .profile = GetProfile()}))
      << message_;
}

IN_PROC_BROWSER_TEST_F(BluetoothLowEnergyApiKioskTestChromeOs,
                       NotifyCharacteristicValueChanged) {
  SetAutoLaunchApp();
  ASSERT_TRUE(RunExtensionTest(
      "api_test/bluetooth_low_energy/"
      "notify_characteristic_value_changed",
      {.launch_as_platform_app = true, .profile = GetProfile()}))
      << message_;
}

IN_PROC_BROWSER_TEST_F(BluetoothLowEnergyApiKioskTestChromeOs,
                       NotifyCharacteristicValueChanged_ErrorConditions) {
  SetAutoLaunchApp();
  ASSERT_TRUE(RunExtensionTest(
      "api_test/bluetooth_low_energy/"
      "notify_characteristic_value_changed_error_conditions",
      {.launch_as_platform_app = true, .profile = GetProfile()}))
      << message_;
}

// TODO(rkc): Figure out how to integrate with BluetoothTestBlueZ and write
// comprehensive tests for GATT server events. See http://crbug.com/40467075 for
// details.

}  // namespace
}  // namespace extensions