michael@0: #include <stdio.h>
michael@0: #include <stdlib.h>
michael@0: 
michael@0: #include "TestCommon.h"
michael@0: #include "nsCOMPtr.h"
michael@0: #include "nsIServiceManager.h"
michael@0: #include "nsNetCID.h"
michael@0: #include "nsIURL.h"
michael@0: #include "prinrval.h"
michael@0: #include "nsStringAPI.h"
michael@0: #include "nsComponentManagerUtils.h"
michael@0: 
michael@0: static nsIURL     *test_url = 0;
michael@0: static nsCString   test_param;
michael@0: 
michael@0: static void run_test(const char *testname, int count, void (* testfunc)())
michael@0: {
michael@0:     PRIntervalTime start, end;
michael@0:     start = PR_IntervalNow();
michael@0:     for (; count; --count)
michael@0:         testfunc();
michael@0:     end = PR_IntervalNow();
michael@0:     printf("completed %s test in %u milliseconds\n", testname,
michael@0:             PR_IntervalToMilliseconds(end - start));
michael@0: }
michael@0: 
michael@0: static void set_spec_test()
michael@0: {
michael@0:     test_url->SetSpec(test_param);
michael@0: }
michael@0: 
michael@0: static void get_spec_test()
michael@0: {
michael@0:     nsAutoCString spec;
michael@0:     test_url->GetSpec(spec);
michael@0: }
michael@0: 
michael@0: static void resolve_test()
michael@0: {
michael@0:     nsAutoCString spec;
michael@0:     test_url->Resolve(NS_LITERAL_CSTRING("foo.html?q=45"), spec);
michael@0: }
michael@0: 
michael@0: static void set_scheme_test()
michael@0: {
michael@0:     test_url->SetScheme(NS_LITERAL_CSTRING("foo"));
michael@0: }
michael@0: 
michael@0: static void get_scheme_test()
michael@0: {
michael@0:     nsAutoCString scheme;
michael@0:     test_url->GetScheme(scheme);
michael@0: }
michael@0: 
michael@0: static void host_test()
michael@0: {
michael@0:     nsAutoCString host;
michael@0:     test_url->GetHost(host);
michael@0:     test_url->SetHost(NS_LITERAL_CSTRING("www.yahoo.com"));
michael@0:     test_url->SetHost(host);
michael@0: }
michael@0: 
michael@0: static void set_path_test()
michael@0: {
michael@0:     test_url->SetPath(NS_LITERAL_CSTRING("/some-path/one-the-net/about.html?with-a-query#for-you"));
michael@0: }
michael@0: 
michael@0: static void get_path_test()
michael@0: {
michael@0:     nsAutoCString path;
michael@0:     test_url->GetPath(path);
michael@0: }
michael@0: 
michael@0: static void query_test()
michael@0: {
michael@0:     nsAutoCString query;
michael@0:     test_url->GetQuery(query);
michael@0:     test_url->SetQuery(NS_LITERAL_CSTRING("a=b&d=c&what-ever-you-want-to-be-called=45"));
michael@0:     test_url->SetQuery(query);
michael@0: }
michael@0: 
michael@0: static void ref_test()
michael@0: {
michael@0:     nsAutoCString ref;
michael@0:     test_url->GetRef(ref);
michael@0:     test_url->SetRef(NS_LITERAL_CSTRING("#some-book-mark"));
michael@0:     test_url->SetRef(ref);
michael@0: }
michael@0: 
michael@0: int main(int argc, char **argv)
michael@0: {
michael@0:     if (test_common_init(&argc, &argv) != 0)
michael@0:         return -1;
michael@0: 
michael@0:     if (argc < 2) {
michael@0:         printf("usage: TestURL url [count]\n");
michael@0:         return -1;
michael@0:     }
michael@0: 
michael@0:     int count = 1000;
michael@0:     if (argc == 3)
michael@0:         count = atoi(argv[2]);
michael@0:     else
michael@0:         printf("using a default count of %d\n", count);
michael@0: 
michael@0:     nsCOMPtr<nsIURL> url( do_CreateInstance(NS_STANDARDURL_CONTRACTID) );
michael@0:     if (!url) {
michael@0:         printf("failed to instantiate component: %s\n", NS_STANDARDURL_CONTRACTID);
michael@0:         return -1;
michael@0:     }
michael@0: 
michael@0:     test_url = url;
michael@0:     test_param = argv[1];
michael@0: 
michael@0:     run_test("SetSpec", count, set_spec_test);
michael@0:     run_test("GetSpec", count, get_spec_test);
michael@0:     run_test("Resolve", count, resolve_test);
michael@0:     run_test("SetScheme", count, set_scheme_test);
michael@0:     run_test("GetScheme", count, get_scheme_test);
michael@0:     run_test("[GS]etHost", count, host_test);
michael@0:     run_test("SetPath", count, set_path_test);
michael@0:     run_test("GetPath", count, get_path_test);
michael@0:     run_test("[GS]etQuery", count, query_test);
michael@0:     run_test("[GS]etRef", count, ref_test);
michael@0: 
michael@0:     return 0;
michael@0: }