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
11using blink::WebFrame;
12
13namespace extensions {
14namespace {
15
16class 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
24TEST_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
94