1 /*
2 * Unit tests to document InternetShortcut's behaviour
3 *
4 * Copyright 2008 Damjan Jovanovic
5 *
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19 */
20
21 #include <stdarg.h>
22 #include <stdio.h>
23
24 #define COBJMACROS
25
26 #include "windef.h"
27 #include "winbase.h"
28 #include "winreg.h"
29 #include "winerror.h"
30
31 #include "shlobj.h"
32 #include "shobjidl.h"
33 #include "shlguid.h"
34 #include "ole2.h"
35 #include "initguid.h"
36 #include "isguids.h"
37 #include "intshcut.h"
38
39 #include "wine/test.h"
40
41 static HRESULT WINAPI Unknown_QueryInterface(IUnknown *pUnknown, REFIID riid, void **ppvObject)
42 {
43 if (IsEqualGUID(&IID_IUnknown, riid))
44 {
45 *ppvObject = pUnknown;
46 return S_OK;
47 }
48 return E_NOINTERFACE;
49 }
50
51 static ULONG WINAPI Unknown_AddRef(IUnknown *pUnknown)
52 {
53 return 2;
54 }
55
56 static ULONG WINAPI Unknown_Release(IUnknown *pUnknown)
57 {
58 return 1;
59 }
60
61 static IUnknownVtbl unknownVtbl = {
62 Unknown_QueryInterface,
63 Unknown_AddRef,
64 Unknown_Release
65 };
66
67 static IUnknown unknown = {
68 &unknownVtbl
69 };
70
71 static const char *printGUID(const GUID *guid)
72 {
73 static char guidSTR[39];
74
75 if (!guid) return NULL;
76
77 sprintf(guidSTR, "{%08x-%04x-%04x-%02x%02x-%02x%02x%02x%02x%02x%02x}",
78 guid->Data1, guid->Data2, guid->Data3,
79 guid->Data4[0], guid->Data4[1], guid->Data4[2], guid->Data4[3],
80 guid->Data4[4], guid->Data4[5], guid->Data4[6], guid->Data4[7]);
81 return guidSTR;
82 }
83
84 static void test_Aggregability(void)
85 {
86 HRESULT hr;
87 IUnknown *pUnknown = NULL;
88
89 hr = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IUnknown, (void**)&pUnknown);
90 ok(SUCCEEDED(hr), "could not create instance of CLSID_InternetShortcut with IID_IUnknown, hr = 0x%x\n", hr);
91 if (pUnknown)
92 IUnknown_Release(pUnknown);
93
94 hr = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IUniformResourceLocatorA, (void**)&pUnknown);
95 ok(SUCCEEDED(hr), "could not create instance of CLSID_InternetShortcut with IID_IUniformResourceLocatorA, hr = 0x%x\n", hr);
96 if (pUnknown)
97 IUnknown_Release(pUnknown);
98
99 hr = CoCreateInstance(&CLSID_InternetShortcut, &unknown, CLSCTX_ALL, &IID_IUnknown, (void**)&pUnknown);
100 ok(FAILED(hr), "aggregation didn't fail like it should, hr = 0x%x\n", hr);
101 if (pUnknown)
102 IUnknown_Release(pUnknown);
103 }
104
105 static void can_query_interface(IUnknown *pUnknown, REFIID riid)
106 {
107 HRESULT hr;
108 IUnknown *newInterface;
109 hr = IUnknown_QueryInterface(pUnknown, riid, (void**)&newInterface);
110 ok(SUCCEEDED(hr), "interface %s could not be queried\n", printGUID(riid));
111 if (SUCCEEDED(hr))
112 IUnknown_Release(newInterface);
113 }
114
115 static void test_QueryInterface(void)
116 {
117 HRESULT hr;
118 IUnknown *pUnknown;
119
120 hr = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IUnknown, (void**)&pUnknown);
121 if (SUCCEEDED(hr))
122 {
123 can_query_interface(pUnknown, &IID_IUniformResourceLocatorA);
124 can_query_interface(pUnknown, &IID_IUniformResourceLocatorW);
125 can_query_interface(pUnknown, &IID_IPersistFile);
126 IUnknown_Release(pUnknown);
127 }
128 else
129 skip("could not create a CLSID_InternetShortcut for QueryInterface tests, hr=0x%x\n", hr);
130 }
131
132 static CHAR *set_and_get_url(IUniformResourceLocatorA *urlA, LPCSTR input, DWORD flags)
133 {
134 HRESULT hr;
135 hr = urlA->lpVtbl->SetURL(urlA, input, flags);
136 if (SUCCEEDED(hr))
137 {
138 CHAR *output;
139 hr = urlA->lpVtbl->GetURL(urlA, &output);
140 if (SUCCEEDED(hr))
141 return output;
142 else
143 skip("GetUrl failed, hr=0x%x\n", hr);
144 }
145 else
146 skip("SetUrl (%s, 0x%x) failed, hr=0x%x\n", input, flags, hr);
147 return NULL;
148 }
149
150 static void check_string_transform(IUniformResourceLocatorA *urlA, LPCSTR input, DWORD flags, LPCSTR expectedOutput)
151 {
152 CHAR *output = set_and_get_url(urlA, input, flags);
153 if (output != NULL)
154 {
155 ok(lstrcmpA(output, expectedOutput) == 0, "unexpected URL change %s -> %s (expected %s)\n",
156 input, output, expectedOutput);
157 CoTaskMemFree(output);
158 }
159 }
160
161 static void test_NullURLs(void)
162 {
163 HRESULT hr;
164 IUniformResourceLocatorA *urlA;
165
166 hr = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IUniformResourceLocatorA, (void**)&urlA);
167 if (SUCCEEDED(hr))
168 {
169 LPSTR url = NULL;
170
171 hr = urlA->lpVtbl->GetURL(urlA, &url);
172 ok(SUCCEEDED(hr), "getting uninitialized URL unexpectedly failed, hr=0x%x\n", hr);
173 ok(url == NULL, "uninitialized URL is not NULL but %s\n", url);
174
175 hr = urlA->lpVtbl->SetURL(urlA, NULL, 0);
176 ok(SUCCEEDED(hr), "setting NULL URL unexpectedly failed, hr=0x%x\n", hr);
177
178 hr = urlA->lpVtbl->GetURL(urlA, &url);
179 ok(SUCCEEDED(hr), "getting NULL URL unexpectedly failed, hr=0x%x\n", hr);
180 ok(url == NULL, "URL unexpectedly not NULL but %s\n", url);
181
182 urlA->lpVtbl->Release(urlA);
183 }
184 else
185 skip("could not create a CLSID_InternetShortcut for NullURL tests, hr=0x%x\n", hr);
186 }
187
188 static void test_SetURLFlags(void)
189 {
190 HRESULT hr;
191 IUniformResourceLocatorA *urlA;
192
193 hr = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IUniformResourceLocatorA, (void**)&urlA);
194 if (SUCCEEDED(hr))
195 {
196 check_string_transform(urlA, "somerandomstring", 0, "somerandomstring");
197 check_string_transform(urlA, "www.winehq.org", 0, "www.winehq.org");
198
199 todo_wine
200 {
201 check_string_transform(urlA, "www.winehq.org", IURL_SETURL_FL_GUESS_PROTOCOL, "http://www.winehq.org/");
202 check_string_transform(urlA, "ftp.winehq.org", IURL_SETURL_FL_GUESS_PROTOCOL, "ftp://ftp.winehq.org/");
203 }
204
205 urlA->lpVtbl->Release(urlA);
206 }
207 else
208 skip("could not create a CLSID_InternetShortcut for SetUrl tests, hr=0x%x\n", hr);
209 }
210
211 static void test_InternetShortcut(void)
212 {
213 test_Aggregability();
214 test_QueryInterface();
215 test_NullURLs();
216 test_SetURLFlags();
217 }
218
219 START_TEST(intshcut)
220 {
221 OleInitialize(NULL);
222 test_InternetShortcut();
223 OleUninitialize();
224 }
225
This page was automatically generated by the
LXR engine.
Visit the LXR main site for more
information.