dlls/ieframe/tests/intshcut.c - wine - Git at Google

 /*
  * Unit tests to document InternetShortcut's behaviour
  *
  * Copyright 2008 Damjan Jovanovic
  *
  * 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.
  *
  * 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
  * Lesser General Public License for more details.
  *
  * You should have received a copy of the GNU Lesser General Public
  * License along with this library; if not, write to the Free Software
  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  */

 #include <stdarg.h>
 #include <stdio.h>

 #define COBJMACROS

 #include "windef.h"
 #include "winbase.h"
 #include "winreg.h"
 #include "winerror.h"

 #include "shlobj.h"
 #include "shobjidl.h"
 #include "shlguid.h"
 #include "ole2.h"
 #include "mshtml.h"
 #include "initguid.h"
 #include "isguids.h"
 #include "intshcut.h"

 #include "wine/test.h"

 static HRESULT WINAPI Unknown_QueryInterface(IUnknown *pUnknown, REFIID riid, void **ppvObject)
 {
     if (IsEqualGUID(&IID_IUnknown, riid))
     {
         *ppvObject = pUnknown;
         return S_OK;
     }
     return E_NOINTERFACE;
 }

 static ULONG WINAPI Unknown_AddRef(IUnknown *pUnknown)
 {
     return 2;
 }

 static ULONG WINAPI Unknown_Release(IUnknown *pUnknown)
 {
     return 1;
 }

 static IUnknownVtbl unknownVtbl = {
     Unknown_QueryInterface,
     Unknown_AddRef,
     Unknown_Release
 };

 static IUnknown unknown = {
     &unknownVtbl
 };

 static void test_Aggregability(void)
 {
     HRESULT hr;
     IUnknown *pUnknown = NULL;

     hr = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IUnknown, (void**)&pUnknown);
     ok(hr == S_OK, "could not create instance of CLSID_InternetShortcut with IID_IUnknown, hr = 0x%x\n", hr);
     if (pUnknown)
         IUnknown_Release(pUnknown);

     hr = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IUniformResourceLocatorA, (void**)&pUnknown);
     ok(hr == S_OK, "could not create instance of CLSID_InternetShortcut with IID_IUniformResourceLocatorA, hr = 0x%x\n", hr);
     if (pUnknown)
         IUnknown_Release(pUnknown);

     hr = CoCreateInstance(&CLSID_InternetShortcut, &unknown, CLSCTX_ALL, &IID_IUnknown, (void**)&pUnknown);
     ok(hr == CLASS_E_NOAGGREGATION, "aggregation didn't fail like it should, hr = 0x%x\n", hr);
     if (pUnknown)
         IUnknown_Release(pUnknown);
 }

 static void can_query_interface(IUnknown *pUnknown, REFIID riid)
 {
     HRESULT hr;
     IUnknown *newInterface;
     hr = IUnknown_QueryInterface(pUnknown, riid, (void**)&newInterface);
     ok(hr == S_OK, "interface %s could not be queried\n", wine_dbgstr_guid(riid));
     if (SUCCEEDED(hr))
         IUnknown_Release(newInterface);
 }

 static void test_QueryInterface(void)
 {
     HRESULT hr;
     IUnknown *pUnknown;

     hr = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IUnknown, (void**)&pUnknown);
     ok(hr == S_OK, "Could not create InternetShortcut object: %08x\n", hr);

     can_query_interface(pUnknown, &IID_IUniformResourceLocatorA);
     can_query_interface(pUnknown, &IID_IUniformResourceLocatorW);
     can_query_interface(pUnknown, &IID_IPersistFile);
     IUnknown_Release(pUnknown);
 }

 #define test_shortcut_url(a,b) _test_shortcut_url(__LINE__,a,b)
 static void _test_shortcut_url(unsigned line, IUnknown *unk, const char *exurl)
 {
     IUniformResourceLocatorA *locator_a;
     char *url_a;
     HRESULT hres;

     hres = IUnknown_QueryInterface(unk, &IID_IUniformResourceLocatorA, (void**)&locator_a);
     ok_(__FILE__,line)(hres == S_OK, "Could not get IUniformResourceLocatorA iface: %08x\n", hres);

     hres = locator_a->lpVtbl->GetURL(locator_a, &url_a);
     ok_(__FILE__,line)(hres == S_OK, "GetURL failed: %08x\n", hres);
     ok_(__FILE__,line)(!strcmp(url_a, exurl), "unexpected URL, got %s, expected %s\n", url_a, exurl);
     CoTaskMemFree(url_a);

     IUnknown_Release((IUnknown*)locator_a);
 }

 #define check_string_transform(a,b,c,d,e) _check_string_transform(__LINE__,a,b,c,d,e)
 static void _check_string_transform(unsigned line, IUniformResourceLocatorA *urlA, LPCSTR input, DWORD flags,
         LPCSTR expectedOutput, BOOL is_todo)
 {
     CHAR *output;
     HRESULT hr;

     hr = urlA->lpVtbl->SetURL(urlA, input, flags);
     ok_(__FILE__,line)(hr == S_OK, "SetUrl failed, hr=0x%x\n", hr);
     if (FAILED(hr))
         return;

     output = (void*)0xdeadbeef;
     hr = urlA->lpVtbl->GetURL(urlA, &output);
     if(expectedOutput) {
         todo_wine_if(is_todo) {
             ok_(__FILE__,line)(hr == S_OK, "GetUrl failed, hr=0x%x\n", hr);
         }
         todo_wine
         ok_(__FILE__,line)(!lstrcmpA(output, expectedOutput), "unexpected URL change %s -> %s (expected %s)\n",
             input, output, expectedOutput);
     }else {
         todo_wine
         ok_(__FILE__,line)(hr == S_FALSE, "GetUrl failed, hr=0x%x\n", hr);
         todo_wine
         ok_(__FILE__,line)(!output, "GetUrl returned %s\n", output);
     }
     if (hr == S_OK)
         CoTaskMemFree(output);
 }

 static void test_ReadAndWriteProperties(void)
 {
     int iconIndex = 7;
     PROPSPEC ps[2];
     HRESULT hr;
     IUniformResourceLocatorA *urlA;
     IUniformResourceLocatorA *urlAFromFile;
     WCHAR fileNameW[MAX_PATH];

     static const WCHAR shortcutW[] = {'t','e','s','t','s','h','o','r','t','c','u','t','.','u','r','l',0};
     WCHAR iconPath[] = {'f','i','l','e',':','/','/','/','C',':','/','a','r','b','i','t','r','a','r','y','/','i','c','o','n','/','p','a','t','h',0};
     char testurl[] = "http://some/bogus/url.html";

     ps[0].ulKind = PRSPEC_PROPID;
     U(ps[0]).propid = PID_IS_ICONFILE;
     ps[1].ulKind = PRSPEC_PROPID;
     U(ps[1]).propid = PID_IS_ICONINDEX;

     /* Make sure we have a valid temporary directory */
     GetTempPathW(MAX_PATH, fileNameW);
     lstrcatW(fileNameW, shortcutW);

     hr = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IUniformResourceLocatorA, (void**)&urlA);
     ok(hr == S_OK, "Could not create CLSID_InternetShortcut instance: %08x\n", hr);
     if (hr == S_OK)
     {
         IPersistFile *pf;
         IPropertyStorage *pPropStgWrite;
         IPropertySetStorage *pPropSetStg;
         PROPVARIANT pv[2];

         /* We need to set a URL -- IPersistFile refuses to save without one. */
         hr = urlA->lpVtbl->SetURL(urlA, testurl, 0);
         ok(hr == S_OK, "Failed to set a URL.  hr=0x%x\n", hr);

         /* Write this shortcut out to a file so that we can test reading it in again. */
         hr = urlA->lpVtbl->QueryInterface(urlA, &IID_IPersistFile, (void **) &pf);
         ok(hr == S_OK, "Failed to get the IPersistFile for writing.  hr=0x%x\n", hr);

         hr = IPersistFile_Save(pf, fileNameW, TRUE);
         ok(hr == S_OK, "Failed to save via IPersistFile. hr=0x%x\n", hr);

         IPersistFile_Release(pf);

         pv[0].vt = VT_LPWSTR;
         U(pv[0]).pwszVal = (void *) iconPath;
         pv[1].vt = VT_I4;
         U(pv[1]).lVal = iconIndex;
         hr = urlA->lpVtbl->QueryInterface(urlA, &IID_IPropertySetStorage, (void **) &pPropSetStg);
         ok(hr == S_OK, "Unable to get an IPropertySetStorage, hr=0x%x\n", hr);

         hr = IPropertySetStorage_Open(pPropSetStg, &FMTID_Intshcut, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, &pPropStgWrite);
         ok(hr == S_OK, "Unable to get an IPropertyStorage for writing, hr=0x%x\n", hr);

         hr = IPropertyStorage_WriteMultiple(pPropStgWrite, 2, ps, pv, 0);
         ok(hr == S_OK, "Unable to set properties, hr=0x%x\n", hr);

         hr = IPropertyStorage_Commit(pPropStgWrite, STGC_DEFAULT);
         ok(hr == S_OK, "Failed to commit properties, hr=0x%x\n", hr);

         pPropStgWrite->lpVtbl->Release(pPropStgWrite);
         urlA->lpVtbl->Release(urlA);
         IPropertySetStorage_Release(pPropSetStg);
     }

     hr = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IUniformResourceLocatorA, (void**)&urlAFromFile);
     ok(hr == S_OK, "Could not create CLSID_InternetShortcut instance: %08x\n", hr);
     if (hr == S_OK)
     {
         IPropertySetStorage *pPropSetStg;
         IPropertyStorage *pPropStgRead;
         PROPVARIANT pvread[2];
         IPersistFile *pf;
         LPSTR url = NULL;

         /* Now read that .url file back in. */
         hr = urlAFromFile->lpVtbl->QueryInterface(urlAFromFile, &IID_IPersistFile, (void **) &pf);
         ok(hr == S_OK, "Failed to get the IPersistFile for reading.  hr=0x%x\n", hr);

         hr = IPersistFile_Load(pf, fileNameW, 0);
         ok(hr == S_OK, "Failed to load via IPersistFile. hr=0x%x\n", hr);
         IPersistFile_Release(pf);


         hr = urlAFromFile->lpVtbl->GetURL(urlAFromFile, &url);
         ok(hr == S_OK, "Unable to get url from file, hr=0x%x\n", hr);
         ok(lstrcmpA(url, testurl) == 0, "Wrong url read from file: %s\n",url);
         CoTaskMemFree(url);

         hr = urlAFromFile->lpVtbl->QueryInterface(urlAFromFile, &IID_IPropertySetStorage, (void **) &pPropSetStg);
         ok(hr == S_OK, "Unable to get an IPropertySetStorage, hr=0x%x\n", hr);

         hr = IPropertySetStorage_Open(pPropSetStg, &FMTID_Intshcut, STGM_READ | STGM_SHARE_EXCLUSIVE, &pPropStgRead);
         ok(hr == S_OK, "Unable to get an IPropertyStorage for reading, hr=0x%x\n", hr);

         memset(pvread, 0, sizeof(pvread));
         hr = IPropertyStorage_ReadMultiple(pPropStgRead, 2, ps, pvread);
     todo_wine /* Wine doesn't yet support setting properties after save */
     {
         ok(hr == S_OK, "Unable to read properties, hr=0x%x\n", hr);
         ok(pvread[1].vt == VT_I4, "got %d\n", pvread[1].vt);
         ok(U(pvread[1]).lVal == iconIndex, "Read wrong icon index: %d\n", U(pvread[1]).iVal);
         ok(pvread[0].vt == VT_LPWSTR, "got %d\n", pvread[0].vt);
         ok(lstrcmpW(U(pvread[0]).pwszVal, iconPath) == 0, "Wrong icon path read: %s\n", wine_dbgstr_w(U(pvread[0]).pwszVal));
     }
         PropVariantClear(&pvread[0]);
         PropVariantClear(&pvread[1]);
         IPropertyStorage_Release(pPropStgRead);
         IPropertySetStorage_Release(pPropSetStg);
         urlAFromFile->lpVtbl->Release(urlAFromFile);
         DeleteFileW(fileNameW);
     }
 }

 static void test_NullURLs(void)
 {
     LPSTR url = NULL;
     HRESULT hr;
     IUniformResourceLocatorA *urlA;

     hr = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IUniformResourceLocatorA, (void**)&urlA);
     ok(hr == S_OK, "Could not create InternetShortcut object: %08x\n", hr);

     hr = urlA->lpVtbl->GetURL(urlA, &url);
     ok(hr == S_FALSE, "getting uninitialized URL unexpectedly failed, hr=0x%x\n", hr);
     ok(url == NULL, "uninitialized URL is not NULL but %s\n", url);

     hr = urlA->lpVtbl->SetURL(urlA, NULL, 0);
     ok(hr == S_OK, "setting NULL URL unexpectedly failed, hr=0x%x\n", hr);

     hr = urlA->lpVtbl->GetURL(urlA, &url);
     ok(hr == S_FALSE, "getting NULL URL unexpectedly failed, hr=0x%x\n", hr);
     ok(url == NULL, "URL unexpectedly not NULL but %s\n", url);

     urlA->lpVtbl->Release(urlA);
 }

 typedef struct {
     const char *data;
     const char *url;
 } load_test_t;

 static const load_test_t load_tests[] = {
     {"[InternetShortcut]\n"
      "URL=http://www.winehq.org/\n"
      "HotKey=0\n"
      "IDList=\n"
      "[{000214A0-0000-0000-C000-000000000046}]\n"
      "Prop0=1,2\n",

      "http://www.winehq.org/"
     }
 };

 static void test_Load(void)
 {
     IPersistFile *persist_file;
     const load_test_t *test;
     WCHAR file_path[MAX_PATH];
     DWORD size;
     HANDLE file;
     HRESULT hres;

     static const WCHAR test_urlW[] = {'t','e','s','t','.','u','r','l',0};

     GetTempPathW(MAX_PATH, file_path);
     lstrcatW(file_path, test_urlW);

     for(test = load_tests; test < load_tests + sizeof(load_tests)/sizeof(*load_tests); test++) {
         IPropertySetStorage *propsetstorage;
         IPropertyStorage *propstorage;
         PROPVARIANT v;
         PROPSPEC ps;

         file = CreateFileW(file_path, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
         ok(file != INVALID_HANDLE_VALUE, "could not create test file\n");
         if(file == INVALID_HANDLE_VALUE)
             continue;

         WriteFile(file, test->data, strlen(test->data), &size, NULL);
         CloseHandle(file);

         hres = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IPersistFile, (void**)&persist_file);
         ok(hres == S_OK, "Could not create InternetShortcut instance: %08x\n", hres);

         hres = IPersistFile_Load(persist_file, file_path, 0);
         ok(hres == S_OK, "Load failed: %08x\n", hres);

         test_shortcut_url((IUnknown*)persist_file, test->url);

         hres = IPersistFile_QueryInterface(persist_file, &IID_IPropertySetStorage, (void **)&propsetstorage);
         ok(hres == S_OK, "Unable to get an IPropertySetStorage, hr=0x%x\n", hres);

         hres = IPropertySetStorage_Open(propsetstorage, &FMTID_Intshcut, STGM_READ | STGM_SHARE_EXCLUSIVE, &propstorage);
         ok(hres == S_OK, "Unable to get an IPropertyStorage for reading, hr=0x%x\n", hres);

         ps.ulKind = PRSPEC_PROPID;
         U(ps).propid = PID_IS_ICONFILE;
         v.vt = VT_NULL;
         hres = IPropertyStorage_ReadMultiple(propstorage, 1, &ps, &v);
         ok(hres == S_FALSE, "got 0x%08x\n", hres);
         ok(v.vt == VT_EMPTY, "got %d\n", v.vt);

         ps.ulKind = PRSPEC_PROPID;
         U(ps).propid = PID_IS_ICONINDEX;
         v.vt = VT_EMPTY;
         hres = IPropertyStorage_ReadMultiple(propstorage, 1, &ps, &v);
         ok(hres == S_FALSE, "got 0x%08x\n", hres);
         ok(v.vt == VT_EMPTY, "got %d\n", v.vt);

         IPropertyStorage_Release(propstorage);
         IPropertySetStorage_Release(propsetstorage);

         IPersistFile_Release(persist_file);
         DeleteFileW(file_path);
     }
 }

 static void test_SetURLFlags(void)
 {
     HRESULT hr;
     IUniformResourceLocatorA *urlA;

     hr = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IUniformResourceLocatorA, (void**)&urlA);
     ok(hr == S_OK, "Could not create InternetShortcut object: %08x\n", hr);

     check_string_transform(urlA, "somerandomstring", 0, NULL, TRUE);
     check_string_transform(urlA, "www.winehq.org", 0, NULL, TRUE);

     check_string_transform(urlA, "www.winehq.org", IURL_SETURL_FL_GUESS_PROTOCOL, "http://www.winehq.org/", FALSE);
     check_string_transform(urlA, "ftp.winehq.org", IURL_SETURL_FL_GUESS_PROTOCOL, "ftp://ftp.winehq.org/", FALSE);

     urlA->lpVtbl->Release(urlA);
 }

 static void test_InternetShortcut(void)
 {
     IUniformResourceLocatorA *url;
     HRESULT hres;

     hres = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IUniformResourceLocatorA, (void**)&url);
     ok(hres == S_OK, "Could not create CLSID_InternetShortcut instance: %08x\n", hres);
     if(FAILED(hres))
         return;
     url->lpVtbl->Release(url);

     test_Aggregability();
     test_QueryInterface();
     test_NullURLs();
     test_SetURLFlags();
     test_ReadAndWriteProperties();
     test_Load();
 }

 START_TEST(intshcut)
 {
     OleInitialize(NULL);

     test_InternetShortcut();

     OleUninitialize();
 }
	/*
	* Unit tests to document InternetShortcut's behaviour
	*
	* Copyright 2008 Damjan Jovanovic
	*
	* 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.
	*
	* 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
	* Lesser General Public License for more details.
	*
	* You should have received a copy of the GNU Lesser General Public
	* License along with this library; if not, write to the Free Software
	* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
	*/

	#include <stdarg.h>
	#include <stdio.h>

	#define COBJMACROS

	#include "windef.h"
	#include "winbase.h"
	#include "winreg.h"
	#include "winerror.h"

	#include "shlobj.h"
	#include "shobjidl.h"
	#include "shlguid.h"
	#include "ole2.h"
	#include "mshtml.h"
	#include "initguid.h"
	#include "isguids.h"
	#include "intshcut.h"

	#include "wine/test.h"

	static HRESULT WINAPI Unknown_QueryInterface(IUnknown pUnknown, REFIID riid, void *ppvObject)
	{
	if (IsEqualGUID(&IID_IUnknown, riid))
	{
	*ppvObject = pUnknown;
	return S_OK;
	}
	return E_NOINTERFACE;
	}

	static ULONG WINAPI Unknown_AddRef(IUnknown *pUnknown)
	{
	return 2;
	}

	static ULONG WINAPI Unknown_Release(IUnknown *pUnknown)
	{
	return 1;
	}

	static IUnknownVtbl unknownVtbl = {
	Unknown_QueryInterface,
	Unknown_AddRef,
	Unknown_Release
	};

	static IUnknown unknown = {
	&unknownVtbl
	};

	static void test_Aggregability(void)
	{
	HRESULT hr;
	IUnknown *pUnknown = NULL;

	hr = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IUnknown, (void**)&pUnknown);
	ok(hr == S_OK, "could not create instance of CLSID_InternetShortcut with IID_IUnknown, hr = 0x%x\n", hr);
	if (pUnknown)
	IUnknown_Release(pUnknown);

	hr = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IUniformResourceLocatorA, (void**)&pUnknown);
	ok(hr == S_OK, "could not create instance of CLSID_InternetShortcut with IID_IUniformResourceLocatorA, hr = 0x%x\n", hr);
	if (pUnknown)
	IUnknown_Release(pUnknown);

	hr = CoCreateInstance(&CLSID_InternetShortcut, &unknown, CLSCTX_ALL, &IID_IUnknown, (void**)&pUnknown);
	ok(hr == CLASS_E_NOAGGREGATION, "aggregation didn't fail like it should, hr = 0x%x\n", hr);
	if (pUnknown)
	IUnknown_Release(pUnknown);
	}

	static void can_query_interface(IUnknown *pUnknown, REFIID riid)
	{
	HRESULT hr;
	IUnknown *newInterface;
	hr = IUnknown_QueryInterface(pUnknown, riid, (void**)&newInterface);
	ok(hr == S_OK, "interface %s could not be queried\n", wine_dbgstr_guid(riid));
	if (SUCCEEDED(hr))
	IUnknown_Release(newInterface);
	}

	static void test_QueryInterface(void)
	{
	HRESULT hr;
	IUnknown *pUnknown;

	hr = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IUnknown, (void**)&pUnknown);
	ok(hr == S_OK, "Could not create InternetShortcut object: %08x\n", hr);

	can_query_interface(pUnknown, &IID_IUniformResourceLocatorA);
	can_query_interface(pUnknown, &IID_IUniformResourceLocatorW);
	can_query_interface(pUnknown, &IID_IPersistFile);
	IUnknown_Release(pUnknown);
	}

	#define test_shortcut_url(a,b) _test_shortcut_url(__LINE__,a,b)
	static void _test_shortcut_url(unsigned line, IUnknown unk, const char exurl)
	{
	IUniformResourceLocatorA *locator_a;
	char *url_a;
	HRESULT hres;

	hres = IUnknown_QueryInterface(unk, &IID_IUniformResourceLocatorA, (void**)&locator_a);
	ok_(__FILE__,line)(hres == S_OK, "Could not get IUniformResourceLocatorA iface: %08x\n", hres);

	hres = locator_a->lpVtbl->GetURL(locator_a, &url_a);
	ok_(__FILE__,line)(hres == S_OK, "GetURL failed: %08x\n", hres);
	ok_(__FILE__,line)(!strcmp(url_a, exurl), "unexpected URL, got %s, expected %s\n", url_a, exurl);
	CoTaskMemFree(url_a);

	IUnknown_Release((IUnknown*)locator_a);
	}

	#define check_string_transform(a,b,c,d,e) _check_string_transform(__LINE__,a,b,c,d,e)
	static void _check_string_transform(unsigned line, IUniformResourceLocatorA *urlA, LPCSTR input, DWORD flags,
	LPCSTR expectedOutput, BOOL is_todo)
	{
	CHAR *output;
	HRESULT hr;

	hr = urlA->lpVtbl->SetURL(urlA, input, flags);
	ok_(__FILE__,line)(hr == S_OK, "SetUrl failed, hr=0x%x\n", hr);
	if (FAILED(hr))
	return;

	output = (void*)0xdeadbeef;
	hr = urlA->lpVtbl->GetURL(urlA, &output);
	if(expectedOutput) {
	todo_wine_if(is_todo) {
	ok_(__FILE__,line)(hr == S_OK, "GetUrl failed, hr=0x%x\n", hr);
	}
	todo_wine
	ok_(__FILE__,line)(!lstrcmpA(output, expectedOutput), "unexpected URL change %s -> %s (expected %s)\n",
	input, output, expectedOutput);
	}else {
	todo_wine
	ok_(__FILE__,line)(hr == S_FALSE, "GetUrl failed, hr=0x%x\n", hr);
	todo_wine
	ok_(__FILE__,line)(!output, "GetUrl returned %s\n", output);
	}
	if (hr == S_OK)
	CoTaskMemFree(output);
	}

	static void test_ReadAndWriteProperties(void)
	{
	int iconIndex = 7;
	PROPSPEC ps[2];
	HRESULT hr;
	IUniformResourceLocatorA *urlA;
	IUniformResourceLocatorA *urlAFromFile;
	WCHAR fileNameW[MAX_PATH];

	static const WCHAR shortcutW[] = {'t','e','s','t','s','h','o','r','t','c','u','t','.','u','r','l',0};
	WCHAR iconPath[] = {'f','i','l','e',':','/','/','/','C',':','/','a','r','b','i','t','r','a','r','y','/','i','c','o','n','/','p','a','t','h',0};
	char testurl[] = "http://some/bogus/url.html";

	ps[0].ulKind = PRSPEC_PROPID;
	U(ps[0]).propid = PID_IS_ICONFILE;
	ps[1].ulKind = PRSPEC_PROPID;
	U(ps[1]).propid = PID_IS_ICONINDEX;

	/* Make sure we have a valid temporary directory */
	GetTempPathW(MAX_PATH, fileNameW);
	lstrcatW(fileNameW, shortcutW);

	hr = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IUniformResourceLocatorA, (void**)&urlA);
	ok(hr == S_OK, "Could not create CLSID_InternetShortcut instance: %08x\n", hr);
	if (hr == S_OK)
	{
	IPersistFile *pf;
	IPropertyStorage *pPropStgWrite;
	IPropertySetStorage *pPropSetStg;
	PROPVARIANT pv[2];

	/* We need to set a URL -- IPersistFile refuses to save without one. */
	hr = urlA->lpVtbl->SetURL(urlA, testurl, 0);
	ok(hr == S_OK, "Failed to set a URL. hr=0x%x\n", hr);

	/* Write this shortcut out to a file so that we can test reading it in again. */
	hr = urlA->lpVtbl->QueryInterface(urlA, &IID_IPersistFile, (void **) &pf);
	ok(hr == S_OK, "Failed to get the IPersistFile for writing. hr=0x%x\n", hr);

	hr = IPersistFile_Save(pf, fileNameW, TRUE);
	ok(hr == S_OK, "Failed to save via IPersistFile. hr=0x%x\n", hr);

	IPersistFile_Release(pf);

	pv[0].vt = VT_LPWSTR;
	U(pv[0]).pwszVal = (void *) iconPath;
	pv[1].vt = VT_I4;
	U(pv[1]).lVal = iconIndex;
	hr = urlA->lpVtbl->QueryInterface(urlA, &IID_IPropertySetStorage, (void **) &pPropSetStg);
	ok(hr == S_OK, "Unable to get an IPropertySetStorage, hr=0x%x\n", hr);

	hr = IPropertySetStorage_Open(pPropSetStg, &FMTID_Intshcut, STGM_READWRITE \| STGM_SHARE_EXCLUSIVE, &pPropStgWrite);
	ok(hr == S_OK, "Unable to get an IPropertyStorage for writing, hr=0x%x\n", hr);

	hr = IPropertyStorage_WriteMultiple(pPropStgWrite, 2, ps, pv, 0);
	ok(hr == S_OK, "Unable to set properties, hr=0x%x\n", hr);

	hr = IPropertyStorage_Commit(pPropStgWrite, STGC_DEFAULT);
	ok(hr == S_OK, "Failed to commit properties, hr=0x%x\n", hr);

	pPropStgWrite->lpVtbl->Release(pPropStgWrite);
	urlA->lpVtbl->Release(urlA);
	IPropertySetStorage_Release(pPropSetStg);
	}

	hr = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IUniformResourceLocatorA, (void**)&urlAFromFile);
	ok(hr == S_OK, "Could not create CLSID_InternetShortcut instance: %08x\n", hr);
	if (hr == S_OK)
	{
	IPropertySetStorage *pPropSetStg;
	IPropertyStorage *pPropStgRead;
	PROPVARIANT pvread[2];
	IPersistFile *pf;
	LPSTR url = NULL;

	/* Now read that .url file back in. */
	hr = urlAFromFile->lpVtbl->QueryInterface(urlAFromFile, &IID_IPersistFile, (void **) &pf);
	ok(hr == S_OK, "Failed to get the IPersistFile for reading. hr=0x%x\n", hr);

	hr = IPersistFile_Load(pf, fileNameW, 0);
	ok(hr == S_OK, "Failed to load via IPersistFile. hr=0x%x\n", hr);
	IPersistFile_Release(pf);


	hr = urlAFromFile->lpVtbl->GetURL(urlAFromFile, &url);
	ok(hr == S_OK, "Unable to get url from file, hr=0x%x\n", hr);
	ok(lstrcmpA(url, testurl) == 0, "Wrong url read from file: %s\n",url);
	CoTaskMemFree(url);

	hr = urlAFromFile->lpVtbl->QueryInterface(urlAFromFile, &IID_IPropertySetStorage, (void **) &pPropSetStg);
	ok(hr == S_OK, "Unable to get an IPropertySetStorage, hr=0x%x\n", hr);

	hr = IPropertySetStorage_Open(pPropSetStg, &FMTID_Intshcut, STGM_READ \| STGM_SHARE_EXCLUSIVE, &pPropStgRead);
	ok(hr == S_OK, "Unable to get an IPropertyStorage for reading, hr=0x%x\n", hr);

	memset(pvread, 0, sizeof(pvread));
	hr = IPropertyStorage_ReadMultiple(pPropStgRead, 2, ps, pvread);
	todo_wine /* Wine doesn't yet support setting properties after save */
	{
	ok(hr == S_OK, "Unable to read properties, hr=0x%x\n", hr);
	ok(pvread[1].vt == VT_I4, "got %d\n", pvread[1].vt);
	ok(U(pvread[1]).lVal == iconIndex, "Read wrong icon index: %d\n", U(pvread[1]).iVal);
	ok(pvread[0].vt == VT_LPWSTR, "got %d\n", pvread[0].vt);
	ok(lstrcmpW(U(pvread[0]).pwszVal, iconPath) == 0, "Wrong icon path read: %s\n", wine_dbgstr_w(U(pvread[0]).pwszVal));
	}
	PropVariantClear(&pvread[0]);
	PropVariantClear(&pvread[1]);
	IPropertyStorage_Release(pPropStgRead);
	IPropertySetStorage_Release(pPropSetStg);
	urlAFromFile->lpVtbl->Release(urlAFromFile);
	DeleteFileW(fileNameW);
	}
	}

	static void test_NullURLs(void)
	{
	LPSTR url = NULL;
	HRESULT hr;
	IUniformResourceLocatorA *urlA;

	hr = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IUniformResourceLocatorA, (void**)&urlA);
	ok(hr == S_OK, "Could not create InternetShortcut object: %08x\n", hr);

	hr = urlA->lpVtbl->GetURL(urlA, &url);
	ok(hr == S_FALSE, "getting uninitialized URL unexpectedly failed, hr=0x%x\n", hr);
	ok(url == NULL, "uninitialized URL is not NULL but %s\n", url);

	hr = urlA->lpVtbl->SetURL(urlA, NULL, 0);
	ok(hr == S_OK, "setting NULL URL unexpectedly failed, hr=0x%x\n", hr);

	hr = urlA->lpVtbl->GetURL(urlA, &url);
	ok(hr == S_FALSE, "getting NULL URL unexpectedly failed, hr=0x%x\n", hr);
	ok(url == NULL, "URL unexpectedly not NULL but %s\n", url);

	urlA->lpVtbl->Release(urlA);
	}

	typedef struct {
	const char *data;
	const char *url;
	} load_test_t;

	static const load_test_t load_tests[] = {
	{"[InternetShortcut]\n"
	"URL=http://www.winehq.org/\n"
	"HotKey=0\n"
	"IDList=\n"
	"[{000214A0-0000-0000-C000-000000000046}]\n"
	"Prop0=1,2\n",

	"http://www.winehq.org/"
	}
	};

	static void test_Load(void)
	{
	IPersistFile *persist_file;
	const load_test_t *test;
	WCHAR file_path[MAX_PATH];
	DWORD size;
	HANDLE file;
	HRESULT hres;

	static const WCHAR test_urlW[] = {'t','e','s','t','.','u','r','l',0};

	GetTempPathW(MAX_PATH, file_path);
	lstrcatW(file_path, test_urlW);

	for(test = load_tests; test < load_tests + sizeof(load_tests)/sizeof(*load_tests); test++) {
	IPropertySetStorage *propsetstorage;
	IPropertyStorage *propstorage;
	PROPVARIANT v;
	PROPSPEC ps;

	file = CreateFileW(file_path, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
	ok(file != INVALID_HANDLE_VALUE, "could not create test file\n");
	if(file == INVALID_HANDLE_VALUE)
	continue;

	WriteFile(file, test->data, strlen(test->data), &size, NULL);
	CloseHandle(file);

	hres = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IPersistFile, (void**)&persist_file);
	ok(hres == S_OK, "Could not create InternetShortcut instance: %08x\n", hres);

	hres = IPersistFile_Load(persist_file, file_path, 0);
	ok(hres == S_OK, "Load failed: %08x\n", hres);

	test_shortcut_url((IUnknown*)persist_file, test->url);

	hres = IPersistFile_QueryInterface(persist_file, &IID_IPropertySetStorage, (void **)&propsetstorage);
	ok(hres == S_OK, "Unable to get an IPropertySetStorage, hr=0x%x\n", hres);

	hres = IPropertySetStorage_Open(propsetstorage, &FMTID_Intshcut, STGM_READ \| STGM_SHARE_EXCLUSIVE, &propstorage);
	ok(hres == S_OK, "Unable to get an IPropertyStorage for reading, hr=0x%x\n", hres);

	ps.ulKind = PRSPEC_PROPID;
	U(ps).propid = PID_IS_ICONFILE;
	v.vt = VT_NULL;
	hres = IPropertyStorage_ReadMultiple(propstorage, 1, &ps, &v);
	ok(hres == S_FALSE, "got 0x%08x\n", hres);
	ok(v.vt == VT_EMPTY, "got %d\n", v.vt);

	ps.ulKind = PRSPEC_PROPID;
	U(ps).propid = PID_IS_ICONINDEX;
	v.vt = VT_EMPTY;
	hres = IPropertyStorage_ReadMultiple(propstorage, 1, &ps, &v);
	ok(hres == S_FALSE, "got 0x%08x\n", hres);
	ok(v.vt == VT_EMPTY, "got %d\n", v.vt);

	IPropertyStorage_Release(propstorage);
	IPropertySetStorage_Release(propsetstorage);

	IPersistFile_Release(persist_file);
	DeleteFileW(file_path);
	}
	}

	static void test_SetURLFlags(void)
	{
	HRESULT hr;
	IUniformResourceLocatorA *urlA;

	hr = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IUniformResourceLocatorA, (void**)&urlA);
	ok(hr == S_OK, "Could not create InternetShortcut object: %08x\n", hr);

	check_string_transform(urlA, "somerandomstring", 0, NULL, TRUE);
	check_string_transform(urlA, "www.winehq.org", 0, NULL, TRUE);

	check_string_transform(urlA, "www.winehq.org", IURL_SETURL_FL_GUESS_PROTOCOL, "http://www.winehq.org/", FALSE);
	check_string_transform(urlA, "ftp.winehq.org", IURL_SETURL_FL_GUESS_PROTOCOL, "ftp://ftp.winehq.org/", FALSE);

	urlA->lpVtbl->Release(urlA);
	}

	static void test_InternetShortcut(void)
	{
	IUniformResourceLocatorA *url;
	HRESULT hres;

	hres = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IUniformResourceLocatorA, (void**)&url);
	ok(hres == S_OK, "Could not create CLSID_InternetShortcut instance: %08x\n", hres);
	if(FAILED(hres))
	return;
	url->lpVtbl->Release(url);

	test_Aggregability();
	test_QueryInterface();
	test_NullURLs();
	test_SetURLFlags();
	test_ReadAndWriteProperties();
	test_Load();
	}

	START_TEST(intshcut)
	{
	OleInitialize(NULL);

	test_InternetShortcut();

	OleUninitialize();
	}