modules/rostests/winetests/shell32/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)
 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     obj->num_resets = 0;
 506
 507     ok(obj->num_expand == 0, "Expected 0 expansions, got %u\n", obj->num_expand);
 508     SendMessageW(hwnd_edit, WM_SETTEXT, 0, (LPARAM)str1);
 509     SendMessageW(hwnd_edit, EM_SETSEL, ARRAY_SIZE(str1) - 1, ARRAY_SIZE(str1) - 1);
 510     SendMessageW(hwnd_edit, WM_CHAR, '\\', 1);
 511     dispatch_messages();
 512     ok(obj->num_expand == 1, "Expected 1 expansion, got %u\n", obj->num_expand);
 513     ok(lstrcmpW(obj->last_expand, str1a) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str1a), wine_dbgstr_w(obj->last_expand));
 514     ok(obj->num_resets == 1, "Expected 1 reset, got %u\n", obj->num_resets);
 515     SendMessageW(hwnd_edit, WM_SETTEXT, 0, (LPARAM)str2);
 516     SendMessageW(hwnd_edit, EM_SETSEL, ARRAY_SIZE(str2) - 1, ARRAY_SIZE(str2) - 1);
 517     SendMessageW(hwnd_edit, WM_CHAR, 'z', 1);
 518     dispatch_messages();
 519     ok(obj->num_expand == 2, "Expected 2 expansions, got %u\n", obj->num_expand);
 520     ok(lstrcmpW(obj->last_expand, str2a) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str2a), wine_dbgstr_w(obj->last_expand));
 521     ok(obj->num_resets == 2, "Expected 2 resets, got %u\n", obj->num_resets);
 522     SetFocus(hwnd_edit);
 523     SendMessageW(hwnd_edit, WM_CHAR, '_', 1);
 524     SendMessageW(hwnd_edit, WM_CHAR, 'b', 1);
 525     SetFocus(0);
 526     SetFocus(hwnd_edit);
 527     SendMessageW(hwnd_edit, WM_CHAR, 'b', 1);
 528     SendMessageW(hwnd_edit, WM_CHAR, 'q', 1);
 529     dispatch_messages();
 530     ok(obj->num_expand == 2, "Expected 2 expansions, got %u\n", obj->num_expand);
 531     ok(obj->num_resets == 2, "Expected 2 resets, got %u\n", obj->num_resets);
 532     SendMessageW(hwnd_edit, WM_CHAR, '\\', 1);
 533     dispatch_messages();
 534     ok(obj->num_expand == 3, "Expected 3 expansions, got %u\n", obj->num_expand);
 535     ok(lstrcmpW(obj->last_expand, str2b) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str2b), wine_dbgstr_w(obj->last_expand));
 536     ok(obj->num_resets == 3, "Expected 3 resets, got %u\n", obj->num_resets);
 537     SendMessageW(hwnd_edit, EM_SETSEL, ARRAY_SIZE(str1a) - 1, -1);
 538     SendMessageW(hwnd_edit, WM_CHAR, 'x', 1);
 539     SendMessageW(hwnd_edit, WM_CHAR, 'y', 1);
 540     dispatch_messages();
 541     ok(obj->num_expand == 4, "Expected 4 expansions, got %u\n", obj->num_expand);
 542     ok(lstrcmpW(obj->last_expand, str1a) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str1a), wine_dbgstr_w(obj->last_expand));
 543     ok(obj->num_resets == 4, "Expected 4 resets, got %u\n", obj->num_resets);
 544     SendMessageW(hwnd_edit, EM_SETSEL, ARRAY_SIZE(str1) - 1, -1);
 545     SendMessageW(hwnd_edit, WM_CHAR, 'x', 1);
 546     dispatch_messages();
 547     ok(obj->num_expand == 4, "Expected 4 expansions, got %u\n", obj->num_expand);
 548     ok(obj->num_resets == 5, "Expected 5 resets, got %u\n", obj->num_resets);
 549 }
 550
 551 static void test_prefix_filtering(HWND hwnd_edit)
 552 {
 553     static WCHAR htt[]  = {'h','t','t',0};
 554     static WCHAR www[]  = {'w','w','w','.',0};
 555     static WCHAR str0[] = {'w','w','w','.','a','x',0};
 556     static WCHAR str1[] = {'h','t','t','p','s',':','/','/','w','w','w','.','a','c',0};
 557     static WCHAR str2[] = {'a','a',0};
 558     static WCHAR str3[] = {'a','b',0};
 559     static WCHAR str4[] = {'h','t','t','p',':','/','/','a','0',0};
 560     static WCHAR str5[] = {'h','t','t','p','s',':','/','/','h','t','a',0};
 561     static WCHAR str6[] = {'h','f','o','o',0};
 562     static WCHAR str7[] = {'h','t','t','p',':','/','/','w','w','w','.','a','d','d',0};
 563     static WCHAR str8[] = {'w','w','w','.','w','w','w','.','?',0};
 564     static WCHAR str9[] = {'h','t','t','p',':','/','/','a','b','c','.','a','a','.','c','o','m',0};
 565     static WCHAR str10[]= {'f','t','p',':','/','/','a','b','c',0};
 566     static WCHAR str11[]= {'f','i','l','e',':','/','/','a','a',0};
 567     static WCHAR str12[]= {'f','t','p',':','/','/','w','w','w','.','a','a',0};
 568     static WCHAR *suggestions[] = { str0, str1, str2, str3, str4, str5, str6, str7, str8, str9, str10, str11, str12 };
 569     static WCHAR *sorted1[] = { str4, str2, str3, str9, str1, str7, str0 };
 570     static WCHAR *sorted2[] = { str3, str9 };
 571     static WCHAR *sorted3[] = { str1, str7, str0 };
 572     static WCHAR *sorted4[] = { str6, str5 };
 573     static WCHAR *sorted5[] = { str5 };
 574     static WCHAR *sorted6[] = { str4, str9 };
 575     static WCHAR *sorted7[] = { str11, str10, str12 };
 576     IUnknown *enumerator;
 577     IAutoComplete2 *autocomplete;
 578     IAutoCompleteDropDown *acdropdown;
 579     WCHAR buffer[20];
 580     HRESULT hr;
 581
 582     hr = CoCreateInstance(&CLSID_AutoComplete, NULL, CLSCTX_INPROC_SERVER, &IID_IAutoComplete2, (void**)&autocomplete);
 583     ok(hr == S_OK, "CoCreateInstance failed: %x\n", hr);
 584
 585     hr = IAutoComplete2_QueryInterface(autocomplete, &IID_IAutoCompleteDropDown, (LPVOID*)&acdropdown);
 586     ok(hr == S_OK, "No IAutoCompleteDropDown interface: %x\n", hr);
 587
 588     string_enumerator_create((void**)&enumerator, suggestions, ARRAY_SIZE(suggestions));
 589
 590     hr = IAutoComplete2_SetOptions(autocomplete, ACO_FILTERPREFIXES | ACO_AUTOSUGGEST | ACO_AUTOAPPEND);
 591     ok(hr == S_OK, "IAutoComplete2_SetOptions failed: %x\n", hr);
 592     hr = IAutoComplete2_Init(autocomplete, hwnd_edit, enumerator, NULL, NULL);
 593     ok(hr == S_OK, "IAutoComplete_Init failed: %x\n", hr);
 594
 595     SendMessageW(hwnd_edit, EM_SETSEL, 0, -1);
 596     SendMessageW(hwnd_edit, WM_CHAR, 'a', 1);
 597     dispatch_messages();
 598     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
 599     ok(lstrcmpW(str4 + 7, buffer) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str4 + 7), wine_dbgstr_w(buffer));
 600     check_dropdown(acdropdown, hwnd_edit, sorted1, ARRAY_SIZE(sorted1));
 601
 602     SendMessageW(hwnd_edit, EM_SETSEL, 0, -1);
 603     SendMessageW(hwnd_edit, WM_CHAR, 'a', 1);
 604     SendMessageW(hwnd_edit, WM_CHAR, 'b', 1);
 605     dispatch_messages();
 606     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
 607     ok(lstrcmpW(str3, buffer) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str3), wine_dbgstr_w(buffer));
 608     check_dropdown(acdropdown, hwnd_edit, sorted2, ARRAY_SIZE(sorted2));
 609     SendMessageW(hwnd_edit, EM_SETSEL, 0, -1);
 610     SendMessageW(hwnd_edit, WM_CHAR, 'a', 1);
 611     SendMessageW(hwnd_edit, WM_CHAR, 'b', 1);
 612     SendMessageW(hwnd_edit, WM_CHAR, 'c', 1);
 613     dispatch_messages();
 614     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
 615     ok(lstrcmpW(str9 + 7, buffer) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str9 + 7), wine_dbgstr_w(buffer));
 616
 617     SendMessageW(hwnd_edit, WM_SETTEXT, 0, (LPARAM)www);
 618     SendMessageW(hwnd_edit, EM_SETSEL, ARRAY_SIZE(www) - 1, ARRAY_SIZE(www) - 1);
 619     SendMessageW(hwnd_edit, WM_CHAR, 'a', 1);
 620     dispatch_messages();
 621     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
 622     ok(lstrcmpW(str1 + 8, buffer) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str1 + 8), wine_dbgstr_w(buffer));
 623     check_dropdown(acdropdown, hwnd_edit, sorted3, ARRAY_SIZE(sorted3));
 624     SendMessageW(hwnd_edit, WM_SETTEXT, 0, (LPARAM)www);
 625     SendMessageW(hwnd_edit, EM_SETSEL, ARRAY_SIZE(www) - 1, ARRAY_SIZE(www) - 1);
 626     SendMessageW(hwnd_edit, WM_CHAR, 'w', 1);
 627     dispatch_messages();
 628     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
 629     ok(lstrcmpW(str8, buffer) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str8), wine_dbgstr_w(buffer));
 630
 631     SendMessageW(hwnd_edit, EM_SETSEL, 0, -1);
 632     SendMessageW(hwnd_edit, WM_CHAR, 'h', 1);
 633     dispatch_messages();
 634     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
 635     ok(lstrcmpW(str6, buffer) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str6), wine_dbgstr_w(buffer));
 636     check_dropdown(acdropdown, hwnd_edit, sorted4, ARRAY_SIZE(sorted4));
 637     SendMessageW(hwnd_edit, WM_CHAR, 't', 1);
 638     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
 639     ok(lstrcmpW(str5 + 8, buffer) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str5 + 8), wine_dbgstr_w(buffer));
 640     check_dropdown(acdropdown, hwnd_edit, sorted5, ARRAY_SIZE(sorted5));
 641     SendMessageW(hwnd_edit, WM_CHAR, 't', 1);
 642     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
 643     ok(lstrcmpW(htt, buffer) == 0, "Expected %s, got %s\n", wine_dbgstr_w(htt), wine_dbgstr_w(buffer));
 644     check_dropdown(acdropdown, hwnd_edit, NULL, 0);
 645     SendMessageW(hwnd_edit, WM_CHAR, 'p', 1);
 646     SendMessageW(hwnd_edit, WM_CHAR, ':', 1);
 647     SendMessageW(hwnd_edit, WM_CHAR, '/', 1);
 648     SendMessageW(hwnd_edit, WM_CHAR, '/', 1);
 649     SendMessageW(hwnd_edit, WM_CHAR, 'a', 1);
 650     dispatch_messages();
 651     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
 652     ok(lstrcmpW(str4, buffer) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str4), wine_dbgstr_w(buffer));
 653     check_dropdown(acdropdown, hwnd_edit, sorted6, ARRAY_SIZE(sorted6));
 654     SendMessageW(hwnd_edit, EM_SETSEL, 0, 2);
 655     SendMessageW(hwnd_edit, WM_CHAR, 'H', 1);
 656     dispatch_messages();
 657     check_dropdown(acdropdown, hwnd_edit, NULL, 0);
 658     SendMessageW(hwnd_edit, WM_CHAR, 't', 1);
 659     dispatch_messages();
 660     check_dropdown(acdropdown, hwnd_edit, sorted6, ARRAY_SIZE(sorted6));
 661
 662     SendMessageW(hwnd_edit, EM_SETSEL, 0, -1);
 663     SendMessageW(hwnd_edit, WM_CHAR, 'F', 1);
 664     dispatch_messages();
 665     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
 666     check_dropdown(acdropdown, hwnd_edit, sorted7, ARRAY_SIZE(sorted7));
 667     SendMessageW(hwnd_edit, WM_CHAR, 'i', 1);
 668     SendMessageW(hwnd_edit, WM_CHAR, 'L', 1);
 669     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
 670     check_dropdown(acdropdown, hwnd_edit, sorted7, 1);
 671
 672     IAutoCompleteDropDown_Release(acdropdown);
 673     IAutoComplete2_Release(autocomplete);
 674     IUnknown_Release(enumerator);
 675 }
 676
 677 static void test_custom_source(void)
 678 {
 679     static WCHAR str_alpha[] = {'t','e','s','t','1',0};
 680     static WCHAR str_alpha2[] = {'t','e','s','t','2',0};
 681     static WCHAR str_beta[] = {'a','u','t','o',' ','c','o','m','p','l','e','t','e',0};
 682     static WCHAR str_au[] = {'a','u',0};
 683     static WCHAR str_aut[] = {'a','u','t',0};
 684     static WCHAR *suggestions[] = { str_alpha, str_alpha2, str_beta };
 685     struct string_enumerator *obj;
 686     IUnknown *enumerator;
 687     IAutoComplete2 *autocomplete;
 688     IAutoCompleteDropDown *acdropdown;
 689     HWND hwnd_edit;
 690     DWORD flags = 0;
 691     WCHAR buffer[20];
 692     HRESULT hr;
 693
 694     ShowWindow(hMainWnd, SW_SHOW);
 695
 696     hwnd_edit = CreateWindowA("Edit", "", WS_OVERLAPPED | WS_VISIBLE | WS_CHILD | WS_BORDER, 50, 5, 200, 20, hMainWnd, 0, NULL, 0);
 697
 698     hr = CoCreateInstance(&CLSID_AutoComplete, NULL, CLSCTX_INPROC_SERVER, &IID_IAutoComplete2, (void**)&autocomplete);
 699     ok(hr == S_OK, "CoCreateInstance failed: %x\n", hr);
 700
 701     hr = IAutoComplete2_QueryInterface(autocomplete, &IID_IAutoCompleteDropDown, (LPVOID*)&acdropdown);
 702     ok(hr == S_OK, "No IAutoCompleteDropDown interface: %x\n", hr);
 703
 704     string_enumerator_create((void**)&enumerator, suggestions, ARRAY_SIZE(suggestions));
 705     obj = (struct string_enumerator*)enumerator;
 706
 707     hr = IAutoComplete2_SetOptions(autocomplete, ACO_AUTOSUGGEST | ACO_AUTOAPPEND);
 708     ok(hr == S_OK, "IAutoComplete2_SetOptions failed: %x\n", hr);
 709     hr = IAutoCompleteDropDown_ResetEnumerator(acdropdown);
 710     ok(hr == S_OK, "IAutoCompleteDropDown_ResetEnumerator failed: %x\n", hr);
 711     hr = IAutoComplete2_Init(autocomplete, hwnd_edit, enumerator, NULL, NULL);
 712     ok(hr == S_OK, "IAutoComplete_Init failed: %x\n", hr);
 713
 714     SetFocus(hwnd_edit);
 715     SendMessageW(hwnd_edit, WM_CHAR, 'a', 1);
 716     SendMessageW(hwnd_edit, WM_CHAR, 'u', 1);
 717     dispatch_messages();
 718     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
 719     ok(lstrcmpW(str_beta, buffer) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str_beta), wine_dbgstr_w(buffer));
 720     ok(obj->num_resets == 1, "Expected 1 reset, got %u\n", obj->num_resets);
 721     SendMessageW(hwnd_edit, EM_SETSEL, 0, -1);
 722     SendMessageW(hwnd_edit, WM_CHAR, '\b', 1);
 723     dispatch_messages();
 724     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
 725     ok(buffer[0] == '\0', "Expected empty string, got %s\n", wine_dbgstr_w(buffer));
 726     ok(obj->num_resets == 1, "Expected 1 reset, got %u\n", obj->num_resets);
 727     hr = IAutoCompleteDropDown_ResetEnumerator(acdropdown);
 728     ok(hr == S_OK, "IAutoCompleteDropDown_ResetEnumerator failed: %x\n", hr);
 729     ok(obj->num_resets == 1, "Expected 1 reset, got %u\n", obj->num_resets);
 730     obj->num_resets = 0;
 731
 732     /* hijack the window procedure */
 733     HijackerWndProc_prev = (WNDPROC)SetWindowLongPtrW(hwnd_edit, GWLP_WNDPROC, (LONG_PTR)HijackerWndProc);
 734     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
 735     ok(lstrcmpW(HijackerWndProc_txt, buffer) == 0, "Expected %s, got %s\n", wine_dbgstr_w(HijackerWndProc_txt), wine_dbgstr_w(buffer));
 736
 737     SendMessageW(hwnd_edit, WM_CHAR, 'a', 1);
 738     SendMessageW(hwnd_edit, WM_CHAR, 'u', 1);
 739     SetWindowLongPtrW(hwnd_edit, GWLP_WNDPROC, (LONG_PTR)HijackerWndProc_prev);
 740     dispatch_messages();
 741     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
 742     ok(lstrcmpW(str_au, buffer) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str_au), wine_dbgstr_w(buffer));
 743     ok(obj->num_resets == 1, "Expected 1 reset, got %u\n", obj->num_resets);
 744     SendMessageW(hwnd_edit, EM_SETSEL, 0, -1);
 745     SendMessageW(hwnd_edit, WM_CHAR, '\b', 1);
 746     dispatch_messages();
 747     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
 748     ok(buffer[0] == '\0', "Expected empty string, got %s\n", wine_dbgstr_w(buffer));
 749     hr = IAutoCompleteDropDown_ResetEnumerator(acdropdown);
 750     ok(hr == S_OK, "IAutoCompleteDropDown_ResetEnumerator failed: %x\n", hr);
 751
 752     HijackerWndProc_prev = (WNDPROC)SetWindowLongPtrW(hwnd_edit, GWLP_WNDPROC, (LONG_PTR)HijackerWndProc2);
 753     SendMessageW(hwnd_edit, WM_CHAR, 'a', 1);
 754     SendMessageW(hwnd_edit, WM_CHAR, 'u', 1);
 755     SetWindowLongPtrW(hwnd_edit, GWLP_WNDPROC, (LONG_PTR)HijackerWndProc_prev);
 756     dispatch_messages();
 757     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
 758     ok(lstrcmpW(str_beta, buffer) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str_beta), wine_dbgstr_w(buffer));
 759     ok(obj->num_resets == 2, "Expected 2 resets, got %u\n", obj->num_resets);
 760     /* end of hijacks */
 761
 762     hr = IAutoCompleteDropDown_GetDropDownStatus(acdropdown, &flags, NULL);
 763     ok(hr == S_OK, "IAutoCompleteDropDown_GetDropDownStatus failed: %x\n", hr);
 764     ok(flags & ACDD_VISIBLE, "AutoComplete DropDown should be visible\n");
 765     SendMessageW(hwnd_edit, WM_SETTEXT, 0, (LPARAM)str_au);
 766     dispatch_messages();
 767     hr = IAutoCompleteDropDown_GetDropDownStatus(acdropdown, &flags, NULL);
 768     ok(hr == S_OK, "IAutoCompleteDropDown_GetDropDownStatus failed: %x\n", hr);
 769     ok(!(flags & ACDD_VISIBLE), "AutoComplete DropDown should have been hidden\n");
 770     SendMessageW(hwnd_edit, WM_SETTEXT, 0, (LPARAM)str_aut);
 771     dispatch_messages();
 772     hr = IAutoCompleteDropDown_GetDropDownStatus(acdropdown, &flags, NULL);
 773     ok(hr == S_OK, "IAutoCompleteDropDown_GetDropDownStatus failed: %x\n", hr);
 774     ok(!(flags & ACDD_VISIBLE), "AutoComplete DropDown should be hidden\n");
 775     SendMessageW(hwnd_edit, WM_GETTEXT, ARRAY_SIZE(buffer), (LPARAM)buffer);
 776     ok(lstrcmpW(str_aut, buffer) == 0, "Expected %s, got %s\n", wine_dbgstr_w(str_aut), wine_dbgstr_w(buffer));
 777
 778     test_aclist_expand(hwnd_edit, enumerator);
 779     obj->num_resets = 0;
 780
 781     hr = IAutoCompleteDropDown_ResetEnumerator(acdropdown);
 782     ok(hr == S_OK, "IAutoCompleteDropDown_ResetEnumerator failed: %x\n", hr);
 783     SendMessageW(hwnd_edit, WM_CHAR, 'x', 1);
 784     dispatch_messages();
 785     ok(obj->num_resets == 1, "Expected 1 reset, got %u\n", obj->num_resets);
 786     SendMessageW(hwnd_edit, WM_CHAR, 'x', 1);
 787     dispatch_messages();
 788     ok(obj->num_resets == 1, "Expected 1 reset, got %u\n", obj->num_resets);
 789
 790     IAutoCompleteDropDown_Release(acdropdown);
 791     IAutoComplete2_Release(autocomplete);
 792     IUnknown_Release(enumerator);
 793
 794     test_prefix_filtering(hwnd_edit);
 795
 796     ShowWindow(hMainWnd, SW_HIDE);
 797     DestroyWindow(hwnd_edit);
 798 }
 799
 800 START_TEST(autocomplete)
 801 {
 802     HRESULT r;
 803     MSG msg;
 804     IAutoComplete* ac;
 805     RECT win_rect;
 806     POINT orig_pos;
 807
 808     r = CoInitialize(NULL);
 809     ok(r == S_OK, "CoInitialize failed (0x%08x). Tests aborted.\n", r);
 810     if (r != S_OK)
 811         return;
 812
 813     createMainWnd();
 814     ok(hMainWnd != NULL, "Failed to create parent window. Tests aborted.\n");
 815     if (!hMainWnd) return;
 816
 817     /* Move the cursor away from the dropdown listbox */
 818     GetWindowRect(hMainWnd, &win_rect);
 819     GetCursorPos(&orig_pos);
 820     SetCursorPos(win_rect.left, win_rect.top);
 821
 822     test_invalid_init();
 823     ac = test_init();
 824     if (!ac)
 825         goto cleanup;
 826     test_killfocus();
 827
 828     test_custom_source();
 829
 830     PostQuitMessage(0);
 831     while(GetMessageA(&msg,0,0,0)) {
 832         TranslateMessage(&msg);
 833         DispatchMessageA(&msg);
 834     }
 835
 836     IAutoComplete_Release(ac);
 837
 838 cleanup:
 839     SetCursorPos(orig_pos.x, orig_pos.y);
 840     DestroyWindow(hEdit);
 841     DestroyWindow(hMainWnd);
 842
 843     CoUninitialize();
 844 }