msctf: Implement ITfKeystrokeMgr::UnpreserveKey.
[wine] / 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     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     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 static void test_ClientId(void)
652 {
653     ITfClientId *pcid;
654     TfClientId id1,id2;
655     HRESULT hr;
656     GUID g2;
657
658     hr = ITfThreadMgr_QueryInterface(g_tm, &IID_ITfClientId, (LPVOID*)&pcid);
659     ok(SUCCEEDED(hr),"Unable to aquire ITfClientId interface\n");
660
661     CoCreateGuid(&g2);
662
663     hr = ITfClientId_GetClientId(pcid,&GUID_NULL,&id1);
664     ok(SUCCEEDED(hr),"GetClientId failed\n");
665     hr = ITfClientId_GetClientId(pcid,&GUID_NULL,&id2);
666     ok(SUCCEEDED(hr),"GetClientId failed\n");
667     ok(id1==id2,"Id's for GUID_NULL do not match\n");
668     hr = ITfClientId_GetClientId(pcid,&CLSID_FakeService,&id2);
669     ok(SUCCEEDED(hr),"GetClientId failed\n");
670     ok(id2!=id1,"Id matches GUID_NULL\n");
671     todo_wine ok(id2==tid,"Id for CLSID_FakeService not matching tid\n");
672     ok(id2!=cid,"Id for CLSID_FakeService matching cid\n");
673     hr = ITfClientId_GetClientId(pcid,&g2,&id2);
674     ok(SUCCEEDED(hr),"GetClientId failed\n");
675     ok(id2!=id1,"Id matches GUID_NULL\n");
676     ok(id2!=tid,"Id for random guid matching tid\n");
677     ok(id2!=cid,"Id for random guid matching cid\n");
678     ITfClientId_Release(pcid);
679 }
680
681 START_TEST(inputprocessor)
682 {
683     if (SUCCEEDED(initialize()))
684     {
685         test_Register();
686         test_RegisterCategory();
687         test_EnumInputProcessorInfo();
688         test_Enable();
689         test_ThreadMgrAdviseSinks();
690         test_Activate();
691         test_startSession();
692         test_TfGuidAtom();
693         test_ClientId();
694         test_KeystrokeMgr();
695         test_endSession();
696         test_EnumLanguageProfiles();
697         test_FindClosestCategory();
698         test_Disable();
699         test_ThreadMgrUnadviseSinks();
700         test_UnregisterCategory();
701         test_Unregister();
702     }
703     else
704         skip("Unable to create InputProcessor\n");
705     cleanup();
706 }
707
708 /**********************************************************************
709  * ITextStoreACP
710  **********************************************************************/
711 typedef struct tagTextStoreACP
712 {
713     const ITextStoreACPVtbl *TextStoreACPVtbl;
714     LONG refCount;
715 } TextStoreACP;
716
717 static void TextStoreACP_Destructor(TextStoreACP *This)
718 {
719     HeapFree(GetProcessHeap(),0,This);
720 }
721
722 static HRESULT WINAPI TextStoreACP_QueryInterface(ITextStoreACP *iface, REFIID iid, LPVOID *ppvOut)
723 {
724     TextStoreACP *This = (TextStoreACP *)iface;
725     *ppvOut = NULL;
726
727     if (IsEqualIID(iid, &IID_IUnknown) || IsEqualIID(iid, &IID_ITextStoreACP))
728     {
729         *ppvOut = This;
730     }
731
732     if (*ppvOut)
733     {
734         IUnknown_AddRef(iface);
735         return S_OK;
736     }
737
738     return E_NOINTERFACE;
739 }
740
741 static ULONG WINAPI TextStoreACP_AddRef(ITextStoreACP *iface)
742 {
743     TextStoreACP *This = (TextStoreACP *)iface;
744     return InterlockedIncrement(&This->refCount);
745 }
746
747 static ULONG WINAPI TextStoreACP_Release(ITextStoreACP *iface)
748 {
749     TextStoreACP *This = (TextStoreACP *)iface;
750     ULONG ret;
751
752     ret = InterlockedDecrement(&This->refCount);
753     if (ret == 0)
754         TextStoreACP_Destructor(This);
755     return ret;
756 }
757
758 static HRESULT WINAPI TextStoreACP_AdviseSink(ITextStoreACP *iface,
759     REFIID riid, IUnknown *punk, DWORD dwMask)
760 {
761     trace("\n");
762     return S_OK;
763 }
764
765 static HRESULT WINAPI TextStoreACP_UnadviseSink(ITextStoreACP *iface,
766     IUnknown *punk)
767 {
768     trace("\n");
769     return S_OK;
770 }
771 static HRESULT WINAPI TextStoreACP_RequestLock(ITextStoreACP *iface,
772     DWORD dwLockFlags, HRESULT *phrSession)
773 {
774     trace("\n");
775     return S_OK;
776 }
777 static HRESULT WINAPI TextStoreACP_GetStatus(ITextStoreACP *iface,
778     TS_STATUS *pdcs)
779 {
780     trace("\n");
781     return S_OK;
782 }
783 static HRESULT WINAPI TextStoreACP_QueryInsert(ITextStoreACP *iface,
784     LONG acpTestStart, LONG acpTestEnd, ULONG cch, LONG *pacpResultStart,
785     LONG *pacpResultEnd)
786 {
787     trace("\n");
788     return S_OK;
789 }
790 static HRESULT WINAPI TextStoreACP_GetSelection(ITextStoreACP *iface,
791     ULONG ulIndex, ULONG ulCount, TS_SELECTION_ACP *pSelection, ULONG *pcFetched)
792 {
793     trace("\n");
794     return S_OK;
795 }
796 static HRESULT WINAPI TextStoreACP_SetSelection(ITextStoreACP *iface,
797     ULONG ulCount, const TS_SELECTION_ACP *pSelection)
798 {
799     trace("\n");
800     return S_OK;
801 }
802 static HRESULT WINAPI TextStoreACP_GetText(ITextStoreACP *iface,
803     LONG acpStart, LONG acpEnd, WCHAR *pchPlain, ULONG cchPlainReq,
804     ULONG *pcchPlainRet, TS_RUNINFO *prgRunInfo, ULONG cRunInfoReq,
805     ULONG *pcRunInfoRet, LONG *pacpNext)
806 {
807     trace("\n");
808     return S_OK;
809 }
810 static HRESULT WINAPI TextStoreACP_SetText(ITextStoreACP *iface,
811     DWORD dwFlags, LONG acpStart, LONG acpEnd, const WCHAR *pchText,
812     ULONG cch, TS_TEXTCHANGE *pChange)
813 {
814     trace("\n");
815     return S_OK;
816 }
817 static HRESULT WINAPI TextStoreACP_GetFormattedText(ITextStoreACP *iface,
818     LONG acpStart, LONG acpEnd, IDataObject **ppDataObject)
819 {
820     trace("\n");
821     return S_OK;
822 }
823 static HRESULT WINAPI TextStoreACP_GetEmbedded(ITextStoreACP *iface,
824     LONG acpPos, REFGUID rguidService, REFIID riid, IUnknown **ppunk)
825 {
826     trace("\n");
827     return S_OK;
828 }
829 static HRESULT WINAPI TextStoreACP_QueryInsertEmbedded(ITextStoreACP *iface,
830     const GUID *pguidService, const FORMATETC *pFormatEtc, BOOL *pfInsertable)
831 {
832     trace("\n");
833     return S_OK;
834 }
835 static HRESULT WINAPI TextStoreACP_InsertEmbedded(ITextStoreACP *iface,
836     DWORD dwFlags, LONG acpStart, LONG acpEnd, IDataObject *pDataObject,
837     TS_TEXTCHANGE *pChange)
838 {
839     trace("\n");
840     return S_OK;
841 }
842 static HRESULT WINAPI TextStoreACP_InsertTextAtSelection(ITextStoreACP *iface,
843     DWORD dwFlags, const WCHAR *pchText, ULONG cch, LONG *pacpStart,
844     LONG *pacpEnd, TS_TEXTCHANGE *pChange)
845 {
846     trace("\n");
847     return S_OK;
848 }
849 static HRESULT WINAPI TextStoreACP_InsertEmbeddedAtSelection(ITextStoreACP *iface,
850     DWORD dwFlags, IDataObject *pDataObject, LONG *pacpStart, LONG *pacpEnd,
851     TS_TEXTCHANGE *pChange)
852 {
853     trace("\n");
854     return S_OK;
855 }
856 static HRESULT WINAPI TextStoreACP_RequestSupportedAttrs(ITextStoreACP *iface,
857     DWORD dwFlags, ULONG cFilterAttrs, const TS_ATTRID *paFilterAttrs)
858 {
859     trace("\n");
860     return S_OK;
861 }
862 static HRESULT WINAPI TextStoreACP_RequestAttrsAtPosition(ITextStoreACP *iface,
863     LONG acpPos, ULONG cFilterAttrs, const TS_ATTRID *paFilterAttrs,
864     DWORD dwFlags)
865 {
866     trace("\n");
867     return S_OK;
868 }
869 static HRESULT WINAPI TextStoreACP_RequestAttrsTransitioningAtPosition(ITextStoreACP *iface,
870     LONG acpPos, ULONG cFilterAttrs, const TS_ATTRID *paFilterAttrs,
871     DWORD dwFlags)
872 {
873     trace("\n");
874     return S_OK;
875 }
876 static HRESULT WINAPI TextStoreACP_FindNextAttrTransition(ITextStoreACP *iface,
877     LONG acpStart, LONG acpHalt, ULONG cFilterAttrs, const TS_ATTRID *paFilterAttrs,
878     DWORD dwFlags, LONG *pacpNext, BOOL *pfFound, LONG *plFoundOffset)
879 {
880     trace("\n");
881     return S_OK;
882 }
883 static HRESULT WINAPI TextStoreACP_RetrieveRequestedAttrs(ITextStoreACP *iface,
884     ULONG ulCount, TS_ATTRVAL *paAttrVals, ULONG *pcFetched)
885 {
886     trace("\n");
887     return S_OK;
888 }
889 static HRESULT WINAPI TextStoreACP_GetEndACP(ITextStoreACP *iface,
890     LONG *pacp)
891 {
892     trace("\n");
893     return S_OK;
894 }
895 static HRESULT WINAPI TextStoreACP_GetActiveView(ITextStoreACP *iface,
896     TsViewCookie *pvcView)
897 {
898     trace("\n");
899     return S_OK;
900 }
901 static HRESULT WINAPI TextStoreACP_GetACPFromPoint(ITextStoreACP *iface,
902     TsViewCookie vcView, const POINT *ptScreen, DWORD dwFlags,
903     LONG *pacp)
904 {
905     trace("\n");
906     return S_OK;
907 }
908 static HRESULT WINAPI TextStoreACP_GetTextExt(ITextStoreACP *iface,
909     TsViewCookie vcView, LONG acpStart, LONG acpEnd, RECT *prc,
910     BOOL *pfClipped)
911 {
912     trace("\n");
913     return S_OK;
914 }
915 static HRESULT WINAPI TextStoreACP_GetScreenExt(ITextStoreACP *iface,
916     TsViewCookie vcView, RECT *prc)
917 {
918     trace("\n");
919     return S_OK;
920 }
921 static HRESULT WINAPI TextStoreACP_GetWnd(ITextStoreACP *iface,
922     TsViewCookie vcView, HWND *phwnd)
923 {
924     trace("\n");
925     return S_OK;
926 }
927
928 static const ITextStoreACPVtbl TextStoreACP_TextStoreACPVtbl =
929 {
930     TextStoreACP_QueryInterface,
931     TextStoreACP_AddRef,
932     TextStoreACP_Release,
933
934     TextStoreACP_AdviseSink,
935     TextStoreACP_UnadviseSink,
936     TextStoreACP_RequestLock,
937     TextStoreACP_GetStatus,
938     TextStoreACP_QueryInsert,
939     TextStoreACP_GetSelection,
940     TextStoreACP_SetSelection,
941     TextStoreACP_GetText,
942     TextStoreACP_SetText,
943     TextStoreACP_GetFormattedText,
944     TextStoreACP_GetEmbedded,
945     TextStoreACP_QueryInsertEmbedded,
946     TextStoreACP_InsertEmbedded,
947     TextStoreACP_InsertTextAtSelection,
948     TextStoreACP_InsertEmbeddedAtSelection,
949     TextStoreACP_RequestSupportedAttrs,
950     TextStoreACP_RequestAttrsAtPosition,
951     TextStoreACP_RequestAttrsTransitioningAtPosition,
952     TextStoreACP_FindNextAttrTransition,
953     TextStoreACP_RetrieveRequestedAttrs,
954     TextStoreACP_GetEndACP,
955     TextStoreACP_GetActiveView,
956     TextStoreACP_GetACPFromPoint,
957     TextStoreACP_GetTextExt,
958     TextStoreACP_GetScreenExt,
959     TextStoreACP_GetWnd
960 };
961
962 HRESULT TextStoreACP_Constructor(IUnknown **ppOut)
963 {
964     TextStoreACP *This;
965
966     This = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(TextStoreACP));
967     if (This == NULL)
968         return E_OUTOFMEMORY;
969
970     This->TextStoreACPVtbl = &TextStoreACP_TextStoreACPVtbl;
971     This->refCount = 1;
972
973     *ppOut = (IUnknown *)This;
974     return S_OK;
975 }
976
977 /**********************************************************************
978  * ITfThreadMgrEventSink
979  **********************************************************************/
980 typedef struct tagThreadMgrEventSink
981 {
982     const ITfThreadMgrEventSinkVtbl *ThreadMgrEventSinkVtbl;
983     LONG refCount;
984 } ThreadMgrEventSink;
985
986 static void ThreadMgrEventSink_Destructor(ThreadMgrEventSink *This)
987 {
988     HeapFree(GetProcessHeap(),0,This);
989 }
990
991 static HRESULT WINAPI ThreadMgrEventSink_QueryInterface(ITfThreadMgrEventSink *iface, REFIID iid, LPVOID *ppvOut)
992 {
993     ThreadMgrEventSink *This = (ThreadMgrEventSink *)iface;
994     *ppvOut = NULL;
995
996     if (IsEqualIID(iid, &IID_IUnknown) || IsEqualIID(iid, &IID_ITfThreadMgrEventSink))
997     {
998         *ppvOut = This;
999     }
1000
1001     if (*ppvOut)
1002     {
1003         IUnknown_AddRef(iface);
1004         return S_OK;
1005     }
1006
1007     return E_NOINTERFACE;
1008 }
1009
1010 static ULONG WINAPI ThreadMgrEventSink_AddRef(ITfThreadMgrEventSink *iface)
1011 {
1012     ThreadMgrEventSink *This = (ThreadMgrEventSink *)iface;
1013     ok (tmSinkRefCount == This->refCount,"ThreadMgrEventSink refcount off %i vs %i\n",This->refCount,tmSinkRefCount);
1014     return InterlockedIncrement(&This->refCount);
1015 }
1016
1017 static ULONG WINAPI ThreadMgrEventSink_Release(ITfThreadMgrEventSink *iface)
1018 {
1019     ThreadMgrEventSink *This = (ThreadMgrEventSink *)iface;
1020     ULONG ret;
1021
1022     ok (tmSinkRefCount == This->refCount,"ThreadMgrEventSink refcount off %i vs %i\n",This->refCount,tmSinkRefCount);
1023     ret = InterlockedDecrement(&This->refCount);
1024     if (ret == 0)
1025         ThreadMgrEventSink_Destructor(This);
1026     return ret;
1027 }
1028
1029 static HRESULT WINAPI ThreadMgrEventSink_OnInitDocumentMgr(ITfThreadMgrEventSink *iface,
1030 ITfDocumentMgr *pdim)
1031 {
1032     ok(test_OnInitDocumentMgr == SINK_EXPECTED, "Unexpected OnInitDocumentMgr sink\n");
1033     test_OnInitDocumentMgr = SINK_FIRED;
1034     return S_OK;
1035 }
1036
1037 static HRESULT WINAPI ThreadMgrEventSink_OnUninitDocumentMgr(ITfThreadMgrEventSink *iface,
1038 ITfDocumentMgr *pdim)
1039 {
1040     trace("\n");
1041     return S_OK;
1042 }
1043
1044 static HRESULT WINAPI ThreadMgrEventSink_OnSetFocus(ITfThreadMgrEventSink *iface,
1045 ITfDocumentMgr *pdimFocus, ITfDocumentMgr *pdimPrevFocus)
1046 {
1047     ok(test_OnSetFocus == SINK_EXPECTED, "Unexpected OnSetFocus sink\n");
1048     ok(pdimFocus == test_CurrentFocus,"Sink reports wrong focus\n");
1049     ok(pdimPrevFocus == test_PrevFocus,"Sink reports wrong previous focus\n");
1050     test_OnSetFocus = SINK_FIRED;
1051     return S_OK;
1052 }
1053
1054 static HRESULT WINAPI ThreadMgrEventSink_OnPushContext(ITfThreadMgrEventSink *iface,
1055 ITfContext *pic)
1056 {
1057     ok(test_OnPushContext == SINK_EXPECTED, "Unexpected OnPushContext sink\n");
1058     test_OnPushContext = SINK_FIRED;
1059     return S_OK;
1060 }
1061
1062 static HRESULT WINAPI ThreadMgrEventSink_OnPopContext(ITfThreadMgrEventSink *iface,
1063 ITfContext *pic)
1064 {
1065     ok(test_OnPopContext == SINK_EXPECTED, "Unexpected OnPopContext sink\n");
1066     test_OnPopContext = SINK_FIRED;
1067     return S_OK;
1068 }
1069
1070 static const ITfThreadMgrEventSinkVtbl ThreadMgrEventSink_ThreadMgrEventSinkVtbl =
1071 {
1072     ThreadMgrEventSink_QueryInterface,
1073     ThreadMgrEventSink_AddRef,
1074     ThreadMgrEventSink_Release,
1075
1076     ThreadMgrEventSink_OnInitDocumentMgr,
1077     ThreadMgrEventSink_OnUninitDocumentMgr,
1078     ThreadMgrEventSink_OnSetFocus,
1079     ThreadMgrEventSink_OnPushContext,
1080     ThreadMgrEventSink_OnPopContext
1081 };
1082
1083 HRESULT ThreadMgrEventSink_Constructor(IUnknown **ppOut)
1084 {
1085     ThreadMgrEventSink *This;
1086
1087     This = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(ThreadMgrEventSink));
1088     if (This == NULL)
1089         return E_OUTOFMEMORY;
1090
1091     This->ThreadMgrEventSinkVtbl = &ThreadMgrEventSink_ThreadMgrEventSinkVtbl;
1092     This->refCount = 1;
1093
1094     *ppOut = (IUnknown *)This;
1095     return S_OK;
1096 }
1097
1098
1099 /********************************************************************************************
1100  * Stub text service for testing
1101  ********************************************************************************************/
1102
1103 static LONG TS_refCount;
1104 static IClassFactory *cf;
1105 static DWORD regid;
1106
1107 typedef HRESULT (*LPFNCONSTRUCTOR)(IUnknown *pUnkOuter, IUnknown **ppvOut);
1108
1109 typedef struct tagClassFactory
1110 {
1111     const IClassFactoryVtbl *vtbl;
1112     LONG   ref;
1113     LPFNCONSTRUCTOR ctor;
1114 } ClassFactory;
1115
1116 typedef struct tagTextService
1117 {
1118     const ITfTextInputProcessorVtbl *TextInputProcessorVtbl;
1119     LONG refCount;
1120 } TextService;
1121
1122 static void ClassFactory_Destructor(ClassFactory *This)
1123 {
1124     HeapFree(GetProcessHeap(),0,This);
1125     TS_refCount--;
1126 }
1127
1128 static HRESULT WINAPI ClassFactory_QueryInterface(IClassFactory *iface, REFIID riid, LPVOID *ppvOut)
1129 {
1130     *ppvOut = NULL;
1131     if (IsEqualIID(riid, &IID_IClassFactory) || IsEqualIID(riid, &IID_IUnknown))
1132     {
1133         IClassFactory_AddRef(iface);
1134         *ppvOut = iface;
1135         return S_OK;
1136     }
1137
1138     return E_NOINTERFACE;
1139 }
1140
1141 static ULONG WINAPI ClassFactory_AddRef(IClassFactory *iface)
1142 {
1143     ClassFactory *This = (ClassFactory *)iface;
1144     return InterlockedIncrement(&This->ref);
1145 }
1146
1147 static ULONG WINAPI ClassFactory_Release(IClassFactory *iface)
1148 {
1149     ClassFactory *This = (ClassFactory *)iface;
1150     ULONG ret = InterlockedDecrement(&This->ref);
1151
1152     if (ret == 0)
1153         ClassFactory_Destructor(This);
1154     return ret;
1155 }
1156
1157 static HRESULT WINAPI ClassFactory_CreateInstance(IClassFactory *iface, IUnknown *punkOuter, REFIID iid, LPVOID *ppvOut)
1158 {
1159     ClassFactory *This = (ClassFactory *)iface;
1160     HRESULT ret;
1161     IUnknown *obj;
1162
1163     ret = This->ctor(punkOuter, &obj);
1164     if (FAILED(ret))
1165         return ret;
1166     ret = IUnknown_QueryInterface(obj, iid, ppvOut);
1167     IUnknown_Release(obj);
1168     return ret;
1169 }
1170
1171 static HRESULT WINAPI ClassFactory_LockServer(IClassFactory *iface, BOOL fLock)
1172 {
1173     if(fLock)
1174         InterlockedIncrement(&TS_refCount);
1175     else
1176         InterlockedDecrement(&TS_refCount);
1177
1178     return S_OK;
1179 }
1180
1181 static const IClassFactoryVtbl ClassFactoryVtbl = {
1182     /* IUnknown */
1183     ClassFactory_QueryInterface,
1184     ClassFactory_AddRef,
1185     ClassFactory_Release,
1186
1187     /* IClassFactory*/
1188     ClassFactory_CreateInstance,
1189     ClassFactory_LockServer
1190 };
1191
1192 static HRESULT ClassFactory_Constructor(LPFNCONSTRUCTOR ctor, LPVOID *ppvOut)
1193 {
1194     ClassFactory *This = HeapAlloc(GetProcessHeap(),0,sizeof(ClassFactory));
1195     This->vtbl = &ClassFactoryVtbl;
1196     This->ref = 1;
1197     This->ctor = ctor;
1198     *ppvOut = (LPVOID)This;
1199     TS_refCount++;
1200     return S_OK;
1201 }
1202
1203 static void TextService_Destructor(TextService *This)
1204 {
1205     HeapFree(GetProcessHeap(),0,This);
1206 }
1207
1208 static HRESULT WINAPI TextService_QueryInterface(ITfTextInputProcessor *iface, REFIID iid, LPVOID *ppvOut)
1209 {
1210     TextService *This = (TextService *)iface;
1211     *ppvOut = NULL;
1212
1213     if (IsEqualIID(iid, &IID_IUnknown) || IsEqualIID(iid, &IID_ITfTextInputProcessor))
1214     {
1215         *ppvOut = This;
1216     }
1217
1218     if (*ppvOut)
1219     {
1220         IUnknown_AddRef(iface);
1221         return S_OK;
1222     }
1223
1224     return E_NOINTERFACE;
1225 }
1226
1227 static ULONG WINAPI TextService_AddRef(ITfTextInputProcessor *iface)
1228 {
1229     TextService *This = (TextService *)iface;
1230     return InterlockedIncrement(&This->refCount);
1231 }
1232
1233 static ULONG WINAPI TextService_Release(ITfTextInputProcessor *iface)
1234 {
1235     TextService *This = (TextService *)iface;
1236     ULONG ret;
1237
1238     ret = InterlockedDecrement(&This->refCount);
1239     if (ret == 0)
1240         TextService_Destructor(This);
1241     return ret;
1242 }
1243
1244 static HRESULT WINAPI TextService_Activate(ITfTextInputProcessor *iface,
1245         ITfThreadMgr *ptim, TfClientId id)
1246 {
1247     trace("TextService_Activate\n");
1248     ok(test_ShouldActivate,"Activation came unexpectedly\n");
1249     tid = id;
1250     return S_OK;
1251 }
1252
1253 static HRESULT WINAPI TextService_Deactivate(ITfTextInputProcessor *iface)
1254 {
1255     trace("TextService_Deactivate\n");
1256     ok(test_ShouldDeactivate,"Deactivation came unexpectedly\n");
1257     return S_OK;
1258 }
1259
1260 static const ITfTextInputProcessorVtbl TextService_TextInputProcessorVtbl=
1261 {
1262     TextService_QueryInterface,
1263     TextService_AddRef,
1264     TextService_Release,
1265
1266     TextService_Activate,
1267     TextService_Deactivate
1268 };
1269
1270 HRESULT TextService_Constructor(IUnknown *pUnkOuter, IUnknown **ppOut)
1271 {
1272     TextService *This;
1273     if (pUnkOuter)
1274         return CLASS_E_NOAGGREGATION;
1275
1276     This = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(TextService));
1277     if (This == NULL)
1278         return E_OUTOFMEMORY;
1279
1280     This->TextInputProcessorVtbl= &TextService_TextInputProcessorVtbl;
1281     This->refCount = 1;
1282
1283     *ppOut = (IUnknown *)This;
1284     return S_OK;
1285 }
1286
1287 HRESULT RegisterTextService(REFCLSID rclsid)
1288 {
1289     ClassFactory_Constructor( TextService_Constructor ,(LPVOID*)&cf);
1290     return CoRegisterClassObject(rclsid, (IUnknown*) cf, CLSCTX_INPROC_SERVER, REGCLS_MULTIPLEUSE, &regid);
1291 }
1292
1293 HRESULT UnregisterTextService()
1294 {
1295     return CoRevokeClassObject(regid);
1296 }