// 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 <memory>

#include "base/strings/utf_string_conversions.h"
#include "chrome/browser/ui/browser.h"
#include "chrome/browser/ui/dialogs/browser_dialogs.h"
#include "chrome/browser/ui/window_metadata/window_metadata_controller.h"
#include "chrome/test/base/in_process_browser_test.h"
#include "content/public/test/browser_test.h"
#include "ui/base/models/dialog_model_field.h"
#include "ui/base/test/test_dialog_model_host.h"

using WindowNamePromptTest = InProcessBrowserTest;

namespace {

void SetTextfieldContents(ui::TestDialogModelHost* host,
                          const std::string& text) {
  host->SetSingleTextfieldText(base::UTF8ToUTF16(text));
}

std::string GetTextfieldContents(ui::TestDialogModelHost* host) {
  return base::UTF16ToUTF8(host->FindSingleTextfield()->text());
}

IN_PROC_BROWSER_TEST_F(WindowNamePromptTest, OpensWithInitialName) {
  WindowMetadataController::From(browser())->SetWindowUserTitle("foobar");

  auto host = std::make_unique<ui::TestDialogModelHost>(
      chrome::CreateWindowNamePromptDialogModelForTesting(browser()));

  EXPECT_EQ(GetTextfieldContents(host.get()), "foobar");
}

IN_PROC_BROWSER_TEST_F(WindowNamePromptTest, AcceptNonemptySetsName) {
  auto host = std::make_unique<ui::TestDialogModelHost>(
      chrome::CreateWindowNamePromptDialogModelForTesting(browser()));

  EXPECT_EQ(GetTextfieldContents(host.get()), "");
  SetTextfieldContents(host.get(), "foo");

  ui::TestDialogModelHost::Accept(std::move(host));

  EXPECT_EQ(WindowMetadataController::From(browser())->user_title(), "foo");
}

IN_PROC_BROWSER_TEST_F(WindowNamePromptTest, AcceptEmptyClearsName) {
  WindowMetadataController::From(browser())->SetWindowUserTitle("foo");

  auto host = std::make_unique<ui::TestDialogModelHost>(
      chrome::CreateWindowNamePromptDialogModelForTesting(browser()));

  EXPECT_EQ(GetTextfieldContents(host.get()), "foo");
  SetTextfieldContents(host.get(), "");

  ui::TestDialogModelHost::Accept(std::move(host));

  EXPECT_EQ(WindowMetadataController::From(browser())->user_title(), "");
}

IN_PROC_BROWSER_TEST_F(WindowNamePromptTest, CancelDoesntTouchName) {
  auto host = std::make_unique<ui::TestDialogModelHost>(
      chrome::CreateWindowNamePromptDialogModelForTesting(browser()));
  SetTextfieldContents(host.get(), "foo");

  ui::TestDialogModelHost::Cancel(std::move(host));

  EXPECT_EQ(WindowMetadataController::From(browser())->user_title(), "");
}

}  // namespace