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 = (WCHAR*)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 }