Index: third_party/WebKit/Source/core/dom/DocumentTest.cpp |
diff --git a/third_party/WebKit/Source/core/dom/DocumentTest.cpp b/third_party/WebKit/Source/core/dom/DocumentTest.cpp |
index 673536c1c9554689d462dfbacefdfc037a5a4480..d06b908fbdbaf9f1fe161442c382920a4ad7efb7 100644 |
--- a/third_party/WebKit/Source/core/dom/DocumentTest.cpp |
+++ b/third_party/WebKit/Source/core/dom/DocumentTest.cpp |
@@ -258,8 +258,12 @@ |
MockValidationMessageClient() { Reset(); } |
void Reset() { |
show_validation_message_was_called = false; |
+ will_unload_document_was_called = false; |
+ document_detached_was_called = false; |
} |
bool show_validation_message_was_called; |
+ bool will_unload_document_was_called; |
+ bool document_detached_was_called; |
// ValidationMessageClient functions. |
void ShowValidationMessage(const Element& anchor, |
@@ -272,6 +276,12 @@ |
void HideValidationMessage(const Element& anchor) override {} |
bool IsValidationMessageVisible(const Element& anchor) override { |
return true; |
+ } |
+ void WillUnloadDocument(const Document&) override { |
+ will_unload_document_was_called = true; |
+ } |
+ void DocumentDetached(const Document&) override { |
+ document_detached_was_called = true; |
} |
void WillBeDestroyed() override {} |
@@ -769,6 +779,8 @@ |
// prepareForCommit() unloads the document, and shutdown. |
GetDocument().GetFrame()->PrepareForCommit(); |
+ EXPECT_TRUE(mock_client->will_unload_document_was_called); |
+ EXPECT_TRUE(mock_client->document_detached_was_called); |
// Unload handler tried to show a validation message, but it should fail. |
EXPECT_FALSE(mock_client->show_validation_message_was_called); |