| [email protected] | ae26b28 | 2014-05-15 16:40:16 | [diff] [blame^] | 1 | // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 | // Use of this source code is governed by a BSD-style license that can be |
| 3 | // found in the LICENSE file. |
| 4 | |
| 5 | #include "content/public/test/render_view_test.h" |
| 6 | #include "extensions/renderer/script_context.h" |
| 7 | #include "third_party/WebKit/public/web/WebDocument.h" |
| 8 | #include "third_party/WebKit/public/web/WebLocalFrame.h" |
| 9 | #include "url/gurl.h" |
| 10 | |
| 11 | using blink::WebFrame; |
| 12 | |
| 13 | namespace extensions { |
| 14 | namespace { |
| 15 | |
| 16 | class ScriptContextTest : public content::RenderViewTest { |
| 17 | protected: |
| 18 | GURL GetEffectiveDocumentURL(const WebFrame* frame) { |
| 19 | return ScriptContext::GetEffectiveDocumentURL( |
| 20 | frame, frame->document().url(), true); |
| 21 | } |
| 22 | }; |
| 23 | |
| 24 | TEST_F(ScriptContextTest, GetEffectiveDocumentURL) { |
| 25 | GURL top_url("http://example.com/"); |
| 26 | GURL different_url("http://example.net/"); |
| 27 | GURL blank_url("about:blank"); |
| 28 | GURL srcdoc_url("about:srcdoc"); |
| 29 | |
| 30 | const char frame_html[] = |
| 31 | "<iframe name='frame1' srcdoc=\"" |
| 32 | " <iframe name='frame1_1'></iframe>" |
| 33 | " <iframe name='frame1_2' sandbox=''></iframe>" |
| 34 | "\"></iframe>" |
| 35 | "<iframe name='frame2' sandbox='' srcdoc=\"" |
| 36 | " <iframe name='frame2_1'></iframe>" |
| 37 | "\"></iframe>" |
| 38 | "<iframe name='frame3'></iframe>"; |
| 39 | |
| 40 | const char frame3_html[] = "<iframe name='frame3_1'></iframe>"; |
| 41 | |
| 42 | WebFrame* frame = GetMainFrame(); |
| 43 | ASSERT_TRUE(frame); |
| 44 | |
| 45 | frame->loadHTMLString(frame_html, top_url); |
| 46 | do { |
| 47 | ProcessPendingMessages(); |
| 48 | } while (frame->isLoading()); |
| 49 | |
| 50 | WebFrame* frame1 = frame->findChildByName("frame1"); |
| 51 | ASSERT_TRUE(frame1); |
| 52 | WebFrame* frame1_1 = frame1->findChildByName("frame1_1"); |
| 53 | ASSERT_TRUE(frame1_1); |
| 54 | WebFrame* frame1_2 = frame1->findChildByName("frame1_2"); |
| 55 | ASSERT_TRUE(frame1_2); |
| 56 | WebFrame* frame2 = frame->findChildByName("frame2"); |
| 57 | ASSERT_TRUE(frame2); |
| 58 | WebFrame* frame2_1 = frame2->findChildByName("frame2_1"); |
| 59 | ASSERT_TRUE(frame2_1); |
| 60 | WebFrame* frame3 = frame->findChildByName("frame3"); |
| 61 | ASSERT_TRUE(frame3); |
| 62 | |
| 63 | // Load a blank document in a frame from a different origin. |
| 64 | frame3->loadHTMLString(frame3_html, different_url); |
| 65 | do { |
| 66 | ProcessPendingMessages(); |
| 67 | } while (frame3->isLoading()); |
| 68 | |
| 69 | WebFrame* frame3_1 = frame->findChildByName("frame3"); |
| 70 | ASSERT_TRUE(frame3_1); |
| 71 | |
| 72 | // Top-level frame |
| 73 | EXPECT_EQ(GetEffectiveDocumentURL(frame), top_url); |
| 74 | // top -> srcdoc = inherit |
| 75 | EXPECT_EQ(GetEffectiveDocumentURL(frame1), top_url); |
| 76 | // top -> srcdoc -> about:blank = inherit |
| 77 | EXPECT_EQ(GetEffectiveDocumentURL(frame1_1), top_url); |
| 78 | // top -> srcdoc -> about:blank sandboxed = same URL |
| 79 | EXPECT_EQ(GetEffectiveDocumentURL(frame1_2), blank_url); |
| 80 | |
| 81 | // top -> srcdoc [sandboxed] = same URL |
| 82 | EXPECT_EQ(GetEffectiveDocumentURL(frame2), srcdoc_url); |
| 83 | // top -> srcdoc [sandboxed] -> about:blank = same URL |
| 84 | EXPECT_EQ(GetEffectiveDocumentURL(frame2_1), blank_url); |
| 85 | |
| 86 | // top -> different origin = different origin |
| 87 | EXPECT_EQ(GetEffectiveDocumentURL(frame3), different_url); |
| 88 | // top -> different origin -> about:blank = inherit |
| 89 | EXPECT_EQ(GetEffectiveDocumentURL(frame3_1), different_url); |
| 90 | } |
| 91 | |
| 92 | } // namespace |
| 93 | } // namespace extensions |
