// Copyright 2022 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/ui/supervised_user/extension_install_blocked_by_parent_dialog.h"

#include <string>

#include "base/functional/callback_helpers.h"
#include "base/memory/scoped_refptr.h"
#include "chrome/browser/ui/browser.h"
#include "chrome/browser/ui/tabs/tab_strip_model.h"
#include "chrome/browser/ui/test/test_browser_dialog.h"
#include "content/public/browser/web_contents.h"
#include "content/public/test/browser_test.h"
#include "extensions/common/extension.h"
#include "extensions/common/extension_builder.h"

// TODO(crbug.com/437238062): Remove browser dependency and enable test for
// Desktop Android
class ExtensionInstallBlockedByParentDialogTest : public DialogBrowserTest {
 public:
  ExtensionInstallBlockedByParentDialogTest() = default;
  ~ExtensionInstallBlockedByParentDialogTest() override = default;

  void ShowUi(const std::string& name) override {
    extensions::ExtensionBuilder::Type type =
        extensions::ExtensionBuilder::Type::EXTENSION;
    if (name == "extension") {
      // type EXTENSION is correct.
    } else {
      CHECK_EQ(name, "app");
      type = extensions::ExtensionBuilder::Type::PLATFORM_APP;
    }
    extension_ = extensions::ExtensionBuilder("test extension", type).Build();

    extensions::ShowExtensionInstallBlockedByParentDialog(
        extensions::ExtensionInstalledBlockedByParentDialogAction::kAdd,
        extension_.get(), browser()->tab_strip_model()->GetWebContentsAt(0),
        base::DoNothing());
  }

 private:
  scoped_refptr<const extensions::Extension> extension_;
};

IN_PROC_BROWSER_TEST_F(ExtensionInstallBlockedByParentDialogTest,
                       InvokeUi_extension) {
  ShowAndVerifyUi();
}

IN_PROC_BROWSER_TEST_F(ExtensionInstallBlockedByParentDialogTest,
                       InvokeUi_app) {
  ShowAndVerifyUi();
}