| Index: chrome/browser/extensions/api/web_navigation/frame_navigation_state_unittest.cc
|
| diff --git a/chrome/browser/extensions/api/web_navigation/frame_navigation_state_unittest.cc b/chrome/browser/extensions/api/web_navigation/frame_navigation_state_unittest.cc
|
| index 1642f6fe6ad93fd50395c48819f12cc3e09a548b..3d998792d3e4d9fb8fe118500982a7a38419dce5 100644
|
| --- a/chrome/browser/extensions/api/web_navigation/frame_navigation_state_unittest.cc
|
| +++ b/chrome/browser/extensions/api/web_navigation/frame_navigation_state_unittest.cc
|
| @@ -2,204 +2,177 @@
|
| // Use of this source code is governed by a BSD-style license that can be
|
| // found in the LICENSE file.
|
|
|
| +#include "base/macros.h"
|
| #include "base/values.h"
|
| #include "chrome/browser/extensions/api/web_navigation/frame_navigation_state.h"
|
| +#include "chrome/test/base/chrome_render_view_host_test_harness.h"
|
| #include "chrome/test/base/testing_profile.h"
|
| #include "testing/gtest/include/gtest/gtest.h"
|
|
|
| namespace extensions {
|
|
|
| -content::RenderViewHost* fake_rvh =
|
| - reinterpret_cast<content::RenderViewHost*>(31337);
|
| +class FrameNavigationStateTest : public ChromeRenderViewHostTestHarness {
|
| + protected:
|
| + FrameNavigationStateTest() {}
|
| + virtual ~FrameNavigationStateTest() {}
|
| +
|
| + FrameNavigationState navigation_state_;
|
| +
|
| + private:
|
| + DISALLOW_COPY_AND_ASSIGN(FrameNavigationStateTest);
|
| +};
|
|
|
| // Test that a frame is correctly tracked, and removed once the tab contents
|
| // goes away.
|
| -TEST(FrameNavigationStateTest, TrackFrame) {
|
| - FrameNavigationState navigation_state;
|
| - const FrameNavigationState::FrameID frame_id0(-1, fake_rvh);
|
| - const FrameNavigationState::FrameID frame_id1(23, fake_rvh);
|
| - const FrameNavigationState::FrameID frame_id2(42, fake_rvh);
|
| +TEST_F(FrameNavigationStateTest, TrackFrame) {
|
| const GURL url1("http://www.google.com/");
|
| const GURL url2("http://mail.google.com/");
|
|
|
| // Create a main frame.
|
| - EXPECT_FALSE(navigation_state.CanSendEvents(frame_id1));
|
| - EXPECT_FALSE(navigation_state.IsValidFrame(frame_id1));
|
| - navigation_state.TrackFrame(frame_id1, frame_id0, url1, true, false, false);
|
| - navigation_state.SetNavigationCommitted(frame_id1);
|
| - EXPECT_TRUE(navigation_state.CanSendEvents(frame_id1));
|
| - EXPECT_TRUE(navigation_state.IsValidFrame(frame_id1));
|
| + EXPECT_FALSE(navigation_state_.CanSendEvents(main_rfh()));
|
| + EXPECT_FALSE(navigation_state_.IsValidFrame(main_rfh()));
|
| + navigation_state_.TrackFrame(main_rfh(), url1, false, false);
|
| + navigation_state_.SetNavigationCommitted(main_rfh());
|
| + EXPECT_TRUE(navigation_state_.CanSendEvents(main_rfh()));
|
| + EXPECT_TRUE(navigation_state_.IsValidFrame(main_rfh()));
|
|
|
| // Add a sub frame.
|
| - EXPECT_FALSE(navigation_state.CanSendEvents(frame_id2));
|
| - EXPECT_FALSE(navigation_state.IsValidFrame(frame_id2));
|
| - navigation_state.TrackFrame(frame_id2, frame_id1, url2, false, false, false);
|
| - navigation_state.SetNavigationCommitted(frame_id2);
|
| - EXPECT_TRUE(navigation_state.CanSendEvents(frame_id2));
|
| - EXPECT_TRUE(navigation_state.IsValidFrame(frame_id2));
|
| + content::RenderFrameHost* sub_frame =
|
| + content::RenderFrameHostTester::For(main_rfh())->AppendChild("child");
|
| + EXPECT_FALSE(navigation_state_.CanSendEvents(sub_frame));
|
| + EXPECT_FALSE(navigation_state_.IsValidFrame(sub_frame));
|
| + navigation_state_.TrackFrame(sub_frame, url2, false, false);
|
| + navigation_state_.SetNavigationCommitted(sub_frame);
|
| + EXPECT_TRUE(navigation_state_.CanSendEvents(sub_frame));
|
| + EXPECT_TRUE(navigation_state_.IsValidFrame(sub_frame));
|
|
|
| // Check frame state.
|
| - EXPECT_TRUE(navigation_state.IsMainFrame(frame_id1));
|
| - EXPECT_EQ(url1, navigation_state.GetUrl(frame_id1));
|
| - EXPECT_FALSE(navigation_state.IsMainFrame(frame_id2));
|
| - EXPECT_EQ(url2, navigation_state.GetUrl(frame_id2));
|
| - EXPECT_EQ(frame_id1, navigation_state.GetMainFrameID());
|
| + EXPECT_EQ(url1, navigation_state_.GetUrl(main_rfh()));
|
| + EXPECT_EQ(url2, navigation_state_.GetUrl(sub_frame));
|
| + EXPECT_EQ(main_rfh(), navigation_state_.GetLastCommittedMainFrameHost());
|
|
|
| // Drop the frames.
|
| - navigation_state.StopTrackingFramesInRVH(fake_rvh,
|
| - FrameNavigationState::FrameID());
|
| - EXPECT_FALSE(navigation_state.CanSendEvents(frame_id1));
|
| - EXPECT_FALSE(navigation_state.IsValidFrame(frame_id1));
|
| - EXPECT_FALSE(navigation_state.CanSendEvents(frame_id2));
|
| - EXPECT_FALSE(navigation_state.IsValidFrame(frame_id2));
|
| + navigation_state_.StopTrackingFramesInRVH(rvh(), NULL);
|
| + EXPECT_FALSE(navigation_state_.CanSendEvents(main_rfh()));
|
| + EXPECT_FALSE(navigation_state_.IsValidFrame(main_rfh()));
|
| + EXPECT_FALSE(navigation_state_.CanSendEvents(sub_frame));
|
| + EXPECT_FALSE(navigation_state_.IsValidFrame(sub_frame));
|
| }
|
|
|
| // Test that no events can be sent for a frame after an error occurred, but
|
| // before a new navigation happened in this frame.
|
| -TEST(FrameNavigationStateTest, ErrorState) {
|
| - FrameNavigationState navigation_state;
|
| - const FrameNavigationState::FrameID frame_id0(-1, fake_rvh);
|
| - const FrameNavigationState::FrameID frame_id1(42, fake_rvh);
|
| +TEST_F(FrameNavigationStateTest, ErrorState) {
|
| const GURL url("http://www.google.com/");
|
|
|
| - navigation_state.TrackFrame(frame_id1, frame_id0, url, true, false, false);
|
| - EXPECT_TRUE(navigation_state.CanSendEvents(frame_id1));
|
| - EXPECT_FALSE(navigation_state.GetErrorOccurredInFrame(frame_id1));
|
| + navigation_state_.TrackFrame(main_rfh(), url, false, false);
|
| + EXPECT_TRUE(navigation_state_.CanSendEvents(main_rfh()));
|
| + EXPECT_FALSE(navigation_state_.GetErrorOccurredInFrame(main_rfh()));
|
|
|
| // After an error occurred, no further events should be sent.
|
| - navigation_state.SetErrorOccurredInFrame(frame_id1);
|
| - EXPECT_FALSE(navigation_state.CanSendEvents(frame_id1));
|
| - EXPECT_TRUE(navigation_state.GetErrorOccurredInFrame(frame_id1));
|
| + navigation_state_.SetErrorOccurredInFrame(main_rfh());
|
| + EXPECT_FALSE(navigation_state_.CanSendEvents(main_rfh()));
|
| + EXPECT_TRUE(navigation_state_.GetErrorOccurredInFrame(main_rfh()));
|
|
|
| // Navigations to a network error page should be ignored.
|
| - navigation_state.TrackFrame(frame_id1, frame_id0, GURL(), true, true, false);
|
| - EXPECT_FALSE(navigation_state.CanSendEvents(frame_id1));
|
| - EXPECT_TRUE(navigation_state.GetErrorOccurredInFrame(frame_id1));
|
| + navigation_state_.TrackFrame(main_rfh(), GURL(), true, false);
|
| + EXPECT_FALSE(navigation_state_.CanSendEvents(main_rfh()));
|
| + EXPECT_TRUE(navigation_state_.GetErrorOccurredInFrame(main_rfh()));
|
|
|
| // However, when the frame navigates again, it should send events again.
|
| - navigation_state.TrackFrame(frame_id1, frame_id0, url, true, false, false);
|
| - EXPECT_TRUE(navigation_state.CanSendEvents(frame_id1));
|
| - EXPECT_FALSE(navigation_state.GetErrorOccurredInFrame(frame_id1));
|
| + navigation_state_.TrackFrame(main_rfh(), url, false, false);
|
| + EXPECT_TRUE(navigation_state_.CanSendEvents(main_rfh()));
|
| + EXPECT_FALSE(navigation_state_.GetErrorOccurredInFrame(main_rfh()));
|
| }
|
|
|
| // Tests that for a sub frame, no events are send after an error occurred, but
|
| // before a new navigation happened in this frame.
|
| -TEST(FrameNavigationStateTest, ErrorStateFrame) {
|
| - FrameNavigationState navigation_state;
|
| - const FrameNavigationState::FrameID frame_id0(-1, fake_rvh);
|
| - const FrameNavigationState::FrameID frame_id1(23, fake_rvh);
|
| - const FrameNavigationState::FrameID frame_id2(42, fake_rvh);
|
| +TEST_F(FrameNavigationStateTest, ErrorStateFrame) {
|
| const GURL url("http://www.google.com/");
|
|
|
| - navigation_state.TrackFrame(frame_id1, frame_id0, url, true, false, false);
|
| - navigation_state.TrackFrame(frame_id2, frame_id1, url, false, false, false);
|
| - EXPECT_TRUE(navigation_state.CanSendEvents(frame_id1));
|
| - EXPECT_TRUE(navigation_state.CanSendEvents(frame_id2));
|
| + content::RenderFrameHost* sub_frame =
|
| + content::RenderFrameHostTester::For(main_rfh())->AppendChild("child");
|
| + navigation_state_.TrackFrame(main_rfh(), url, false, false);
|
| + navigation_state_.TrackFrame(sub_frame, url, false, false);
|
| + EXPECT_TRUE(navigation_state_.CanSendEvents(main_rfh()));
|
| + EXPECT_TRUE(navigation_state_.CanSendEvents(sub_frame));
|
|
|
| // After an error occurred, no further events should be sent.
|
| - navigation_state.SetErrorOccurredInFrame(frame_id2);
|
| - EXPECT_TRUE(navigation_state.CanSendEvents(frame_id1));
|
| - EXPECT_FALSE(navigation_state.CanSendEvents(frame_id2));
|
| + navigation_state_.SetErrorOccurredInFrame(sub_frame);
|
| + EXPECT_TRUE(navigation_state_.CanSendEvents(main_rfh()));
|
| + EXPECT_FALSE(navigation_state_.CanSendEvents(sub_frame));
|
|
|
| // Navigations to a network error page should be ignored.
|
| - navigation_state.TrackFrame(frame_id2, frame_id1, GURL(), false, true, false);
|
| - EXPECT_TRUE(navigation_state.CanSendEvents(frame_id1));
|
| - EXPECT_FALSE(navigation_state.CanSendEvents(frame_id2));
|
| + navigation_state_.TrackFrame(sub_frame, GURL(), true, false);
|
| + EXPECT_TRUE(navigation_state_.CanSendEvents(main_rfh()));
|
| + EXPECT_FALSE(navigation_state_.CanSendEvents(sub_frame));
|
|
|
| // However, when the frame navigates again, it should send events again.
|
| - navigation_state.TrackFrame(frame_id2, frame_id1, url, false, false, false);
|
| - EXPECT_TRUE(navigation_state.CanSendEvents(frame_id1));
|
| - EXPECT_TRUE(navigation_state.CanSendEvents(frame_id2));
|
| + navigation_state_.TrackFrame(sub_frame, url, false, false);
|
| + EXPECT_TRUE(navigation_state_.CanSendEvents(main_rfh()));
|
| + EXPECT_TRUE(navigation_state_.CanSendEvents(sub_frame));
|
| }
|
|
|
| // Tests that no events are send for a not web-safe scheme.
|
| -TEST(FrameNavigationStateTest, WebSafeScheme) {
|
| - FrameNavigationState navigation_state;
|
| - const FrameNavigationState::FrameID frame_id0(-1, fake_rvh);
|
| - const FrameNavigationState::FrameID frame_id1(23, fake_rvh);
|
| +TEST_F(FrameNavigationStateTest, WebSafeScheme) {
|
| const GURL url("unsafe://www.google.com/");
|
|
|
| - navigation_state.TrackFrame(frame_id1, frame_id0, url, true, false, false);
|
| - EXPECT_FALSE(navigation_state.CanSendEvents(frame_id1));
|
| -}
|
| -
|
| -// Test that parent frame IDs are tracked.
|
| -TEST(FrameNavigationStateTest, ParentFrameID) {
|
| - FrameNavigationState navigation_state;
|
| - const FrameNavigationState::FrameID frame_id0(-1, fake_rvh);
|
| - const FrameNavigationState::FrameID frame_id1(23, fake_rvh);
|
| - const FrameNavigationState::FrameID frame_id2(42, fake_rvh);
|
| - const GURL url("http://www.google.com/");
|
| -
|
| - navigation_state.TrackFrame(frame_id1, frame_id0, url, true, false, false);
|
| - navigation_state.TrackFrame(frame_id2, frame_id1, url, false, false, false);
|
| - EXPECT_TRUE(navigation_state.CanSendEvents(frame_id1));
|
| - EXPECT_TRUE(navigation_state.CanSendEvents(frame_id2));
|
| -
|
| - EXPECT_TRUE(navigation_state.GetParentFrameID(frame_id1) == frame_id0);
|
| - EXPECT_TRUE(navigation_state.GetParentFrameID(frame_id2) == frame_id1);
|
| + navigation_state_.TrackFrame(main_rfh(), url, false, false);
|
| + EXPECT_FALSE(navigation_state_.CanSendEvents(main_rfh()));
|
| }
|
|
|
| // Test for <iframe srcdoc=""> frames.
|
| -TEST(FrameNavigationStateTest, SrcDoc) {
|
| - FrameNavigationState navigation_state;
|
| - const FrameNavigationState::FrameID frame_id0(-1, fake_rvh);
|
| - const FrameNavigationState::FrameID frame_id1(23, fake_rvh);
|
| - const FrameNavigationState::FrameID frame_id2(42, fake_rvh);
|
| +TEST_F(FrameNavigationStateTest, SrcDoc) {
|
| const GURL url("http://www.google.com/");
|
| const GURL blank("about:blank");
|
| const GURL srcdoc("about:srcdoc");
|
|
|
| - navigation_state.TrackFrame(frame_id1, frame_id0, url, true, false, false);
|
| - navigation_state.TrackFrame(frame_id2, frame_id1, blank, false, false, true);
|
| - EXPECT_TRUE(navigation_state.CanSendEvents(frame_id1));
|
| - EXPECT_TRUE(navigation_state.CanSendEvents(frame_id2));
|
| + content::RenderFrameHost* sub_frame =
|
| + content::RenderFrameHostTester::For(main_rfh())->AppendChild("child");
|
| + navigation_state_.TrackFrame(main_rfh(), url, false, false);
|
| + navigation_state_.TrackFrame(sub_frame, blank, false, true);
|
| + EXPECT_TRUE(navigation_state_.CanSendEvents(main_rfh()));
|
| + EXPECT_TRUE(navigation_state_.CanSendEvents(sub_frame));
|
|
|
| - EXPECT_TRUE(navigation_state.GetUrl(frame_id1) == url);
|
| - EXPECT_TRUE(navigation_state.GetUrl(frame_id2) == srcdoc);
|
| + EXPECT_EQ(url, navigation_state_.GetUrl(main_rfh()));
|
| + EXPECT_EQ(srcdoc, navigation_state_.GetUrl(sub_frame));
|
|
|
| - EXPECT_TRUE(navigation_state.IsValidUrl(srcdoc));
|
| + EXPECT_TRUE(navigation_state_.IsValidUrl(srcdoc));
|
| }
|
|
|
| // Test that an individual frame can be detached.
|
| -TEST(FrameNavigationStateTest, DetachFrame) {
|
| - FrameNavigationState navigation_state;
|
| - const FrameNavigationState::FrameID frame_id0(-1, fake_rvh);
|
| - const FrameNavigationState::FrameID frame_id1(23, fake_rvh);
|
| - const FrameNavigationState::FrameID frame_id2(42, fake_rvh);
|
| +TEST_F(FrameNavigationStateTest, DetachFrame) {
|
| const GURL url1("http://www.google.com/");
|
| const GURL url2("http://mail.google.com/");
|
|
|
| // Create a main frame.
|
| - EXPECT_FALSE(navigation_state.CanSendEvents(frame_id1));
|
| - EXPECT_FALSE(navigation_state.IsValidFrame(frame_id1));
|
| - navigation_state.TrackFrame(frame_id1, frame_id0, url1, true, false, false);
|
| - navigation_state.SetNavigationCommitted(frame_id1);
|
| - EXPECT_TRUE(navigation_state.CanSendEvents(frame_id1));
|
| - EXPECT_TRUE(navigation_state.IsValidFrame(frame_id1));
|
| + EXPECT_FALSE(navigation_state_.CanSendEvents(main_rfh()));
|
| + EXPECT_FALSE(navigation_state_.IsValidFrame(main_rfh()));
|
| + navigation_state_.TrackFrame(main_rfh(), url1, false, false);
|
| + navigation_state_.SetNavigationCommitted(main_rfh());
|
| + EXPECT_TRUE(navigation_state_.CanSendEvents(main_rfh()));
|
| + EXPECT_TRUE(navigation_state_.IsValidFrame(main_rfh()));
|
|
|
| // Add a sub frame.
|
| - EXPECT_FALSE(navigation_state.CanSendEvents(frame_id2));
|
| - EXPECT_FALSE(navigation_state.IsValidFrame(frame_id2));
|
| - navigation_state.TrackFrame(frame_id2, frame_id1, url2, false, false, false);
|
| - navigation_state.SetNavigationCommitted(frame_id2);
|
| - EXPECT_TRUE(navigation_state.CanSendEvents(frame_id2));
|
| - EXPECT_TRUE(navigation_state.IsValidFrame(frame_id2));
|
| + content::RenderFrameHost* sub_frame =
|
| + content::RenderFrameHostTester::For(main_rfh())->AppendChild("child");
|
| + EXPECT_FALSE(navigation_state_.CanSendEvents(sub_frame));
|
| + EXPECT_FALSE(navigation_state_.IsValidFrame(sub_frame));
|
| + navigation_state_.TrackFrame(sub_frame, url2, false, false);
|
| + navigation_state_.SetNavigationCommitted(sub_frame);
|
| + EXPECT_TRUE(navigation_state_.CanSendEvents(sub_frame));
|
| + EXPECT_TRUE(navigation_state_.IsValidFrame(sub_frame));
|
|
|
| // Check frame state.
|
| - EXPECT_TRUE(navigation_state.IsMainFrame(frame_id1));
|
| - EXPECT_EQ(url1, navigation_state.GetUrl(frame_id1));
|
| - EXPECT_FALSE(navigation_state.IsMainFrame(frame_id2));
|
| - EXPECT_EQ(url2, navigation_state.GetUrl(frame_id2));
|
| - EXPECT_EQ(frame_id1, navigation_state.GetMainFrameID());
|
| + EXPECT_EQ(url1, navigation_state_.GetUrl(main_rfh()));
|
| + EXPECT_EQ(url2, navigation_state_.GetUrl(sub_frame));
|
| + EXPECT_EQ(main_rfh(), navigation_state_.GetLastCommittedMainFrameHost());
|
|
|
| // Drop one frame.
|
| - navigation_state.FrameDetached(frame_id2);
|
| - EXPECT_TRUE(navigation_state.IsMainFrame(frame_id1));
|
| - EXPECT_EQ(url1, navigation_state.GetUrl(frame_id1));
|
| - EXPECT_EQ(frame_id1, navigation_state.GetMainFrameID());
|
| - EXPECT_FALSE(navigation_state.CanSendEvents(frame_id2));
|
| - EXPECT_FALSE(navigation_state.IsValidFrame(frame_id2));
|
| + navigation_state_.FrameDetached(sub_frame);
|
| + EXPECT_EQ(url1, navigation_state_.GetUrl(main_rfh()));
|
| + EXPECT_EQ(main_rfh(), navigation_state_.GetLastCommittedMainFrameHost());
|
| + EXPECT_FALSE(navigation_state_.CanSendEvents(sub_frame));
|
| + EXPECT_FALSE(navigation_state_.IsValidFrame(sub_frame));
|
| }
|
|
|
| } // namespace extensions
|
|
|