dlls/shell32/tests/autocomplete.c

   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
  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 }
 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
 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
 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
 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};
 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
 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
 282 static struct string_enumerator *impl_from_IEnumString(IEnumString *iface)
 283 {
 284     return CONTAINING_RECORD(iface, struct string_enumerator, IEnumString_iface);
 285 }
 286
 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
 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
 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
 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
 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
 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
 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
 384 static struct string_enumerator *impl_from_IACList(IACList *iface)
 385 {
 386     return CONTAINING_RECORD(iface, struct string_enumerator, IACList_iface);
 387 }
 388
 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
 394 static ULONG WINAPI aclist_AddRef(IACList *iface)
 395 {
 396     return string_enumerator_AddRef(&impl_from_IACList(iface)->IEnumString_iface);
 397 }
 398
 399 static ULONG WINAPI aclist_Release(IACList *iface)
 400 {
 401     return string_enumerator_Release(&impl_from_IACList(iface)->IEnumString_iface);
 402 }
 403
 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
 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
 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)
 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
 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
 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
 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
 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 }