// Copyright (c) 2013 The Chromium Embedded Framework 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 "include/base/cef_callback.h" #include "include/test/cef_test_helpers.h" #include "include/wrapper/cef_closure_task.h" #include "tests/ceftests/routing_test_handler.h" #include "tests/gtest/include/gtest/gtest.h" namespace { const char kLifeSpanUrl[] = "https://tests-life-span/test.html"; const char kUnloadDialogText[] = "Are you sure?"; const char kUnloadMsg[] = "LifeSpanTestHandler.Unload"; // Browser side. class LifeSpanTestHandler : public RoutingTestHandler { public: struct Settings { Settings() = default; bool force_close = false; bool add_onunload_handler = false; bool allow_do_close = true; bool accept_before_unload_dialog = true; }; explicit LifeSpanTestHandler(const Settings& settings) : settings_(settings) { // By default no LifeSpan tests call DestroyTest(). SetDestroyTestExpected(false); } void RunTest() override { // Add the resources that we will navigate to/from. std::string page = "Page"; AddResource(kLifeSpanUrl, page, "text/html"); // Create the browser. CreateBrowser(kLifeSpanUrl); // Intentionally don't call SetTestTimeout() for these tests. } void OnAfterCreated(CefRefPtr browser) override { got_after_created_.yes(); RoutingTestHandler::OnAfterCreated(browser); } bool DoClose(CefRefPtr browser) override { if (executing_delay_close_) { return false; } EXPECT_TRUE(browser->IsSame(GetBrowser())); got_do_close_.yes(); if (!settings_.allow_do_close) { // The close will be canceled. ScheduleDelayClose(); } return !settings_.allow_do_close; } void OnBeforeClose(CefRefPtr browser) override { if (!executing_delay_close_) { got_before_close_.yes(); EXPECT_TRUE(browser->IsSame(GetBrowser())); } RoutingTestHandler::OnBeforeClose(browser); } bool OnBeforeUnloadDialog(CefRefPtr browser, const CefString& message_text, bool is_reload, CefRefPtr callback) override { if (executing_delay_close_) { callback->Continue(true, CefString()); return true; } EXPECT_TRUE(browser->IsSame(GetBrowser())); // The message is no longer configurable via JavaScript. // See https://crbug.com/587940. EXPECT_STREQ("Is it OK to leave/reload this page?", message_text.ToString().c_str()); EXPECT_FALSE(is_reload); EXPECT_TRUE(callback.get()); if (!settings_.accept_before_unload_dialog) { // The close will be canceled. ScheduleDelayClose(); } got_before_unload_dialog_.yes(); callback->Continue(settings_.accept_before_unload_dialog, CefString()); return true; } void OnLoadEnd(CefRefPtr browser, CefRefPtr frame, int httpStatusCode) override { got_load_end_.yes(); EXPECT_TRUE(browser->IsSame(GetBrowser())); if (settings_.add_onunload_handler) { // Send the page a user gesture to enable firing of the onbefureunload // handler. See https://crbug.com/707007. CefExecuteJavaScriptWithUserGestureForTests(frame, CefString()); } // Attempt to close the browser. CloseBrowser(browser, settings_.force_close); } bool OnQuery(CefRefPtr browser, CefRefPtr frame, int64_t query_id, const CefString& request, bool persistent, CefRefPtr callback) override { if (request.ToString() == kUnloadMsg) { if (!executing_delay_close_) { got_unload_message_.yes(); } } callback->Success(""); return true; } TrackCallback got_after_created_; TrackCallback got_do_close_; TrackCallback got_before_close_; TrackCallback got_before_unload_dialog_; TrackCallback got_unload_message_; TrackCallback got_load_end_; TrackCallback got_delay_close_; private: // Wait a bit to make sure no additional events are received and then close // the window. void ScheduleDelayClose() { // This test will call DestroyTest(). SetDestroyTestExpected(true); CefPostDelayedTask( TID_UI, base::BindOnce(&LifeSpanTestHandler::DelayClose, this), 100); } void DelayClose() { got_delay_close_.yes(); executing_delay_close_ = true; DestroyTest(); } Settings settings_; // Forces the window to close (bypasses test conditions). bool executing_delay_close_ = false; IMPLEMENT_REFCOUNTING(LifeSpanTestHandler); }; } // namespace TEST(LifeSpanTest, DoCloseAllow) { LifeSpanTestHandler::Settings settings; settings.allow_do_close = true; CefRefPtr handler = new LifeSpanTestHandler(settings); handler->ExecuteTest(); EXPECT_TRUE(handler->got_after_created_); EXPECT_TRUE(handler->got_do_close_); EXPECT_TRUE(handler->got_before_close_); EXPECT_FALSE(handler->got_before_unload_dialog_); EXPECT_TRUE(handler->got_unload_message_); EXPECT_TRUE(handler->got_load_end_); EXPECT_FALSE(handler->got_delay_close_); ReleaseAndWaitForDestructor(handler); } TEST(LifeSpanTest, DoCloseAllowForce) { LifeSpanTestHandler::Settings settings; settings.allow_do_close = true; settings.force_close = true; CefRefPtr handler = new LifeSpanTestHandler(settings); handler->ExecuteTest(); EXPECT_TRUE(handler->got_after_created_); EXPECT_TRUE(handler->got_do_close_); EXPECT_TRUE(handler->got_before_close_); EXPECT_FALSE(handler->got_before_unload_dialog_); EXPECT_TRUE(handler->got_unload_message_); EXPECT_TRUE(handler->got_load_end_); EXPECT_FALSE(handler->got_delay_close_); ReleaseAndWaitForDestructor(handler); } TEST(LifeSpanTest, DoCloseDisallow) { LifeSpanTestHandler::Settings settings; settings.allow_do_close = false; CefRefPtr handler = new LifeSpanTestHandler(settings); handler->ExecuteTest(); EXPECT_TRUE(handler->got_after_created_); EXPECT_TRUE(handler->got_do_close_); EXPECT_FALSE(handler->got_before_close_); EXPECT_FALSE(handler->got_before_unload_dialog_); EXPECT_TRUE(handler->got_unload_message_); EXPECT_TRUE(handler->got_load_end_); EXPECT_TRUE(handler->got_delay_close_); ReleaseAndWaitForDestructor(handler); } TEST(LifeSpanTest, DoCloseDisallowForce) { LifeSpanTestHandler::Settings settings; settings.allow_do_close = false; settings.force_close = true; CefRefPtr handler = new LifeSpanTestHandler(settings); handler->ExecuteTest(); EXPECT_TRUE(handler->got_after_created_); EXPECT_TRUE(handler->got_do_close_); EXPECT_FALSE(handler->got_before_close_); EXPECT_FALSE(handler->got_before_unload_dialog_); EXPECT_TRUE(handler->got_unload_message_); EXPECT_TRUE(handler->got_load_end_); EXPECT_TRUE(handler->got_delay_close_); ReleaseAndWaitForDestructor(handler); } TEST(LifeSpanTest, DoCloseDisallowWithOnUnloadAllow) { LifeSpanTestHandler::Settings settings; settings.allow_do_close = false; settings.add_onunload_handler = true; settings.accept_before_unload_dialog = true; CefRefPtr handler = new LifeSpanTestHandler(settings); handler->ExecuteTest(); EXPECT_TRUE(handler->got_after_created_); EXPECT_TRUE(handler->got_do_close_); EXPECT_FALSE(handler->got_before_close_); EXPECT_TRUE(handler->got_before_unload_dialog_); EXPECT_TRUE(handler->got_unload_message_); EXPECT_TRUE(handler->got_load_end_); EXPECT_TRUE(handler->got_delay_close_); ReleaseAndWaitForDestructor(handler); } TEST(LifeSpanTest, DoCloseAllowWithOnUnloadForce) { LifeSpanTestHandler::Settings settings; settings.allow_do_close = true; settings.add_onunload_handler = true; settings.force_close = true; CefRefPtr handler = new LifeSpanTestHandler(settings); handler->ExecuteTest(); EXPECT_TRUE(handler->got_after_created_); EXPECT_TRUE(handler->got_do_close_); EXPECT_TRUE(handler->got_before_close_); EXPECT_FALSE(handler->got_before_unload_dialog_); EXPECT_TRUE(handler->got_unload_message_); EXPECT_TRUE(handler->got_load_end_); EXPECT_FALSE(handler->got_delay_close_); ReleaseAndWaitForDestructor(handler); } TEST(LifeSpanTest, DoCloseDisallowWithOnUnloadForce) { LifeSpanTestHandler::Settings settings; settings.allow_do_close = false; settings.add_onunload_handler = true; settings.force_close = true; CefRefPtr handler = new LifeSpanTestHandler(settings); handler->ExecuteTest(); EXPECT_TRUE(handler->got_after_created_); EXPECT_TRUE(handler->got_do_close_); EXPECT_FALSE(handler->got_before_close_); EXPECT_FALSE(handler->got_before_unload_dialog_); EXPECT_TRUE(handler->got_unload_message_); EXPECT_TRUE(handler->got_load_end_); EXPECT_TRUE(handler->got_delay_close_); ReleaseAndWaitForDestructor(handler); } TEST(LifeSpanTest, OnUnloadAllow) { LifeSpanTestHandler::Settings settings; settings.add_onunload_handler = true; settings.accept_before_unload_dialog = true; CefRefPtr handler = new LifeSpanTestHandler(settings); handler->ExecuteTest(); EXPECT_TRUE(handler->got_after_created_); EXPECT_TRUE(handler->got_do_close_); EXPECT_TRUE(handler->got_before_close_); EXPECT_TRUE(handler->got_before_unload_dialog_); EXPECT_TRUE(handler->got_unload_message_); EXPECT_TRUE(handler->got_load_end_); EXPECT_FALSE(handler->got_delay_close_); ReleaseAndWaitForDestructor(handler); } TEST(LifeSpanTest, OnUnloadDisallow) { LifeSpanTestHandler::Settings settings; settings.add_onunload_handler = true; settings.accept_before_unload_dialog = false; CefRefPtr handler = new LifeSpanTestHandler(settings); handler->ExecuteTest(); EXPECT_TRUE(handler->got_after_created_); EXPECT_FALSE(handler->got_do_close_); EXPECT_FALSE(handler->got_before_close_); EXPECT_TRUE(handler->got_before_unload_dialog_); EXPECT_FALSE(handler->got_unload_message_); EXPECT_TRUE(handler->got_load_end_); EXPECT_TRUE(handler->got_delay_close_); ReleaseAndWaitForDestructor(handler); }