1 /*
2  * Tests for autocomplete
3  *
4  * Copyright 2008 Jan de Mooij
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 #define COBJMACROS
22 
23 #include <stdarg.h>
24 
25 #include "windows.h"
26 #include "shobjidl.h"
27 #include "shlguid.h"
28 #include "shldisp.h"
29 #include "shlobj.h"
30 
31 #include "wine/heap.h"
32 #include "wine/test.h"
33 
34 static HWND hMainWnd, hEdit;
35 static HINSTANCE hinst;
36 static int killfocus_count;
37 
test_invalid_init(void)38 static void test_invalid_init(void)
39 {
40     HRESULT hr;
41     IAutoComplete *ac;
42     IUnknown *acSource;
43     HWND edit_control;
44 
45     /* AutoComplete instance */
46     hr = CoCreateInstance(&CLSID_AutoComplete, NULL, CLSCTX_INPROC_SERVER,
47                          &IID_IAutoComplete, (void **)&ac);
48     if (hr == REGDB_E_CLASSNOTREG)
49     {
50         win_skip("CLSID_AutoComplete is not registered\n");
51         return;
52     }
53     ok(hr == S_OK, "no IID_IAutoComplete (0x%08x)\n", hr);
54 
55     /* AutoComplete source */
56     hr = CoCreateInstance(&CLSID_ACLMulti, NULL, CLSCTX_INPROC_SERVER,
57                         &IID_IACList, (void **)&acSource);
58     if (hr == REGDB_E_CLASSNOTREG)
59     {
60         win_skip("CLSID_ACLMulti is not registered\n");
61         IAutoComplete_Release(ac);
62         return;
63     }
64     ok(hr == S_OK, "no IID_IACList (0x%08x)\n", hr);
65 
66     edit_control = CreateWindowExA(0, "EDIT", "Some text", 0, 10, 10, 300, 300,
67                        hMainWnd, NULL, hinst, NULL);
68     ok(edit_control != NULL, "Can't create edit control\n");
69 
70     /* The refcount of acSource would be incremented on older Windows. */
71     hr = IAutoComplete_Init(ac, NULL, acSource, NULL, NULL);
72     ok(hr == E_INVALIDARG ||
73        broken(hr == S_OK), /* Win2k/XP/Win2k3 */
74        "Init returned 0x%08x\n", hr);
75     if (hr == E_INVALIDARG)
76     {
77         LONG ref;
78 
79         IUnknown_AddRef(acSource);
80         ref = IUnknown_Release(acSource);
81         ok(ref == 1, "Expected AutoComplete source refcount to be 1, got %d\n", ref);
82     }
83 
84 if (0)
85 {
86     /* Older Windows versions never check the window handle, while newer
87      * versions only check for NULL. Subsequent attempts to initialize the
88      * object after this call succeeds would fail, because initialization
89      * state is determined by whether a non-NULL window handle is stored. */
90     hr = IAutoComplete_Init(ac, (HWND)0xdeadbeef, acSource, NULL, NULL);
91     ok(hr == S_OK, "Init returned 0x%08x\n", hr);
92 
93     /* Tests crash on older Windows. */
94     hr = IAutoComplete_Init(ac, NULL, NULL, NULL, NULL);
95     ok(hr == E_INVALIDARG, "Init returned 0x%08x\n", hr);
96 
97     hr = IAutoComplete_Init(ac, edit_control, NULL, NULL, NULL);
98     ok(hr == E_INVALIDARG, "Init returned 0x%08x\n", hr);
99 }
100 
101     /* bind to edit control */
102     hr = IAutoComplete_Init(ac, edit_control, acSource, NULL, NULL);
103     ok(hr == S_OK, "Init returned 0x%08x\n", hr);
104 
105     /* try invalid parameters after successful initialization .*/
106     hr = IAutoComplete_Init(ac, NULL, NULL, NULL, NULL);
107     ok(hr == E_INVALIDARG ||
108        hr == E_FAIL, /* Win2k/XP/Win2k3 */
109        "Init returned 0x%08x\n", hr);
110 
111     hr = IAutoComplete_Init(ac, NULL, acSource, NULL, NULL);
112     ok(hr == E_INVALIDARG ||
113        hr == E_FAIL, /* Win2k/XP/Win2k3 */
114        "Init returned 0x%08x\n", hr);
115 
116     hr = IAutoComplete_Init(ac, edit_control, NULL, NULL, NULL);
117     ok(hr == E_INVALIDARG ||
118        hr == E_FAIL, /* Win2k/XP/Win2k3 */
119        "Init returned 0x%08x\n", hr);
120 
121     /* try initializing twice on the same control */
122     hr = IAutoComplete_Init(ac, edit_control, acSource, NULL, NULL);
123     ok(hr == E_FAIL, "Init returned 0x%08x\n", hr);
124 
125     /* try initializing with a different control */
126     hr = IAutoComplete_Init(ac, hEdit, acSource, NULL, NULL);
127     ok(hr == E_FAIL, "Init returned 0x%08x\n", hr);
128 
129     DestroyWindow(edit_control);
130 
131     /* try initializing with a different control after
132      * destroying the original initialization control */
133     hr = IAutoComplete_Init(ac, hEdit, acSource, NULL, NULL);
134     ok(hr == E_UNEXPECTED ||
135        hr == E_FAIL, /* Win2k/XP/Win2k3 */
136        "Init returned 0x%08x\n", hr);
137 
138     IUnknown_Release(acSource);
139     IAutoComplete_Release(ac);
140 }
test_init(void)141 static IAutoComplete *test_init(void)
142 {
143     HRESULT r;
144     IAutoComplete *ac, *ac2;
145     IUnknown *acSource;
146     LONG_PTR user_data;
147 
148     /* AutoComplete instance */
149     r = CoCreateInstance(&CLSID_AutoComplete, NULL, CLSCTX_INPROC_SERVER,
150                          &IID_IAutoComplete, (LPVOID*)&ac);
151     if (r == REGDB_E_CLASSNOTREG)
152     {
153         win_skip("CLSID_AutoComplete is not registered\n");
154         return NULL;
155     }
156     ok(r == S_OK, "no IID_IAutoComplete (0x%08x)\n", r);
157 
158     /* AutoComplete source */
159     r = CoCreateInstance(&CLSID_ACLMulti, NULL, CLSCTX_INPROC_SERVER,
160                         &IID_IACList, (LPVOID*)&acSource);
161     if (r == REGDB_E_CLASSNOTREG)
162     {
163         win_skip("CLSID_ACLMulti is not registered\n");
164         IAutoComplete_Release(ac);
165         return NULL;
166     }
167     ok(r == S_OK, "no IID_IACList (0x%08x)\n", r);
168 
169     user_data = GetWindowLongPtrA(hEdit, GWLP_USERDATA);
170     ok(user_data == 0, "Expected the edit control user data to be zero\n");
171 
172     /* bind to edit control */
173     r = IAutoComplete_Init(ac, hEdit, acSource, NULL, NULL);
174     ok(r == S_OK, "Init returned 0x%08x\n", r);
175 
176     user_data = GetWindowLongPtrA(hEdit, GWLP_USERDATA);
177     ok(user_data == 0, "Expected the edit control user data to be zero\n");
178 
179     /* bind a different object to the same edit control */
180     r = CoCreateInstance(&CLSID_AutoComplete, NULL, CLSCTX_INPROC_SERVER,
181                          &IID_IAutoComplete, (LPVOID*)&ac2);
182     ok(r == S_OK, "no IID_IAutoComplete (0x%08x)\n", r);
183 
184     r = IAutoComplete_Init(ac2, hEdit, acSource, NULL, NULL);
185     ok(r == S_OK, "Init returned 0x%08x\n", r);
186     IAutoComplete_Release(ac2);
187 
188     IUnknown_Release(acSource);
189 
190     return ac;
191 }
192 
test_killfocus(void)193 static void test_killfocus(void)
194 {
195     /* Test if WM_KILLFOCUS messages are handled properly by checking if
196      * the parent receives an EN_KILLFOCUS message. */
197     SetFocus(hEdit);
198     killfocus_count = 0;
199     SetFocus(0);
200     ok(killfocus_count == 1, "Expected one EN_KILLFOCUS message, got: %d\n", killfocus_count);
201 }
202 
MyWndProc(HWND hWnd,UINT msg,WPARAM wParam,LPARAM lParam)203 static LRESULT CALLBACK MyWndProc(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam)
204 {
205     switch(msg) {
206     case WM_CREATE:
207         /* create edit control */
208         hEdit = CreateWindowExA(0, "EDIT", "Some text", 0, 10, 10, 300, 300,
209                     hWnd, NULL, hinst, NULL);
210         ok(hEdit != NULL, "Can't create edit control\n");
211         break;
212     case WM_COMMAND:
213         if(HIWORD(wParam) == EN_KILLFOCUS)
214             killfocus_count++;
215         break;
216     }
217     return DefWindowProcA(hWnd, msg, wParam, lParam);
218 }
219 
createMainWnd(void)220 static void createMainWnd(void)
221 {
222     WNDCLASSA wc;
223     wc.style = CS_HREDRAW | CS_VREDRAW;
224     wc.cbClsExtra = 0;
225     wc.cbWndExtra = 0;
226     wc.hInstance = GetModuleHandleA(NULL);
227     wc.hIcon = NULL;
228     wc.hCursor = LoadCursorA(NULL, (LPSTR)IDC_IBEAM);
229     wc.hbrBackground = GetSysColorBrush(COLOR_WINDOW);
230     wc.lpszMenuName = NULL;
231     wc.lpszClassName = "MyTestWnd";
232     wc.lpfnWndProc = MyWndProc;
233     RegisterClassA(&wc);
234 
235     hMainWnd = CreateWindowExA(0, "MyTestWnd", "Blah", WS_OVERLAPPEDWINDOW,
236       CW_USEDEFAULT, CW_USEDEFAULT, 130, 105, NULL, NULL, GetModuleHandleA(NULL), 0);
237 }
238 
239 static WNDPROC HijackerWndProc_prev;
240 static const WCHAR HijackerWndProc_txt[] = {'H','i','j','a','c','k','e','d',0};
HijackerWndProc(HWND hWnd,UINT msg,WPARAM wParam,LPARAM lParam)241 static LRESULT CALLBACK HijackerWndProc(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam)
242 {
243     switch(msg) {
244     case WM_GETTEXT:
245     {
246         size_t len = min(wParam, ARRAY_SIZE(HijackerWndProc_txt));
247         memcpy((void*)lParam, HijackerWndProc_txt, len * sizeof(WCHAR));
248         return len;
249     }
250     case WM_GETTEXTLENGTH:
251         return ARRAY_SIZE(HijackerWndProc_txt) - 1;
252     }
253     return CallWindowProcW(HijackerWndProc_prev, hWnd, msg, wParam, lParam);
254 }
255 
HijackerWndProc2(HWND hWnd,UINT msg,WPARAM wParam,LPARAM lParam)256 static LRESULT CALLBACK HijackerWndProc2(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam)
257 {
258     switch(msg) {
259     case EM_SETSEL:
260         lParam = wParam;
261         break;
262     case WM_SETTEXT:
263         lParam = (LPARAM)HijackerWndProc_txt;
264         break;
265     }
266     return CallWindowProcW(HijackerWndProc_prev, hWnd, msg, wParam, lParam);
267 }
268 
269 struct string_enumerator
270 {
271     IEnumString IEnumString_iface;
272     IACList IACList_iface;
273     LONG ref;
274     WCHAR **data;
275     int data_len;
276     int cur;
277     UINT num_resets;
278     UINT num_expand;
279     WCHAR last_expand[32];
280 };
281 
impl_from_IEnumString(IEnumString * iface)282 static struct string_enumerator *impl_from_IEnumString(IEnumString *iface)
283 {
284     return CONTAINING_RECORD(iface, struct string_enumerator, IEnumString_iface);
285 }
286 
string_enumerator_QueryInterface(IEnumString * iface,REFIID riid,void ** ppv)287 static HRESULT WINAPI string_enumerator_QueryInterface(IEnumString *iface, REFIID riid, void **ppv)
288 {
289     struct string_enumerator *this = impl_from_IEnumString(iface);
290     if (IsEqualGUID(riid, &IID_IEnumString) || IsEqualGUID(riid, &IID_IUnknown))
291         *ppv = &this->IEnumString_iface;
292     else if (IsEqualGUID(riid, &IID_IACList))
293         *ppv = &this->IACList_iface;
294     else
295     {
296         *ppv = NULL;
297         return E_NOINTERFACE;
298     }
299 
300     IUnknown_AddRef(&this->IEnumString_iface);
301     return S_OK;
302 }
303 
string_enumerator_AddRef(IEnumString * iface)304 static ULONG WINAPI string_enumerator_AddRef(IEnumString *iface)
305 {
306     struct string_enumerator *this = impl_from_IEnumString(iface);
307 
308     ULONG ref = InterlockedIncrement(&this->ref);
309 
310     return ref;
311 }
312 
string_enumerator_Release(IEnumString * iface)313 static ULONG WINAPI string_enumerator_Release(IEnumString *iface)
314 {
315     struct string_enumerator *this = impl_from_IEnumString(iface);
316 
317     ULONG ref = InterlockedDecrement(&this->ref);
318 
319     if (!ref)
320         heap_free(this);
321 
322     return ref;
323 }
324 
string_enumerator_Next(IEnumString * iface,ULONG num,LPOLESTR * strings,ULONG * num_returned)325 static HRESULT WINAPI string_enumerator_Next(IEnumString *iface, ULONG num, LPOLESTR *strings, ULONG *num_returned)
326 {
327     struct string_enumerator *this = impl_from_IEnumString(iface);
328     int i, len;
329 
330     *num_returned = 0;
331     for (i = 0; i < num; i++)
332     {
333         if (this->cur >= this->data_len)
334             return S_FALSE;
335 
336         len = lstrlenW(this->data[this->cur]) + 1;
337 
338         strings[i] = CoTaskMemAlloc(len * sizeof(WCHAR));
339         memcpy(strings[i], this->data[this->cur], len * sizeof(WCHAR));
340 
341         (*num_returned)++;
342         this->cur++;
343     }
344 
345     return S_OK;
346 }
347 
string_enumerator_Reset(IEnumString * iface)348 static HRESULT WINAPI string_enumerator_Reset(IEnumString *iface)
349 {
350     struct string_enumerator *this = impl_from_IEnumString(iface);
351 
352     this->cur = 0;
353     this->num_resets++;
354 
355     return S_OK;
356 }
357 
string_enumerator_Skip(IEnumString * iface,ULONG num)358 static HRESULT WINAPI string_enumerator_Skip(IEnumString *iface, ULONG num)
359 {
360     struct string_enumerator *this = impl_from_IEnumString(iface);
361 
362     this->cur += num;
363 
364     return S_OK;
365 }
366 
string_enumerator_Clone(IEnumString * iface,IEnumString ** out)367 static HRESULT WINAPI string_enumerator_Clone(IEnumString *iface, IEnumString **out)
368 {
369     *out = NULL;
370     return E_NOTIMPL;
371 }
372 
373 static IEnumStringVtbl string_enumerator_vtbl =
374 {
375     string_enumerator_QueryInterface,
376     string_enumerator_AddRef,
377     string_enumerator_Release,
378     string_enumerator_Next,
379     string_enumerator_Skip,
380     string_enumerator_Reset,
381     string_enumerator_Clone
382 };
383 
impl_from_IACList(IACList * iface)384 static struct string_enumerator *impl_from_IACList(IACList *iface)
385 {
386     return CONTAINING_RECORD(iface, struct string_enumerator, IACList_iface);
387 }
388 
aclist_QueryInterface(IACList * iface,REFIID riid,void ** ppv)389 static HRESULT WINAPI aclist_QueryInterface(IACList *iface, REFIID riid, void **ppv)
390 {
391     return string_enumerator_QueryInterface(&impl_from_IACList(iface)->IEnumString_iface, riid, ppv);
392 }
393 
aclist_AddRef(IACList * iface)394 static ULONG WINAPI aclist_AddRef(IACList *iface)
395 {
396     return string_enumerator_AddRef(&impl_from_IACList(iface)->IEnumString_iface);
397 }
398 
aclist_Release(IACList * iface)399 static ULONG WINAPI aclist_Release(IACList *iface)
400 {
401     return string_enumerator_Release(&impl_from_IACList(iface)->IEnumString_iface);
402 }
403 
aclist_Expand(IACList * iface,const WCHAR * expand)404 static HRESULT WINAPI aclist_Expand(IACList *iface, const WCHAR *expand)
405 {
406     struct string_enumerator *this = impl_from_IACList(iface);
407 
408     /* see what we get called with and how many times,
409        don't actually do any expansion of the strings */
410     memcpy(this->last_expand, expand, min((lstrlenW(expand) + 1)*sizeof(WCHAR), sizeof(this->last_expand)));
411     this->last_expand[ARRAY_SIZE(this->last_expand) - 1] = '\0';
412     this->num_expand++;
413 
414     return S_OK;
415 }
416 
417 static IACListVtbl aclist_vtbl =
418 {
419     aclist_QueryInterface,
420     aclist_AddRef,
421     aclist_Release,
422     aclist_Expand
423 };
424 
string_enumerator_create(void ** ppv,WCHAR ** suggestions,int count)425 static HRESULT string_enumerator_create(void **ppv, WCHAR **suggestions, int count)
426 {
427     struct string_enumerator *object;
428 
429     object = heap_alloc_zero(sizeof(*object));
430     object->IEnumString_iface.lpVtbl = &string_enumerator_vtbl;
431     object->IACList_iface.lpVtbl = &aclist_vtbl;
432     object->ref = 1;
433     object->data = suggestions;
434     object->data_len = count;
435     object->cur = 0;
436 
437     *ppv = &object->IEnumString_iface;
438 
439     return S_OK;
440 }
441 
dispatch_messages(void)442 static void dispatch_messages(void)
443 {
444     MSG msg;
445     Sleep(33);
446     while (PeekMessageA(&msg, 0, 0, 0, PM_REMOVE))
447     {
448         TranslateMessage(&msg);
449         DispatchMessageA(&msg);
450     }
451 }
452 
453 #define check_dropdown(acdropdown, hwnd_edit, list, list_num) check_dropdown_(__FILE__, __LINE__, acdropdown, hwnd_edit, list, list_num)
check_dropdown_(const char * file,UINT line,IAutoCompleteDropDown * acdropdown,HWND hwnd_edit,WCHAR ** list,UINT list_num)454 static void check_dropdown_(const char *file, UINT line, IAutoCompleteDropDown *acdropdown, HWND hwnd_edit, WCHAR **list, UINT list_num)
455 {
456     UINT i;
457     DWORD flags = 0;
458     LPWSTR str;
459     HRESULT hr;
460 
461     hr = IAutoCompleteDropDown_GetDropDownStatus(acdropdown, &flags, &str);
462     ok_(file, line)(hr == S_OK, "IAutoCompleteDropDown_GetDropDownStatus failed: %x\n", hr);
463     if (hr != S_OK) return;
464     if (list_num) ok_(file, line)(flags & ACDD_VISIBLE, "AutoComplete DropDown not visible\n");
465     else
466     {
467         ok_(file, line)(!(flags & ACDD_VISIBLE), "AutoComplete DropDown visible\n");
468         return;
469     }
470     ok_(file, line)(str == NULL, "Expected (null), got %s\n", wine_dbgstr_w(str));
471     if (str)
472     {
473         CoTaskMemFree(str);
474         return;
475     }
476 
477     for (i = 0; i <= list_num; i++)
478     {
479         flags = 0;
480         SendMessageW(hwnd_edit, WM_KEYDOWN, VK_DOWN, 0);
481         SendMessageW(hwnd_edit, WM_KEYUP, VK_DOWN, 0xc0000000);
482         hr = IAutoCompleteDropDown_GetDropDownStatus(acdropdown, &flags, &str);
483         ok_(file, line)(hr == S_OK, "IAutoCompleteDropDown_GetDropDownStatus failed: %x\n", hr);
484         ok_(file, line)(flags & ACDD_VISIBLE, "AutoComplete DropDown not visible\n");
485         if (hr == S_OK)
486         {
487             if (i < list_num)
488                 ok_(file, line)(str && !lstrcmpW(list[i], str), "Expected %s, got %s\n",
489                                 wine_dbgstr_w(list[i]), wine_dbgstr_w(str));
490             else
491                 ok_(file, line)(str == NULL, "Expected (null), got %s\n", wine_dbgstr_w(str));
492         }
493         CoTaskMemFree(str);
494     }
495 }
496 
test_aclist_expand(HWND hwnd_edit,void * enumerator,IAutoCompleteDropDown * acdropdown)497 static void test_aclist_expand(HWND hwnd_edit, void *enumerator, IAutoCompleteDropDown *acdropdown)
498 {
499     struct string_enumerator *obj = (struct string_enumerator*)enumerator;
500     static WCHAR str1[] = {'t','e','s','t',0};
501     static WCHAR str1a[] = {'t','e','s','t','\\',0};
502     static WCHAR str2[] = {'t','e','s','t','\\','f','o','o','\\','b','a','r','\\','b','a',0};
503     static WCHAR str2a[] = {'t','e','s','t','\\','f','o','o','\\','b','a','r','\\',0};
504     static WCHAR str2b[] = {'t','e','s','t','\\','f','o','o','\\','b','a','r','\\','b','a','z','_','b','b','q','\\',0};
505     HRESULT hr;
506     obj->num_resets = 0;
507 
508     ok(obj->num_expand == 0, "Expected 0 expansions, got %u\n", obj->num_expand);
509     SendMessageW(hwnd_edit, WM_SETTEXT, 0, (LPARAM)str1);
510     SendMessageW(hwnd_edit, EM_SETSEL, ARRAY_SIZE(str1) - 1, ARRAY_SIZE(str1) - 1);
511     SendMessageW(hwnd_edit, WM_CHAR, '\\', 1);
512     dispatch_messages();
513     ok(obj->num_expand == 1, "Expected 1 expansion, got %u\n", obj->num_expand);
514     ok(lstrcmpW(obj->last_expand, str1a) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str1a), wine_dbgstr_w(obj->last_expand));
515     ok(obj->num_resets == 1, "Expected 1 reset, got %u\n", obj->num_resets);
516     SendMessageW(hwnd_edit, WM_SETTEXT, 0, (LPARAM)str2);
517     SendMessageW(hwnd_edit, EM_SETSEL, ARRAY_SIZE(str2) - 1, ARRAY_SIZE(str2) - 1);
518     SendMessageW(hwnd_edit, WM_CHAR, 'z', 1);
519     dispatch_messages();
520     ok(obj->num_expand == 2, "Expected 2 expansions, got %u\n", obj->num_expand);
521     ok(lstrcmpW(obj->last_expand, str2a) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str2a), wine_dbgstr_w(obj->last_expand));
522     ok(obj->num_resets == 2, "Expected 2 resets, got %u\n", obj->num_resets);
523     SetFocus(hwnd_edit);
524     SendMessageW(hwnd_edit, WM_CHAR, '_', 1);
525     SendMessageW(hwnd_edit, WM_CHAR, 'b', 1);
526     SetFocus(0);
527     SetFocus(hwnd_edit);
528     SendMessageW(hwnd_edit, WM_CHAR, 'b', 1);
529     SendMessageW(hwnd_edit, WM_CHAR, 'q', 1);
530     dispatch_messages();
531     ok(obj->num_expand == 2, "Expected 2 expansions, got %u\n", obj->num_expand);
532     ok(obj->num_resets == 2, "Expected 2 resets, got %u\n", obj->num_resets);
533     SendMessageW(hwnd_edit, WM_CHAR, '\\', 1);
534     dispatch_messages();
535     ok(obj->num_expand == 3, "Expected 3 expansions, got %u\n", obj->num_expand);
536     ok(lstrcmpW(obj->last_expand, str2b) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str2b), wine_dbgstr_w(obj->last_expand));
537     ok(obj->num_resets == 3, "Expected 3 resets, got %u\n", obj->num_resets);
538     SendMessageW(hwnd_edit, EM_SETSEL, ARRAY_SIZE(str1a) - 1, -1);
539     SendMessageW(hwnd_edit, WM_CHAR, 'x', 1);
540     SendMessageW(hwnd_edit, WM_CHAR, 'y', 1);
541     dispatch_messages();
542     ok(obj->num_expand == 4, "Expected 4 expansions, got %u\n", obj->num_expand);
543     ok(lstrcmpW(obj->last_expand, str1a) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str1a), wine_dbgstr_w(obj->last_expand));
544     ok(obj->num_resets == 4, "Expected 4 resets, got %u\n", obj->num_resets);
545     SendMessageW(hwnd_edit, EM_SETSEL, ARRAY_SIZE(str1) - 1, -1);
546     SendMessageW(hwnd_edit, WM_CHAR, 'x', 1);
547     dispatch_messages();
548     ok(obj->num_expand == 4, "Expected 4 expansions, got %u\n", obj->num_expand);
549     ok(obj->num_resets == 5, "Expected 5 resets, got %u\n", obj->num_resets);
550     SendMessageW(hwnd_edit, WM_SETTEXT, 0, (LPARAM)str1a);
551     SendMessageW(hwnd_edit, EM_SETSEL, ARRAY_SIZE(str1a) - 1, ARRAY_SIZE(str1a) - 1);
552     SendMessageW(hwnd_edit, WM_CHAR, 'f', 1);
553     dispatch_messages();
554     ok(obj->num_expand == 5, "Expected 5 expansions, got %u\n", obj->num_expand);
555     ok(lstrcmpW(obj->last_expand, str1a) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str1a), wine_dbgstr_w(obj->last_expand));
556     ok(obj->num_resets == 6, "Expected 6 resets, got %u\n", obj->num_resets);
557     hr = IAutoCompleteDropDown_ResetEnumerator(acdropdown);
558     ok(hr == S_OK, "IAutoCompleteDropDown_ResetEnumerator failed: %x\n", hr);
559     SendMessageW(hwnd_edit, WM_CHAR, 'o', 1);
560     dispatch_messages();
561     ok(obj->num_expand == 6, "Expected 6 expansions, got %u\n", obj->num_expand);
562     ok(lstrcmpW(obj->last_expand, str1a) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str1a), wine_dbgstr_w(obj->last_expand));
563     ok(obj->num_resets == 7, "Expected 7 resets, got %u\n", obj->num_resets);
564 }
565 
test_prefix_filtering(HWND hwnd_edit)566 static void test_prefix_filtering(HWND hwnd_edit)
567 {
568     static WCHAR htt[]  = {'h','t','t',0};
569     static WCHAR www[]  = {'w','w','w','.',0};
570     static WCHAR str0[] = {'w','w','w','.','a','x',0};
571     static WCHAR str1[] = {'h','t','t','p','s',':','/','/','w','w','w','.','a','c',0};
572     static WCHAR str2[] = {'a','a',0};
573     static WCHAR str3[] = {'a','b',0};
574     static WCHAR str4[] = {'h','t','t','p',':','/','/','a','0',0};
575     static WCHAR str5[] = {'h','t','t','p','s',':','/','/','h','t','a',0};
576     static WCHAR str6[] = {'h','f','o','o',0};
577     static WCHAR str7[] = {'h','t','t','p',':','/','/','w','w','w','.','a','d','d',0};
578     static WCHAR str8[] = {'w','w','w','.','w','w','w','.','?',0};
579     static WCHAR str9[] = {'h','t','t','p',':','/','/','a','b','c','.','a','a','.','c','o','m',0};
580     static WCHAR str10[]= {'f','t','p',':','/','/','a','b','c',0};
581     static WCHAR str11[]= {'f','i','l','e',':','/','/','a','a',0};
582     static WCHAR str12[]= {'f','t','p',':','/','/','w','w','w','.','a','a',0};
583     static WCHAR *suggestions[] = { str0, str1, str2, str3, str4, str5, str6, str7, str8, str9, str10, str11, str12 };
584     static WCHAR *sorted1[] = { str4, str2, str3, str9, str1, str7, str0 };
585     static WCHAR *sorted2[] = { str3, str9 };
586     static WCHAR *sorted3[] = { str1, str7, str0 };
587     static WCHAR *sorted4[] = { str6, str5 };
588     static WCHAR *sorted5[] = { str5 };
589     static WCHAR *sorted6[] = { str4, str9 };
590     static WCHAR *sorted7[] = { str11, str10, str12 };
591     IUnknown *enumerator;
592     IAutoComplete2 *autocomplete;
593     IAutoCompleteDropDown *acdropdown;
594     WCHAR buffer[20];
595     HRESULT hr;
596 
597     hr = CoCreateInstance(&CLSID_AutoComplete, NULL, CLSCTX_INPROC_SERVER, &IID_IAutoComplete2, (void**)&autocomplete);
598     ok(hr == S_OK, "CoCreateInstance failed: %x\n", hr);
599 
600     hr = IAutoComplete2_QueryInterface(autocomplete, &IID_IAutoCompleteDropDown, (LPVOID*)&acdropdown);
601     ok(hr == S_OK, "No IAutoCompleteDropDown interface: %x\n", hr);
602 
603     string_enumerator_create((void**)&enumerator, suggestions, ARRAY_SIZE(suggestions));
604 
605     hr = IAutoComplete2_SetOptions(autocomplete, ACO_FILTERPREFIXES | ACO_AUTOSUGGEST | ACO_AUTOAPPEND);
606     ok(hr == S_OK, "IAutoComplete2_SetOptions failed: %x\n", hr);
607     hr = IAutoComplete2_Init(autocomplete, hwnd_edit, enumerator, NULL, NULL);
608     ok(hr == S_OK, "IAutoComplete_Init failed: %x\n", hr);
609 
610     SendMessageW(hwnd_edit, EM_SETSEL, 0, -1);
611     SendMessageW(hwnd_edit, WM_CHAR, 'a', 1);
612     dispatch_messages();
613     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
614     ok(lstrcmpW(str4 + 7, buffer) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str4 + 7), wine_dbgstr_w(buffer));
615     check_dropdown(acdropdown, hwnd_edit, sorted1, ARRAY_SIZE(sorted1));
616 
617     SendMessageW(hwnd_edit, EM_SETSEL, 0, -1);
618     SendMessageW(hwnd_edit, WM_CHAR, 'a', 1);
619     SendMessageW(hwnd_edit, WM_CHAR, 'b', 1);
620     dispatch_messages();
621     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
622     ok(lstrcmpW(str3, buffer) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str3), wine_dbgstr_w(buffer));
623     check_dropdown(acdropdown, hwnd_edit, sorted2, ARRAY_SIZE(sorted2));
624     SendMessageW(hwnd_edit, EM_SETSEL, 0, -1);
625     SendMessageW(hwnd_edit, WM_CHAR, 'a', 1);
626     SendMessageW(hwnd_edit, WM_CHAR, 'b', 1);
627     SendMessageW(hwnd_edit, WM_CHAR, 'c', 1);
628     dispatch_messages();
629     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
630     ok(lstrcmpW(str9 + 7, buffer) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str9 + 7), wine_dbgstr_w(buffer));
631 
632     SendMessageW(hwnd_edit, WM_SETTEXT, 0, (LPARAM)www);
633     SendMessageW(hwnd_edit, EM_SETSEL, ARRAY_SIZE(www) - 1, ARRAY_SIZE(www) - 1);
634     SendMessageW(hwnd_edit, WM_CHAR, 'a', 1);
635     dispatch_messages();
636     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
637     ok(lstrcmpW(str1 + 8, buffer) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str1 + 8), wine_dbgstr_w(buffer));
638     check_dropdown(acdropdown, hwnd_edit, sorted3, ARRAY_SIZE(sorted3));
639     SendMessageW(hwnd_edit, WM_SETTEXT, 0, (LPARAM)www);
640     SendMessageW(hwnd_edit, EM_SETSEL, ARRAY_SIZE(www) - 1, ARRAY_SIZE(www) - 1);
641     SendMessageW(hwnd_edit, WM_CHAR, 'w', 1);
642     dispatch_messages();
643     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
644     ok(lstrcmpW(str8, buffer) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str8), wine_dbgstr_w(buffer));
645 
646     SendMessageW(hwnd_edit, EM_SETSEL, 0, -1);
647     SendMessageW(hwnd_edit, WM_CHAR, 'h', 1);
648     dispatch_messages();
649     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
650     ok(lstrcmpW(str6, buffer) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str6), wine_dbgstr_w(buffer));
651     check_dropdown(acdropdown, hwnd_edit, sorted4, ARRAY_SIZE(sorted4));
652     SendMessageW(hwnd_edit, WM_CHAR, 't', 1);
653     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
654     ok(lstrcmpW(str5 + 8, buffer) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str5 + 8), wine_dbgstr_w(buffer));
655     check_dropdown(acdropdown, hwnd_edit, sorted5, ARRAY_SIZE(sorted5));
656     SendMessageW(hwnd_edit, WM_CHAR, 't', 1);
657     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
658     ok(lstrcmpW(htt, buffer) == 0, "Expected %s, got %s\n", wine_dbgstr_w(htt), wine_dbgstr_w(buffer));
659     check_dropdown(acdropdown, hwnd_edit, NULL, 0);
660     SendMessageW(hwnd_edit, WM_CHAR, 'p', 1);
661     SendMessageW(hwnd_edit, WM_CHAR, ':', 1);
662     SendMessageW(hwnd_edit, WM_CHAR, '/', 1);
663     SendMessageW(hwnd_edit, WM_CHAR, '/', 1);
664     SendMessageW(hwnd_edit, WM_CHAR, 'a', 1);
665     dispatch_messages();
666     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
667     ok(lstrcmpW(str4, buffer) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str4), wine_dbgstr_w(buffer));
668     check_dropdown(acdropdown, hwnd_edit, sorted6, ARRAY_SIZE(sorted6));
669     SendMessageW(hwnd_edit, EM_SETSEL, 0, 2);
670     SendMessageW(hwnd_edit, WM_CHAR, 'H', 1);
671     dispatch_messages();
672     check_dropdown(acdropdown, hwnd_edit, NULL, 0);
673     SendMessageW(hwnd_edit, WM_CHAR, 't', 1);
674     dispatch_messages();
675     check_dropdown(acdropdown, hwnd_edit, sorted6, ARRAY_SIZE(sorted6));
676 
677     SendMessageW(hwnd_edit, EM_SETSEL, 0, -1);
678     SendMessageW(hwnd_edit, WM_CHAR, 'F', 1);
679     dispatch_messages();
680     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
681     check_dropdown(acdropdown, hwnd_edit, sorted7, ARRAY_SIZE(sorted7));
682     SendMessageW(hwnd_edit, WM_CHAR, 'i', 1);
683     SendMessageW(hwnd_edit, WM_CHAR, 'L', 1);
684     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
685     check_dropdown(acdropdown, hwnd_edit, sorted7, 1);
686 
687     IAutoCompleteDropDown_Release(acdropdown);
688     IAutoComplete2_Release(autocomplete);
689     IUnknown_Release(enumerator);
690 }
691 
test_custom_source(void)692 static void test_custom_source(void)
693 {
694     static WCHAR str_alpha[] = {'t','e','s','t','1',0};
695     static WCHAR str_alpha2[] = {'t','e','s','t','2',0};
696     static WCHAR str_beta[] = {'a','u','t','o',' ','c','o','m','p','l','e','t','e',0};
697     static WCHAR str_au[] = {'a','u',0};
698     static WCHAR str_aut[] = {'a','u','t',0};
699     static WCHAR *suggestions[] = { str_alpha, str_alpha2, str_beta };
700     struct string_enumerator *obj;
701     IUnknown *enumerator;
702     IAutoComplete2 *autocomplete;
703     IAutoCompleteDropDown *acdropdown;
704     HWND hwnd_edit;
705     DWORD flags = 0;
706     WCHAR buffer[20];
707     HRESULT hr;
708 
709     ShowWindow(hMainWnd, SW_SHOW);
710 
711     hwnd_edit = CreateWindowA("Edit", "", WS_OVERLAPPED | WS_VISIBLE | WS_CHILD | WS_BORDER, 50, 5, 200, 20, hMainWnd, 0, NULL, 0);
712 
713     hr = CoCreateInstance(&CLSID_AutoComplete, NULL, CLSCTX_INPROC_SERVER, &IID_IAutoComplete2, (void**)&autocomplete);
714     ok(hr == S_OK, "CoCreateInstance failed: %x\n", hr);
715 
716     hr = IAutoComplete2_QueryInterface(autocomplete, &IID_IAutoCompleteDropDown, (LPVOID*)&acdropdown);
717     ok(hr == S_OK, "No IAutoCompleteDropDown interface: %x\n", hr);
718 
719     string_enumerator_create((void**)&enumerator, suggestions, ARRAY_SIZE(suggestions));
720     obj = (struct string_enumerator*)enumerator;
721 
722     hr = IAutoComplete2_SetOptions(autocomplete, ACO_AUTOSUGGEST | ACO_AUTOAPPEND);
723     ok(hr == S_OK, "IAutoComplete2_SetOptions failed: %x\n", hr);
724     hr = IAutoCompleteDropDown_ResetEnumerator(acdropdown);
725     ok(hr == S_OK, "IAutoCompleteDropDown_ResetEnumerator failed: %x\n", hr);
726     hr = IAutoComplete2_Init(autocomplete, hwnd_edit, enumerator, NULL, NULL);
727     ok(hr == S_OK, "IAutoComplete_Init failed: %x\n", hr);
728 
729     SetFocus(hwnd_edit);
730     SendMessageW(hwnd_edit, WM_CHAR, 'a', 1);
731     SendMessageW(hwnd_edit, WM_CHAR, 'u', 1);
732     dispatch_messages();
733     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
734     ok(lstrcmpW(str_beta, buffer) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str_beta), wine_dbgstr_w(buffer));
735     ok(obj->num_resets == 1, "Expected 1 reset, got %u\n", obj->num_resets);
736     SendMessageW(hwnd_edit, EM_SETSEL, 0, -1);
737     SendMessageW(hwnd_edit, WM_CHAR, '\b', 1);
738     dispatch_messages();
739     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
740     ok(buffer[0] == '\0', "Expected empty string, got %s\n", wine_dbgstr_w(buffer));
741     ok(obj->num_resets == 1, "Expected 1 reset, got %u\n", obj->num_resets);
742     hr = IAutoCompleteDropDown_ResetEnumerator(acdropdown);
743     ok(hr == S_OK, "IAutoCompleteDropDown_ResetEnumerator failed: %x\n", hr);
744     ok(obj->num_resets == 1, "Expected 1 reset, got %u\n", obj->num_resets);
745     obj->num_resets = 0;
746 
747     /* hijack the window procedure */
748     HijackerWndProc_prev = (WNDPROC)SetWindowLongPtrW(hwnd_edit, GWLP_WNDPROC, (LONG_PTR)HijackerWndProc);
749     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
750     ok(lstrcmpW(HijackerWndProc_txt, buffer) == 0, "Expected %s, got %s\n", wine_dbgstr_w(HijackerWndProc_txt), wine_dbgstr_w(buffer));
751 
752     SendMessageW(hwnd_edit, WM_CHAR, 'a', 1);
753     SendMessageW(hwnd_edit, WM_CHAR, 'u', 1);
754     SetWindowLongPtrW(hwnd_edit, GWLP_WNDPROC, (LONG_PTR)HijackerWndProc_prev);
755     dispatch_messages();
756     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
757     ok(lstrcmpW(str_au, buffer) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str_au), wine_dbgstr_w(buffer));
758     ok(obj->num_resets == 1, "Expected 1 reset, got %u\n", obj->num_resets);
759     SendMessageW(hwnd_edit, EM_SETSEL, 0, -1);
760     SendMessageW(hwnd_edit, WM_CHAR, '\b', 1);
761     dispatch_messages();
762     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
763     ok(buffer[0] == '\0', "Expected empty string, got %s\n", wine_dbgstr_w(buffer));
764     hr = IAutoCompleteDropDown_ResetEnumerator(acdropdown);
765     ok(hr == S_OK, "IAutoCompleteDropDown_ResetEnumerator failed: %x\n", hr);
766 
767     HijackerWndProc_prev = (WNDPROC)SetWindowLongPtrW(hwnd_edit, GWLP_WNDPROC, (LONG_PTR)HijackerWndProc2);
768     SendMessageW(hwnd_edit, WM_CHAR, 'a', 1);
769     SendMessageW(hwnd_edit, WM_CHAR, 'u', 1);
770     SetWindowLongPtrW(hwnd_edit, GWLP_WNDPROC, (LONG_PTR)HijackerWndProc_prev);
771     dispatch_messages();
772     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
773     ok(lstrcmpW(str_beta, buffer) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str_beta), wine_dbgstr_w(buffer));
774     ok(obj->num_resets == 2, "Expected 2 resets, got %u\n", obj->num_resets);
775     /* end of hijacks */
776 
777     hr = IAutoCompleteDropDown_GetDropDownStatus(acdropdown, &flags, NULL);
778     ok(hr == S_OK, "IAutoCompleteDropDown_GetDropDownStatus failed: %x\n", hr);
779     ok(flags & ACDD_VISIBLE, "AutoComplete DropDown should be visible\n");
780     SendMessageW(hwnd_edit, WM_SETTEXT, 0, (LPARAM)str_au);
781     dispatch_messages();
782     hr = IAutoCompleteDropDown_GetDropDownStatus(acdropdown, &flags, NULL);
783     ok(hr == S_OK, "IAutoCompleteDropDown_GetDropDownStatus failed: %x\n", hr);
784     ok(!(flags & ACDD_VISIBLE), "AutoComplete DropDown should have been hidden\n");
785     SendMessageW(hwnd_edit, WM_SETTEXT, 0, (LPARAM)str_aut);
786     dispatch_messages();
787     hr = IAutoCompleteDropDown_GetDropDownStatus(acdropdown, &flags, NULL);
788     ok(hr == S_OK, "IAutoCompleteDropDown_GetDropDownStatus failed: %x\n", hr);
789     ok(!(flags & ACDD_VISIBLE), "AutoComplete DropDown should be hidden\n");
790     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
791     ok(lstrcmpW(str_aut, buffer) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str_aut), wine_dbgstr_w(buffer));
792 
793     test_aclist_expand(hwnd_edit, enumerator, acdropdown);
794     obj->num_resets = 0;
795 
796     hr = IAutoCompleteDropDown_ResetEnumerator(acdropdown);
797     ok(hr == S_OK, "IAutoCompleteDropDown_ResetEnumerator failed: %x\n", hr);
798     SendMessageW(hwnd_edit, WM_CHAR, 'x', 1);
799     dispatch_messages();
800     ok(obj->num_resets == 1, "Expected 1 reset, got %u\n", obj->num_resets);
801     SendMessageW(hwnd_edit, WM_CHAR, 'x', 1);
802     dispatch_messages();
803     ok(obj->num_resets == 1, "Expected 1 reset, got %u\n", obj->num_resets);
804 
805     IAutoCompleteDropDown_Release(acdropdown);
806     IAutoComplete2_Release(autocomplete);
807     IUnknown_Release(enumerator);
808 
809     test_prefix_filtering(hwnd_edit);
810 
811     ShowWindow(hMainWnd, SW_HIDE);
812     DestroyWindow(hwnd_edit);
813 }
814 
START_TEST(autocomplete)815 START_TEST(autocomplete)
816 {
817     HRESULT r;
818     MSG msg;
819     IAutoComplete* ac;
820     RECT win_rect;
821     POINT orig_pos;
822 
823     r = CoInitialize(NULL);
824     ok(r == S_OK, "CoInitialize failed (0x%08x). Tests aborted.\n", r);
825     if (r != S_OK)
826         return;
827 
828     createMainWnd();
829     ok(hMainWnd != NULL, "Failed to create parent window. Tests aborted.\n");
830     if (!hMainWnd) return;
831 
832     /* Move the cursor away from the dropdown listbox */
833     GetWindowRect(hMainWnd, &win_rect);
834     GetCursorPos(&orig_pos);
835     SetCursorPos(win_rect.left, win_rect.top);
836 
837     test_invalid_init();
838     ac = test_init();
839     if (!ac)
840         goto cleanup;
841     test_killfocus();
842 
843     test_custom_source();
844 
845     PostQuitMessage(0);
846     while(GetMessageA(&msg,0,0,0)) {
847         TranslateMessage(&msg);
848         DispatchMessageA(&msg);
849     }
850 
851     IAutoComplete_Release(ac);
852 
853 cleanup:
854     SetCursorPos(orig_pos.x, orig_pos.y);
855     DestroyWindow(hEdit);
856     DestroyWindow(hMainWnd);
857 
858     CoUninitialize();
859 }
860