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
server_callback(SoupServer * server,SoupMessage * msg,const char * path,GHashTable * query,SoupClientContext * context,gpointer data)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
web_resource_fixture_setup(WebResourceFixture * fixture,gconstpointer data)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
web_resource_fixture_teardown(WebResourceFixture * fixture,gconstpointer data)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
test_webkit_web_resource_get_url(WebResourceFixture * fixture,gconstpointer data)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
test_webkit_web_resource_get_data(WebResourceFixture * fixture,gconstpointer data)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
test_webkit_web_resource_get_mime_type(WebResourceFixture * fixture,gconstpointer data)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
test_webkit_web_resource_get_encoding(WebResourceFixture * fixture,gconstpointer data)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
test_webkit_web_resource_get_frame_name(WebResourceFixture * fixture,gconstpointer data)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
resource_request_starting_cb(WebKitWebView * web_view,WebKitWebFrame * web_frame,WebKitWebResource * web_resource,WebKitNetworkRequest * request,WebKitNetworkResponse * response,gpointer data)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
notify_load_status_cb(WebKitWebView * web_view,GParamSpec * pspec,gpointer data)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
test_web_resource_loading()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
resource_request_starting_sub_cb(WebKitWebView * web_view,WebKitWebFrame * web_frame,WebKitWebResource * web_resource,WebKitNetworkRequest * request,WebKitNetworkResponse * response,gpointer data)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
notify_load_status_sub_cb(WebKitWebView * web_view,GParamSpec * pspec,gpointer data)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
idle_quit_loop_cb(gpointer data)231 static gboolean idle_quit_loop_cb(gpointer data)
232 {
233 g_main_loop_quit(loop);
234 return FALSE;
235 }
236
test_web_resource_sub_resource_loading()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
main(int argc,char ** argv)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
main(int argc,char ** argv)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
338