text: // Copyright (c) 2012 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 "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 {
  
  class FrameNavigationStateTest : public ChromeRenderViewHostTestHarness {
   protected:
    FrameNavigationStateTest() {}
    ~FrameNavigationStateTest() override {}
  
    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_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(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.
    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_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(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_F(FrameNavigationStateTest, ErrorState) {
    const GURL url("http://www.google.com/");
  
    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(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(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(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_F(FrameNavigationStateTest, ErrorStateFrame) {
    const GURL url("http://www.google.com/");
  
    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(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(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(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_F(FrameNavigationStateTest, WebSafeScheme) {
    const GURL url("unsafe://www.google.com/");
  
    navigation_state_.TrackFrame(main_rfh(), url, false, false);
    EXPECT_FALSE(navigation_state_.CanSendEvents(main_rfh()));
  }
  
  // Test for <iframe srcdoc=""> frames.
  TEST_F(FrameNavigationStateTest, SrcDoc) {
    const GURL url("http://www.google.com/");
    const GURL blank("about:blank");
    const GURL srcdoc("about:srcdoc");
  
    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_EQ(url, navigation_state_.GetUrl(main_rfh()));
    EXPECT_EQ(srcdoc, navigation_state_.GetUrl(sub_frame));
  
    EXPECT_TRUE(navigation_state_.IsValidUrl(srcdoc));
  }
  
  // Test that an individual frame can be detached.
  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(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.
    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_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(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
  