dlls/inetcomm/tests/mimeintl.c

   1 /*
   2  * MimeInternational tests
   3  *
   4  * Copyright 2008 Huw Davies
   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 #define COBJMACROS
  22 #define NONAMELESSUNION
  23
  24 #include "windows.h"
  25 #include "ole2.h"
  26 #include "ocidl.h"
  27
  28 #include "mimeole.h"
  29
  30 #include "initguid.h"
  31 #include "mlang.h"
  32
  33 #include <stdio.h>
  34
  35 #include "wine/test.h"
  36
  37 static void test_create(void)
  38 {
  39     IMimeInternational *internat, *internat2;
  40     HRESULT hr;
  41     ULONG ref;
  42
  43     hr = MimeOleGetInternat(&internat);
  44     ok(hr == S_OK, "ret %08x\n", hr);
  45     hr = MimeOleGetInternat(&internat2);
  46     ok(hr == S_OK, "ret %08x\n", hr);
  47
  48     /* Under w2k8 it's no longer a singleton */
  49     if(internat == internat2)
  50     {
  51         /* test to show that the object is a singleton with
  52            a reference held by the dll. */
  53         ref = IMimeInternational_Release(internat2);
  54         ok(ref == 2 ||
  55            ref == 1, /* win95 - object is a static singleton */
  56            "got %d\n", ref);
  57
  58         ref = IMimeInternational_Release(internat);
  59         ok(ref == 1, "got %d\n", ref);
  60     }
  61     else
  62     {
  63         ref = IMimeInternational_Release(internat2);
  64         ok(ref == 0, "got %d\n", ref);
  65
  66         ref = IMimeInternational_Release(internat);
  67         ok(ref == 0, "got %d\n", ref);
  68     }
  69
  70 }
  71
  72 static inline HRESULT get_mlang(IMultiLanguage **ml)
  73 {
  74     return CoCreateInstance(&CLSID_CMultiLanguage, NULL,  CLSCTX_INPROC_SERVER | CLSCTX_INPROC_HANDLER,
  75                             &IID_IMultiLanguage, (void **)ml);
  76 }
  77
  78 static HRESULT mlang_getcsetinfo(const char *charset, MIMECSETINFO *mlang_info)
  79 {
  80     DWORD len = MultiByteToWideChar(CP_ACP, 0, charset, -1, NULL, 0);
  81     BSTR bstr = SysAllocStringLen(NULL, len - 1);
  82     HRESULT hr;
  83     IMultiLanguage *ml;
  84
  85     MultiByteToWideChar(CP_ACP, 0, charset, -1, bstr, len);
  86
  87     hr = get_mlang(&ml);
  88
  89     if(SUCCEEDED(hr))
  90     {
  91         hr = IMultiLanguage_GetCharsetInfo(ml, bstr, mlang_info);
  92         IMultiLanguage_Release(ml);
  93     }
  94     SysFreeString(bstr);
  95     if(FAILED(hr)) hr = MIME_E_NOT_FOUND;
  96     return hr;
  97 }
  98
  99 static HRESULT mlang_getcodepageinfo(UINT cp, MIMECPINFO *mlang_cp_info)
 100 {
 101     HRESULT hr;
 102     IMultiLanguage *ml;
 103
 104     hr = get_mlang(&ml);
 105
 106     if(SUCCEEDED(hr))
 107     {
 108         hr = IMultiLanguage_GetCodePageInfo(ml, cp, mlang_cp_info);
 109         IMultiLanguage_Release(ml);
 110     }
 111     return hr;
 112 }
 113
 114 static HRESULT mlang_getcsetinfo_from_cp(UINT cp, CHARSETTYPE charset_type, MIMECSETINFO *mlang_info)
 115 {
 116     MIMECPINFO mlang_cp_info;
 117     WCHAR *charset_name;
 118     HRESULT hr;
 119     IMultiLanguage *ml;
 120
 121     hr = mlang_getcodepageinfo(cp, &mlang_cp_info);
 122     if(FAILED(hr)) return hr;
 123
 124     switch(charset_type)
 125     {
 126     case CHARSET_BODY:
 127         charset_name = mlang_cp_info.wszBodyCharset;
 128         break;
 129     case CHARSET_HEADER:
 130         charset_name = mlang_cp_info.wszHeaderCharset;
 131         break;
 132     case CHARSET_WEB:
 133         charset_name = mlang_cp_info.wszWebCharset;
 134         break;
 135     }
 136
 137     hr = get_mlang(&ml);
 138
 139     if(SUCCEEDED(hr))
 140     {
 141         hr = IMultiLanguage_GetCharsetInfo(ml, charset_name, mlang_info);
 142         IMultiLanguage_Release(ml);
 143     }
 144     return hr;
 145 }
 146
 147 static void test_charset(void)
 148 {
 149     IMimeInternational *internat;
 150     HRESULT hr;
 151     HCHARSET hcs, hcs_windows_1252, hcs_windows_1251;
 152     INETCSETINFO cs_info;
 153     MIMECSETINFO mlang_cs_info;
 154
 155     hr = MimeOleGetInternat(&internat);
 156     ok(hr == S_OK, "ret %08x\n", hr);
 157
 158     hr = IMimeInternational_FindCharset(internat, "nonexistent", &hcs);
 159     ok(hr == MIME_E_NOT_FOUND, "got %08x\n", hr);
 160
 161     hr = IMimeInternational_FindCharset(internat, "windows-1252", &hcs_windows_1252);
 162     ok(hr == S_OK, "got %08x\n", hr);
 163     hr = IMimeInternational_FindCharset(internat, "windows-1252", &hcs);
 164     ok(hr == S_OK, "got %08x\n", hr);
 165     ok(hcs_windows_1252 == hcs, "got different hcharsets for the same name\n");
 166     hr = IMimeInternational_FindCharset(internat, "WiNdoWs-1252", &hcs);
 167     ok(hr == S_OK, "got %08x\n", hr);
 168     ok(hcs_windows_1252 == hcs, "got different hcharsets for the same name\n");
 169
 170     hr = IMimeInternational_FindCharset(internat, "windows-1251", &hcs_windows_1251);
 171     ok(hr == S_OK, "got %08x\n", hr);
 172     ok(hcs_windows_1252 != hcs_windows_1251, "got the same hcharset for the different names\n");
 173
 174     hr = IMimeInternational_GetCharsetInfo(internat, hcs_windows_1252, &cs_info);
 175     ok(hr == S_OK, "got %08x\n", hr);
 176
 177     hr = mlang_getcsetinfo("windows-1252", &mlang_cs_info);
 178     ok(hr == S_OK, "got %08x\n", hr);
 179     ok(cs_info.cpiWindows == mlang_cs_info.uiCodePage, "cpiWindows %d while mlang uiCodePage %d\n",
 180        cs_info.cpiWindows, mlang_cs_info.uiCodePage);
 181     ok(cs_info.cpiInternet == mlang_cs_info.uiInternetEncoding, "cpiInternet %d while mlang uiInternetEncoding %d\n",
 182        cs_info.cpiInternet, mlang_cs_info.uiInternetEncoding);
 183     ok(cs_info.hCharset == hcs_windows_1252, "hCharset doesn't match requested\n");
 184     ok(!strcmp(cs_info.szName, "windows-1252"), "szName doesn't match requested\n");
 185
 186     hr = IMimeInternational_GetCodePageCharset(internat, 1252, CHARSET_BODY, &hcs);
 187     ok(hr == S_OK, "got %08x\n", hr);
 188     hr = IMimeInternational_GetCharsetInfo(internat, hcs, &cs_info);
 189     ok(hr == S_OK, "got %08x\n", hr);
 190
 191     hr = mlang_getcsetinfo_from_cp(1252, CHARSET_BODY, &mlang_cs_info);
 192     ok(hr == S_OK, "got %08x\n", hr);
 193     ok(cs_info.cpiWindows == mlang_cs_info.uiCodePage, "cpiWindows %d while mlang uiCodePage %d\n",
 194        cs_info.cpiWindows, mlang_cs_info.uiCodePage);
 195     ok(cs_info.cpiInternet == mlang_cs_info.uiInternetEncoding, "cpiInternet %d while mlang uiInternetEncoding %d\n",
 196        cs_info.cpiInternet, mlang_cs_info.uiInternetEncoding);
 197
 198     IMimeInternational_Release(internat);
 199 }
 200
 201 static void test_defaultcharset(void)
 202 {
 203     IMimeInternational *internat;
 204     HRESULT hr;
 205     HCHARSET hcs_default, hcs, hcs_windows_1251;
 206
 207     hr = MimeOleGetInternat(&internat);
 208     ok(hr == S_OK, "ret %08x\n", hr);
 209
 210     hr = IMimeInternational_GetDefaultCharset(internat, &hcs_default);
 211     ok(hr == S_OK, "ret %08x\n", hr);
 212     hr = IMimeInternational_GetCodePageCharset(internat, GetACP(), CHARSET_BODY, &hcs);
 213     ok(hr == S_OK, "ret %08x\n", hr);
 214     ok(hcs_default == hcs, "Unexpected default charset\n");
 215
 216     hr = IMimeInternational_FindCharset(internat, "windows-1251", &hcs_windows_1251);
 217     ok(hr == S_OK, "got %08x\n", hr);
 218     hr = IMimeInternational_SetDefaultCharset(internat, hcs_windows_1251);
 219     ok(hr == S_OK, "ret %08x\n", hr);
 220     hr = IMimeInternational_GetDefaultCharset(internat, &hcs);
 221     ok(hr == S_OK, "ret %08x\n", hr);
 222     ok(hcs == hcs_windows_1251, "didn't retrieve recently set default\n");
 223     /* Set the old default back again */
 224     hr = IMimeInternational_SetDefaultCharset(internat, hcs_default);
 225     ok(hr == S_OK, "ret %08x\n", hr);
 226
 227     IMimeInternational_Release(internat);
 228 }
 229
 230 static void test_convert(void)
 231 {
 232     IMimeInternational *internat;
 233     HRESULT hr;
 234     BLOB src, dst;
 235     ULONG read;
 236     PROPVARIANT prop_in, prop_out;
 237     static char test_string[] = "test string";
 238     static WCHAR test_stringW[] = L"test string";
 239
 240     hr = MimeOleGetInternat(&internat);
 241     ok(hr == S_OK, "ret %08x\n", hr);
 242
 243     src.pBlobData = (BYTE*)test_string;
 244     src.cbSize = sizeof(test_string);
 245     hr = IMimeInternational_ConvertBuffer(internat, 1252, 28591, &src, &dst, &read);
 246     ok(hr == S_OK, "ret %08x\n", hr);
 247     ok(read == sizeof(test_string), "got %d\n", read);
 248     ok(dst.cbSize == sizeof(test_string), "got %d\n", dst.cbSize);
 249     CoTaskMemFree(dst.pBlobData);
 250
 251     src.cbSize = 2;
 252     hr = IMimeInternational_ConvertBuffer(internat, 1252, 28591, &src, &dst, &read);
 253     ok(hr == S_OK, "ret %08x\n", hr);
 254     ok(read == 2, "got %d\n", read);
 255     ok(dst.cbSize == 2, "got %d\n", dst.cbSize);
 256     CoTaskMemFree(dst.pBlobData);
 257
 258     prop_in.vt = VT_LPWSTR;
 259     prop_in.u.pwszVal = test_stringW;
 260     hr = IMimeInternational_ConvertString(internat, CP_UNICODE, 1252, &prop_in, &prop_out);
 261     ok(hr == S_OK, "ret %08x\n", hr);
 262     ok(prop_out.vt == VT_LPSTR, "got %d\n", prop_out.vt);
 263     ok(!strcmp(prop_out.u.pszVal, test_string), "got %s\n", prop_out.u.pszVal);
 264     PropVariantClear(&prop_out);
 265
 266     /* If in.vt is VT_LPWSTR, ignore cpiSrc */
 267     prop_in.vt = VT_LPWSTR;
 268     prop_in.u.pwszVal = test_stringW;
 269     hr = IMimeInternational_ConvertString(internat, 28591, 1252, &prop_in, &prop_out);
 270     ok(hr == S_OK, "ret %08x\n", hr);
 271     ok(prop_out.vt == VT_LPSTR, "got %d\n", prop_out.vt);
 272     ok(!strcmp(prop_out.u.pszVal, test_string), "got %s\n", prop_out.u.pszVal);
 273     PropVariantClear(&prop_out);
 274
 275     prop_in.vt = VT_LPSTR;
 276     prop_in.u.pszVal = test_string;
 277     hr = IMimeInternational_ConvertString(internat, 28591, CP_UNICODE, &prop_in, &prop_out);
 278     ok(hr == S_OK, "ret %08x\n", hr);
 279     ok(prop_out.vt == VT_LPWSTR, "got %d\n", prop_out.vt);
 280     ok(!lstrcmpW(prop_out.u.pwszVal, test_stringW), "mismatched strings\n");
 281     PropVariantClear(&prop_out);
 282
 283     /* If in.vt is VT_LPSTR and cpiSrc is CP_UNICODE, use another multibyte codepage (probably GetACP()) */
 284     prop_in.vt = VT_LPSTR;
 285     prop_in.u.pszVal = test_string;
 286     hr = IMimeInternational_ConvertString(internat, CP_UNICODE, CP_UNICODE, &prop_in, &prop_out);
 287     ok(hr == S_OK, "ret %08x\n", hr);
 288     ok(prop_out.vt == VT_LPWSTR, "got %d\n", prop_out.vt);
 289     ok(!lstrcmpW(prop_out.u.pwszVal, test_stringW), "mismatched strings\n");
 290     PropVariantClear(&prop_out);
 291
 292     IMimeInternational_Release(internat);
 293 }
 294
 295 START_TEST(mimeintl)
 296 {
 297     OleInitialize(NULL);
 298     test_create();
 299     test_charset();
 300     test_defaultcharset();
 301     test_convert();
 302     OleUninitialize();
 303 }