/* * Copyright (C) 2012 Igalia S.L. * Copyright (C) 2012 Intel Corporation * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Library General Public * License as published by the Free Software Foundation; either * version 2 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Library General Public License for more details. * * You should have received a copy of the GNU Library General Public License * along with this library; see the file COPYING.LIB. If not, write to * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, * Boston, MA 02110-1301, USA. */ #include "config.h" #include "UnitTestUtils/EWK2UnitTestBase.h" #include "UnitTestUtils/EWK2UnitTestServer.h" #include #include #include #include using namespace EWK2UnitTest; using namespace WTF; extern EWK2UnitTestEnvironment* environment; static const char FIRST_PARTY_DOMAIN[] = "127.0.0.1"; static const char THIRD_PARTY_DOMAIN[] = "localhost"; static const char INDEX_HTML_STRING[] = "" "

EFLWebKit2 Cookie Manager test

" "

" ""; static void serverCallback(SoupServer* server, SoupMessage* message, const char* path, GHashTable*, SoupClientContext*, gpointer) { 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 (!strcmp(path, "/index.html")) { Eina_Strbuf* buffer = eina_strbuf_new(); eina_strbuf_append_printf(buffer, INDEX_HTML_STRING, soup_server_get_port(server)); soup_message_headers_replace(message->response_headers, "Set-Cookie", "foo=bar; Max-Age=60"); soup_message_body_append(message->response_body, SOUP_MEMORY_TAKE, eina_strbuf_string_steal(buffer), eina_strbuf_length_get(buffer)); eina_strbuf_free(buffer); } else if (!strcmp(path, "/image.png")) soup_message_headers_replace(message->response_headers, "Set-Cookie", "baz=qux; Max-Age=60"); else soup_message_set_status(message, SOUP_STATUS_NOT_FOUND); soup_message_body_complete(message->response_body); } static void getAcceptPolicyCallback(Ewk_Cookie_Accept_Policy policy, Ewk_Error* error, void* event_info) { ASSERT_FALSE(error); Ewk_Cookie_Accept_Policy* ret = static_cast(event_info); *ret = policy; ecore_main_loop_quit(); } static Ewk_Cookie_Accept_Policy getAcceptPolicy(Ewk_Cookie_Manager* manager) { Ewk_Cookie_Accept_Policy policy = EWK_COOKIE_ACCEPT_POLICY_ALWAYS; ewk_cookie_manager_async_accept_policy_get(manager, getAcceptPolicyCallback, &policy); ecore_main_loop_begin(); return policy; } static void getHostnamesWithCookiesCallback(Eina_List* hostnames, Ewk_Error* error, void* event_info) { ASSERT_FALSE(error); Eina_List** ret = static_cast(event_info); Eina_List* l; void* data; EINA_LIST_FOREACH(hostnames, l, data) *ret = eina_list_append(*ret, eina_stringshare_ref(static_cast(data))); ecore_main_loop_quit(); } static Eina_List* getHostnamesWithCookies(Ewk_Cookie_Manager* manager) { Eina_List* ret = 0; ewk_cookie_manager_async_hostnames_with_cookies_get(manager, getHostnamesWithCookiesCallback, &ret); ecore_main_loop_begin(); return ret; } static void freeHostNames(Eina_List* hostnames) { void* data; EINA_LIST_FREE(hostnames, data) eina_stringshare_del(static_cast(data)); } static int countHostnamesWithCookies(Ewk_Cookie_Manager* manager) { Eina_List* hostnames = getHostnamesWithCookies(manager); int count = eina_list_count(hostnames); freeHostNames(hostnames); return count; } static int compareHostNames(const void* hostName1, const void* hostName2) { return strcmp(static_cast(hostName1), static_cast(hostName2)); } TEST_F(EWK2UnitTestBase, ewk_cookie_manager_accept_policy) { OwnPtr httpServer = adoptPtr(new EWK2UnitTestServer); httpServer->run(serverCallback); Ewk_Cookie_Manager* cookieManager = ewk_context_cookie_manager_get(ewk_view_context_get(webView())); ASSERT_TRUE(cookieManager); // Default policy is EWK_COOKIE_ACCEPT_POLICY_NO_THIRD_PARTY. ASSERT_EQ(EWK_COOKIE_ACCEPT_POLICY_NO_THIRD_PARTY, getAcceptPolicy(cookieManager)); ASSERT_TRUE(loadUrlSync(httpServer->getURLForPath("/index.html").data())); Eina_List* hostnames = getHostnamesWithCookies(cookieManager); ASSERT_EQ(1, eina_list_count(hostnames)); ASSERT_STREQ(FIRST_PARTY_DOMAIN, static_cast(eina_list_nth(hostnames, 0))); freeHostNames(hostnames); ewk_cookie_manager_cookies_clear(cookieManager); // Change policy to EWK_COOKIE_ACCEPT_POLICY_ALWAYS ewk_cookie_manager_accept_policy_set(cookieManager, EWK_COOKIE_ACCEPT_POLICY_ALWAYS); ASSERT_EQ(EWK_COOKIE_ACCEPT_POLICY_ALWAYS, getAcceptPolicy(cookieManager)); ASSERT_TRUE(loadUrlSync(httpServer->getURLForPath("/index.html").data())); hostnames = getHostnamesWithCookies(cookieManager); ASSERT_EQ(2, eina_list_count(hostnames)); hostnames = eina_list_sort(hostnames, eina_list_count(hostnames), compareHostNames); ASSERT_STREQ(FIRST_PARTY_DOMAIN, static_cast(eina_list_nth(hostnames, 0))); ASSERT_STREQ(THIRD_PARTY_DOMAIN, static_cast(eina_list_nth(hostnames, 1))); freeHostNames(hostnames); ewk_cookie_manager_cookies_clear(cookieManager); // Change policy to EWK_COOKIE_ACCEPT_POLICY_NEVER ewk_cookie_manager_accept_policy_set(cookieManager, EWK_COOKIE_ACCEPT_POLICY_NEVER); ASSERT_EQ(EWK_COOKIE_ACCEPT_POLICY_NEVER, getAcceptPolicy(cookieManager)); ASSERT_TRUE(loadUrlSync(httpServer->getURLForPath("/index.html").data())); ASSERT_EQ(0, countHostnamesWithCookies(cookieManager)); } void onCookiesChanged(void *eventInfo) { bool* cookiesChanged = static_cast(eventInfo); *cookiesChanged = true; } TEST_F(EWK2UnitTestBase, ewk_cookie_manager_changes_watch) { OwnPtr httpServer = adoptPtr(new EWK2UnitTestServer); httpServer->run(serverCallback); Ewk_Cookie_Manager* cookieManager = ewk_context_cookie_manager_get(ewk_view_context_get(webView())); ASSERT_TRUE(cookieManager); ewk_cookie_manager_accept_policy_set(cookieManager, EWK_COOKIE_ACCEPT_POLICY_ALWAYS); ASSERT_EQ(EWK_COOKIE_ACCEPT_POLICY_ALWAYS, getAcceptPolicy(cookieManager)); // Watch for changes bool cookiesChanged = false; ewk_cookie_manager_changes_watch(cookieManager, onCookiesChanged, &cookiesChanged); // Check for cookie changes notifications ASSERT_TRUE(loadUrlSync(httpServer->getURLForPath("/index.html").data())); while (!cookiesChanged) ecore_main_loop_iterate(); ASSERT_TRUE(cookiesChanged); cookiesChanged = false; ewk_cookie_manager_cookies_clear(cookieManager); while (!cookiesChanged) ecore_main_loop_iterate(); ASSERT_TRUE(cookiesChanged); // Stop watching for notifications ewk_cookie_manager_changes_watch(cookieManager, 0, 0); cookiesChanged = false; ASSERT_TRUE(loadUrlSync(httpServer->getURLForPath("/index.html").data())); ASSERT_EQ(2, countHostnamesWithCookies(cookieManager)); ASSERT_FALSE(cookiesChanged); // Watch again for notifications ewk_cookie_manager_changes_watch(cookieManager, onCookiesChanged, &cookiesChanged); // Make sure we don't get notifications when loading setting an existing persistent storage char textStorage1[] = "/tmp/txt-cookie.XXXXXX"; ASSERT_TRUE(mktemp(textStorage1)); char textStorage2[] = "/tmp/txt-cookie.XXXXXX"; ASSERT_TRUE(mktemp(textStorage2)); ewk_cookie_manager_persistent_storage_set(cookieManager, textStorage1, EWK_COOKIE_PERSISTENT_STORAGE_TEXT); ASSERT_TRUE(loadUrlSync(httpServer->getURLForPath("/index.html").data())); ASSERT_EQ(2, countHostnamesWithCookies(cookieManager)); cookiesChanged = false; ewk_cookie_manager_persistent_storage_set(cookieManager, textStorage2, EWK_COOKIE_PERSISTENT_STORAGE_TEXT); ASSERT_EQ(0, countHostnamesWithCookies(cookieManager)); ewk_cookie_manager_persistent_storage_set(cookieManager, textStorage1, EWK_COOKIE_PERSISTENT_STORAGE_TEXT); ASSERT_EQ(2, countHostnamesWithCookies(cookieManager)); ASSERT_FALSE(cookiesChanged); // Final clean up. ewk_cookie_manager_changes_watch(cookieManager, 0, 0); unlink(textStorage1); unlink(textStorage2); } TEST_F(EWK2UnitTestBase, ewk_cookie_manager_cookies_delete) { OwnPtr httpServer = adoptPtr(new EWK2UnitTestServer); httpServer->run(serverCallback); Ewk_Cookie_Manager* cookieManager = ewk_context_cookie_manager_get(ewk_view_context_get(webView())); ASSERT_TRUE(cookieManager); ewk_cookie_manager_accept_policy_set(cookieManager, EWK_COOKIE_ACCEPT_POLICY_ALWAYS); ASSERT_EQ(EWK_COOKIE_ACCEPT_POLICY_ALWAYS, getAcceptPolicy(cookieManager)); ASSERT_TRUE(loadUrlSync(httpServer->getURLForPath("/index.html").data())); Eina_List* hostnames = getHostnamesWithCookies(cookieManager); ASSERT_EQ(2, eina_list_count(hostnames)); freeHostNames(hostnames); // Delete first party cookie ewk_cookie_manager_hostname_cookies_clear(cookieManager, FIRST_PARTY_DOMAIN); hostnames = getHostnamesWithCookies(cookieManager); ASSERT_EQ(1, eina_list_count(hostnames)); ASSERT_STREQ(THIRD_PARTY_DOMAIN, static_cast(eina_list_nth(hostnames, 0))); freeHostNames(hostnames); // Delete third party cookie ewk_cookie_manager_hostname_cookies_clear(cookieManager, THIRD_PARTY_DOMAIN); ASSERT_EQ(0, countHostnamesWithCookies(cookieManager)); // Get all cookies again ASSERT_TRUE(loadUrlSync(httpServer->getURLForPath("/index.html").data())); ASSERT_EQ(2, countHostnamesWithCookies(cookieManager)); // Clear all cookies ewk_cookie_manager_cookies_clear(cookieManager); ASSERT_EQ(0, countHostnamesWithCookies(cookieManager)); } TEST_F(EWK2UnitTestBase, DISABLED_ewk_cookie_manager_permanent_storage) { OwnPtr httpServer = adoptPtr(new EWK2UnitTestServer); httpServer->run(serverCallback); // Generate unique names for cookie storages. char textStorage[] = "/tmp/txt-cookie.XXXXXX"; ASSERT_TRUE(mktemp(textStorage)); char sqliteStorage[] = "/tmp/sqlite-cookie.XXXXXX"; ASSERT_TRUE(mktemp(sqliteStorage)); Ewk_Cookie_Manager* cookieManager = ewk_context_cookie_manager_get(ewk_view_context_get(webView())); ASSERT_TRUE(cookieManager); ewk_cookie_manager_accept_policy_set(cookieManager, EWK_COOKIE_ACCEPT_POLICY_ALWAYS); ASSERT_EQ(EWK_COOKIE_ACCEPT_POLICY_ALWAYS, getAcceptPolicy(cookieManager)); // Text storage using a new file. ewk_cookie_manager_persistent_storage_set(cookieManager, textStorage, EWK_COOKIE_PERSISTENT_STORAGE_TEXT); ASSERT_EQ(0, countHostnamesWithCookies(cookieManager)); ASSERT_TRUE(loadUrlSync(httpServer->getURLForPath("/index.html").data())); ASSERT_EQ(2, countHostnamesWithCookies(cookieManager)); // SQLite storage using a new file. ewk_cookie_manager_persistent_storage_set(cookieManager, sqliteStorage, EWK_COOKIE_PERSISTENT_STORAGE_SQLITE); ASSERT_EQ(0, countHostnamesWithCookies(cookieManager)); ASSERT_TRUE(loadUrlSync(httpServer->getURLForPath("/index.html").data())); ASSERT_EQ(2, countHostnamesWithCookies(cookieManager)); // Text storage using an existing file. ewk_cookie_manager_persistent_storage_set(cookieManager, textStorage, EWK_COOKIE_PERSISTENT_STORAGE_TEXT); ASSERT_EQ(2, countHostnamesWithCookies(cookieManager)); ewk_cookie_manager_cookies_clear(cookieManager); ASSERT_EQ(0, countHostnamesWithCookies(cookieManager)); // SQLite storage with an existing file. ewk_cookie_manager_persistent_storage_set(cookieManager, sqliteStorage, EWK_COOKIE_PERSISTENT_STORAGE_SQLITE); ASSERT_EQ(2, countHostnamesWithCookies(cookieManager)); ewk_cookie_manager_cookies_clear(cookieManager); ASSERT_EQ(0, countHostnamesWithCookies(cookieManager)); // Final clean up. unlink(textStorage); unlink(sqliteStorage); }