git.oblomov.eu Git - wine/blob - dlls/msctf/tests/inputprocessor.c

   1 /*
   2  * Unit tests for ITfInputProcessor
   3  *
   4  * Copyright 2009 Aric Stewart, CodeWeavers
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include <stdio.h>
  22
  23 #define COBJMACROS
  24 #include "wine/test.h"
  25 #include "winuser.h"
  26 #include "initguid.h"
  27 #include "shlwapi.h"
  28 #include "shlguid.h"
  29 #include "comcat.h"
  30 #include "msctf.h"
  31 #include "olectl.h"
  32
  33 static ITfInputProcessorProfiles* g_ipp;
  34 static LANGID gLangid;
  35 static ITfCategoryMgr * g_cm = NULL;
  36 static ITfThreadMgr* g_tm = NULL;
  37 static ITfDocumentMgr *g_dm = NULL;
  38 static TfClientId cid = 0;
  39 static TfClientId tid = 0;
  40
  41 #define SINK_UNEXPECTED 0
  42 #define SINK_EXPECTED 1
  43 #define SINK_FIRED 2
  44
  45 static BOOL test_ShouldActivate = FALSE;
  46 static BOOL test_ShouldDeactivate = FALSE;
  47
  48 static DWORD tmSinkCookie;
  49 static DWORD tmSinkRefCount;
  50 static ITfDocumentMgr *test_CurrentFocus = NULL;
  51 static ITfDocumentMgr *test_PrevFocus = NULL;
  52 static INT  test_OnSetFocus = SINK_UNEXPECTED;
  53 static INT  test_OnInitDocumentMgr = SINK_UNEXPECTED;
  54 static INT  test_OnPushContext = SINK_UNEXPECTED;
  55 static INT  test_OnPopContext = SINK_UNEXPECTED;
  56 static INT  test_KEV_OnSetFocus = SINK_UNEXPECTED;
  57
  58 HRESULT RegisterTextService(REFCLSID rclsid);
  59 HRESULT UnregisterTextService();
  60 HRESULT ThreadMgrEventSink_Constructor(IUnknown **ppOut);
  61 HRESULT TextStoreACP_Constructor(IUnknown **ppOut);
  62
  63 DEFINE_GUID(CLSID_FakeService, 0xEDE1A7AD,0x66DE,0x47E0,0xB6,0x20,0x3E,0x92,0xF8,0x24,0x6B,0xF3);
  64 DEFINE_GUID(CLSID_TF_InputProcessorProfiles, 0x33c53a50,0xf456,0x4884,0xb0,0x49,0x85,0xfd,0x64,0x3e,0xcf,0xed);
  65 DEFINE_GUID(CLSID_TF_CategoryMgr,         0xA4B544A1,0x438D,0x4B41,0x93,0x25,0x86,0x95,0x23,0xE2,0xD6,0xC7);
  66 DEFINE_GUID(GUID_TFCAT_TIP_KEYBOARD,     0x34745c63,0xb2f0,0x4784,0x8b,0x67,0x5e,0x12,0xc8,0x70,0x1a,0x31);
  67 DEFINE_GUID(GUID_TFCAT_TIP_SPEECH,       0xB5A73CD1,0x8355,0x426B,0xA1,0x61,0x25,0x98,0x08,0xF2,0x6B,0x14);
  68 DEFINE_GUID(GUID_TFCAT_TIP_HANDWRITING,  0x246ecb87,0xc2f2,0x4abe,0x90,0x5b,0xc8,0xb3,0x8a,0xdd,0x2c,0x43);
  69 DEFINE_GUID (GUID_TFCAT_DISPLAYATTRIBUTEPROVIDER,  0x046B8C80,0x1647,0x40F7,0x9B,0x21,0xB9,0x3B,0x81,0xAA,0xBC,0x1B);
  70 DEFINE_GUID(GUID_NULL,0,0,0,0,0,0,0,0,0,0,0);
  71 DEFINE_GUID(CLSID_TF_ThreadMgr,           0x529a9e6b,0x6587,0x4f23,0xab,0x9e,0x9c,0x7d,0x68,0x3e,0x3c,0x50);
  72 DEFINE_GUID(CLSID_PreservedKey,           0xA0ED8E55,0xCD3B,0x4274,0xB2,0x95,0xF6,0xC9,0xBA,0x2B,0x84,0x72);
  73
  74
  75 static HRESULT initialize(void)
  76 {
  77     HRESULT hr;
  78     CoInitialize(NULL);
  79     hr = CoCreateInstance (&CLSID_TF_InputProcessorProfiles, NULL,
  80           CLSCTX_INPROC_SERVER, &IID_ITfInputProcessorProfiles, (void**)&g_ipp);
  81     if (SUCCEEDED(hr))
  82         hr = CoCreateInstance (&CLSID_TF_CategoryMgr, NULL,
  83           CLSCTX_INPROC_SERVER, &IID_ITfCategoryMgr, (void**)&g_cm);
  84     if (SUCCEEDED(hr))
  85         hr = CoCreateInstance (&CLSID_TF_ThreadMgr, NULL,
  86           CLSCTX_INPROC_SERVER, &IID_ITfThreadMgr, (void**)&g_tm);
  87     return hr;
  88 }
  89
  90 static void cleanup(void)
  91 {
  92     if (g_ipp)
  93         ITfInputProcessorProfiles_Release(g_ipp);
  94     if (g_cm)
  95         ITfCategoryMgr_Release(g_cm);
  96     if (g_tm)
  97         ITfThreadMgr_Release(g_tm);
  98     CoUninitialize();
  99 }
 100
 101 static void test_Register(void)
 102 {
 103     HRESULT hr;
 104
 105     static const WCHAR szDesc[] = {'F','a','k','e',' ','W','i','n','e',' ','S','e','r','v','i','c','e',0};
 106     static const WCHAR szFile[] = {'F','a','k','e',' ','W','i','n','e',' ','S','e','r','v','i','c','e',' ','F','i','l','e',0};
 107
 108     hr = ITfInputProcessorProfiles_GetCurrentLanguage(g_ipp,&gLangid);
 109     ok(SUCCEEDED(hr),"Unable to get current language id\n");
 110     trace("Current Language %x\n",gLangid);
 111
 112     hr = RegisterTextService(&CLSID_FakeService);
 113     ok(SUCCEEDED(hr),"Unable to register COM for TextService\n");
 114     hr = ITfInputProcessorProfiles_Register(g_ipp, &CLSID_FakeService);
 115     ok(SUCCEEDED(hr),"Unable to register text service(%x)\n",hr);
 116     hr = ITfInputProcessorProfiles_AddLanguageProfile(g_ipp, &CLSID_FakeService, gLangid, &CLSID_FakeService, szDesc, sizeof(szDesc)/sizeof(WCHAR), szFile, sizeof(szFile)/sizeof(WCHAR), 1);
 117     ok(SUCCEEDED(hr),"Unable to add Language Profile (%x)\n",hr);
 118 }
 119
 120 static void test_Unregister(void)
 121 {
 122     HRESULT hr;
 123     hr = ITfInputProcessorProfiles_Unregister(g_ipp, &CLSID_FakeService);
 124     ok(SUCCEEDED(hr),"Unable to unregister text service(%x)\n",hr);
 125     UnregisterTextService();
 126 }
 127
 128 static void test_EnumInputProcessorInfo(void)
 129 {
 130     IEnumGUID *ppEnum;
 131     BOOL found = FALSE;
 132
 133     if (SUCCEEDED(ITfInputProcessorProfiles_EnumInputProcessorInfo(g_ipp, &ppEnum)))
 134     {
 135         ULONG fetched;
 136         GUID g;
 137         while (IEnumGUID_Next(ppEnum, 1, &g, &fetched) == S_OK)
 138         {
 139             if(IsEqualGUID(&g,&CLSID_FakeService))
 140                 found = TRUE;
 141         }
 142     }
 143     ok(found,"Did not find registered text service\n");
 144 }
 145
 146 static void test_EnumLanguageProfiles(void)
 147 {
 148     BOOL found = FALSE;
 149     IEnumTfLanguageProfiles *ppEnum;
 150     if (SUCCEEDED(ITfInputProcessorProfiles_EnumLanguageProfiles(g_ipp,gLangid,&ppEnum)))
 151     {
 152         TF_LANGUAGEPROFILE profile;
 153         while (IEnumTfLanguageProfiles_Next(ppEnum,1,&profile,NULL)==S_OK)
 154         {
 155             if (IsEqualGUID(&profile.clsid,&CLSID_FakeService))
 156             {
 157                 found = TRUE;
 158                 ok(profile.langid == gLangid, "LangId Incorrect\n");
 159                 ok(IsEqualGUID(&profile.catid,&GUID_TFCAT_TIP_KEYBOARD), "CatId Incorrect\n");
 160                 ok(IsEqualGUID(&profile.guidProfile,&CLSID_FakeService), "guidProfile Incorrect\n");
 161             }
 162         }
 163     }
 164     ok(found,"Registered text service not found\n");
 165 }
 166
 167 static void test_RegisterCategory(void)
 168 {
 169     HRESULT hr;
 170     hr = ITfCategoryMgr_RegisterCategory(g_cm, &CLSID_FakeService, &GUID_TFCAT_TIP_KEYBOARD, &CLSID_FakeService);
 171     ok(SUCCEEDED(hr),"ITfCategoryMgr_RegisterCategory failed\n");
 172     hr = ITfCategoryMgr_RegisterCategory(g_cm, &CLSID_FakeService, &GUID_TFCAT_DISPLAYATTRIBUTEPROVIDER, &CLSID_FakeService);
 173     ok(SUCCEEDED(hr),"ITfCategoryMgr_RegisterCategory failed\n");
 174 }
 175
 176 static void test_UnregisterCategory(void)
 177 {
 178     HRESULT hr;
 179     hr = ITfCategoryMgr_UnregisterCategory(g_cm, &CLSID_FakeService, &GUID_TFCAT_TIP_KEYBOARD, &CLSID_FakeService);
 180     ok(SUCCEEDED(hr),"ITfCategoryMgr_UnregisterCategory failed\n");
 181     hr = ITfCategoryMgr_UnregisterCategory(g_cm, &CLSID_FakeService, &GUID_TFCAT_DISPLAYATTRIBUTEPROVIDER, &CLSID_FakeService);
 182     ok(SUCCEEDED(hr),"ITfCategoryMgr_UnregisterCategory failed\n");
 183 }
 184
 185 static void test_FindClosestCategory(void)
 186 {
 187     GUID output;
 188     HRESULT hr;
 189     const GUID *list[3] = {&GUID_TFCAT_TIP_SPEECH, &GUID_TFCAT_TIP_KEYBOARD, &GUID_TFCAT_TIP_HANDWRITING};
 190
 191     hr = ITfCategoryMgr_FindClosestCategory(g_cm, &CLSID_FakeService, &output, NULL, 0);
 192     ok(SUCCEEDED(hr),"ITfCategoryMgr_FindClosestCategory failed (%x)\n",hr);
 193     ok(IsEqualGUID(&output,&GUID_TFCAT_DISPLAYATTRIBUTEPROVIDER),"Wrong GUID\n");
 194
 195     hr = ITfCategoryMgr_FindClosestCategory(g_cm, &CLSID_FakeService, &output, list, 1);
 196     ok(SUCCEEDED(hr),"ITfCategoryMgr_FindClosestCategory failed (%x)\n",hr);
 197     ok(IsEqualGUID(&output,&GUID_NULL),"Wrong GUID\n");
 198
 199     hr = ITfCategoryMgr_FindClosestCategory(g_cm, &CLSID_FakeService, &output, list, 3);
 200     ok(SUCCEEDED(hr),"ITfCategoryMgr_FindClosestCategory failed (%x)\n",hr);
 201     ok(IsEqualGUID(&output,&GUID_TFCAT_TIP_KEYBOARD),"Wrong GUID\n");
 202 }
 203
 204 static void test_Enable(void)
 205 {
 206     HRESULT hr;
 207     BOOL enabled = FALSE;
 208
 209     hr = ITfInputProcessorProfiles_EnableLanguageProfile(g_ipp,&CLSID_FakeService, gLangid, &CLSID_FakeService, TRUE);
 210     ok(SUCCEEDED(hr),"Failed to enable text service\n");
 211     hr = ITfInputProcessorProfiles_IsEnabledLanguageProfile(g_ipp,&CLSID_FakeService, gLangid, &CLSID_FakeService, &enabled);
 212     ok(SUCCEEDED(hr),"Failed to get enabled state\n");
 213     ok(enabled == TRUE,"enabled state incorrect\n");
 214 }
 215
 216 static void test_Disable(void)
 217 {
 218     HRESULT hr;
 219
 220     trace("Disabling\n");
 221     hr = ITfInputProcessorProfiles_EnableLanguageProfile(g_ipp,&CLSID_FakeService, gLangid, &CLSID_FakeService, FALSE);
 222     ok(SUCCEEDED(hr),"Failed to disable text service\n");
 223 }
 224
 225 static void test_ThreadMgrAdviseSinks(void)
 226 {
 227     ITfSource *source = NULL;
 228     HRESULT hr;
 229     IUnknown *sink;
 230
 231     hr = ITfThreadMgr_QueryInterface(g_tm, &IID_ITfSource, (LPVOID*)&source);
 232     ok(SUCCEEDED(hr),"Failed to get IID_ITfSource for ThreadMgr\n");
 233     if (!source)
 234         return;
 235
 236     ThreadMgrEventSink_Constructor(&sink);
 237
 238     tmSinkRefCount = 1;
 239     tmSinkCookie = 0;
 240     hr = ITfSource_AdviseSink(source,&IID_ITfThreadMgrEventSink, sink, &tmSinkCookie);
 241     ok(SUCCEEDED(hr),"Failed to Advise Sink\n");
 242     ok(tmSinkCookie!=0,"Failed to get sink cookie\n");
 243
 244     /* Advising the sink adds a ref, Relesing here lets the object be deleted
 245        when unadvised */
 246     tmSinkRefCount = 2;
 247     IUnknown_Release(sink);
 248     ITfSource_Release(source);
 249 }
 250
 251 static void test_ThreadMgrUnadviseSinks(void)
 252 {
 253     ITfSource *source = NULL;
 254     HRESULT hr;
 255
 256     hr = ITfThreadMgr_QueryInterface(g_tm, &IID_ITfSource, (LPVOID*)&source);
 257     ok(SUCCEEDED(hr),"Failed to get IID_ITfSource for ThreadMgr\n");
 258     if (!source)
 259         return;
 260
 261     tmSinkRefCount = 1;
 262     hr = ITfSource_UnadviseSink(source, tmSinkCookie);
 263     ok(SUCCEEDED(hr),"Failed to unadvise Sink\n");
 264     ITfSource_Release(source);
 265 }
 266
 267 /**********************************************************************
 268  * ITfKeyEventSink
 269  **********************************************************************/
 270 typedef struct tagKeyEventSink
 271 {
 272     const ITfKeyEventSinkVtbl *KeyEventSinkVtbl;
 273     LONG refCount;
 274 } KeyEventSink;
 275
 276 static void KeyEventSink_Destructor(KeyEventSink *This)
 277 {
 278     HeapFree(GetProcessHeap(),0,This);
 279 }
 280
 281 static HRESULT WINAPI KeyEventSink_QueryInterface(ITfKeyEventSink *iface, REFIID iid, LPVOID *ppvOut)
 282 {
 283     KeyEventSink *This = (KeyEventSink *)iface;
 284     *ppvOut = NULL;
 285
 286     if (IsEqualIID(iid, &IID_IUnknown) || IsEqualIID(iid, &IID_ITfKeyEventSink))
 287     {
 288         *ppvOut = This;
 289     }
 290
 291     if (*ppvOut)
 292     {
 293         IUnknown_AddRef(iface);
 294         return S_OK;
 295     }
 296
 297     return E_NOINTERFACE;
 298 }
 299
 300 static ULONG WINAPI KeyEventSink_AddRef(ITfKeyEventSink *iface)
 301 {
 302     KeyEventSink *This = (KeyEventSink *)iface;
 303     return InterlockedIncrement(&This->refCount);
 304 }
 305
 306 static ULONG WINAPI KeyEventSink_Release(ITfKeyEventSink *iface)
 307 {
 308     KeyEventSink *This = (KeyEventSink *)iface;
 309     ULONG ret;
 310
 311     ret = InterlockedDecrement(&This->refCount);
 312     if (ret == 0)
 313         KeyEventSink_Destructor(This);
 314     return ret;
 315 }
 316
 317 static HRESULT WINAPI KeyEventSink_OnSetFocus(ITfKeyEventSink *iface,
 318         BOOL fForeground)
 319 {
 320     ok(test_KEV_OnSetFocus == SINK_EXPECTED,"Unexpected KeyEventSink_OnSetFocus\n");
 321     test_KEV_OnSetFocus = SINK_FIRED;
 322     return S_OK;
 323 }
 324
 325 static HRESULT WINAPI KeyEventSink_OnTestKeyDown(ITfKeyEventSink *iface,
 326         ITfContext *pic, WPARAM wParam, LPARAM lParam, BOOL *pfEaten)
 327 {
 328     trace("\n");
 329     return S_OK;
 330 }
 331
 332 static HRESULT WINAPI KeyEventSink_OnTestKeyUp(ITfKeyEventSink *iface,
 333         ITfContext *pic, WPARAM wParam, LPARAM lParam, BOOL *pfEaten)
 334 {
 335     trace("\n");
 336     return S_OK;
 337 }
 338
 339 static HRESULT WINAPI KeyEventSink_OnKeyDown(ITfKeyEventSink *iface,
 340         ITfContext *pic, WPARAM wParam, LPARAM lParam, BOOL *pfEaten)
 341 {
 342     trace("\n");
 343     return S_OK;
 344 }
 345
 346 static HRESULT WINAPI KeyEventSink_OnKeyUp(ITfKeyEventSink *iface,
 347         ITfContext *pic, WPARAM wParam, LPARAM lParam, BOOL *pfEaten)
 348 {
 349     trace("\n");
 350     return S_OK;
 351 }
 352
 353 static HRESULT WINAPI KeyEventSink_OnPreservedKey(ITfKeyEventSink *iface,
 354     ITfContext *pic, REFGUID rguid, BOOL *pfEaten)
 355 {
 356     trace("\n");
 357     return S_OK;
 358 }
 359
 360 static const ITfKeyEventSinkVtbl KeyEventSink_KeyEventSinkVtbl =
 361 {
 362     KeyEventSink_QueryInterface,
 363     KeyEventSink_AddRef,
 364     KeyEventSink_Release,
 365
 366     KeyEventSink_OnSetFocus,
 367     KeyEventSink_OnTestKeyDown,
 368     KeyEventSink_OnTestKeyUp,
 369     KeyEventSink_OnKeyDown,
 370     KeyEventSink_OnKeyUp,
 371     KeyEventSink_OnPreservedKey
 372 };
 373
 374 HRESULT KeyEventSink_Constructor(ITfKeyEventSink **ppOut)
 375 {
 376     KeyEventSink *This;
 377
 378     This = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(KeyEventSink));
 379     if (This == NULL)
 380         return E_OUTOFMEMORY;
 381
 382     This->KeyEventSinkVtbl = &KeyEventSink_KeyEventSinkVtbl;
 383     This->refCount = 1;
 384
 385     *ppOut = (ITfKeyEventSink*)This;
 386     return S_OK;
 387 }
 388
 389
 390 static void test_KeystrokeMgr(void)
 391 {
 392     ITfKeystrokeMgr *keymgr= NULL;
 393     HRESULT hr;
 394     TF_PRESERVEDKEY tfpk;
 395     BOOL preserved;
 396     ITfKeyEventSink *sink;
 397
 398     KeyEventSink_Constructor(&sink);
 399
 400     hr = ITfThreadMgr_QueryInterface(g_tm, &IID_ITfKeystrokeMgr, (LPVOID*)&keymgr);
 401     ok(SUCCEEDED(hr),"Failed to get IID_ITfKeystrokeMgr for ThreadMgr\n");
 402
 403     tfpk.uVKey = 'A';
 404     tfpk.uModifiers = TF_MOD_SHIFT;
 405
 406     test_KEV_OnSetFocus = SINK_EXPECTED;
 407     hr = ITfKeystrokeMgr_AdviseKeyEventSink(keymgr,tid,sink,TRUE);
 408     todo_wine ok(SUCCEEDED(hr),"ITfKeystrokeMgr_AdviseKeyEventSink failed\n");
 409     todo_wine ok(test_KEV_OnSetFocus == SINK_FIRED, "KeyEventSink_OnSetFocus not fired as expected\n");
 410
 411     hr =ITfKeystrokeMgr_PreserveKey(keymgr, 0, &CLSID_PreservedKey, &tfpk, NULL, 0);
 412     todo_wine ok(hr==E_INVALIDARG,"ITfKeystrokeMgr_PreserveKey inproperly succeeded\n");
 413
 414     hr =ITfKeystrokeMgr_PreserveKey(keymgr, tid, &CLSID_PreservedKey, &tfpk, NULL, 0);
 415     todo_wine ok(SUCCEEDED(hr),"ITfKeystrokeMgr_PreserveKey failed\n");
 416
 417     hr =ITfKeystrokeMgr_PreserveKey(keymgr, tid, &CLSID_PreservedKey, &tfpk, NULL, 0);
 418     todo_wine ok(hr == TF_E_ALREADY_EXISTS,"ITfKeystrokeMgr_PreserveKey inproperly succeeded\n");
 419
 420     preserved = FALSE;
 421     hr = ITfKeystrokeMgr_IsPreservedKey(keymgr, &CLSID_PreservedKey, &tfpk, &preserved);
 422     todo_wine ok(hr == S_OK, "ITfKeystrokeMgr_IsPreservedKey failed\n");
 423     if (hr == S_OK) todo_wine ok(preserved == TRUE,"misreporting preserved key\n");
 424
 425     hr = ITfKeystrokeMgr_UnpreserveKey(keymgr, &CLSID_PreservedKey,&tfpk);
 426     todo_wine ok(SUCCEEDED(hr),"ITfKeystrokeMgr_UnpreserveKey failed\n");
 427
 428     hr = ITfKeystrokeMgr_IsPreservedKey(keymgr, &CLSID_PreservedKey, &tfpk, &preserved);
 429     todo_wine ok(hr == S_FALSE, "ITfKeystrokeMgr_IsPreservedKey failed\n");
 430     if (hr == S_FALSE) todo_wine ok(preserved == FALSE,"misreporting preserved key\n");
 431
 432     hr = ITfKeystrokeMgr_UnpreserveKey(keymgr, &CLSID_PreservedKey,&tfpk);
 433     todo_wine ok(hr==CONNECT_E_NOCONNECTION,"ITfKeystrokeMgr_UnpreserveKey inproperly succeeded\n");
 434
 435     hr = ITfKeystrokeMgr_UnadviseKeyEventSink(keymgr,tid);
 436     todo_wine ok(SUCCEEDED(hr),"ITfKeystrokeMgr_UnadviseKeyEventSink failed\n");
 437
 438     ITfKeystrokeMgr_Release(keymgr);
 439     ITfKeyEventSink_Release(sink);
 440 }
 441
 442 static void test_Activate(void)
 443 {
 444     HRESULT hr;
 445
 446     hr = ITfInputProcessorProfiles_ActivateLanguageProfile(g_ipp,&CLSID_FakeService,gLangid,&CLSID_FakeService);
 447     todo_wine ok(SUCCEEDED(hr),"Failed to Activate text service\n");
 448 }
 449
 450 static inline int check_context_refcount(ITfContext *iface)
 451 {
 452     IUnknown_AddRef(iface);
 453     return IUnknown_Release(iface);
 454 }
 455
 456 static void test_startSession(void)
 457 {
 458     HRESULT hr;
 459     DWORD cnt;
 460     DWORD editCookie;
 461     ITfDocumentMgr *dmtest;
 462     ITfContext *cxt,*cxt2,*cxt3,*cxtTest;
 463     ITextStoreACP *ts;
 464
 465     test_ShouldActivate = TRUE;
 466     ITfThreadMgr_Activate(g_tm,&cid);
 467     todo_wine ok(cid != tid,"TextService id mistakenly matches Client id\n");
 468
 469     hr = ITfThreadMgr_CreateDocumentMgr(g_tm,&g_dm);
 470     ok(SUCCEEDED(hr),"CreateDocumentMgr failed\n");
 471
 472     hr = ITfThreadMgr_GetFocus(g_tm,&dmtest);
 473     ok(SUCCEEDED(hr),"GetFocus Failed\n");
 474     ok(dmtest == NULL,"Initial focus not null\n");
 475
 476     test_CurrentFocus = g_dm;
 477     test_PrevFocus = NULL;
 478     test_OnSetFocus  = SINK_EXPECTED;
 479     hr = ITfThreadMgr_SetFocus(g_tm,g_dm);
 480     ok(SUCCEEDED(hr),"SetFocus Failed\n");
 481     ok(test_OnSetFocus == SINK_FIRED, "OnSetFocus sink not called\n");
 482     test_OnSetFocus  = SINK_UNEXPECTED;
 483
 484     hr = ITfThreadMgr_GetFocus(g_tm,&dmtest);
 485     ok(SUCCEEDED(hr),"GetFocus Failed\n");
 486     ok(g_dm == dmtest,"Expected DocumentMgr not focused\n");
 487
 488     cnt = ITfDocumentMgr_Release(g_dm);
 489     ok(cnt == 2,"DocumentMgr refcount not expected (2 vs %i)\n",cnt);
 490
 491     hr = ITfThreadMgr_GetFocus(g_tm,&dmtest);
 492     ok(SUCCEEDED(hr),"GetFocus Failed\n");
 493     ok(g_dm == dmtest,"Expected DocumentMgr not focused\n");
 494
 495     TextStoreACP_Constructor((IUnknown**)&ts);
 496
 497     hr = ITfDocumentMgr_CreateContext(g_dm, cid, 0, (IUnknown*)ts, &cxt, &editCookie);
 498     ok(SUCCEEDED(hr),"CreateContext Failed\n");
 499
 500     hr = ITfDocumentMgr_CreateContext(g_dm, cid, 0, NULL, &cxt2, &editCookie);
 501     ok(SUCCEEDED(hr),"CreateContext Failed\n");
 502
 503     hr = ITfDocumentMgr_CreateContext(g_dm, cid, 0, NULL, &cxt3, &editCookie);
 504     ok(SUCCEEDED(hr),"CreateContext Failed\n");
 505
 506     cnt = check_context_refcount(cxt);
 507     test_OnPushContext = SINK_EXPECTED;
 508     test_OnInitDocumentMgr = SINK_EXPECTED;
 509     hr = ITfDocumentMgr_Push(g_dm, cxt);
 510     ok(SUCCEEDED(hr),"Push Failed\n");
 511     ok(check_context_refcount(cxt) > cnt, "Ref count did not increase\n");
 512     ok(test_OnPushContext == SINK_FIRED, "OnPushContext sink not fired\n");
 513     ok(test_OnInitDocumentMgr == SINK_FIRED, "OnInitDocumentMgr sink not fired\n");
 514
 515     hr = ITfDocumentMgr_GetTop(g_dm, &cxtTest);
 516     ok(SUCCEEDED(hr),"GetTop Failed\n");
 517     ok(cxtTest == cxt, "Wrong context on top\n");
 518     ok(check_context_refcount(cxt) > cnt, "Ref count did not increase\n");
 519     cnt = ITfContext_Release(cxtTest);
 520
 521     hr = ITfDocumentMgr_GetBase(g_dm, &cxtTest);
 522     ok(SUCCEEDED(hr),"GetBase Failed\n");
 523     ok(cxtTest == cxt, "Wrong context on Base\n");
 524     ok(check_context_refcount(cxt) > cnt, "Ref count did not increase\n");
 525     ITfContext_Release(cxtTest);
 526
 527     check_context_refcount(cxt2);
 528     test_OnPushContext = SINK_EXPECTED;
 529     hr = ITfDocumentMgr_Push(g_dm, cxt2);
 530     ok(SUCCEEDED(hr),"Push Failed\n");
 531     ok(test_OnPushContext == SINK_FIRED, "OnPushContext sink not fired\n");
 532
 533     cnt = check_context_refcount(cxt2);
 534     hr = ITfDocumentMgr_GetTop(g_dm, &cxtTest);
 535     ok(SUCCEEDED(hr),"GetTop Failed\n");
 536     ok(cxtTest == cxt2, "Wrong context on top\n");
 537     ok(check_context_refcount(cxt2) > cnt, "Ref count did not increase\n");
 538     ITfContext_Release(cxtTest);
 539
 540     cnt = check_context_refcount(cxt);
 541     hr = ITfDocumentMgr_GetBase(g_dm, &cxtTest);
 542     ok(SUCCEEDED(hr),"GetBase Failed\n");
 543     ok(cxtTest == cxt, "Wrong context on Base\n");
 544     ok(check_context_refcount(cxt) > cnt, "Ref count did not increase\n");
 545     ITfContext_Release(cxtTest);
 546
 547     cnt = check_context_refcount(cxt3);
 548     hr = ITfDocumentMgr_Push(g_dm, cxt3);
 549     ok(!SUCCEEDED(hr),"Push Succeeded\n");
 550     ok(check_context_refcount(cxt3) == cnt, "Ref changed\n");
 551
 552     cnt = check_context_refcount(cxt2);
 553     hr = ITfDocumentMgr_GetTop(g_dm, &cxtTest);
 554     ok(SUCCEEDED(hr),"GetTop Failed\n");
 555     ok(cxtTest == cxt2, "Wrong context on top\n");
 556     ok(check_context_refcount(cxt2) > cnt, "Ref count did not increase\n");
 557     ITfContext_Release(cxtTest);
 558
 559     cnt = check_context_refcount(cxt);
 560     hr = ITfDocumentMgr_GetBase(g_dm, &cxtTest);
 561     ok(SUCCEEDED(hr),"GetBase Failed\n");
 562     ok(cxtTest == cxt, "Wrong context on Base\n");
 563     ok(check_context_refcount(cxt) > cnt, "Ref count did not increase\n");
 564     ITfContext_Release(cxtTest);
 565
 566     cnt = check_context_refcount(cxt2);
 567     test_OnPopContext = SINK_EXPECTED;
 568     hr = ITfDocumentMgr_Pop(g_dm, 0);
 569     ok(SUCCEEDED(hr),"Pop Failed\n");
 570     ok(check_context_refcount(cxt2) < cnt, "Ref count did not decrease\n");
 571     ok(test_OnPopContext == SINK_FIRED, "OnPopContext sink not fired\n");
 572
 573     hr = ITfDocumentMgr_GetTop(g_dm, &cxtTest);
 574     ok(SUCCEEDED(hr),"GetTop Failed\n");
 575     ok(cxtTest == cxt, "Wrong context on top\n");
 576     ITfContext_Release(cxtTest);
 577
 578     hr = ITfDocumentMgr_GetBase(g_dm, &cxtTest);
 579     ok(SUCCEEDED(hr),"GetBase Failed\n");
 580     ok(cxtTest == cxt, "Wrong context on base\n");
 581     ITfContext_Release(cxtTest);
 582
 583     hr = ITfDocumentMgr_Pop(g_dm, 0);
 584     ok(!SUCCEEDED(hr),"Pop Succeeded\n");
 585
 586     hr = ITfDocumentMgr_GetTop(g_dm, &cxtTest);
 587     ok(SUCCEEDED(hr),"GetTop Failed\n");
 588     ok(cxtTest == cxt, "Wrong context on top\n");
 589     ITfContext_Release(cxtTest);
 590
 591     hr = ITfDocumentMgr_GetBase(g_dm, &cxtTest);
 592     ok(SUCCEEDED(hr),"GetBase Failed\n");
 593     ok(cxtTest == cxt, "Wrong context on base\n");
 594     ITfContext_Release(cxtTest);
 595
 596     ITfContext_Release(cxt);
 597     ITfContext_Release(cxt2);
 598     ITfContext_Release(cxt3);
 599 }
 600
 601 static void test_endSession(void)
 602 {
 603     test_ShouldDeactivate = TRUE;
 604     test_CurrentFocus = NULL;
 605     test_PrevFocus = g_dm;
 606     test_OnSetFocus  = SINK_EXPECTED;
 607     ITfThreadMgr_Deactivate(g_tm);
 608     ok(test_OnSetFocus == SINK_FIRED, "OnSetFocus sink not called\n");
 609     test_OnSetFocus  = SINK_UNEXPECTED;
 610 }
 611
 612 static void test_TfGuidAtom(void)
 613 {
 614     GUID gtest,g1;
 615     HRESULT hr;
 616     TfGuidAtom atom1,atom2;
 617     BOOL equal;
 618
 619     CoCreateGuid(&gtest);
 620
 621     /* msdn reports this should return E_INVALIDARG.  However my test show it crashing (winxp)*/
 622     /*
 623     hr = ITfCategoryMgr_RegisterGUID(g_cm,&gtest,NULL);
 624     ok(hr==E_INVALIDARG,"ITfCategoryMgr_RegisterGUID should have failed\n");
 625     */
 626     hr = ITfCategoryMgr_RegisterGUID(g_cm,&gtest,&atom1);
 627     ok(SUCCEEDED(hr),"ITfCategoryMgr_RegisterGUID failed\n");
 628     hr = ITfCategoryMgr_RegisterGUID(g_cm,&gtest,&atom2);
 629     ok(SUCCEEDED(hr),"ITfCategoryMgr_RegisterGUID failed\n");
 630     ok(atom1 == atom2,"atoms do not match\n");
 631     hr = ITfCategoryMgr_GetGUID(g_cm,atom2,NULL);
 632     ok(hr==E_INVALIDARG,"ITfCategoryMgr_GetGUID should have failed\n");
 633     hr = ITfCategoryMgr_GetGUID(g_cm,atom2,&g1);
 634     ok(SUCCEEDED(hr),"ITfCategoryMgr_GetGUID failed\n");
 635     ok(IsEqualGUID(&g1,&gtest),"guids do not match\n");
 636     hr = ITfCategoryMgr_IsEqualTfGuidAtom(g_cm,atom1,&gtest,NULL);
 637     ok(hr==E_INVALIDARG,"ITfCategoryMgr_IsEqualTfGuidAtom should have failed\n");
 638     hr = ITfCategoryMgr_IsEqualTfGuidAtom(g_cm,atom1,&gtest,&equal);
 639     ok(SUCCEEDED(hr),"ITfCategoryMgr_IsEqualTfGuidAtom failed\n");
 640     ok(equal == TRUE,"Equal value invalid\n");
 641
 642     /* show that cid and tid TfClientIds are also TfGuidAtoms */
 643     hr = ITfCategoryMgr_IsEqualTfGuidAtom(g_cm,tid,&CLSID_FakeService,&equal);
 644     ok(SUCCEEDED(hr),"ITfCategoryMgr_IsEqualTfGuidAtom failed\n");
 645     todo_wine ok(equal == TRUE,"Equal value invalid\n");
 646     hr = ITfCategoryMgr_GetGUID(g_cm,cid,&g1);
 647     ok(SUCCEEDED(hr),"ITfCategoryMgr_GetGUID failed\n");
 648     todo_wine ok(!IsEqualGUID(&g1,&GUID_NULL),"guid should not be NULL\n");
 649 }
 650
 651 START_TEST(inputprocessor)
 652 {
 653     if (SUCCEEDED(initialize()))
 654     {
 655         test_Register();
 656         test_RegisterCategory();
 657         test_EnumInputProcessorInfo();
 658         test_Enable();
 659         test_ThreadMgrAdviseSinks();
 660         test_Activate();
 661         test_startSession();
 662         test_TfGuidAtom();
 663         test_KeystrokeMgr();
 664         test_endSession();
 665         test_EnumLanguageProfiles();
 666         test_FindClosestCategory();
 667         test_Disable();
 668         test_ThreadMgrUnadviseSinks();
 669         test_UnregisterCategory();
 670         test_Unregister();
 671     }
 672     else
 673         skip("Unable to create InputProcessor\n");
 674     cleanup();
 675 }
 676
 677 /**********************************************************************
 678  * ITextStoreACP
 679  **********************************************************************/
 680 typedef struct tagTextStoreACP
 681 {
 682     const ITextStoreACPVtbl *TextStoreACPVtbl;
 683     LONG refCount;
 684 } TextStoreACP;
 685
 686 static void TextStoreACP_Destructor(TextStoreACP *This)
 687 {
 688     HeapFree(GetProcessHeap(),0,This);
 689 }
 690
 691 static HRESULT WINAPI TextStoreACP_QueryInterface(ITextStoreACP *iface, REFIID iid, LPVOID *ppvOut)
 692 {
 693     TextStoreACP *This = (TextStoreACP *)iface;
 694     *ppvOut = NULL;
 695
 696     if (IsEqualIID(iid, &IID_IUnknown) || IsEqualIID(iid, &IID_ITextStoreACP))
 697     {
 698         *ppvOut = This;
 699     }
 700
 701     if (*ppvOut)
 702     {
 703         IUnknown_AddRef(iface);
 704         return S_OK;
 705     }
 706
 707     return E_NOINTERFACE;
 708 }
 709
 710 static ULONG WINAPI TextStoreACP_AddRef(ITextStoreACP *iface)
 711 {
 712     TextStoreACP *This = (TextStoreACP *)iface;
 713     return InterlockedIncrement(&This->refCount);
 714 }
 715
 716 static ULONG WINAPI TextStoreACP_Release(ITextStoreACP *iface)
 717 {
 718     TextStoreACP *This = (TextStoreACP *)iface;
 719     ULONG ret;
 720
 721     ret = InterlockedDecrement(&This->refCount);
 722     if (ret == 0)
 723         TextStoreACP_Destructor(This);
 724     return ret;
 725 }
 726
 727 static HRESULT WINAPI TextStoreACP_AdviseSink(ITextStoreACP *iface,
 728     REFIID riid, IUnknown *punk, DWORD dwMask)
 729 {
 730     trace("\n");
 731     return S_OK;
 732 }
 733
 734 static HRESULT WINAPI TextStoreACP_UnadviseSink(ITextStoreACP *iface,
 735     IUnknown *punk)
 736 {
 737     trace("\n");
 738     return S_OK;
 739 }
 740 static HRESULT WINAPI TextStoreACP_RequestLock(ITextStoreACP *iface,
 741     DWORD dwLockFlags, HRESULT *phrSession)
 742 {
 743     trace("\n");
 744     return S_OK;
 745 }
 746 static HRESULT WINAPI TextStoreACP_GetStatus(ITextStoreACP *iface,
 747     TS_STATUS *pdcs)
 748 {
 749     trace("\n");
 750     return S_OK;
 751 }
 752 static HRESULT WINAPI TextStoreACP_QueryInsert(ITextStoreACP *iface,
 753     LONG acpTestStart, LONG acpTestEnd, ULONG cch, LONG *pacpResultStart,
 754     LONG *pacpResultEnd)
 755 {
 756     trace("\n");
 757     return S_OK;
 758 }
 759 static HRESULT WINAPI TextStoreACP_GetSelection(ITextStoreACP *iface,
 760     ULONG ulIndex, ULONG ulCount, TS_SELECTION_ACP *pSelection, ULONG *pcFetched)
 761 {
 762     trace("\n");
 763     return S_OK;
 764 }
 765 static HRESULT WINAPI TextStoreACP_SetSelection(ITextStoreACP *iface,
 766     ULONG ulCount, const TS_SELECTION_ACP *pSelection)
 767 {
 768     trace("\n");
 769     return S_OK;
 770 }
 771 static HRESULT WINAPI TextStoreACP_GetText(ITextStoreACP *iface,
 772     LONG acpStart, LONG acpEnd, WCHAR *pchPlain, ULONG cchPlainReq,
 773     ULONG *pcchPlainRet, TS_RUNINFO *prgRunInfo, ULONG cRunInfoReq,
 774     ULONG *pcRunInfoRet, LONG *pacpNext)
 775 {
 776     trace("\n");
 777     return S_OK;
 778 }
 779 static HRESULT WINAPI TextStoreACP_SetText(ITextStoreACP *iface,
 780     DWORD dwFlags, LONG acpStart, LONG acpEnd, const WCHAR *pchText,
 781     ULONG cch, TS_TEXTCHANGE *pChange)
 782 {
 783     trace("\n");
 784     return S_OK;
 785 }
 786 static HRESULT WINAPI TextStoreACP_GetFormattedText(ITextStoreACP *iface,
 787     LONG acpStart, LONG acpEnd, IDataObject **ppDataObject)
 788 {
 789     trace("\n");
 790     return S_OK;
 791 }
 792 static HRESULT WINAPI TextStoreACP_GetEmbedded(ITextStoreACP *iface,
 793     LONG acpPos, REFGUID rguidService, REFIID riid, IUnknown **ppunk)
 794 {
 795     trace("\n");
 796     return S_OK;
 797 }
 798 static HRESULT WINAPI TextStoreACP_QueryInsertEmbedded(ITextStoreACP *iface,
 799     const GUID *pguidService, const FORMATETC *pFormatEtc, BOOL *pfInsertable)
 800 {
 801     trace("\n");
 802     return S_OK;
 803 }
 804 static HRESULT WINAPI TextStoreACP_InsertEmbedded(ITextStoreACP *iface,
 805     DWORD dwFlags, LONG acpStart, LONG acpEnd, IDataObject *pDataObject,
 806     TS_TEXTCHANGE *pChange)
 807 {
 808     trace("\n");
 809     return S_OK;
 810 }
 811 static HRESULT WINAPI TextStoreACP_InsertTextAtSelection(ITextStoreACP *iface,
 812     DWORD dwFlags, const WCHAR *pchText, ULONG cch, LONG *pacpStart,
 813     LONG *pacpEnd, TS_TEXTCHANGE *pChange)
 814 {
 815     trace("\n");
 816     return S_OK;
 817 }
 818 static HRESULT WINAPI TextStoreACP_InsertEmbeddedAtSelection(ITextStoreACP *iface,
 819     DWORD dwFlags, IDataObject *pDataObject, LONG *pacpStart, LONG *pacpEnd,
 820     TS_TEXTCHANGE *pChange)
 821 {
 822     trace("\n");
 823     return S_OK;
 824 }
 825 static HRESULT WINAPI TextStoreACP_RequestSupportedAttrs(ITextStoreACP *iface,
 826     DWORD dwFlags, ULONG cFilterAttrs, const TS_ATTRID *paFilterAttrs)
 827 {
 828     trace("\n");
 829     return S_OK;
 830 }
 831 static HRESULT WINAPI TextStoreACP_RequestAttrsAtPosition(ITextStoreACP *iface,
 832     LONG acpPos, ULONG cFilterAttrs, const TS_ATTRID *paFilterAttrs,
 833     DWORD dwFlags)
 834 {
 835     trace("\n");
 836     return S_OK;
 837 }
 838 static HRESULT WINAPI TextStoreACP_RequestAttrsTransitioningAtPosition(ITextStoreACP *iface,
 839     LONG acpPos, ULONG cFilterAttrs, const TS_ATTRID *paFilterAttrs,
 840     DWORD dwFlags)
 841 {
 842     trace("\n");
 843     return S_OK;
 844 }
 845 static HRESULT WINAPI TextStoreACP_FindNextAttrTransition(ITextStoreACP *iface,
 846     LONG acpStart, LONG acpHalt, ULONG cFilterAttrs, const TS_ATTRID *paFilterAttrs,
 847     DWORD dwFlags, LONG *pacpNext, BOOL *pfFound, LONG *plFoundOffset)
 848 {
 849     trace("\n");
 850     return S_OK;
 851 }
 852 static HRESULT WINAPI TextStoreACP_RetrieveRequestedAttrs(ITextStoreACP *iface,
 853     ULONG ulCount, TS_ATTRVAL *paAttrVals, ULONG *pcFetched)
 854 {
 855     trace("\n");
 856     return S_OK;
 857 }
 858 static HRESULT WINAPI TextStoreACP_GetEndACP(ITextStoreACP *iface,
 859     LONG *pacp)
 860 {
 861     trace("\n");
 862     return S_OK;
 863 }
 864 static HRESULT WINAPI TextStoreACP_GetActiveView(ITextStoreACP *iface,
 865     TsViewCookie *pvcView)
 866 {
 867     trace("\n");
 868     return S_OK;
 869 }
 870 static HRESULT WINAPI TextStoreACP_GetACPFromPoint(ITextStoreACP *iface,
 871     TsViewCookie vcView, const POINT *ptScreen, DWORD dwFlags,
 872     LONG *pacp)
 873 {
 874     trace("\n");
 875     return S_OK;
 876 }
 877 static HRESULT WINAPI TextStoreACP_GetTextExt(ITextStoreACP *iface,
 878     TsViewCookie vcView, LONG acpStart, LONG acpEnd, RECT *prc,
 879     BOOL *pfClipped)
 880 {
 881     trace("\n");
 882     return S_OK;
 883 }
 884 static HRESULT WINAPI TextStoreACP_GetScreenExt(ITextStoreACP *iface,
 885     TsViewCookie vcView, RECT *prc)
 886 {
 887     trace("\n");
 888     return S_OK;
 889 }
 890 static HRESULT WINAPI TextStoreACP_GetWnd(ITextStoreACP *iface,
 891     TsViewCookie vcView, HWND *phwnd)
 892 {
 893     trace("\n");
 894     return S_OK;
 895 }
 896
 897 static const ITextStoreACPVtbl TextStoreACP_TextStoreACPVtbl =
 898 {
 899     TextStoreACP_QueryInterface,
 900     TextStoreACP_AddRef,
 901     TextStoreACP_Release,
 902
 903     TextStoreACP_AdviseSink,
 904     TextStoreACP_UnadviseSink,
 905     TextStoreACP_RequestLock,
 906     TextStoreACP_GetStatus,
 907     TextStoreACP_QueryInsert,
 908     TextStoreACP_GetSelection,
 909     TextStoreACP_SetSelection,
 910     TextStoreACP_GetText,
 911     TextStoreACP_SetText,
 912     TextStoreACP_GetFormattedText,
 913     TextStoreACP_GetEmbedded,
 914     TextStoreACP_QueryInsertEmbedded,
 915     TextStoreACP_InsertEmbedded,
 916     TextStoreACP_InsertTextAtSelection,
 917     TextStoreACP_InsertEmbeddedAtSelection,
 918     TextStoreACP_RequestSupportedAttrs,
 919     TextStoreACP_RequestAttrsAtPosition,
 920     TextStoreACP_RequestAttrsTransitioningAtPosition,
 921     TextStoreACP_FindNextAttrTransition,
 922     TextStoreACP_RetrieveRequestedAttrs,
 923     TextStoreACP_GetEndACP,
 924     TextStoreACP_GetActiveView,
 925     TextStoreACP_GetACPFromPoint,
 926     TextStoreACP_GetTextExt,
 927     TextStoreACP_GetScreenExt,
 928     TextStoreACP_GetWnd
 929 };
 930
 931 HRESULT TextStoreACP_Constructor(IUnknown **ppOut)
 932 {
 933     TextStoreACP *This;
 934
 935     This = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(TextStoreACP));
 936     if (This == NULL)
 937         return E_OUTOFMEMORY;
 938
 939     This->TextStoreACPVtbl = &TextStoreACP_TextStoreACPVtbl;
 940     This->refCount = 1;
 941
 942     *ppOut = (IUnknown *)This;
 943     return S_OK;
 944 }
 945
 946 /**********************************************************************
 947  * ITfThreadMgrEventSink
 948  **********************************************************************/
 949 typedef struct tagThreadMgrEventSink
 950 {
 951     const ITfThreadMgrEventSinkVtbl *ThreadMgrEventSinkVtbl;
 952     LONG refCount;
 953 } ThreadMgrEventSink;
 954
 955 static void ThreadMgrEventSink_Destructor(ThreadMgrEventSink *This)
 956 {
 957     HeapFree(GetProcessHeap(),0,This);
 958 }
 959
 960 static HRESULT WINAPI ThreadMgrEventSink_QueryInterface(ITfThreadMgrEventSink *iface, REFIID iid, LPVOID *ppvOut)
 961 {
 962     ThreadMgrEventSink *This = (ThreadMgrEventSink *)iface;
 963     *ppvOut = NULL;
 964
 965     if (IsEqualIID(iid, &IID_IUnknown) || IsEqualIID(iid, &IID_ITfThreadMgrEventSink))
 966     {
 967         *ppvOut = This;
 968     }
 969
 970     if (*ppvOut)
 971     {
 972         IUnknown_AddRef(iface);
 973         return S_OK;
 974     }
 975
 976     return E_NOINTERFACE;
 977 }
 978
 979 static ULONG WINAPI ThreadMgrEventSink_AddRef(ITfThreadMgrEventSink *iface)
 980 {
 981     ThreadMgrEventSink *This = (ThreadMgrEventSink *)iface;
 982     ok (tmSinkRefCount == This->refCount,"ThreadMgrEventSink refcount off %i vs %i\n",This->refCount,tmSinkRefCount);
 983     return InterlockedIncrement(&This->refCount);
 984 }
 985
 986 static ULONG WINAPI ThreadMgrEventSink_Release(ITfThreadMgrEventSink *iface)
 987 {
 988     ThreadMgrEventSink *This = (ThreadMgrEventSink *)iface;
 989     ULONG ret;
 990
 991     ok (tmSinkRefCount == This->refCount,"ThreadMgrEventSink refcount off %i vs %i\n",This->refCount,tmSinkRefCount);
 992     ret = InterlockedDecrement(&This->refCount);
 993     if (ret == 0)
 994         ThreadMgrEventSink_Destructor(This);
 995     return ret;
 996 }
 997
 998 static HRESULT WINAPI ThreadMgrEventSink_OnInitDocumentMgr(ITfThreadMgrEventSink *iface,
 999 ITfDocumentMgr *pdim)
1000 {
1001     ok(test_OnInitDocumentMgr == SINK_EXPECTED, "Unexpected OnInitDocumentMgr sink\n");
1002     test_OnInitDocumentMgr = SINK_FIRED;
1003     return S_OK;
1004 }
1005
1006 static HRESULT WINAPI ThreadMgrEventSink_OnUninitDocumentMgr(ITfThreadMgrEventSink *iface,
1007 ITfDocumentMgr *pdim)
1008 {
1009     trace("\n");
1010     return S_OK;
1011 }
1012
1013 static HRESULT WINAPI ThreadMgrEventSink_OnSetFocus(ITfThreadMgrEventSink *iface,
1014 ITfDocumentMgr *pdimFocus, ITfDocumentMgr *pdimPrevFocus)
1015 {
1016     ok(test_OnSetFocus == SINK_EXPECTED, "Unexpected OnSetFocus sink\n");
1017     ok(pdimFocus == test_CurrentFocus,"Sink reports wrong focus\n");
1018     ok(pdimPrevFocus == test_PrevFocus,"Sink reports wrong previous focus\n");
1019     test_OnSetFocus = SINK_FIRED;
1020     return S_OK;
1021 }
1022
1023 static HRESULT WINAPI ThreadMgrEventSink_OnPushContext(ITfThreadMgrEventSink *iface,
1024 ITfContext *pic)
1025 {
1026     ok(test_OnPushContext == SINK_EXPECTED, "Unexpected OnPushContext sink\n");
1027     test_OnPushContext = SINK_FIRED;
1028     return S_OK;
1029 }
1030
1031 static HRESULT WINAPI ThreadMgrEventSink_OnPopContext(ITfThreadMgrEventSink *iface,
1032 ITfContext *pic)
1033 {
1034     ok(test_OnPopContext == SINK_EXPECTED, "Unexpected OnPopContext sink\n");
1035     test_OnPopContext = SINK_FIRED;
1036     return S_OK;
1037 }
1038
1039 static const ITfThreadMgrEventSinkVtbl ThreadMgrEventSink_ThreadMgrEventSinkVtbl =
1040 {
1041     ThreadMgrEventSink_QueryInterface,
1042     ThreadMgrEventSink_AddRef,
1043     ThreadMgrEventSink_Release,
1044
1045     ThreadMgrEventSink_OnInitDocumentMgr,
1046     ThreadMgrEventSink_OnUninitDocumentMgr,
1047     ThreadMgrEventSink_OnSetFocus,
1048     ThreadMgrEventSink_OnPushContext,
1049     ThreadMgrEventSink_OnPopContext
1050 };
1051
1052 HRESULT ThreadMgrEventSink_Constructor(IUnknown **ppOut)
1053 {
1054     ThreadMgrEventSink *This;
1055
1056     This = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(ThreadMgrEventSink));
1057     if (This == NULL)
1058         return E_OUTOFMEMORY;
1059
1060     This->ThreadMgrEventSinkVtbl = &ThreadMgrEventSink_ThreadMgrEventSinkVtbl;
1061     This->refCount = 1;
1062
1063     *ppOut = (IUnknown *)This;
1064     return S_OK;
1065 }
1066
1067
1068 /********************************************************************************************
1069  * Stub text service for testing
1070  ********************************************************************************************/
1071
1072 static LONG TS_refCount;
1073 static IClassFactory *cf;
1074 static DWORD regid;
1075
1076 typedef HRESULT (*LPFNCONSTRUCTOR)(IUnknown *pUnkOuter, IUnknown **ppvOut);
1077
1078 typedef struct tagClassFactory
1079 {
1080     const IClassFactoryVtbl *vtbl;
1081     LONG   ref;
1082     LPFNCONSTRUCTOR ctor;
1083 } ClassFactory;
1084
1085 typedef struct tagTextService
1086 {
1087     const ITfTextInputProcessorVtbl *TextInputProcessorVtbl;
1088     LONG refCount;
1089 } TextService;
1090
1091 static void ClassFactory_Destructor(ClassFactory *This)
1092 {
1093     HeapFree(GetProcessHeap(),0,This);
1094     TS_refCount--;
1095 }
1096
1097 static HRESULT WINAPI ClassFactory_QueryInterface(IClassFactory *iface, REFIID riid, LPVOID *ppvOut)
1098 {
1099     *ppvOut = NULL;
1100     if (IsEqualIID(riid, &IID_IClassFactory) || IsEqualIID(riid, &IID_IUnknown))
1101     {
1102         IClassFactory_AddRef(iface);
1103         *ppvOut = iface;
1104         return S_OK;
1105     }
1106
1107     return E_NOINTERFACE;
1108 }
1109
1110 static ULONG WINAPI ClassFactory_AddRef(IClassFactory *iface)
1111 {
1112     ClassFactory *This = (ClassFactory *)iface;
1113     return InterlockedIncrement(&This->ref);
1114 }
1115
1116 static ULONG WINAPI ClassFactory_Release(IClassFactory *iface)
1117 {
1118     ClassFactory *This = (ClassFactory *)iface;
1119     ULONG ret = InterlockedDecrement(&This->ref);
1120
1121     if (ret == 0)
1122         ClassFactory_Destructor(This);
1123     return ret;
1124 }
1125
1126 static HRESULT WINAPI ClassFactory_CreateInstance(IClassFactory *iface, IUnknown *punkOuter, REFIID iid, LPVOID *ppvOut)
1127 {
1128     ClassFactory *This = (ClassFactory *)iface;
1129     HRESULT ret;
1130     IUnknown *obj;
1131
1132     ret = This->ctor(punkOuter, &obj);
1133     if (FAILED(ret))
1134         return ret;
1135     ret = IUnknown_QueryInterface(obj, iid, ppvOut);
1136     IUnknown_Release(obj);
1137     return ret;
1138 }
1139
1140 static HRESULT WINAPI ClassFactory_LockServer(IClassFactory *iface, BOOL fLock)
1141 {
1142     if(fLock)
1143         InterlockedIncrement(&TS_refCount);
1144     else
1145         InterlockedDecrement(&TS_refCount);
1146
1147     return S_OK;
1148 }
1149
1150 static const IClassFactoryVtbl ClassFactoryVtbl = {
1151     /* IUnknown */
1152     ClassFactory_QueryInterface,
1153     ClassFactory_AddRef,
1154     ClassFactory_Release,
1155
1156     /* IClassFactory*/
1157     ClassFactory_CreateInstance,
1158     ClassFactory_LockServer
1159 };
1160
1161 static HRESULT ClassFactory_Constructor(LPFNCONSTRUCTOR ctor, LPVOID *ppvOut)
1162 {
1163     ClassFactory *This = HeapAlloc(GetProcessHeap(),0,sizeof(ClassFactory));
1164     This->vtbl = &ClassFactoryVtbl;
1165     This->ref = 1;
1166     This->ctor = ctor;
1167     *ppvOut = (LPVOID)This;
1168     TS_refCount++;
1169     return S_OK;
1170 }
1171
1172 static void TextService_Destructor(TextService *This)
1173 {
1174     HeapFree(GetProcessHeap(),0,This);
1175 }
1176
1177 static HRESULT WINAPI TextService_QueryInterface(ITfTextInputProcessor *iface, REFIID iid, LPVOID *ppvOut)
1178 {
1179     TextService *This = (TextService *)iface;
1180     *ppvOut = NULL;
1181
1182     if (IsEqualIID(iid, &IID_IUnknown) || IsEqualIID(iid, &IID_ITfTextInputProcessor))
1183     {
1184         *ppvOut = This;
1185     }
1186
1187     if (*ppvOut)
1188     {
1189         IUnknown_AddRef(iface);
1190         return S_OK;
1191     }
1192
1193     return E_NOINTERFACE;
1194 }
1195
1196 static ULONG WINAPI TextService_AddRef(ITfTextInputProcessor *iface)
1197 {
1198     TextService *This = (TextService *)iface;
1199     return InterlockedIncrement(&This->refCount);
1200 }
1201
1202 static ULONG WINAPI TextService_Release(ITfTextInputProcessor *iface)
1203 {
1204     TextService *This = (TextService *)iface;
1205     ULONG ret;
1206
1207     ret = InterlockedDecrement(&This->refCount);
1208     if (ret == 0)
1209         TextService_Destructor(This);
1210     return ret;
1211 }
1212
1213 static HRESULT WINAPI TextService_Activate(ITfTextInputProcessor *iface,
1214         ITfThreadMgr *ptim, TfClientId id)
1215 {
1216     trace("TextService_Activate\n");
1217     ok(test_ShouldActivate,"Activation came unexpectedly\n");
1218     tid = id;
1219     return S_OK;
1220 }
1221
1222 static HRESULT WINAPI TextService_Deactivate(ITfTextInputProcessor *iface)
1223 {
1224     trace("TextService_Deactivate\n");
1225     ok(test_ShouldDeactivate,"Deactivation came unexpectedly\n");
1226     return S_OK;
1227 }
1228
1229 static const ITfTextInputProcessorVtbl TextService_TextInputProcessorVtbl=
1230 {
1231     TextService_QueryInterface,
1232     TextService_AddRef,
1233     TextService_Release,
1234
1235     TextService_Activate,
1236     TextService_Deactivate
1237 };
1238
1239 HRESULT TextService_Constructor(IUnknown *pUnkOuter, IUnknown **ppOut)
1240 {
1241     TextService *This;
1242     if (pUnkOuter)
1243         return CLASS_E_NOAGGREGATION;
1244
1245     This = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(TextService));
1246     if (This == NULL)
1247         return E_OUTOFMEMORY;
1248
1249     This->TextInputProcessorVtbl= &TextService_TextInputProcessorVtbl;
1250     This->refCount = 1;
1251
1252     *ppOut = (IUnknown *)This;
1253     return S_OK;
1254 }
1255
1256 HRESULT RegisterTextService(REFCLSID rclsid)
1257 {
1258     ClassFactory_Constructor( TextService_Constructor ,(LPVOID*)&cf);
1259     return CoRegisterClassObject(rclsid, (IUnknown*) cf, CLSCTX_INPROC_SERVER, REGCLS_MULTIPLEUSE, &regid);
1260 }
1261
1262 HRESULT UnregisterTextService()
1263 {
1264     return CoRevokeClassObject(regid);
1265 }