1/*
2 * Copyright (C) 2010 Igalia S.L.
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 "test_utils.h"
21
22#include <glib.h>
23#include <glib/gstdio.h>
24#include <gtk/gtk.h>
25#include <webkit/webkit.h>
26
27#if GTK_CHECK_VERSION(2, 14, 0)
28
29#define HTML_DOCUMENT_TITLE "<html><head><title>This is the title</title></head><body></body></html>"
30#define HTML_DOCUMENT_ELEMENTS "<html><body><ul><li>1</li><li>2</li><li>3</li></ul></body></html>"
31#define HTML_DOCUMENT_ELEMENTS_CLASS "<html><body><div class=\"test\"></div><div class=\"strange\"></div><div class=\"test\"></div></body></html>"
32#define HTML_DOCUMENT_ELEMENTS_ID "<html><body><div id=\"testok\"></div><div id=\"testbad\">first</div><div id=\"testbad\">second</div></body></html>"
33#define HTML_DOCUMENT_LINKS "<html><head><title>Title</title></head><body><a href=\"about:blank\">blank</a><a href=\"http://www.google.com\">google</a><a href=\"http://www.webkit.org\">webkit</a></body></html>"
34#define HTML_DOCUMENT_IFRAME "<html><head><title>IFrame</title></head><body><iframe id='iframe'></iframe><div id='test'></div></body></html>"
35
36typedef struct {
37    GtkWidget* webView;
38    GMainLoop* loop;
39} DomDocumentFixture;
40
41static gboolean finish_loading(DomDocumentFixture* fixture)
42{
43    if (g_main_loop_is_running(fixture->loop))
44        g_main_loop_quit(fixture->loop);
45
46    return FALSE;
47}
48
49static void dom_document_fixture_setup(DomDocumentFixture* fixture, gconstpointer data)
50{
51    fixture->loop = g_main_loop_new(NULL, TRUE);
52    fixture->webView = webkit_web_view_new();
53    g_object_ref_sink(fixture->webView);
54
55    if (data != NULL)
56        webkit_web_view_load_string(WEBKIT_WEB_VIEW (fixture->webView), (const char*) data, NULL, NULL, NULL);
57
58    g_idle_add((GSourceFunc)finish_loading, fixture);
59    g_main_loop_run(fixture->loop);
60}
61
62static void dom_document_fixture_teardown(DomDocumentFixture* fixture, gconstpointer data)
63{
64    if (fixture->webView)
65        g_object_unref(fixture->webView);
66    g_main_loop_unref(fixture->loop);
67}
68
69static void test_dom_document_title(DomDocumentFixture* fixture, gconstpointer data)
70{
71    g_assert(fixture);
72    WebKitWebView* view = (WebKitWebView*)fixture->webView;
73    g_assert(view);
74    WebKitDOMDocument* document = webkit_web_view_get_dom_document(view);
75    g_assert(document);
76    gchar* title = webkit_dom_document_get_title(document);
77    g_assert(title);
78    g_assert_cmpstr(title, ==, "This is the title");
79    g_free(title);
80    webkit_dom_document_set_title(document, "This is the second title");
81    title = webkit_dom_document_get_title(document);
82    g_assert(title);
83    g_assert_cmpstr(title, ==, "This is the second title");
84    g_free(title);
85}
86
87static void test_dom_document_get_elements_by_tag_name(DomDocumentFixture* fixture, gconstpointer data)
88{
89    g_assert(fixture);
90    WebKitWebView* view = (WebKitWebView*)fixture->webView;
91    g_assert(view);
92    WebKitDOMDocument* document = webkit_web_view_get_dom_document(view);
93    g_assert(document);
94    WebKitDOMNodeList* list = webkit_dom_document_get_elements_by_tag_name(document, "li");
95    g_assert(list);
96    gulong length = webkit_dom_node_list_get_length(list);
97    g_assert_cmpint(length, ==, 3);
98
99    guint i;
100
101    for (i = 0; i < length; i++) {
102        WebKitDOMNode* item = webkit_dom_node_list_item(list, i);
103        g_assert(item);
104        WebKitDOMElement* element = (WebKitDOMElement*)item;
105        g_assert(element);
106        g_assert_cmpstr(webkit_dom_element_get_tag_name(element), ==, "LI");
107        WebKitDOMHTMLElement* htmlElement = (WebKitDOMHTMLElement*)element;
108        char* n = g_strdup_printf("%d", i+1);
109        g_assert_cmpstr(webkit_dom_html_element_get_inner_text(htmlElement), ==, n);
110        g_free(n);
111    }
112
113    g_object_unref(list);
114}
115
116static void test_dom_document_get_elements_by_class_name(DomDocumentFixture* fixture, gconstpointer data)
117{
118    g_assert(fixture);
119    WebKitWebView* view = (WebKitWebView*)fixture->webView;
120    g_assert(view);
121    WebKitDOMDocument* document = webkit_web_view_get_dom_document(view);
122    g_assert(document);
123    WebKitDOMNodeList* list = webkit_dom_document_get_elements_by_class_name(document, "test");
124    g_assert(list);
125    gulong length = webkit_dom_node_list_get_length(list);
126    g_assert_cmpint(length, ==, 2);
127
128    guint i;
129
130    for (i = 0; i < length; i++) {
131        WebKitDOMNode* item = webkit_dom_node_list_item(list, i);
132        g_assert(item);
133        WebKitDOMElement* element = (WebKitDOMElement*)item;
134        g_assert(element);
135        g_assert_cmpstr(webkit_dom_element_get_tag_name(element), ==, "DIV");
136    }
137
138    g_object_unref(list);
139}
140
141static void test_dom_document_get_element_by_id(DomDocumentFixture* fixture, gconstpointer data)
142{
143    g_assert(fixture);
144    WebKitWebView* view = (WebKitWebView*)fixture->webView;
145    g_assert(view);
146    WebKitDOMDocument* document = webkit_web_view_get_dom_document(view);
147    g_assert(document);
148    WebKitDOMElement* element = webkit_dom_document_get_element_by_id(document, "testok");
149    g_assert(element);
150    element = webkit_dom_document_get_element_by_id(document, "this-id-does-not-exist");
151    g_assert(element == 0);
152    /* The DOM spec says the return value is undefined when there's
153     * more than one element with the same id; in our case the first
154     * one will be returned */
155    element = webkit_dom_document_get_element_by_id(document, "testbad");
156    g_assert(element);
157    WebKitDOMHTMLElement* htmlElement = (WebKitDOMHTMLElement*)element;
158    g_assert_cmpstr(webkit_dom_html_element_get_inner_text(htmlElement), ==, "first");
159}
160
161static void test_dom_document_get_links(DomDocumentFixture* fixture, gconstpointer data)
162{
163    g_assert(fixture);
164    WebKitWebView* view = (WebKitWebView*)fixture->webView;
165    g_assert(view);
166    WebKitDOMDocument* document = webkit_web_view_get_dom_document(view);
167    g_assert(document);
168    WebKitDOMHTMLCollection *collection = webkit_dom_document_get_links(document);
169    g_assert(collection);
170    gulong length = webkit_dom_html_collection_get_length(collection);
171    g_assert_cmpint(length, ==, 3);
172
173    guint i;
174
175    for (i = 0; i < length; i++) {
176        static const char* names[] = { "blank", "google", "webkit" };
177        static const char* uris[] = { "about:blank", "http://www.google.com/", "http://www.webkit.org/" };
178        WebKitDOMNode *node = webkit_dom_html_collection_item(collection, i);
179        g_assert(node);
180        WebKitDOMElement* element = (WebKitDOMElement*)node;
181        g_assert_cmpstr(webkit_dom_element_get_tag_name(element), ==, "A");
182        WebKitDOMHTMLElement *htmlElement = (WebKitDOMHTMLElement*)element;
183        g_assert_cmpstr(webkit_dom_html_element_get_inner_text(htmlElement), ==, names[i]);
184        WebKitDOMHTMLAnchorElement *anchor = (WebKitDOMHTMLAnchorElement*)element;
185        g_assert_cmpstr(webkit_dom_html_anchor_element_get_href(anchor), ==, uris[i]);
186    }
187    g_object_unref(collection);
188}
189
190static void weak_notify(gpointer data, GObject* zombie)
191{
192    guint* count = (guint*)data;
193    (*count)++;
194}
195
196static void test_dom_document_garbage_collection(DomDocumentFixture* fixture, gconstpointer data)
197{
198    guint count = 0;
199    g_assert(fixture);
200    WebKitWebView* view = (WebKitWebView*)fixture->webView;
201    g_assert(view);
202    WebKitDOMDocument* document = webkit_web_view_get_dom_document(view);
203    g_assert(document);
204    g_object_weak_ref(G_OBJECT(document), (GWeakNotify)weak_notify, &count);
205    WebKitDOMHTMLHeadElement* head = webkit_dom_document_get_head(document);
206    g_assert(head);
207    g_object_weak_ref(G_OBJECT(head), (GWeakNotify)weak_notify, &count);
208    WebKitDOMHTMLElement* body = webkit_dom_document_get_body(document);
209    g_assert(body);
210    g_object_weak_ref(G_OBJECT(body), (GWeakNotify)weak_notify, &count);
211    WebKitDOMHTMLCollection *collection = webkit_dom_document_get_links(document);
212    g_assert(collection);
213    g_object_weak_ref(G_OBJECT(collection), (GWeakNotify)weak_notify, &count);
214
215    webkit_web_view_load_string(WEBKIT_WEB_VIEW(view), HTML_DOCUMENT_LINKS, NULL, NULL, NULL);
216
217    while (g_main_context_pending(NULL))
218        g_main_context_iteration(NULL, FALSE);
219
220    g_assert_cmpuint(count, ==, 3);
221
222    g_object_unref(collection);
223    g_assert_cmpuint(count, ==, 4);
224
225    count = 0;
226
227    document = webkit_web_view_get_dom_document(view);
228    g_assert(document);
229    g_object_weak_ref(G_OBJECT(document), (GWeakNotify)weak_notify, &count);
230    head = webkit_dom_document_get_head(document);
231    g_assert(head);
232    g_object_weak_ref(G_OBJECT(head), (GWeakNotify)weak_notify, &count);
233    body = webkit_dom_document_get_body(document);
234    g_assert(body);
235    g_object_weak_ref(G_OBJECT(body), (GWeakNotify)weak_notify, &count);
236    collection = webkit_dom_document_get_links(document);
237    g_assert(collection);
238    g_object_weak_ref(G_OBJECT(collection), (GWeakNotify)weak_notify, &count);
239    /* Ask twice for the same object */
240    WebKitDOMHTMLCollection* collection2 = webkit_dom_document_get_links(document);
241    g_assert(collection2);
242    g_object_weak_ref(G_OBJECT(collection2), (GWeakNotify)weak_notify, &count);
243
244    g_object_unref(document);
245    g_object_unref(head);
246    g_object_unref(body);
247    g_object_unref(collection);
248    g_object_unref(collection2);
249
250    g_assert_cmpuint(count, ==, 5);
251
252    webkit_web_view_load_string(WEBKIT_WEB_VIEW(view), HTML_DOCUMENT_IFRAME, NULL, NULL, NULL);
253
254    while (g_main_context_pending(NULL))
255        g_main_context_iteration(NULL, FALSE);
256
257    count = 0;
258
259    document = webkit_web_view_get_dom_document(view);
260    WebKitDOMElement* div = webkit_dom_document_get_element_by_id(document, "test");
261    g_assert(div);
262    g_object_weak_ref(G_OBJECT(div), (GWeakNotify)weak_notify, &count);
263    WebKitDOMElement* iframe = webkit_dom_document_get_element_by_id(document, "iframe");
264    g_assert(iframe);
265
266    webkit_dom_element_set_attribute(iframe, "src", "data:<html><head></head></html>", NULL);
267
268    while (g_main_context_pending(NULL))
269        g_main_context_iteration(NULL, FALSE);
270
271    WebKitDOMDocument* iframeDocument = webkit_dom_html_iframe_element_get_content_document(WEBKIT_DOM_HTML_IFRAME_ELEMENT(iframe));
272    g_assert(iframeDocument);
273    head = webkit_dom_document_get_head(iframeDocument);
274    g_assert(head);
275    g_object_weak_ref(G_OBJECT(head), (GWeakNotify)weak_notify, &count);
276
277    webkit_dom_element_set_attribute(iframe, "src", "about:blank", NULL);
278
279    while (g_main_context_pending(NULL))
280        g_main_context_iteration(NULL, FALSE);
281
282    g_assert_cmpuint(count, ==, 1);
283
284    webkit_web_view_load_string(WEBKIT_WEB_VIEW(view), HTML_DOCUMENT_LINKS, NULL, NULL, NULL);
285
286    while (g_main_context_pending(NULL))
287        g_main_context_iteration(NULL, FALSE);
288
289    g_assert_cmpuint(count, ==, 2);
290
291    count = 0;
292
293    document = webkit_web_view_get_dom_document(view);
294    g_assert(document);
295    g_object_weak_ref(G_OBJECT(document), (GWeakNotify)weak_notify, &count);
296    /* Ask twice for the Document */
297    WebKitDOMDocument* document2 = webkit_web_view_get_dom_document(view);
298    g_assert(document2);
299    g_object_weak_ref(G_OBJECT(document2), (GWeakNotify)weak_notify, &count);
300    head = webkit_dom_document_get_head(document);
301    g_assert(head);
302    g_object_weak_ref(G_OBJECT(head), (GWeakNotify)weak_notify, &count);
303    body = webkit_dom_document_get_body(document);
304    g_assert(body);
305    g_object_weak_ref(G_OBJECT(body), (GWeakNotify)weak_notify, &count);
306    collection = webkit_dom_document_get_links(document);
307    g_assert(collection);
308    g_object_weak_ref(G_OBJECT(collection), (GWeakNotify)weak_notify, &count);
309
310    gtk_widget_destroy(fixture->webView);
311    fixture->webView = NULL;
312
313    g_assert_cmpuint(count, ==, 4);
314
315    g_object_unref(collection);
316
317    g_assert_cmpuint(count, ==, 5);
318}
319
320int main(int argc, char** argv)
321{
322    if (!g_thread_supported())
323        g_thread_init(NULL);
324
325    gtk_test_init(&argc, &argv, NULL);
326
327    g_test_bug_base("https://bugs.webkit.org/");
328
329    g_test_add("/webkit/domdocument/test_title",
330               DomDocumentFixture, HTML_DOCUMENT_TITLE,
331               dom_document_fixture_setup,
332               test_dom_document_title,
333               dom_document_fixture_teardown);
334
335    g_test_add("/webkit/domdocument/test_get_elements_by_tag_name",
336               DomDocumentFixture, HTML_DOCUMENT_ELEMENTS,
337               dom_document_fixture_setup,
338               test_dom_document_get_elements_by_tag_name,
339               dom_document_fixture_teardown);
340
341    g_test_add("/webkit/domdocument/test_get_elements_by_class_name",
342               DomDocumentFixture, HTML_DOCUMENT_ELEMENTS_CLASS,
343               dom_document_fixture_setup,
344               test_dom_document_get_elements_by_class_name,
345               dom_document_fixture_teardown);
346
347    g_test_add("/webkit/domdocument/test_get_element_by_id",
348               DomDocumentFixture, HTML_DOCUMENT_ELEMENTS_ID,
349               dom_document_fixture_setup,
350               test_dom_document_get_element_by_id,
351               dom_document_fixture_teardown);
352
353    g_test_add("/webkit/domdocument/test_get_links",
354               DomDocumentFixture, HTML_DOCUMENT_LINKS,
355               dom_document_fixture_setup,
356               test_dom_document_get_links,
357               dom_document_fixture_teardown);
358
359    g_test_add("/webkit/domdocument/test_garbage_collection",
360               DomDocumentFixture, HTML_DOCUMENT_LINKS,
361               dom_document_fixture_setup,
362               test_dom_document_garbage_collection,
363               dom_document_fixture_teardown);
364
365    return g_test_run();
366}
367
368#else
369int main(int argc, char** argv)
370{
371    g_critical("You will gtk-2.14.0 to run the unit tests. Doing nothing now.");
372    return 0;
373}
374
375#endif
376