code.vuplus.com Git - vuplus_webkit/blob - Source/WebKit/gtk/tests/testmimehandling.c

   1 /*
   2  * Copyright (C) 2009 Jan Michael Alonzo
   3  * Copyright (C) 2009 Gustavo Noronha Silva
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Library General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Library General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Library General Public License
  16  * along with this library; see the file COPYING.LIB.  If not, write to
  17  * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
  18  * Boston, MA 02110-1301, USA.
  19  */
  20
  21 #include "test_utils.h"
  22
  23 #include <glib.h>
  24 #include <glib/gstdio.h>
  25 #include <libsoup/soup.h>
  26 #include <string.h>
  27 #include <webkit/webkit.h>
  28 #include <unistd.h>
  29
  30 #if GTK_CHECK_VERSION(2, 14, 0)
  31
  32 GMainLoop* loop;
  33 SoupSession *session;
  34 char* base_uri;
  35
  36 /* For real request testing */
  37 static void
  38 server_callback(SoupServer *server, SoupMessage *msg,
  39                  const char *path, GHashTable *query,
  40                  SoupClientContext *context, gpointer data)
  41 {
  42     if (msg->method != SOUP_METHOD_GET) {
  43         soup_message_set_status(msg, SOUP_STATUS_NOT_IMPLEMENTED);
  44         return;
  45     }
  46
  47     soup_message_set_status(msg, SOUP_STATUS_OK);
  48
  49     /* PDF */
  50     if (g_str_equal(path, "/pdf")) {
  51         char* contents;
  52         gsize length;
  53         GError* error = NULL;
  54
  55         g_file_get_contents("test.pdf", &contents, &length, &error);
  56         g_assert(!error);
  57
  58         soup_message_body_append(msg->response_body, SOUP_MEMORY_TAKE, contents, length);
  59     } else if (g_str_equal(path, "/html")) {
  60         char* contents;
  61         gsize length;
  62         GError* error = NULL;
  63
  64         g_file_get_contents("test.html", &contents, &length, &error);
  65         g_assert(!error);
  66
  67         soup_message_body_append(msg->response_body, SOUP_MEMORY_TAKE, contents, length);
  68     } else if (g_str_equal(path, "/text")) {
  69         char* contents;
  70         gsize length;
  71         GError* error = NULL;
  72
  73         soup_message_headers_append(msg->response_headers, "Content-Disposition", "attachment; filename=test.txt");
  74
  75         g_file_get_contents("test.txt", &contents, &length, &error);
  76         g_assert(!error);
  77
  78         soup_message_body_append(msg->response_body, SOUP_MEMORY_TAKE, contents, length);
  79     } else if (g_str_equal(path, "/ogg")) {
  80         char* contents;
  81         gsize length;
  82         GError* error = NULL;
  83
  84         g_file_get_contents("test.ogg", &contents, &length, &error);
  85         g_assert(!error);
  86
  87         soup_message_body_append(msg->response_body, SOUP_MEMORY_TAKE, contents, length);
  88     }
  89
  90     soup_message_body_complete(msg->response_body);
  91 }
  92
  93 static void idle_quit_loop_cb(WebKitWebView* web_view, GParamSpec* pspec, gpointer data)
  94 {
  95     if (webkit_web_view_get_load_status(web_view) == WEBKIT_LOAD_FINISHED ||
  96         webkit_web_view_get_load_status(web_view) == WEBKIT_LOAD_FAILED)
  97         g_main_loop_quit(loop);
  98 }
  99
 100 static gboolean mime_type_policy_decision_requested_cb(WebKitWebView* view, WebKitWebFrame* frame,
 101                                                        WebKitNetworkRequest* request, const char* mime_type,
 102                                                        WebKitWebPolicyDecision* decision, gpointer data)
 103 {
 104     char* type = (char*)data;
 105
 106     if (g_str_equal(type, "pdf")) {
 107         g_assert_cmpstr(mime_type, ==, "application/pdf");
 108         g_assert(!webkit_web_view_can_show_mime_type(view, mime_type));
 109     } else if (g_str_equal(type, "html")) {
 110         g_assert_cmpstr(mime_type, ==, "text/html");
 111         g_assert(webkit_web_view_can_show_mime_type(view, mime_type));
 112     } else if (g_str_equal(type, "text")) {
 113         WebKitNetworkResponse* response = webkit_web_frame_get_network_response(frame);
 114         SoupMessage* message = webkit_network_response_get_message(response);
 115         char* disposition;
 116
 117         g_assert(message);
 118         soup_message_headers_get_content_disposition(message->response_headers,
 119                                                      &disposition, NULL);
 120         g_object_unref(response);
 121
 122         g_assert_cmpstr(disposition, ==, "attachment");
 123         g_free(disposition);
 124
 125         g_assert_cmpstr(mime_type, ==, "text/plain");
 126         g_assert(webkit_web_view_can_show_mime_type(view, mime_type));
 127     } else if (g_str_equal(type, "ogg")) {
 128         g_assert_cmpstr(mime_type, ==, "audio/x-vorbis+ogg");
 129         g_assert(webkit_web_view_can_show_mime_type(view, mime_type));
 130     }
 131
 132     g_free(type);
 133
 134     return FALSE;
 135 }
 136
 137 static void testRemoteMimeType(const void* data)
 138 {
 139     const char* name = (const char*) data;
 140     WebKitWebView* view = WEBKIT_WEB_VIEW(webkit_web_view_new());
 141     g_object_ref_sink(G_OBJECT(view));
 142
 143     loop = g_main_loop_new(NULL, TRUE);
 144
 145     g_object_connect(G_OBJECT(view),
 146                      "signal::notify::load-status", idle_quit_loop_cb, NULL,
 147                      "signal::mime-type-policy-decision-requested", mime_type_policy_decision_requested_cb, g_strdup(name),
 148                      NULL);
 149
 150     char* effective_uri = g_strdup_printf("%s%s", base_uri, name);
 151     webkit_web_view_load_uri(view, effective_uri);
 152     g_free(effective_uri);
 153
 154     g_main_loop_run(loop);
 155
 156     g_object_unref(view);
 157 }
 158
 159 static void testLocalMimeType(const void* data)
 160 {
 161      const char* typeName = (const char*) data;
 162      WebKitWebView* view = WEBKIT_WEB_VIEW(webkit_web_view_new());
 163      g_object_ref_sink(G_OBJECT(view));
 164
 165      loop = g_main_loop_new(NULL, TRUE);
 166
 167      g_object_connect(G_OBJECT(view),
 168                       "signal::notify::load-status", idle_quit_loop_cb, NULL,
 169                       "signal::mime-type-policy-decision-requested", mime_type_policy_decision_requested_cb, g_strdup(typeName),
 170                       NULL);
 171
 172      gchar* filename = g_strdup_printf("test.%s", typeName);
 173      GFile* file = g_file_new_for_path(filename);
 174      g_free(filename);
 175
 176      gchar* fileURI = g_file_get_uri(file);
 177      g_object_unref(file);
 178
 179      webkit_web_view_load_uri(view, fileURI);
 180      g_free(fileURI);
 181
 182      g_main_loop_run(loop);
 183      g_object_unref(view);
 184 }
 185
 186 int main(int argc, char** argv)
 187 {
 188     SoupServer* server;
 189     SoupURI* soup_uri;
 190
 191     g_thread_init(NULL);
 192     gtk_test_init(&argc, &argv, NULL);
 193
 194     /* Hopefully make test independent of the path it's called from. */
 195     testutils_relative_chdir("Source/WebKit/gtk/tests/resources/test.html", argv[0]);
 196
 197     server = soup_server_new(SOUP_SERVER_PORT, 0, NULL);
 198     soup_server_run_async(server);
 199
 200     soup_server_add_handler(server, NULL, server_callback, NULL, NULL);
 201
 202     soup_uri = soup_uri_new("http://127.0.0.1/");
 203     soup_uri_set_port(soup_uri, soup_server_get_port(server));
 204
 205     base_uri = soup_uri_to_string(soup_uri, FALSE);
 206     soup_uri_free(soup_uri);
 207
 208     g_test_bug_base("https://bugs.webkit.org/");
 209     g_test_add_data_func("/webkit/mime/remote-PDF", "pdf", testRemoteMimeType);
 210     g_test_add_data_func("/webkit/mime/remote-HTML", "html", testRemoteMimeType);
 211     g_test_add_data_func("/webkit/mime/remote-TEXT", "text", testRemoteMimeType);
 212     g_test_add_data_func("/webkit/mime/remote-OGG", "ogg", testRemoteMimeType);
 213     g_test_add_data_func("/webkit/mime/local-PDF", "pdf", testLocalMimeType);
 214     g_test_add_data_func("/webkit/mime/local-HTML", "html", testLocalMimeType);
 215     g_test_add_data_func("/webkit/mime/local-TEXT", "text", testLocalMimeType);
 216     g_test_add_data_func("/webkit/mime/local-OGG", "ogg", testLocalMimeType);
 217
 218     return g_test_run();
 219 }
 220
 221 #else
 222 int main(int argc, char** argv)
 223 {
 224     g_critical("You will need gtk-2.14.0 to run the unit tests. Doing nothing now.");
 225     return 0;
 226 }
 227
 228 #endif