dlls/mshtml/tests/htmllocation.c

   1 /*
   2  * Copyright 2009 Andrew Eikum for CodeWeavers
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  17  */
  18
  19 #define COBJMACROS
  20 #define CONST_VTABLE
  21
  22 #include <wine/test.h>
  23
  24 #include "mshtml.h"
  25 #include "wininet.h"
  26
  27 struct location_test {
  28     const char *name;
  29     const char *url;
  30
  31     const char *href;
  32     const char *protocol;
  33     const char *host;
  34     const char *hostname;
  35     const char *port;
  36     const char *pathname;
  37     const char *search;
  38     const char *hash;
  39 };
  40
  41 static const struct location_test location_tests[] = {
  42     {
  43         "HTTP",
  44         "http://www.winehq.org?search#hash",
  45         "http://www.winehq.org/?search#hash",
  46         "http:",
  47         "www.winehq.org:80",
  48         "www.winehq.org",
  49         "80",
  50         "",
  51         "?search",
  52         "#hash"
  53     },
  54     {
  55         "HTTP with file",
  56         "http://www.winehq.org/file?search#hash",
  57         "http://www.winehq.org/file?search#hash",
  58         "http:",
  59         "www.winehq.org:80",
  60         "www.winehq.org",
  61         "80",
  62         "file",
  63         "?search",
  64         "#hash"
  65     },
  66     {
  67         "FTP",
  68         "ftp://ftp.winehq.org/",
  69         "ftp://ftp.winehq.org/",
  70         "ftp:",
  71         "ftp.winehq.org:21",
  72         "ftp.winehq.org",
  73         "21",
  74         "",
  75         NULL,
  76         NULL
  77     },
  78     {
  79         "FTP with file",
  80         "ftp://ftp.winehq.org/file",
  81         "ftp://ftp.winehq.org/file",
  82         "ftp:",
  83         "ftp.winehq.org:21",
  84         "ftp.winehq.org",
  85         "21",
  86         "file",
  87         NULL,
  88         NULL
  89     },
  90     {
  91         "FILE",
  92         "file://C:\\windows\\win.ini",
  93         "file:///C:/windows/win.ini",
  94         "file:",
  95         NULL,
  96         NULL,
  97         "",
  98         "C:\\windows\\win.ini",
  99         NULL,
 100         NULL
 101     }
 102 };
 103
 104 static int str_eq_wa(LPCWSTR strw, const char *stra)
 105 {
 106     CHAR buf[512];
 107
 108     if(!strw || !stra)
 109         return (void*)strw == (void*)stra;
 110
 111     WideCharToMultiByte(CP_ACP, 0, strw, -1, buf, sizeof(buf), NULL, NULL);
 112     return !lstrcmpA(stra, buf);
 113 }
 114
 115 static void test_href(IHTMLLocation *loc, const struct location_test *test)
 116 {
 117     HRESULT hres;
 118     BSTR str;
 119
 120     hres = IHTMLLocation_get_href(loc, NULL);
 121     ok(hres == E_POINTER,
 122             "%s: get_href should have failed with E_POINTER (0x%08x), was: 0x%08x\n",
 123             test->name, E_POINTER, hres);
 124
 125     hres = IHTMLLocation_get_href(loc, &str);
 126     ok(hres == S_OK, "%s: get_href failed: 0x%08x\n", test->name, hres);
 127     if(hres == S_OK)
 128         ok(str_eq_wa(str, test->href),
 129                 "%s: expected retrieved href to be L\"%s\", was: %s\n",
 130                 test->name, test->href, wine_dbgstr_w(str));
 131     SysFreeString(str);
 132 }
 133
 134 static void test_protocol(IHTMLLocation *loc, const struct location_test *test)
 135 {
 136     HRESULT hres;
 137     BSTR str;
 138
 139     hres = IHTMLLocation_get_protocol(loc, NULL);
 140     ok(hres == E_POINTER,
 141             "%s: get_protocol should have failed with E_POINTER (0x%08x), was: 0x%08x\n",
 142             test->name, E_POINTER, hres);
 143
 144     hres = IHTMLLocation_get_protocol(loc, &str);
 145     ok(hres == S_OK, "%s: get_protocol failed: 0x%08x\n", test->name, hres);
 146     if(hres == S_OK)
 147         ok(str_eq_wa(str, test->protocol),
 148                 "%s: expected retrieved protocol to be L\"%s\", was: %s\n",
 149                 test->name, test->protocol, wine_dbgstr_w(str));
 150     SysFreeString(str);
 151 }
 152
 153 static void test_host(IHTMLLocation *loc, const struct location_test *test)
 154 {
 155     HRESULT hres;
 156     BSTR str;
 157
 158     hres = IHTMLLocation_get_host(loc, NULL);
 159     ok(hres == E_POINTER,
 160             "%s: get_host should have failed with E_POINTER (0x%08x), was: 0x%08x\n",
 161             test->name, E_POINTER, hres);
 162
 163     hres = IHTMLLocation_get_host(loc, &str);
 164     ok(hres == S_OK, "%s: get_host failed: 0x%08x\n", test->name, hres);
 165     if(hres == S_OK)
 166         ok(str_eq_wa(str, test->host),
 167                 "%s: expected retrieved host to be L\"%s\", was: %s\n",
 168                 test->name, test->host, wine_dbgstr_w(str));
 169     SysFreeString(str);
 170 }
 171
 172 static void test_hostname(IHTMLLocation *loc, IHTMLDocument2 *doc, const struct location_test *test)
 173 {
 174     HRESULT hres;
 175     BSTR str;
 176
 177     hres = IHTMLLocation_get_hostname(loc, NULL);
 178     ok(hres == E_POINTER,
 179             "%s: get_hostname should have failed with E_POINTER (0x%08x), was: 0x%08x\n",
 180             test->name, E_POINTER, hres);
 181
 182     hres = IHTMLLocation_get_hostname(loc, &str);
 183     ok(hres == S_OK, "%s: get_hostname failed: 0x%08x\n", test->name, hres);
 184     if(hres == S_OK)
 185         ok(str_eq_wa(str, test->hostname),
 186                 "%s: expected retrieved hostname to be L\"%s\", was: %s\n",
 187                 test->name, test->hostname, wine_dbgstr_w(str));
 188     SysFreeString(str);
 189
 190     hres = IHTMLDocument2_get_domain(doc, &str);
 191     ok(hres == S_OK, "%s: get_domain failed: 0x%08x\n", test->name, hres);
 192     if(hres == S_OK)
 193         ok(str_eq_wa(str, test->hostname ? test->hostname : ""),
 194                 "%s: expected retrieved domain to be L\"%s\", was: %s\n",
 195                 test->name, test->hostname, wine_dbgstr_w(str));
 196     SysFreeString(str);
 197 }
 198
 199 static void test_port(IHTMLLocation *loc, const struct location_test *test)
 200 {
 201     HRESULT hres;
 202     BSTR str;
 203
 204     hres = IHTMLLocation_get_port(loc, NULL);
 205     ok(hres == E_POINTER,
 206             "%s: get_port should have failed with E_POINTER (0x%08x), was: 0x%08x\n",
 207             test->name, E_POINTER, hres);
 208
 209     hres = IHTMLLocation_get_port(loc, &str);
 210     ok(hres == S_OK, "%s: get_port failed: 0x%08x\n", test->name, hres);
 211     if(hres == S_OK)
 212         ok(str_eq_wa(str, test->port),
 213                 "%s: expected retrieved port to be L\"%s\", was: %s\n",
 214                 test->name, test->port, wine_dbgstr_w(str));
 215     SysFreeString(str);
 216 }
 217
 218 static void test_pathname(IHTMLLocation *loc, const struct location_test *test)
 219 {
 220     HRESULT hres;
 221     BSTR str;
 222
 223     hres = IHTMLLocation_get_pathname(loc, NULL);
 224     ok(hres == E_POINTER,
 225             "%s: get_pathname should have failed with E_POINTER (0x%08x), was: 0x%08x\n",
 226             test->name, E_POINTER, hres);
 227
 228     hres = IHTMLLocation_get_pathname(loc, &str);
 229     ok(hres == S_OK, "%s: get_pathname failed: 0x%08x\n", test->name, hres);
 230     if(hres == S_OK)
 231         ok(str_eq_wa(str, test->pathname),
 232                 "%s: expected retrieved pathname to be L\"%s\", was: %s\n",
 233                 test->name, test->pathname, wine_dbgstr_w(str));
 234     SysFreeString(str);
 235 }
 236
 237 static void test_search(IHTMLLocation *loc, const struct location_test *test)
 238 {
 239     HRESULT hres;
 240     BSTR str;
 241
 242     hres = IHTMLLocation_get_search(loc, NULL);
 243     ok(hres == E_POINTER,
 244             "%s: get_search should have failed with E_POINTER (0x%08x), was: 0x%08x\n",
 245             test->name, E_POINTER, hres);
 246
 247     hres = IHTMLLocation_get_search(loc, &str);
 248     ok(hres == S_OK, "%s: get_search failed: 0x%08x\n", test->name, hres);
 249     if(hres == S_OK)
 250         ok(str_eq_wa(str, test->search),
 251                 "%s: expected retrieved search to be L\"%s\", was: %s\n",
 252                 test->name, test->search, wine_dbgstr_w(str));
 253     SysFreeString(str);
 254 }
 255
 256 static void test_hash(IHTMLLocation *loc, const struct location_test *test)
 257 {
 258     HRESULT hres;
 259     BSTR str;
 260
 261     hres = IHTMLLocation_get_hash(loc, NULL);
 262     ok(hres == E_POINTER,
 263             "%s: get_hash should have failed with E_POINTER (0x%08x), was: 0x%08x\n",
 264             test->name, E_POINTER, hres);
 265
 266     hres = IHTMLLocation_get_hash(loc, &str);
 267     ok(hres == S_OK, "%s: get_hash failed: 0x%08x\n", test->name, hres);
 268     if(hres == S_OK)
 269         ok(str_eq_wa(str, test->hash),
 270                 "%s: expected retrieved hash to be L\"%s\", was: %s\n",
 271                 test->name, test->hash, wine_dbgstr_w(str));
 272     SysFreeString(str);
 273 }
 274
 275 static void perform_test(const struct location_test* test)
 276 {
 277     WCHAR url[INTERNET_MAX_URL_LENGTH];
 278     HRESULT hres;
 279     IBindCtx *bc;
 280     IMoniker *url_mon;
 281     IPersistMoniker *persist_mon;
 282     IHTMLDocument2 *doc;
 283     IHTMLDocument6 *doc6;
 284     IHTMLLocation *location;
 285
 286     hres = CreateBindCtx(0, &bc);
 287     ok(hres == S_OK, "%s: CreateBindCtx failed: 0x%08x\n", test->name, hres);
 288     if(FAILED(hres))
 289         return;
 290
 291     MultiByteToWideChar(CP_ACP, 0, test->url, -1, url, sizeof(url)/sizeof(WCHAR));
 292     hres = CreateURLMoniker(NULL, url, &url_mon);
 293     ok(hres == S_OK, "%s: CreateURLMoniker failed: 0x%08x\n", test->name, hres);
 294     if(FAILED(hres)){
 295         IBindCtx_Release(bc);
 296         return;
 297     }
 298
 299     hres = CoCreateInstance(&CLSID_HTMLDocument, NULL,
 300             CLSCTX_INPROC_SERVER | CLSCTX_INPROC_HANDLER, &IID_IHTMLDocument2,
 301             (void**)&doc);
 302     ok(hres == S_OK, "%s: CoCreateInstance failed: 0x%08x\n", test->name, hres);
 303     if(FAILED(hres)){
 304         IMoniker_Release(url_mon);
 305         IBindCtx_Release(bc);
 306         return;
 307     }
 308
 309     hres = IHTMLDocument2_QueryInterface(doc, &IID_IHTMLDocument6, (void**)&doc6);
 310     if(hres == S_OK){
 311         IHTMLDocument6_Release(doc6);
 312     }else{
 313         win_skip("%s: Could not get IHTMLDocument6, probably too old IE. Requires IE 8+\n", test->name);
 314         IMoniker_Release(url_mon);
 315         IBindCtx_Release(bc);
 316         return;
 317     }
 318
 319     hres = IHTMLDocument2_QueryInterface(doc, &IID_IPersistMoniker,
 320             (void**)&persist_mon);
 321     ok(hres == S_OK, "%s: IHTMlDocument2_QueryInterface failed: 0x%08x\n", test->name, hres);
 322     if(FAILED(hres)){
 323         IHTMLDocument2_Release(doc);
 324         IMoniker_Release(url_mon);
 325         IBindCtx_Release(bc);
 326         return;
 327     }
 328
 329     hres = IPersistMoniker_Load(persist_mon, FALSE, url_mon, bc,
 330             STGM_SHARE_EXCLUSIVE | STGM_READWRITE);
 331     ok(hres == S_OK, "%s: IPersistMoniker_Load failed: 0x%08x\n", test->name, hres);
 332     if(FAILED(hres)){
 333         IPersistMoniker_Release(persist_mon);
 334         IHTMLDocument2_Release(doc);
 335         IMoniker_Release(url_mon);
 336         IBindCtx_Release(bc);
 337         return;
 338     }
 339
 340     hres = IHTMLDocument2_get_location(doc, &location);
 341     ok(hres == S_OK, "%s: IHTMLDocument2_get_location failed: 0x%08x\n", test->name, hres);
 342     if(FAILED(hres)){
 343         IPersistMoniker_Release(persist_mon);
 344         IHTMLDocument2_Release(doc);
 345         IMoniker_Release(url_mon);
 346         IBindCtx_Release(bc);
 347         return;
 348     }
 349
 350     test_href(location, test);
 351     test_protocol(location, test);
 352     test_host(location, test);
 353     test_hostname(location, doc, test);
 354     test_port(location, test);
 355     test_pathname(location, test);
 356     test_search(location, test);
 357     test_hash(location, test);
 358
 359     IHTMLLocation_Release(location);
 360     IPersistMoniker_Release(persist_mon);
 361     IHTMLDocument2_Release(doc);
 362     IMoniker_Release(url_mon);
 363     IBindCtx_Release(bc);
 364 }
 365
 366 START_TEST(htmllocation)
 367 {
 368     int i;
 369
 370     CoInitialize(NULL);
 371
 372     for(i=0; i < sizeof(location_tests)/sizeof(*location_tests); i++)
 373         perform_test(location_tests+i);
 374
 375     CoUninitialize();
 376 }