If force-sign-in policy is enabled, popup warning dialog before window closing if auth token becom…

chrome/browser/ui/views/browser_modal_dialog_unittest.cc

+// Copyright 2017 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/views/browser_modal_dialog.h"
+
+#include <memory>
+
+#include "base/memory/ptr_util.h"
+#include "chrome/browser/ui/browser.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+using ::testing::_;
+using ::testing::Return;
+
+class MockBrowserModalDialog : public BrowserModalDialog {
+ public:
+  MockBrowserModalDialog() {
+    BrowserModalDialogList::GetInstance()->AddDialog(this);
+  }
+  ~MockBrowserModalDialog() override {
+    BrowserModalDialogList::GetInstance()->RemoveDialog(this);
+  }
+  MOCK_METHOD1(ActivateModalDialog, void(Browser*));
+  MOCK_METHOD0(IsShowing, bool());
+};
+
+class BrowserModalDialogTest : public ::testing::Test {
+ public:
+  BrowserModalDialogTest() {}
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(BrowserModalDialogTest);
+};
+
+TEST_F(BrowserModalDialogTest, ShowDialog) {
+  BrowserModalDialogList* dialog_list = BrowserModalDialogList::GetInstance();
+  ASSERT_FALSE(dialog_list->IsShowing());
+  MockBrowserModalDialog dialog;
+  EXPECT_CALL(dialog, IsShowing()).Times(2).WillRepeatedly(Return(true));
+  EXPECT_CALL(dialog, ActivateModalDialog(_)).Times(1);
+  ASSERT_TRUE(dialog_list->IsShowing());
+  dialog_list->ActivateModalDialog(nullptr);
+}
+
+TEST_F(BrowserModalDialogTest, RemoveDialog) {
+  BrowserModalDialogList* dialog_list = BrowserModalDialogList::GetInstance();
+  std::unique_ptr<MockBrowserModalDialog> dialog =
+      base::MakeUnique<MockBrowserModalDialog>();
+  EXPECT_CALL(*dialog, IsShowing()).Times(1).WillRepeatedly(Return(true));
+  ASSERT_TRUE(dialog_list->IsShowing());
+  dialog.reset();
+  ASSERT_FALSE(dialog_list->IsShowing());
+}