// Copyright 2014 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/extensions/extension_api_unittest.h"

#include "content/public/test/browser_task_environment.h"
#include "extensions/browser/api_test_utils.h"
#include "extensions/browser/extension_function.h"
#include "extensions/common/extension.h"
#include "extensions/common/extension_builder.h"
#include "extensions/common/manifest.h"
#include "extensions/common/manifest_handlers/background_info.h"

namespace utils = extensions::api_test_utils;

namespace extensions {

ExtensionApiUnittest::ExtensionApiUnittest()
    : ExtensionServiceTestBase(
          std::make_unique<content::BrowserTaskEnvironment>(
              base::test::TaskEnvironment::MainThreadType::UI)) {}
ExtensionApiUnittest::~ExtensionApiUnittest() = default;

void ExtensionApiUnittest::SetUp() {
  ExtensionServiceTestBase::SetUp();
  InitializeEmptyExtensionService();
  extension_ = ExtensionBuilder("Test").Build();
}

std::optional<base::Value> ExtensionApiUnittest::RunFunctionAndReturnValue(
    scoped_refptr<ExtensionFunction> function,
    const std::string& args) {
  function->set_extension(extension());
  return utils::RunFunctionAndReturnSingleResult(std::move(function), args,
                                                 profile());
}

std::optional<base::DictValue>
ExtensionApiUnittest::RunFunctionAndReturnDictionary(
    scoped_refptr<ExtensionFunction> function,
    const std::string& args) {
  std::optional<base::Value> value =
      RunFunctionAndReturnValue(std::move(function), args);
  // We expect to either have successfully retrieved a dictionary from the
  // value or the value to have been nullopt.
  EXPECT_TRUE(!value || value->is_dict());

  if (!value || !value->is_dict()) {
    return std::nullopt;
  }

  return std::move(*value).TakeDict();
}

std::optional<base::ListValue> ExtensionApiUnittest::RunFunctionAndReturnList(
    scoped_refptr<ExtensionFunction> function,
    const std::string& args) {
  std::optional<base::Value> value =
      RunFunctionAndReturnValue(std::move(function), args);

  // We expect to have successfully retrieved a list from the value.
  EXPECT_TRUE(!value || value->is_list());

  if (!value || !value->is_list()) {
    return std::nullopt;
  }

  return std::move(*value).TakeList();
}

std::string ExtensionApiUnittest::RunFunctionAndReturnError(
    scoped_refptr<ExtensionFunction> function,
    const std::string& args) {
  function->set_extension(extension());
  return utils::RunFunctionAndReturnError(std::move(function), args, profile());
}

void ExtensionApiUnittest::RunFunction(
    scoped_refptr<ExtensionFunction> function,
    const std::string& args) {
  RunFunctionAndReturnValue(std::move(function), args);
}

}  // namespace extensions