Index: content/browser/renderer_host/frame_tree_unittest.cc |
diff --git a/content/browser/renderer_host/frame_tree_unittest.cc b/content/browser/renderer_host/frame_tree_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..264e256c41fcbd04f4c3b087d30e7cb3f31f99b5 |
--- /dev/null |
+++ b/content/browser/renderer_host/frame_tree_unittest.cc |
@@ -0,0 +1,147 @@ |
+// Copyright (c) 2013 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 "content/browser/renderer_host/frame_tree.h" |
+ |
+#include "base/run_loop.h" |
+#include "content/browser/renderer_host/render_frame_host_impl.h" |
+#include "content/public/test/mock_render_process_host.h" |
+#include "content/public/test/test_browser_context.h" |
+#include "content/public/test/test_browser_thread_bundle.h" |
+#include "content/public/test/test_renderer_host.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+ |
+namespace content { |
+namespace { |
+ |
+class FrameTreeTest : public RenderViewHostTestHarness { |
+}; |
+ |
+// The root node never changes during navigation even though its |
+// RenderFrameHost does. |
+// - Swapping main frame doesn't change root node. |
+// - Swapping back to NULL doesn't crash (easier tear-down). |
+// - Main frame does not own RenderFrameHost. |
+TEST_F(FrameTreeTest, RootNode) { |
+ FrameTree frame_tree; |
+ |
+ // Initial state has empty node. |
+ FrameTreeNode* root = frame_tree.GetRootForTesting(); |
+ ASSERT_TRUE(root); |
+ EXPECT_FALSE(frame_tree.GetMainFrame()); |
+ |
+ // Swap in main frame. |
+ RenderFrameHostImpl* dummy = reinterpret_cast<RenderFrameHostImpl*>(0x1); |
+ frame_tree.SwapMainFrame(dummy); |
+ EXPECT_EQ(root, frame_tree.GetRootForTesting()); |
+ EXPECT_EQ(dummy, frame_tree.GetMainFrame()); |
+ |
+ // Move back to NULL. |
Charlie Reis
2013/09/27 19:19:21
It's not immediately clear why we need to support
awong
2013/09/27 20:50:09
It's in the top level comment.
|
+ frame_tree.SwapMainFrame(NULL); |
+ EXPECT_EQ(root, frame_tree.GetRootForTesting()); |
+ EXPECT_FALSE(frame_tree.GetMainFrame()); |
+ |
+ // Move back to an invalid pointer, let the FrameTree go out of scope. Test |
+ // should not crash because the main frame isn't owned. |
+ frame_tree.SwapMainFrame(dummy); |
+} |
+ |
+// Test swapping of main frame resets the render-assgined frame id. |
Charlie Reis
2013/09/27 19:19:21
nit: Test that swapping the main frame
(hard to re
awong
2013/09/27 20:50:09
Done.
|
+// - On creation, frame id is unassigned. |
+// - After a swap, frame id is unassigned. |
+TEST_F(FrameTreeTest, FirstNavigationAfterSwap) { |
+ FrameTree frame_tree; |
+ |
+ ASSERT_TRUE(frame_tree.IsFirstNavigationAfterSwap()); |
Charlie Reis
2013/09/27 19:19:21
These should use EXPECT, right? (ASSERT is for th
awong
2013/09/27 20:50:09
Good catch. done.
|
+ frame_tree.OnFirstNavigationAfterSwap(1); |
+ ASSERT_FALSE(frame_tree.IsFirstNavigationAfterSwap()); |
+ |
+ frame_tree.SwapMainFrame(NULL); |
+ ASSERT_TRUE(frame_tree.IsFirstNavigationAfterSwap()); |
Charlie Reis
2013/09/27 19:19:21
We haven't actually tested anything about the fram
awong
2013/09/27 20:50:09
Done.
|
+} |
+ |
+// Exercise tree manipulation routines. |
+// - Add a series of nodes and verify tree structure. |
Charlie Reis
2013/09/27 19:19:21
nit: One less indent space, like the comments abov
awong
2013/09/27 20:50:09
Done.
|
+// - Remove a series of nodes and verify tree structure. |
+TEST_F(FrameTreeTest, Shape) { |
+ FrameTree frame_tree; |
+ std::string no_children_node("no children node"); |
+ std::string deep_subtree("node with deep subtree"); |
+ |
+ // Ensure the top-level node of the FrameTree is initalized by simulating a |
Charlie Reis
2013/09/27 19:19:21
initialized
awong
2013/09/27 20:50:09
Done.
|
+ // main frame swap here. |
+ RenderFrameHostImpl render_frame_host(process(), &frame_tree, |
+ process()->GetNextRoutingID(), false); |
+ frame_tree.SwapMainFrame(&render_frame_host); |
+ frame_tree.OnFirstNavigationAfterSwap(5); |
+ |
+ // Let's send a series of messages for frame attached and build the |
Charlie Reis
2013/09/27 19:19:21
for frame attached -> to simulate attaching a fram
awong
2013/09/27 20:50:09
rewrote the whole comment.
|
+ // frame tree. |
+ frame_tree.AddFrame(process()->GetNextRoutingID(), 5, 14, std::string()); |
+ frame_tree.AddFrame(process()->GetNextRoutingID(), 5, 15, std::string()); |
+ frame_tree.AddFrame(process()->GetNextRoutingID(), 5, 16, std::string()); |
+ |
+ frame_tree.AddFrame(process()->GetNextRoutingID(), 14, 244, std::string()); |
+ frame_tree.AddFrame(process()->GetNextRoutingID(), 14, 245, std::string()); |
+ |
+ frame_tree.AddFrame(process()->GetNextRoutingID(), 15, 255, no_children_node); |
+ |
+ frame_tree.AddFrame(process()->GetNextRoutingID(), 16, 264, std::string()); |
+ frame_tree.AddFrame(process()->GetNextRoutingID(), 16, 265, std::string()); |
+ frame_tree.AddFrame(process()->GetNextRoutingID(), 16, 266, std::string()); |
+ frame_tree.AddFrame(process()->GetNextRoutingID(), 16, 267, deep_subtree); |
+ frame_tree.AddFrame(process()->GetNextRoutingID(), 16, 268, std::string()); |
+ |
+ frame_tree.AddFrame(process()->GetNextRoutingID(), 267, 365, std::string()); |
+ frame_tree.AddFrame(process()->GetNextRoutingID(), 365, 455, std::string()); |
+ frame_tree.AddFrame(process()->GetNextRoutingID(), 455, 555, std::string()); |
+ frame_tree.AddFrame(process()->GetNextRoutingID(), 555, 655, std::string()); |
+ |
+ // Now, verify the tree structure is as expected. |
+ FrameTreeNode* root = frame_tree.GetRootForTesting(); |
+ EXPECT_EQ(5, root->frame_id()); |
+ EXPECT_EQ(3UL, root->child_count()); |
+ |
+ EXPECT_EQ(2UL, root->child_at(0)->child_count()); |
+ EXPECT_EQ(0UL, root->child_at(0)->child_at(0)->child_count()); |
+ EXPECT_EQ(0UL, root->child_at(0)->child_at(1)->child_count()); |
+ |
+ EXPECT_EQ(1UL, root->child_at(1)->child_count()); |
+ EXPECT_EQ(0UL, root->child_at(1)->child_at(0)->child_count()); |
+ EXPECT_STREQ(no_children_node.c_str(), |
+ root->child_at(1)->child_at(0)->frame_name().c_str()); |
+ |
+ EXPECT_EQ(5UL, root->child_at(2)->child_count()); |
+ EXPECT_EQ(0UL, root->child_at(2)->child_at(0)->child_count()); |
+ EXPECT_EQ(0UL, root->child_at(2)->child_at(1)->child_count()); |
+ EXPECT_EQ(0UL, root->child_at(2)->child_at(2)->child_count()); |
+ EXPECT_EQ(1UL, root->child_at(2)->child_at(3)->child_count()); |
+ EXPECT_STREQ(deep_subtree.c_str(), |
+ root->child_at(2)->child_at(3)->frame_name().c_str()); |
+ EXPECT_EQ(0UL, root->child_at(2)->child_at(4)->child_count()); |
+ |
+ FrameTreeNode* deep_tree = root->child_at(2)->child_at(3)->child_at(0); |
+ EXPECT_EQ(365, deep_tree->frame_id()); |
+ EXPECT_EQ(1UL, deep_tree->child_count()); |
+ EXPECT_EQ(455, deep_tree->child_at(0)->frame_id()); |
+ EXPECT_EQ(1UL, deep_tree->child_at(0)->child_count()); |
+ EXPECT_EQ(555, deep_tree->child_at(0)->child_at(0)->frame_id()); |
+ EXPECT_EQ(1UL, deep_tree->child_at(0)->child_at(0)->child_count()); |
+ EXPECT_EQ(655, deep_tree->child_at(0)->child_at(0)->child_at(0)->frame_id()); |
+ EXPECT_EQ(0UL, |
+ deep_tree->child_at(0)->child_at(0)->child_at(0)->child_count()); |
+ |
+ // Test removing of nodes. |
+ frame_tree.RemoveFrame(555, 655); |
+ EXPECT_EQ(0UL, deep_tree->child_at(0)->child_at(0)->child_count()); |
+ |
+ frame_tree.RemoveFrame(16, 265); |
+ EXPECT_EQ(4UL, root->child_at(2)->child_count()); |
+ |
+ frame_tree.RemoveFrame(5, 15); |
+ EXPECT_EQ(2UL, root->child_count()); |
+} |
+ |
+} // namespace |
+} // namespace content |