shell32: Add printers CLSID to test, clean up a bit.
[wine] / dlls / shell32 / tests / string.c
1 /*
2  * Unit tests for shell32 string operations
3  *
4  * Copyright 2004 Jon Griffiths
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 <stdarg.h>
22 #include <stdio.h>
23
24 #define WINE_NOWINSOCK
25 #include "windef.h"
26 #include "winbase.h"
27 #include "wtypes.h"
28 #include "shellapi.h"
29 #include "shtypes.h"
30 #include "objbase.h"
31
32 #include "wine/test.h"
33
34 static HMODULE hShell32;
35 static HRESULT (WINAPI *pStrRetToStrNAW)(LPVOID,DWORD,LPSTRRET,const ITEMIDLIST *);
36
37 static WCHAR *CoDupStrW(const char* src)
38 {
39   INT len = MultiByteToWideChar(CP_ACP, 0, src, -1, NULL, 0);
40   WCHAR* szTemp = CoTaskMemAlloc(len * sizeof(WCHAR));
41   MultiByteToWideChar(CP_ACP, 0, src, -1, szTemp, len);
42   return szTemp;
43 }
44
45 static inline int strcmpW(const WCHAR *str1, const WCHAR *str2)
46 {
47     while (*str1 && (*str1 == *str2)) { str1++; str2++; }
48     return *str1 - *str2;
49 }
50
51 static void test_StrRetToStringNA(void)
52 {
53     trace("StrRetToStringNAW is Ascii\n");
54     /* FIXME */
55 }
56
57 static void test_StrRetToStringNW(void)
58 {
59     static const WCHAR szTestW[] = { 'T','e','s','t','\0' };
60     ITEMIDLIST iidl[10];
61     WCHAR buff[128];
62     STRRET strret;
63     BOOL ret;
64
65     trace("StrRetToStringNAW is Unicode\n");
66
67     strret.uType = STRRET_WSTR;
68     U(strret).pOleStr = CoDupStrW("Test");
69     memset(buff, 0xff, sizeof(buff));
70     ret = pStrRetToStrNAW(buff, sizeof(buff)/sizeof(WCHAR), &strret, NULL);
71     ok(ret == TRUE && !strcmpW(buff, szTestW),
72        "STRRET_WSTR: dup failed, ret=%d\n", ret);
73
74     strret.uType = STRRET_CSTR;
75     lstrcpyA(U(strret).cStr, "Test");
76     memset(buff, 0xff, sizeof(buff));
77     ret = pStrRetToStrNAW(buff, sizeof(buff)/sizeof(WCHAR), &strret, NULL);
78     ok(ret == TRUE && !strcmpW(buff, szTestW),
79        "STRRET_CSTR: dup failed, ret=%d\n", ret);
80
81     strret.uType = STRRET_OFFSET;
82     U(strret).uOffset = 1;
83     strcpy((char*)&iidl, " Test");
84     memset(buff, 0xff, sizeof(buff));
85     ret = pStrRetToStrNAW(buff, sizeof(buff)/sizeof(WCHAR), &strret, iidl);
86     ok(ret == TRUE && !strcmpW(buff, szTestW),
87        "STRRET_OFFSET: dup failed, ret=%d\n", ret);
88
89     /* The next test crashes on W2K, WinXP and W2K3, so we don't test. */
90 #if 0
91     /* Invalid dest - should return FALSE, except NT4 does not, so we don't check. */
92     strret.uType = STRRET_WSTR;
93     U(strret).pOleStr = CoDupStrW("Test");
94     pStrRetToStrNAW(NULL, sizeof(buff)/sizeof(WCHAR), &strret, NULL);
95     trace("NULL dest: ret=%d\n", ret);
96 #endif
97 }
98
99 START_TEST(string)
100 {
101     CoInitialize(0);
102
103     hShell32 = GetModuleHandleA("shell32.dll");
104
105     pStrRetToStrNAW = (void*)GetProcAddress(hShell32, (LPSTR)96);
106     if (pStrRetToStrNAW)
107     {
108         if (!(GetVersion() & 0x80000000))
109             test_StrRetToStringNW();
110         else
111             test_StrRetToStringNA();
112     }
113
114     CoUninitialize();
115 }