|
1 /* |
|
2 * Copyright (C) 2009 Jan Michael Alonzo |
|
3 * |
|
4 * This library is free software; you can redistribute it and/or |
|
5 * modify it under the terms of the GNU Library General Public |
|
6 * License as published by the Free Software Foundation; either |
|
7 * version 2 of the License, or (at your option) any later version. |
|
8 * |
|
9 * This library is distributed in the hope that it will be useful, |
|
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
|
12 * Library General Public License for more details. |
|
13 * |
|
14 * You should have received a copy of the GNU Library General Public License |
|
15 * along with this library; see the file COPYING.LIB. If not, write to |
|
16 * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, |
|
17 * Boston, MA 02110-1301, USA. |
|
18 */ |
|
19 |
|
20 #include <glib.h> |
|
21 #include <gtk/gtk.h> |
|
22 #include <libsoup/soup.h> |
|
23 #include <string.h> |
|
24 #include <webkit/webkit.h> |
|
25 |
|
26 #if GLIB_CHECK_VERSION(2, 16, 0) && GTK_CHECK_VERSION(2, 14, 0) |
|
27 |
|
28 #define INDEX_HTML "<html></html>" |
|
29 #define MAIN_HTML "<html><head><script language=\"javascript\" src=\"/javascript.js\"></script></head><body><h1>hah</h1></html>" |
|
30 #define JAVASCRIPT "function blah () { var a = 1; }" |
|
31 |
|
32 GMainLoop* loop; |
|
33 SoupSession *session; |
|
34 char *base_uri; |
|
35 WebKitWebResource* main_resource; |
|
36 WebKitWebResource* sub_resource; |
|
37 |
|
38 typedef struct { |
|
39 WebKitWebResource* webResource; |
|
40 WebKitWebView* webView; |
|
41 } WebResourceFixture; |
|
42 |
|
43 /* For real request testing */ |
|
44 static void |
|
45 server_callback (SoupServer *server, SoupMessage *msg, |
|
46 const char *path, GHashTable *query, |
|
47 SoupClientContext *context, gpointer data) |
|
48 { |
|
49 if (msg->method != SOUP_METHOD_GET) { |
|
50 soup_message_set_status (msg, SOUP_STATUS_NOT_IMPLEMENTED); |
|
51 return; |
|
52 } |
|
53 |
|
54 soup_message_set_status (msg, SOUP_STATUS_OK); |
|
55 |
|
56 /* Redirect */ |
|
57 if (g_str_equal (path, "/")) { |
|
58 soup_message_set_status (msg, SOUP_STATUS_MOVED_PERMANENTLY); |
|
59 |
|
60 soup_message_headers_append (msg->response_headers, |
|
61 "Location", "/index.html"); |
|
62 } else if (g_str_equal (path, "/index.html")) { |
|
63 soup_message_body_append (msg->response_body, |
|
64 SOUP_MEMORY_COPY, |
|
65 INDEX_HTML, |
|
66 strlen (INDEX_HTML)); |
|
67 } else if (g_str_equal (path, "/main.html")) { |
|
68 soup_message_body_append (msg->response_body, |
|
69 SOUP_MEMORY_COPY, |
|
70 MAIN_HTML, |
|
71 strlen (MAIN_HTML)); |
|
72 } else if (g_str_equal (path, "/javascript.js")) { |
|
73 soup_message_body_append (msg->response_body, |
|
74 SOUP_MEMORY_COPY, |
|
75 JAVASCRIPT, |
|
76 strlen (JAVASCRIPT)); |
|
77 } |
|
78 |
|
79 |
|
80 soup_message_body_complete (msg->response_body); |
|
81 } |
|
82 |
|
83 static void web_resource_fixture_setup(WebResourceFixture* fixture, gconstpointer data) |
|
84 { |
|
85 fixture->webView = WEBKIT_WEB_VIEW(webkit_web_view_new()); |
|
86 g_object_ref_sink(fixture->webView); |
|
87 const gchar* webData = "<html></html>"; |
|
88 fixture->webResource = webkit_web_resource_new(webData, strlen(webData), "http://example.com/", "text/html", "utf8", "Example.com"); |
|
89 g_assert(fixture->webResource); |
|
90 } |
|
91 |
|
92 static void web_resource_fixture_teardown(WebResourceFixture* fixture, gconstpointer data) |
|
93 { |
|
94 g_assert(fixture->webResource); |
|
95 g_object_unref(fixture->webResource); |
|
96 g_object_unref(fixture->webView); |
|
97 } |
|
98 |
|
99 static void test_webkit_web_resource_get_url(WebResourceFixture* fixture, gconstpointer data) |
|
100 { |
|
101 gchar* url; |
|
102 g_object_get(G_OBJECT(fixture->webResource), "uri", &url, NULL); |
|
103 g_assert_cmpstr(url, ==, "http://example.com/"); |
|
104 g_assert_cmpstr(webkit_web_resource_get_uri(fixture->webResource) ,==,"http://example.com/"); |
|
105 g_free(url); |
|
106 } |
|
107 |
|
108 static void test_webkit_web_resource_get_data(WebResourceFixture* fixture, gconstpointer data) |
|
109 { |
|
110 GString* charData = webkit_web_resource_get_data(fixture->webResource); |
|
111 g_assert_cmpstr(charData->str, ==, "<html></html>"); |
|
112 } |
|
113 |
|
114 static void test_webkit_web_resource_get_mime_type(WebResourceFixture* fixture, gconstpointer data) |
|
115 { |
|
116 gchar* mime_type; |
|
117 g_object_get(G_OBJECT(fixture->webResource), "mime-type", &mime_type, NULL); |
|
118 g_assert_cmpstr(mime_type, ==, "text/html"); |
|
119 g_assert_cmpstr(webkit_web_resource_get_mime_type(fixture->webResource),==,"text/html"); |
|
120 g_free(mime_type); |
|
121 } |
|
122 |
|
123 static void test_webkit_web_resource_get_encoding(WebResourceFixture* fixture, gconstpointer data) |
|
124 { |
|
125 gchar* text_encoding; |
|
126 g_object_get(G_OBJECT(fixture->webResource), "encoding", &text_encoding, NULL); |
|
127 g_assert_cmpstr(text_encoding, ==, "utf8"); |
|
128 g_assert_cmpstr(webkit_web_resource_get_encoding(fixture->webResource),==,"utf8"); |
|
129 g_free(text_encoding); |
|
130 } |
|
131 |
|
132 static void test_webkit_web_resource_get_frame_name(WebResourceFixture* fixture, gconstpointer data) |
|
133 { |
|
134 gchar* frame_name; |
|
135 g_object_get(G_OBJECT(fixture->webResource), "frame-name", &frame_name, NULL); |
|
136 g_assert_cmpstr(frame_name, ==, "Example.com"); |
|
137 g_assert_cmpstr(webkit_web_resource_get_frame_name(fixture->webResource),==,"Example.com"); |
|
138 g_free(frame_name); |
|
139 } |
|
140 |
|
141 static void resource_request_starting_cb(WebKitWebView* web_view, WebKitWebFrame* web_frame, WebKitWebResource* web_resource, WebKitNetworkRequest* request, WebKitNetworkResponse* response, gpointer data) |
|
142 { |
|
143 gint* been_there = data; |
|
144 *been_there = *been_there + 1; |
|
145 |
|
146 if (*been_there == 1) { |
|
147 g_assert(!main_resource); |
|
148 main_resource = g_object_ref(web_resource); |
|
149 |
|
150 g_assert_cmpstr(webkit_web_resource_get_uri(web_resource), ==, base_uri); |
|
151 |
|
152 /* This should be a redirect, so the response must be NULL */ |
|
153 g_assert(!response); |
|
154 } else if (*been_there == 2) { |
|
155 char* uri = g_strdup_printf("%sindex.html", base_uri); |
|
156 |
|
157 g_assert_cmpstr(webkit_web_resource_get_uri(web_resource), ==, uri); |
|
158 |
|
159 /* Cancel the request. */ |
|
160 webkit_network_request_set_uri(request, "about:blank"); |
|
161 |
|
162 g_free(uri); |
|
163 } |
|
164 } |
|
165 |
|
166 static void notify_load_status_cb(WebKitWebView* web_view, GParamSpec* pspec, gpointer data) |
|
167 { |
|
168 if (webkit_web_view_get_load_status(web_view) == WEBKIT_LOAD_FINISHED) { |
|
169 gboolean* been_there = data; |
|
170 *been_there = TRUE; |
|
171 |
|
172 g_assert_cmpstr(webkit_web_view_get_uri(web_view), ==, "about:blank"); |
|
173 |
|
174 g_main_loop_quit(loop); |
|
175 } |
|
176 } |
|
177 |
|
178 static void test_web_resource_loading() |
|
179 { |
|
180 WebKitWebView* web_view = WEBKIT_WEB_VIEW(webkit_web_view_new()); |
|
181 gint been_to_resource_request_starting = 0; |
|
182 gboolean been_to_load_finished = FALSE; |
|
183 WebKitWebFrame* web_frame; |
|
184 WebKitWebDataSource* data_source; |
|
185 |
|
186 loop = g_main_loop_new(NULL, TRUE); |
|
187 |
|
188 g_object_ref_sink(web_view); |
|
189 |
|
190 g_signal_connect(web_view, "resource-request-starting", |
|
191 G_CALLBACK(resource_request_starting_cb), |
|
192 &been_to_resource_request_starting); |
|
193 |
|
194 g_signal_connect(web_view, "notify::load-status", |
|
195 G_CALLBACK(notify_load_status_cb), |
|
196 &been_to_load_finished); |
|
197 |
|
198 webkit_web_view_load_uri(web_view, base_uri); |
|
199 |
|
200 /* We won't get finished immediately, because of the redirect */ |
|
201 g_main_loop_run(loop); |
|
202 |
|
203 web_frame = webkit_web_view_get_main_frame(web_view); |
|
204 data_source = webkit_web_frame_get_data_source(web_frame); |
|
205 |
|
206 g_assert(main_resource); |
|
207 g_assert(webkit_web_data_source_get_main_resource(data_source) == main_resource); |
|
208 g_object_unref(main_resource); |
|
209 |
|
210 g_assert_cmpint(been_to_resource_request_starting, ==, 2); |
|
211 g_assert_cmpint(been_to_load_finished, ==, TRUE); |
|
212 |
|
213 g_object_unref(web_view); |
|
214 g_main_loop_unref(loop); |
|
215 } |
|
216 |
|
217 static void resource_request_starting_sub_cb(WebKitWebView* web_view, WebKitWebFrame* web_frame, WebKitWebResource* web_resource, WebKitNetworkRequest* request, WebKitNetworkResponse* response, gpointer data) |
|
218 { |
|
219 if (!main_resource) |
|
220 main_resource = g_object_ref(web_resource); |
|
221 else |
|
222 sub_resource = g_object_ref(web_resource); |
|
223 } |
|
224 |
|
225 static void notify_load_status_sub_cb(WebKitWebView* web_view, GParamSpec* pspec, gpointer data) |
|
226 { |
|
227 if (webkit_web_view_get_load_status(web_view) == WEBKIT_LOAD_FINISHED) |
|
228 g_main_loop_quit(loop); |
|
229 } |
|
230 |
|
231 static gboolean idle_quit_loop_cb(gpointer data) |
|
232 { |
|
233 g_main_loop_quit(loop); |
|
234 return FALSE; |
|
235 } |
|
236 |
|
237 static void test_web_resource_sub_resource_loading() |
|
238 { |
|
239 WebKitWebView* web_view = WEBKIT_WEB_VIEW(webkit_web_view_new()); |
|
240 WebKitWebFrame* web_frame; |
|
241 WebKitWebDataSource* data_source; |
|
242 GList* sub_resources; |
|
243 char* uri = g_strdup_printf("%smain.html", base_uri); |
|
244 |
|
245 main_resource = NULL; |
|
246 |
|
247 loop = g_main_loop_new(NULL, TRUE); |
|
248 |
|
249 g_object_ref_sink(web_view); |
|
250 |
|
251 g_signal_connect(web_view, "resource-request-starting", |
|
252 G_CALLBACK(resource_request_starting_sub_cb), |
|
253 NULL); |
|
254 |
|
255 g_signal_connect(web_view, "notify::load-status", |
|
256 G_CALLBACK(notify_load_status_sub_cb), |
|
257 NULL); |
|
258 |
|
259 webkit_web_view_load_uri(web_view, uri); |
|
260 |
|
261 g_main_loop_run(loop); |
|
262 |
|
263 /* The main resource should be loaded; now let's wait for the sub-resource to load */ |
|
264 g_idle_add(idle_quit_loop_cb, NULL); |
|
265 g_main_loop_run(loop); |
|
266 |
|
267 g_assert(main_resource && sub_resource); |
|
268 g_assert(main_resource != sub_resource); |
|
269 |
|
270 web_frame = webkit_web_view_get_main_frame(web_view); |
|
271 data_source = webkit_web_frame_get_data_source(web_frame); |
|
272 |
|
273 g_assert(webkit_web_data_source_get_main_resource(data_source) == main_resource); |
|
274 g_object_unref(main_resource); |
|
275 |
|
276 sub_resources = webkit_web_data_source_get_subresources(data_source); |
|
277 // Expected resources: javascripts.js, favicon.ico |
|
278 g_assert(sub_resources); |
|
279 g_assert(sub_resources->next); |
|
280 g_assert(!sub_resources->next->next); |
|
281 |
|
282 g_assert(WEBKIT_WEB_RESOURCE(sub_resources->data) == sub_resource); |
|
283 |
|
284 g_object_unref(web_view); |
|
285 g_main_loop_unref(loop); |
|
286 } |
|
287 |
|
288 int main(int argc, char** argv) |
|
289 { |
|
290 SoupServer* server; |
|
291 SoupURI* soup_uri; |
|
292 |
|
293 g_thread_init(NULL); |
|
294 gtk_test_init(&argc, &argv, NULL); |
|
295 |
|
296 server = soup_server_new(SOUP_SERVER_PORT, 0, NULL); |
|
297 soup_server_run_async(server); |
|
298 |
|
299 soup_server_add_handler(server, NULL, server_callback, NULL, NULL); |
|
300 |
|
301 soup_uri = soup_uri_new ("http://127.0.0.1/"); |
|
302 soup_uri_set_port(soup_uri, soup_server_get_port(server)); |
|
303 |
|
304 base_uri = soup_uri_to_string(soup_uri, FALSE); |
|
305 soup_uri_free(soup_uri); |
|
306 |
|
307 g_test_bug_base("https://bugs.webkit.org/"); |
|
308 g_test_add("/webkit/webresource/get_url", |
|
309 WebResourceFixture, 0, web_resource_fixture_setup, |
|
310 test_webkit_web_resource_get_url, web_resource_fixture_teardown); |
|
311 g_test_add("/webkit/webresource/get_mime_type", |
|
312 WebResourceFixture, 0, web_resource_fixture_setup, |
|
313 test_webkit_web_resource_get_mime_type, web_resource_fixture_teardown); |
|
314 g_test_add("/webkit/webresource/get_text_encoding_name", |
|
315 WebResourceFixture, 0, web_resource_fixture_setup, |
|
316 test_webkit_web_resource_get_encoding, web_resource_fixture_teardown); |
|
317 g_test_add("/webkit/webresource/get_frame_name", |
|
318 WebResourceFixture, 0, web_resource_fixture_setup, |
|
319 test_webkit_web_resource_get_frame_name, web_resource_fixture_teardown); |
|
320 g_test_add("/webkit/webresource/get_data", |
|
321 WebResourceFixture, 0, web_resource_fixture_setup, |
|
322 test_webkit_web_resource_get_data, web_resource_fixture_teardown); |
|
323 |
|
324 g_test_add_func("/webkit/webresource/loading", test_web_resource_loading); |
|
325 g_test_add_func("/webkit/webresource/sub_resource_loading", test_web_resource_sub_resource_loading); |
|
326 |
|
327 return g_test_run (); |
|
328 } |
|
329 |
|
330 #else |
|
331 int main(int argc, char** argv) |
|
332 { |
|
333 g_critical("You will need at least glib-2.16.0 and gtk-2.14.0 to run the unit tests. Doing nothing now."); |
|
334 return 0; |
|
335 } |
|
336 |
|
337 #endif |