setupapi: Have each test create and destroy its own string table.
[wine] / dlls / setupapi / tests / stringtable.c
1 /*
2  * Unit test suite for StringTable functions
3  *
4  * Copyright 2005 Steven Edwards for ReactOS
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  * TODO:
22  * Add test for StringTableStringFromIdEx
23  */
24
25 #include <stdarg.h>
26
27 #include "windef.h"
28 #include "winbase.h"
29 #include "wingdi.h"
30 #include "winuser.h"
31 #include "winreg.h"
32 #include "setupapi.h"
33
34 #include "wine/test.h"
35
36
37 DECLARE_HANDLE(HSTRING_TABLE);
38
39 /* Flags for StringTableAddString and StringTableLookUpString */
40 #define ST_CASE_SENSITIVE_COMPARE       0x00000001
41
42 static DWORD    (WINAPI *pStringTableAddString)(HSTRING_TABLE, LPWSTR, DWORD);
43 static VOID     (WINAPI *pStringTableDestroy)(HSTRING_TABLE);
44 static HSTRING_TABLE (WINAPI *pStringTableDuplicate)(HSTRING_TABLE hStringTable);
45 static HSTRING_TABLE (WINAPI *pStringTableInitialize)(VOID);
46 static DWORD    (WINAPI *pStringTableLookUpString)(HSTRING_TABLE, LPWSTR, DWORD);
47 static LPWSTR   (WINAPI *pStringTableStringFromId)(HSTRING_TABLE, DWORD);
48 #if 0
49 static BOOL     (WINAPI *pStringTableStringFromIdEx)(HSTRING_TABLE, DWORD, LPWSTR, LPDWORD);
50 static VOID     (WINAPI *pStringTableTrim)(HSTRING_TABLE);
51 #endif
52
53 HMODULE hdll;
54 static WCHAR string[] = {'s','t','r','i','n','g',0};
55 static WCHAR String[] = {'S','t','r','i','n','g',0};
56 static WCHAR foo[] = {'f','o','o',0};
57
58 static void load_it_up(void)
59 {
60     hdll = GetModuleHandleA("setupapi.dll");
61
62     pStringTableInitialize = (void*)GetProcAddress(hdll, "StringTableInitialize");
63     if (!pStringTableInitialize)
64         pStringTableInitialize = (void*)GetProcAddress(hdll, "pSetupStringTableInitialize");
65
66     pStringTableAddString = (void*)GetProcAddress(hdll, "StringTableAddString");
67     if (!pStringTableAddString)
68         pStringTableAddString = (void*)GetProcAddress(hdll, "pSetupStringTableAddString");
69
70     pStringTableDuplicate = (void*)GetProcAddress(hdll, "StringTableDuplicate");
71     if (!pStringTableDuplicate)
72         pStringTableDuplicate = (void*)GetProcAddress(hdll, "pSetupStringTableDuplicate");
73
74     pStringTableDestroy = (void*)GetProcAddress(hdll, "StringTableDestroy");
75     if (!pStringTableDestroy)
76         pStringTableDestroy = (void*)GetProcAddress(hdll, "pSetupStringTableDestroy");
77
78     pStringTableLookUpString = (void*)GetProcAddress(hdll, "StringTableLookUpString");
79     if (!pStringTableLookUpString)
80         pStringTableLookUpString = (void*)GetProcAddress(hdll, "pSetupStringTableLookUpString");
81
82     pStringTableStringFromId = (void*)GetProcAddress(hdll, "StringTableStringFromId");
83     if (!pStringTableStringFromId)
84         pStringTableStringFromId = (void*)GetProcAddress(hdll, "pSetupStringTableStringFromId");
85 }
86
87 static void test_StringTableAddString(void)
88 {
89     DWORD retval, hstring, hString, hfoo;
90     HANDLE table;
91
92     table = pStringTableInitialize();
93     ok(table != NULL, "failed to initialize string table\n");
94
95     /* case insensitive */
96     hstring=pStringTableAddString(table,string,0);
97     ok(hstring!=-1,"Failed to add string to String Table\n");
98     
99     retval=pStringTableAddString(table,String,0);
100     ok(retval!=-1,"Failed to add String to String Table\n");    
101     ok(hstring==retval,"string handle %x != String handle %x in String Table\n", hstring, retval);        
102     
103     hfoo=pStringTableAddString(table,foo,0);
104     ok(hfoo!=-1,"Failed to add foo to String Table\n");        
105     ok(hfoo!=hstring,"foo and string share the same ID %x in String Table\n", hfoo);            
106     
107     /* case sensitive */    
108     hString=pStringTableAddString(table,String,ST_CASE_SENSITIVE_COMPARE);
109     ok(hstring!=hString,"String handle and string share same ID %x in Table\n", hstring);        
110
111     pStringTableDestroy(table);
112 }
113
114 static void test_StringTableDuplicate(void)
115 {
116     HANDLE table, table2;
117
118     table = pStringTableInitialize();
119     ok(table != NULL,"Failed to Initialize String Table\n");
120
121     table2=pStringTableDuplicate(table);
122     ok(table2!=NULL,"Failed to duplicate String Table\n");
123
124     pStringTableDestroy(table);
125     pStringTableDestroy(table2);
126 }
127
128 static void test_StringTableLookUpString(void)
129 {   
130     DWORD retval, retval2, hstring, hString, hfoo;
131     HANDLE table, table2;
132
133     table = pStringTableInitialize();
134     ok(table != NULL,"failed to initialize string table\n");
135
136     hstring = pStringTableAddString(table, string, 0);
137     ok(hstring != ~0u, "failed to add 'string' to string table\n");
138
139     hString = pStringTableAddString(table, String, 0);
140     ok(hString != ~0u,"failed to add 'String' to string table\n");
141
142     hfoo = pStringTableAddString(table, foo, 0);
143     ok(hfoo != ~0u, "failed to add 'foo' to string table\n");
144
145     table2 = pStringTableDuplicate(table);
146     ok(table2 != NULL, "Failed to duplicate String Table\n");
147
148     /* case insensitive */
149     retval=pStringTableLookUpString(table,string,0);
150     ok(retval!=-1,"Failed find string in String Table 1\n");
151     ok(retval==hstring,
152         "Lookup for string (%x) does not match previous handle (%x) in String Table 1\n",
153         retval, hstring);    
154
155     retval=pStringTableLookUpString(table2,string,0);
156     ok(retval!=-1,"Failed find string in String Table 2\n");
157     
158     retval=pStringTableLookUpString(table,String,0);
159     ok(retval!=-1,"Failed find String in String Table 1\n");
160
161     retval=pStringTableLookUpString(table2,String,0);
162     ok(retval!=-1,"Failed find String in String Table 2\n");    
163     
164     retval=pStringTableLookUpString(table,foo,0);
165     ok(retval!=-1,"Failed find foo in String Table 1\n");    
166     ok(retval==hfoo,
167         "Lookup for foo (%x) does not match previous handle (%x) in String Table 1\n",
168         retval, hfoo);        
169     
170     retval=pStringTableLookUpString(table2,foo,0);
171     ok(retval!=-1,"Failed find foo in String Table 2\n");    
172     
173     /* case sensitive */
174     retval=pStringTableLookUpString(table,string,ST_CASE_SENSITIVE_COMPARE);
175     retval2=pStringTableLookUpString(table,String,ST_CASE_SENSITIVE_COMPARE);    
176     ok(retval!=retval2,"Lookup of string equals String in Table 1\n");
177     ok(retval==hString,
178         "Lookup for String (%x) does not match previous handle (%x) in String Table 1\n",
179         retval, hString);
180
181     pStringTableDestroy(table);
182     pStringTableDestroy(table2);
183 }
184
185 static void test_StringTableStringFromId(void)
186 {
187     HANDLE table;
188     DWORD hstring;
189     WCHAR *string2;
190     int result;
191
192     table = pStringTableInitialize();
193     ok(table != NULL,"Failed to Initialize String Table\n");
194
195     hstring = pStringTableAddString(table, string, 0);
196     ok(hstring != ~0u,"failed to add 'string' to string table\n");
197
198     /* correct */
199     string2=pStringTableStringFromId(table,pStringTableLookUpString(table,string,0));
200     ok(string2!=NULL,"Failed to look up string by ID from String Table\n");
201     
202     result=lstrcmpiW(string, string2);
203     ok(result==0,"StringID %p does not match requested StringID %p\n",string,string2);
204
205     pStringTableDestroy(table);
206 }
207
208 START_TEST(stringtable)
209 {
210     load_it_up();
211
212     test_StringTableAddString();
213     test_StringTableDuplicate();
214     test_StringTableLookUpString();
215     test_StringTableStringFromId();
216 }