midori/tests/download.vala

167 lines
6.2 KiB
Vala

/*
Copyright (C) 2012 Christian Dywan <christian@twotoasts.de>
This library is free software; you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public
License as published by the Free Software Foundation; either
version 2.1 of the License, or (at your option) any later version.
See the file COPYING for the full license text.
*/
struct TestCase {
public string data;
public string mime_type;
public string? expected_filename;
public string? expected_extension;
}
const TestCase[] filenames = {
{ "file:///tmp/midori-user/tumblr123.jpg", "image/jpg", "tumblr123.jpg", ".jpg" },
{ "https://green.cat/8019B6/a.b/500.jpg", "image/jpg", "500.jpg", ".jpg" },
{ "http://example.com/file.png", "image/png", "file.png", ".png" },
{ "http://svn.sf.net/doc/doxy_to_dev.xsl.m4?rev=253", "application/xslt+xml", "doxy_to_dev.xsl.m4", ".m4" }
};
static void download_suggestion () {
foreach (var filename in filenames) {
string? result = Midori.Download.get_filename_suggestion_for_uri (
filename.mime_type, filename.data);
Katze.assert_str_equal (filename.data, result, filename.expected_filename);
}
}
static void download_extension () {
foreach (var filename in filenames) {
string? result = Midori.Download.get_extension_for_uri (filename.data);
Katze.assert_str_equal (filename.data, result, filename.expected_extension);
}
}
static void download_unique () {
string folder = Midori.Paths.make_tmp_dir ("cacheXXXXXX");
string filename = Path.build_path (Path.DIR_SEPARATOR_S, folder, "foo.png");
string org_filename = Path.build_path (Path.DIR_SEPARATOR_S, folder, "foo.png");
string unique = Midori.Download.get_unique_filename (org_filename);
Katze.assert_str_equal (folder, unique, filename);
try {
FileUtils.set_contents (filename, "12345");
unique = Midori.Download.get_unique_filename (org_filename);
filename = Path.build_path (Path.DIR_SEPARATOR_S, folder, "foo-0.png");
Katze.assert_str_equal (folder, unique, filename);
FileUtils.set_contents (filename, "12345");
unique = Midori.Download.get_unique_filename (org_filename);
filename = Path.build_path (Path.DIR_SEPARATOR_S, folder, "foo-1.png");
Katze.assert_str_equal (folder, unique, filename);
for (var i = 0; i < 10; i++) {
filename = Path.build_path (Path.DIR_SEPARATOR_S, folder, "foo-%d.png".printf (i));
FileUtils.set_contents (filename, "12345");
}
}
catch (Error error) {
GLib.error (error.message);
}
unique = Midori.Download.get_unique_filename (org_filename);
filename = Path.build_path (Path.DIR_SEPARATOR_S, folder, "foo-10.png");
Katze.assert_str_equal (folder, unique, filename);
DirUtils.remove (folder);
}
void download_properties () {
#if HAVE_LIBSOUP_2_48_0
var test_server = new Soup.Server ("server-header", null, null);
try {
test_server.listen_local (0, Soup.ServerListenOptions.IPV4_ONLY);
}
catch (Error error) {
GLib.error (error.message);
}
var port = test_server.get_uris ().data.port;
#else
var test_address = new Soup.Address ("127.0.0.1", Soup.ADDRESS_ANY_PORT);
test_address.resolve_sync (null);
var test_server = new Soup.Server ("interface", test_address, null);
test_server.run_async ();
var port = test_server.get_port ();
#endif
string test_url = "http://127.0.0.1:%u".printf (port);
test_server.add_handler ("/", (server, msg, path, query, client)=>{
Thread.usleep (1000000); /* 1 seconds in µs */
if ("/not_found/" in path)
msg.set_status (404);
else
msg.set_status (200);
msg.set_response ("text/html", Soup.MemoryUse.COPY, "<body></body>".data);
});
string was_found_url = test_url + "/cat.png";
string not_found_url = test_url + "/not_found/dog.png";
#if HAVE_WEBKIT2
var download = WebKit.WebContext.get_default ().download_uri (was_found_url);
#else
var download = new WebKit.Download (new WebKit.NetworkRequest (was_found_url));
#endif
assert (Midori.Download.get_type (download) == 0);
Midori.Download.set_type (download, Midori.DownloadType.OPEN);
assert (Midori.Download.get_type (download) == Midori.DownloadType.OPEN);
assert (Midori.Download.get_progress (download) == 0.0);
string filename, uri;
try {
FileUtils.close (FileUtils.open_tmp ("XXXXXX", out filename));
uri = Filename.to_uri (filename, null);
}
catch (Error error) {
GLib.error (error.message);
}
#if HAVE_WEBKIT2
download.set_destination (uri);
#else
download.destination_uri = uri;
download.start ();
#endif
string tee = Midori.Download.calculate_tooltip (download);
assert (tee.contains (Path.get_basename (filename)));
assert (Midori.Download.get_progress (download) == 0.0);
download.notify["progress"].connect ((pspec) => {
string tee2 = Midori.Download.calculate_tooltip (download);
assert (tee2.contains (Path.get_basename (filename)));
});
var loop = MainContext.default ();
do { loop.iteration (true); } while (!Midori.Download.is_finished (download));
#if !HAVE_WEBKIT2
assert (download.status == WebKit.DownloadStatus.FINISHED);
#endif
#if HAVE_WEBKIT2
download = WebKit.WebContext.get_default ().download_uri (not_found_url);
#else
download = new WebKit.Download (new WebKit.NetworkRequest (not_found_url));
#endif
#if HAVE_WEBKIT2
download.set_destination (uri);
#else
download.destination_uri = uri;
download.start ();
#endif
do { loop.iteration (true); } while (!Midori.Download.is_finished (download));
#if !HAVE_WEBKIT2
assert (download.status == WebKit.DownloadStatus.ERROR);
#endif
}
void main (string[] args) {
Test.init (ref args);
Midori.App.setup (ref args, null);
Midori.Paths.init (Midori.RuntimeMode.NORMAL, null);
Test.add_func ("/download/suggestion", download_suggestion);
Test.add_func ("/download/extension", download_extension);
Test.add_func ("/download/unique", download_unique);
Test.add_func ("/download/properties", download_properties);
Test.run ();
}