OLD | NEW |
| (Empty) |
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 "mojo/shell/url_resolver.h" | |
6 | |
7 #include "base/logging.h" | |
8 #include "testing/gtest/include/gtest/gtest.h" | |
9 | |
10 namespace mojo { | |
11 namespace shell { | |
12 namespace test { | |
13 namespace { | |
14 | |
15 typedef testing::Test URLResolverTest; | |
16 | |
17 TEST_F(URLResolverTest, MojoURLsFallThrough) { | |
18 URLResolver resolver; | |
19 resolver.AddURLMapping(GURL("mojo:test"), GURL("mojo:foo")); | |
20 const GURL base_url("file:/base"); | |
21 resolver.SetMojoBaseURL(base_url); | |
22 GURL mapped_url = resolver.ApplyMappings(GURL("mojo:test")); | |
23 std::string resolved(resolver.ResolveMojoURL(mapped_url).spec()); | |
24 // Resolved must start with |base_url|. | |
25 EXPECT_EQ(base_url.spec(), resolved.substr(0, base_url.spec().size())); | |
26 // And must contain foo. | |
27 EXPECT_NE(std::string::npos, resolved.find("foo")); | |
28 } | |
29 | |
30 TEST_F(URLResolverTest, MapURL) { | |
31 URLResolver resolver; | |
32 resolver.AddURLMapping(GURL("https://domokit.org/test.mojo"), | |
33 GURL("file:///mojo/src/out/Debug/test.mojo")); | |
34 GURL mapped_url = | |
35 resolver.ApplyMappings(GURL("https://domokit.org/test.mojo")); | |
36 EXPECT_EQ("file:///mojo/src/out/Debug/test.mojo", mapped_url.spec()); | |
37 } | |
38 | |
39 TEST_F(URLResolverTest, MultipleMapURL) { | |
40 URLResolver resolver; | |
41 resolver.AddURLMapping(GURL("https://a.org/foo"), | |
42 GURL("https://b.org/a/foo")); | |
43 resolver.AddURLMapping(GURL("https://b.org/a/foo"), | |
44 GURL("https://c.org/b/a/foo")); | |
45 GURL mapped_url = resolver.ApplyMappings(GURL("https://a.org/foo")); | |
46 EXPECT_EQ("https://c.org/b/a/foo", mapped_url.spec()); | |
47 } | |
48 | |
49 TEST_F(URLResolverTest, MapOrigin) { | |
50 URLResolver resolver; | |
51 resolver.AddOriginMapping(GURL("https://domokit.org"), | |
52 GURL("file:///mojo/src/out/Debug")); | |
53 GURL mapped_url = | |
54 resolver.ApplyMappings(GURL("https://domokit.org/test.mojo")); | |
55 EXPECT_EQ("file:///mojo/src/out/Debug/test.mojo", mapped_url.spec()); | |
56 } | |
57 | |
58 TEST_F(URLResolverTest, MultipleMapOrigin) { | |
59 URLResolver resolver; | |
60 resolver.AddOriginMapping(GURL("https://a.org"), GURL("https://b.org/a")); | |
61 resolver.AddOriginMapping(GURL("https://b.org"), GURL("https://c.org/b")); | |
62 GURL mapped_url = resolver.ApplyMappings(GURL("https://a.org/foo")); | |
63 EXPECT_EQ("https://c.org/b/a/foo", mapped_url.spec()); | |
64 } | |
65 | |
66 TEST_F(URLResolverTest, MapOriginThenURL) { | |
67 URLResolver resolver; | |
68 resolver.AddOriginMapping(GURL("https://a.org"), GURL("https://b.org/a")); | |
69 resolver.AddURLMapping(GURL("https://b.org/a/foo"), | |
70 GURL("https://c.org/b/a/foo")); | |
71 GURL mapped_url = resolver.ApplyMappings(GURL("https://a.org/foo")); | |
72 EXPECT_EQ("https://c.org/b/a/foo", mapped_url.spec()); | |
73 } | |
74 | |
75 TEST_F(URLResolverTest, MapURLThenOrigin) { | |
76 URLResolver resolver; | |
77 resolver.AddURLMapping(GURL("https://a.org/foo"), | |
78 GURL("https://b.org/a/foo")); | |
79 resolver.AddOriginMapping(GURL("https://b.org"), GURL("https://c.org/b")); | |
80 GURL mapped_url = resolver.ApplyMappings(GURL("https://a.org/foo")); | |
81 EXPECT_EQ("https://c.org/b/a/foo", mapped_url.spec()); | |
82 } | |
83 | |
84 #if defined(OS_POSIX) | |
85 #define ARG_LITERAL(x) x | |
86 #elif defined(OS_WIN) | |
87 #define ARG_LITERAL(x) L ## x | |
88 #endif | |
89 | |
90 TEST_F(URLResolverTest, GetOriginMappings) { | |
91 base::CommandLine::StringVector args; | |
92 args.push_back(ARG_LITERAL("--map-origin=https://a.org=https://b.org/a")); | |
93 std::vector<URLResolver::OriginMapping> mappings = | |
94 URLResolver::GetOriginMappings(args); | |
95 ASSERT_EQ(1U, mappings.size()); | |
96 EXPECT_EQ("https://a.org", mappings[0].origin); | |
97 EXPECT_EQ("https://b.org/a", mappings[0].base_url); | |
98 | |
99 args.clear(); | |
100 args.push_back(ARG_LITERAL("-map-origin=https://a.org=https://b.org/a")); | |
101 mappings = URLResolver::GetOriginMappings(args); | |
102 ASSERT_EQ(1U, mappings.size()); | |
103 EXPECT_EQ("https://a.org", mappings[0].origin); | |
104 EXPECT_EQ("https://b.org/a", mappings[0].base_url); | |
105 | |
106 args.clear(); | |
107 args.push_back(ARG_LITERAL("--map-origin")); | |
108 mappings = URLResolver::GetOriginMappings(args); | |
109 EXPECT_EQ(0U, mappings.size()); | |
110 | |
111 args.clear(); | |
112 args.push_back(ARG_LITERAL("--map-origin=")); | |
113 mappings = URLResolver::GetOriginMappings(args); | |
114 EXPECT_EQ(0U, mappings.size()); | |
115 | |
116 args.clear(); | |
117 args.push_back(ARG_LITERAL("mojo_shell")); | |
118 args.push_back(ARG_LITERAL("--map-origin=https://a.org=https://b.org/a")); | |
119 args.push_back(ARG_LITERAL("--map-origin=https://b.org=https://c.org/b")); | |
120 args.push_back(ARG_LITERAL("https://a.org/foo")); | |
121 mappings = URLResolver::GetOriginMappings(args); | |
122 ASSERT_EQ(2U, mappings.size()); | |
123 EXPECT_EQ("https://a.org", mappings[0].origin); | |
124 EXPECT_EQ("https://b.org/a", mappings[0].base_url); | |
125 EXPECT_EQ("https://b.org", mappings[1].origin); | |
126 EXPECT_EQ("https://c.org/b", mappings[1].base_url); | |
127 } | |
128 | |
129 TEST_F(URLResolverTest, TestQueryForURLMapping) { | |
130 URLResolver resolver; | |
131 resolver.SetMojoBaseURL(GURL("file:/base")); | |
132 resolver.AddURLMapping(GURL("https://a.org/foo"), | |
133 GURL("https://b.org/a/foo")); | |
134 resolver.AddURLMapping(GURL("https://b.org/a/foo"), | |
135 GURL("https://c.org/b/a/foo")); | |
136 GURL mapped_url = resolver.ApplyMappings(GURL("https://a.org/foo?a=b")); | |
137 EXPECT_EQ("https://c.org/b/a/foo?a=b", mapped_url.spec()); | |
138 } | |
139 | |
140 TEST_F(URLResolverTest, TestQueryForBaseURL) { | |
141 URLResolver resolver; | |
142 resolver.SetMojoBaseURL(GURL("file:///base")); | |
143 GURL mapped_url = resolver.ResolveMojoURL(GURL("mojo:foo?a=b")); | |
144 EXPECT_EQ("file:///base/foo.mojo?a=b", mapped_url.spec()); | |
145 } | |
146 | |
147 } // namespace | |
148 } // namespace test | |
149 } // namespace shell | |
150 } // namespace mojo | |
OLD | NEW |