TestLoaderClient.cpp [plain text]
#include "config.h"
#include "LoadTrackingTest.h"
#include "WebKitTestBus.h"
#include "WebKitTestServer.h"
#include "WebViewTest.h"
#include <gtk/gtk.h>
#include <libsoup/soup.h>
#include <wtf/text/CString.h>
static WebKitTestBus* bus;
static WebKitTestServer* kServer;
const char* kDNTHeaderNotPresent = "DNT header not present";
static void testLoadingStatus(LoadTrackingTest* test, gconstpointer data)
{
test->setRedirectURI(kServer->getURIForPath("/normal").data());
test->loadURI(kServer->getURIForPath("/redirect").data());
test->waitUntilLoadFinished();
Vector<LoadTrackingTest::LoadEvents>& events = test->m_loadEvents;
g_assert_cmpint(events.size(), ==, 4);
g_assert_cmpint(events[0], ==, LoadTrackingTest::ProvisionalLoadStarted);
g_assert_cmpint(events[1], ==, LoadTrackingTest::ProvisionalLoadReceivedServerRedirect);
g_assert_cmpint(events[2], ==, LoadTrackingTest::LoadCommitted);
g_assert_cmpint(events[3], ==, LoadTrackingTest::LoadFinished);
}
static void testLoadingError(LoadTrackingTest* test, gconstpointer)
{
test->loadURI(kServer->getURIForPath("/error").data());
test->waitUntilLoadFinished();
Vector<LoadTrackingTest::LoadEvents>& events = test->m_loadEvents;
g_assert_cmpint(events.size(), ==, 3);
g_assert_cmpint(events[0], ==, LoadTrackingTest::ProvisionalLoadStarted);
g_assert_cmpint(events[1], ==, LoadTrackingTest::ProvisionalLoadFailed);
g_assert_cmpint(events[2], ==, LoadTrackingTest::LoadFinished);
}
static void assertNormalLoadHappened(Vector<LoadTrackingTest::LoadEvents>& events)
{
g_assert_cmpint(events.size(), ==, 3);
g_assert_cmpint(events[0], ==, LoadTrackingTest::ProvisionalLoadStarted);
g_assert_cmpint(events[1], ==, LoadTrackingTest::LoadCommitted);
g_assert_cmpint(events[2], ==, LoadTrackingTest::LoadFinished);
}
static void testLoadHtml(LoadTrackingTest* test, gconstpointer)
{
test->loadHtml("<html><body>Hello WebKit-GTK+</body></html>", 0);
test->waitUntilLoadFinished();
assertNormalLoadHappened(test->m_loadEvents);
}
static void testLoadAlternateHTML(LoadTrackingTest* test, gconstpointer)
{
test->loadAlternateHTML("<html><body>Alternate page</body></html>", "http://error-page.foo/", 0);
test->waitUntilLoadFinished();
assertNormalLoadHappened(test->m_loadEvents);
}
static void testLoadPlainText(LoadTrackingTest* test, gconstpointer)
{
test->loadPlainText("Hello WebKit-GTK+");
test->waitUntilLoadFinished();
assertNormalLoadHappened(test->m_loadEvents);
}
static void testLoadRequest(LoadTrackingTest* test, gconstpointer)
{
GRefPtr<WebKitURIRequest> request(webkit_uri_request_new(kServer->getURIForPath("/normal").data()));
test->loadRequest(request.get());
test->waitUntilLoadFinished();
assertNormalLoadHappened(test->m_loadEvents);
}
class LoadStopTrackingTest : public LoadTrackingTest {
public:
MAKE_GLIB_TEST_FIXTURE(LoadStopTrackingTest);
virtual void loadCommitted()
{
LoadTrackingTest::loadCommitted();
webkit_web_view_stop_loading(m_webView);
}
virtual void loadFailed(const gchar* failingURI, GError* error)
{
g_assert(g_error_matches(error, WEBKIT_NETWORK_ERROR, WEBKIT_NETWORK_ERROR_CANCELLED));
LoadTrackingTest::loadFailed(failingURI, error);
}
};
static void testLoadCancelled(LoadStopTrackingTest* test, gconstpointer)
{
test->loadURI(kServer->getURIForPath("/cancelled").data());
test->waitUntilLoadFinished();
Vector<LoadTrackingTest::LoadEvents>& events = test->m_loadEvents;
g_assert_cmpint(events.size(), ==, 4);
g_assert_cmpint(events[0], ==, LoadTrackingTest::ProvisionalLoadStarted);
g_assert_cmpint(events[1], ==, LoadTrackingTest::LoadCommitted);
g_assert_cmpint(events[2], ==, LoadTrackingTest::LoadFailed);
g_assert_cmpint(events[3], ==, LoadTrackingTest::LoadFinished);
}
static void testWebViewTitle(LoadTrackingTest* test, gconstpointer)
{
g_assert(!webkit_web_view_get_title(test->m_webView));
test->loadHtml("<html><head><title>Welcome to WebKit-GTK+!</title></head></html>", 0);
test->waitUntilLoadFinished();
g_assert_cmpstr(webkit_web_view_get_title(test->m_webView), ==, "Welcome to WebKit-GTK+!");
}
static void testWebViewReload(LoadTrackingTest* test, gconstpointer)
{
test->reload();
test->wait(0.25);
test->loadURI(kServer->getURIForPath("/normal").data());
test->waitUntilLoadFinished();
assertNormalLoadHappened(test->m_loadEvents);
test->reload();
test->waitUntilLoadFinished();
assertNormalLoadHappened(test->m_loadEvents);
}
static void testLoadProgress(LoadTrackingTest* test, gconstpointer)
{
test->loadURI(kServer->getURIForPath("/normal").data());
test->waitUntilLoadFinished();
g_assert_cmpfloat(test->m_estimatedProgress, ==, 1);
}
static void testWebViewHistoryLoad(LoadTrackingTest* test, gconstpointer)
{
test->loadURI(kServer->getURIForPath("/normal").data());
test->waitUntilLoadFinished();
assertNormalLoadHappened(test->m_loadEvents);
test->loadURI(kServer->getURIForPath("/normal2").data());
test->waitUntilLoadFinished();
assertNormalLoadHappened(test->m_loadEvents);
test->goBack();
test->waitUntilLoadFinished();
assertNormalLoadHappened(test->m_loadEvents);
test->goForward();
test->waitUntilLoadFinished();
assertNormalLoadHappened(test->m_loadEvents);
}
class ViewURITrackingTest: public LoadTrackingTest {
public:
MAKE_GLIB_TEST_FIXTURE(ViewURITrackingTest);
static void uriChanged(GObject*, GParamSpec*, ViewURITrackingTest* test)
{
g_assert_cmpstr(test->m_activeURI.data(), !=, webkit_web_view_get_uri(test->m_webView));
test->m_activeURI = webkit_web_view_get_uri(test->m_webView);
}
ViewURITrackingTest()
: m_activeURI(webkit_web_view_get_uri(m_webView))
{
g_assert(m_activeURI.isNull());
g_signal_connect(m_webView, "notify::uri", G_CALLBACK(uriChanged), this);
}
void provisionalLoadStarted()
{
checkActiveURI("/redirect");
}
void provisionalLoadReceivedServerRedirect()
{
checkActiveURI("/normal");
}
void loadCommitted()
{
checkActiveURI("/normal");
}
void loadFinished()
{
checkActiveURI("/normal");
LoadTrackingTest::loadFinished();
}
CString m_activeURI;
private:
void checkActiveURI(const char* uri)
{
ASSERT_CMP_CSTRING(m_activeURI, ==, kServer->getURIForPath(uri));
}
};
static void testWebViewActiveURI(ViewURITrackingTest* test, gconstpointer)
{
test->loadURI(kServer->getURIForPath("/redirect").data());
test->waitUntilLoadFinished();
}
class ViewIsLoadingTest: public LoadTrackingTest {
public:
MAKE_GLIB_TEST_FIXTURE(ViewIsLoadingTest);
static void isLoadingChanged(GObject*, GParamSpec*, ViewIsLoadingTest* test)
{
if (webkit_web_view_is_loading(test->m_webView))
test->beginLoad();
else
test->endLoad();
}
ViewIsLoadingTest()
{
g_signal_connect(m_webView, "notify::is-loading", G_CALLBACK(isLoadingChanged), this);
}
void beginLoad()
{
g_assert(m_loadEvents.isEmpty());
g_assert_cmpstr(webkit_web_view_get_uri(m_webView), ==, m_activeURI.data());
}
void endLoad()
{
g_assert(!m_loadEvents.isEmpty());
g_assert(!m_loadEvents.contains(LoadTrackingTest::LoadFinished));
g_assert(!m_loadEvents.contains(LoadTrackingTest::LoadFailed));
}
};
static void testWebViewIsLoading(ViewIsLoadingTest* test, gconstpointer)
{
test->loadURI(kServer->getURIForPath("/normal").data());
test->waitUntilLoadFinished();
test->reload();
test->waitUntilLoadFinished();
test->loadURI(kServer->getURIForPath("/error").data());
test->waitUntilLoadFinished();
test->loadURI(kServer->getURIForPath("/normal").data());
test->waitUntilLoadFinished();
test->loadURI(kServer->getURIForPath("/normal2").data());
test->waitUntilLoadFinished();
test->goBack();
test->waitUntilLoadFinished();
test->goForward();
test->waitUntilLoadFinished();
}
class WebPageURITest: public WebViewTest {
public:
MAKE_GLIB_TEST_FIXTURE(WebPageURITest);
static void webPageURIChangedCallback(GDBusConnection*, const char*, const char*, const char*, const char*, GVariant* result, WebPageURITest* test)
{
const char* uri;
g_variant_get(result, "(&s)", &uri);
test->m_webPageURIs.append(uri);
}
static void webViewURIChanged(GObject*, GParamSpec*, WebPageURITest* test)
{
test->m_webViewURIs.append(webkit_web_view_get_uri(test->m_webView));
}
WebPageURITest()
{
GRefPtr<GDBusProxy> proxy = adoptGRef(bus->createProxy("org.webkit.gtk.WebExtensionTest",
"/org/webkit/gtk/WebExtensionTest", "org.webkit.gtk.WebExtensionTest", m_mainLoop));
m_uriChangedSignalID = g_dbus_connection_signal_subscribe(
g_dbus_proxy_get_connection(proxy.get()),
0,
"org.webkit.gtk.WebExtensionTest",
"URIChanged",
"/org/webkit/gtk/WebExtensionTest",
0,
G_DBUS_SIGNAL_FLAGS_NONE,
reinterpret_cast<GDBusSignalCallback>(webPageURIChangedCallback),
this,
0);
g_assert(m_uriChangedSignalID);
g_signal_connect(m_webView, "notify::uri", G_CALLBACK(webViewURIChanged), this);
}
~WebPageURITest()
{
g_signal_handlers_disconnect_matched(m_webView, G_SIGNAL_MATCH_DATA, 0, 0, 0, 0, this);
g_dbus_connection_signal_unsubscribe(bus->connection(), m_uriChangedSignalID);
}
unsigned m_uriChangedSignalID;
Vector<CString> m_webPageURIs;
Vector<CString> m_webViewURIs;
};
static void testWebPageURI(WebPageURITest* test, gconstpointer)
{
test->loadURI(kServer->getURIForPath("/redirect").data());
test->waitUntilLoadFinished();
g_assert_cmpint(test->m_webPageURIs.size(), ==, test->m_webViewURIs.size());
for (size_t i = 0; i < test->m_webPageURIs.size(); ++i)
ASSERT_CMP_CSTRING(test->m_webPageURIs[i], ==, test->m_webViewURIs[i]);
g_assert_cmpint(test->m_webPageURIs.size(), ==, 2);
ASSERT_CMP_CSTRING(test->m_webPageURIs[0], ==, kServer->getURIForPath("/redirect"));
ASSERT_CMP_CSTRING(test->m_webPageURIs[1], ==, kServer->getURIForPath("/normal"));
}
static void testURIRequestHTTPHeaders(WebViewTest* test, gconstpointer)
{
GRefPtr<WebKitURIRequest> uriRequest = adoptGRef(webkit_uri_request_new("file:///foo/bar"));
g_assert(uriRequest.get());
g_assert_cmpstr(webkit_uri_request_get_uri(uriRequest.get()), ==, "file:///foo/bar");
g_assert(!webkit_uri_request_get_http_headers(uriRequest.get()));
webkit_uri_request_set_uri(uriRequest.get(), kServer->getURIForPath("/do-not-track-header").data());
test->loadRequest(uriRequest.get());
test->waitUntilLoadFinished();
size_t mainResourceDataSize = 0;
const char* mainResourceData = test->mainResourceData(mainResourceDataSize);
g_assert_cmpint(mainResourceDataSize, ==, strlen(kDNTHeaderNotPresent));
g_assert(!strncmp(mainResourceData, kDNTHeaderNotPresent, mainResourceDataSize));
SoupMessageHeaders* headers = webkit_uri_request_get_http_headers(uriRequest.get());
g_assert(headers);
soup_message_headers_append(headers, "DNT", "1");
test->loadRequest(uriRequest.get());
test->waitUntilLoadFinished();
mainResourceData = test->mainResourceData(mainResourceDataSize);
g_assert_cmpint(mainResourceDataSize, ==, 1);
g_assert(!strncmp(mainResourceData, "1", mainResourceDataSize));
test->loadURI(kServer->getURIForPath("/add-do-not-track-header").data());
test->waitUntilLoadFinished();
mainResourceData = test->mainResourceData(mainResourceDataSize);
g_assert_cmpint(mainResourceDataSize, ==, 1);
g_assert(!strncmp(mainResourceData, "1", mainResourceDataSize));
}
static void serverCallback(SoupServer* server, SoupMessage* message, const char* path, GHashTable*, SoupClientContext*, gpointer)
{
static const char* responseString = "<html><body>Testing!Testing!Testing!Testing!Testing!Testing!Testing!"
"Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!"
"Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!"
"Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!"
"Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!"
"Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!"
"Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!"
"Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!Testing!</body></html>";
if (message->method != SOUP_METHOD_GET) {
soup_message_set_status(message, SOUP_STATUS_NOT_IMPLEMENTED);
return;
}
soup_message_set_status(message, SOUP_STATUS_OK);
if (g_str_has_prefix(path, "/normal"))
soup_message_body_append(message->response_body, SOUP_MEMORY_STATIC, responseString, strlen(responseString));
else if (g_str_equal(path, "/error"))
soup_message_set_status(message, SOUP_STATUS_CANT_CONNECT);
else if (g_str_equal(path, "/redirect")) {
soup_message_set_status(message, SOUP_STATUS_MOVED_PERMANENTLY);
soup_message_headers_append(message->response_headers, "Location", "/normal");
} else if (g_str_equal(path, "/cancelled")) {
soup_message_headers_set_encoding(message->response_headers, SOUP_ENCODING_CHUNKED);
soup_message_body_append(message->response_body, SOUP_MEMORY_STATIC, responseString, strlen(responseString));
soup_server_unpause_message(server, message);
return;
} else if (g_str_equal(path, "/do-not-track-header") || g_str_equal(path, "/add-do-not-track-header")) {
const char* doNotTrack = soup_message_headers_get_one(message->request_headers, "DNT");
if (doNotTrack)
soup_message_body_append(message->response_body, SOUP_MEMORY_COPY, doNotTrack, strlen(doNotTrack));
else
soup_message_body_append(message->response_body, SOUP_MEMORY_STATIC, kDNTHeaderNotPresent, strlen(kDNTHeaderNotPresent));
soup_message_set_status(message, SOUP_STATUS_OK);
} else
soup_message_set_status(message, SOUP_STATUS_NOT_FOUND);
soup_message_body_complete(message->response_body);
}
void beforeAll()
{
webkit_web_context_set_web_extensions_directory(webkit_web_context_get_default(), WEBKIT_TEST_WEB_EXTENSIONS_DIR);
bus = new WebKitTestBus();
if (!bus->run())
return;
kServer = new WebKitTestServer();
kServer->run(serverCallback);
LoadTrackingTest::add("WebKitWebView", "loading-status", testLoadingStatus);
LoadTrackingTest::add("WebKitWebView", "loading-error", testLoadingError);
LoadTrackingTest::add("WebKitWebView", "load-html", testLoadHtml);
LoadTrackingTest::add("WebKitWebView", "load-alternate-html", testLoadAlternateHTML);
LoadTrackingTest::add("WebKitWebView", "load-plain-text", testLoadPlainText);
LoadTrackingTest::add("WebKitWebView", "load-request", testLoadRequest);
LoadStopTrackingTest::add("WebKitWebView", "stop-loading", testLoadCancelled);
LoadTrackingTest::add("WebKitWebView", "title", testWebViewTitle);
LoadTrackingTest::add("WebKitWebView", "progress", testLoadProgress);
LoadTrackingTest::add("WebKitWebView", "reload", testWebViewReload);
LoadTrackingTest::add("WebKitWebView", "history-load", testWebViewHistoryLoad);
ViewURITrackingTest::add("WebKitWebView", "active-uri", testWebViewActiveURI);
ViewIsLoadingTest::add("WebKitWebView", "is-loading", testWebViewIsLoading);
WebPageURITest::add("WebKitWebPage", "get-uri", testWebPageURI);
WebViewTest::add("WebKitURIRequest", "http-headers", testURIRequestHTTPHeaders);
}
void afterAll()
{
delete bus;
delete kServer;
}