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

#include "chrome/browser/ui/webui/set_as_default_browser_ui_win.h"

#include "base/command_line.h"
#include "base/win/windows_version.h"
#include "chrome/browser/ui/browser.h"
#include "chrome/browser/ui/browser_window.h"
#include "chrome/common/chrome_switches.h"
#include "chrome/test/base/in_process_browser_test.h"
#include "ui/views/widget/widget.h"

namespace {

bool IsBrowserVisible(Browser* browser) {
  return views::Widget::GetWidgetForNativeWindow(
             browser->window()->GetNativeWindow())
      ->IsVisible();
}

}  // namespace

using SetAsDefaultBrowserUIBrowserTestWithoutFirstRun = InProcessBrowserTest;

class SetAsDefaultBrowserUIBrowserTestWithFirstRun
    : public InProcessBrowserTest {
 public:
  void SetUpCommandLine(base::CommandLine* command_line) override {
    command_line->AppendSwitch(switches::kForceFirstRun);
  }

 protected:
  void TearDownInProcessBrowserTestFixture() override {
    ASSERT_FALSE(SetAsDefaultBrowserUI::GetDialogWidgetForTesting());
  }
};

IN_PROC_BROWSER_TEST_F(SetAsDefaultBrowserUIBrowserTestWithFirstRun, Test) {
  // Windows 8 only test case.
  if (base::win::GetVersion() != base::win::Version::WIN8 &&
      base::win::GetVersion() != base::win::Version::WIN8_1) {
    return;
  }
  ASSERT_FALSE(IsBrowserVisible(browser()));
  views::Widget* dialog_widget =
      SetAsDefaultBrowserUI::GetDialogWidgetForTesting();
  ASSERT_TRUE(dialog_widget);
  ASSERT_TRUE(dialog_widget->IsVisible());
  dialog_widget->CloseNow();
  ASSERT_TRUE(IsBrowserVisible(browser()));
}

IN_PROC_BROWSER_TEST_F(SetAsDefaultBrowserUIBrowserTestWithoutFirstRun,
                       TestWithoutFirstRun) {
  ASSERT_TRUE(IsBrowserVisible(browser()));
  EXPECT_EQ(nullptr, SetAsDefaultBrowserUI::GetDialogWidgetForTesting());
}
