d3d8/tests: Use primary monitor screen size to perform tests.
[wine] / dlls / atl / tests / registrar.c
1 /*
2  * ATL test program
3  *
4  * Copyright 2010 Damjan Jovanovic
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 COBJMACROS
25
26 #include <wine/test.h>
27 #include <windef.h>
28 #include <winbase.h>
29 #include <winuser.h>
30 #include <wingdi.h>
31 #include <winnls.h>
32 #include <winerror.h>
33 #include <winnt.h>
34 #include <wtypes.h>
35 #include <olectl.h>
36 #include <ocidl.h>
37 #include <initguid.h>
38 #include <atliface.h>
39
40 static const char textA[] =
41 "HKCR \n"
42 "{ \n"
43 "    ForceRemove eebf73c4-50fd-478f-bbcf-db212221227a \n"
44 "    { \n"
45 "        val 'string' = s 'string' \n"
46 "        val 'dword_quoted_dec' = d '1' \n"
47 "        val 'dword_unquoted_dec' = d 1 \n"
48 "        val 'dword_quoted_hex' = d '0xA' \n"
49 "        val 'dword_unquoted_hex' = d 0xA \n"
50 "        val 'binary_quoted' = b 'deadbeef' \n"
51 "        val 'binary_unquoted' = b deadbeef \n"
52 "    } \n"
53 "}";
54
55 static void test_registrar(void)
56 {
57     IRegistrar *registrar = NULL;
58     HRESULT hr;
59     INT count;
60     WCHAR *textW = NULL;
61
62     if (!GetProcAddress(GetModuleHandleA("atl.dll"), "AtlAxAttachControl"))
63     {
64         win_skip("Old versions of atl.dll don't support binary values\n");
65         return;
66     }
67
68     hr = CoCreateInstance(&CLSID_Registrar, NULL, CLSCTX_INPROC_SERVER, &IID_IRegistrar, (void**)&registrar);
69     if (FAILED(hr))
70     {
71         skip("creating IRegistrar failed, hr = 0x%08X\n", hr);
72         return;
73     }
74
75     count = MultiByteToWideChar(CP_ACP, 0, textA, -1, NULL, 0);
76     textW = HeapAlloc(GetProcessHeap(), 0, count * sizeof(WCHAR));
77     if (textW)
78     {
79         DWORD dword;
80         DWORD size;
81         LONG lret;
82         HKEY key;
83         BYTE bytes[4];
84
85         MultiByteToWideChar(CP_ACP, 0, textA, -1, textW, count);
86         hr = IRegistrar_StringRegister(registrar, textW);
87         ok(SUCCEEDED(hr), "IRegistrar_StringRegister failed, hr = 0x%08X\n", hr);
88
89         lret = RegOpenKeyA(HKEY_CLASSES_ROOT, "eebf73c4-50fd-478f-bbcf-db212221227a", &key);
90         ok(lret == ERROR_SUCCESS, "error %d opening registry key\n", lret);
91
92         size = sizeof(dword);
93         lret = RegQueryValueExA(key, "dword_unquoted_hex", NULL, NULL, (BYTE*)&dword, &size);
94         ok(lret == ERROR_SUCCESS, "RegQueryValueExA failed, error %d\n", lret);
95         ok(dword != 0xA, "unquoted hex is not supposed to be preserved\n");
96
97         size = sizeof(dword);
98         lret = RegQueryValueExA(key, "dword_quoted_hex", NULL, NULL, (BYTE*)&dword, &size);
99         ok(lret == ERROR_SUCCESS, "RegQueryValueExA failed, error %d\n", lret);
100         ok(dword != 0xA, "quoted hex is not supposed to be preserved\n");
101
102         size = sizeof(dword);
103         lret = RegQueryValueExA(key, "dword_unquoted_dec", NULL, NULL, (BYTE*)&dword, &size);
104         ok(lret == ERROR_SUCCESS, "RegQueryValueExA failed, error %d\n", lret);
105         ok(dword == 1, "unquoted dec is not supposed to be %d\n", dword);
106
107         size = sizeof(dword);
108         lret = RegQueryValueExA(key, "dword_quoted_dec", NULL, NULL, (BYTE*)&dword, &size);
109         ok(lret == ERROR_SUCCESS, "RegQueryValueExA failed, error %d\n", lret);
110         ok(dword == 1, "quoted dec is not supposed to be %d\n", dword);
111
112         size = 4;
113         lret = RegQueryValueExA(key, "binary_quoted", NULL, NULL, bytes, &size);
114         ok(lret == ERROR_SUCCESS, "RegQueryValueA, failed, error %d\n", lret);
115         ok(bytes[0] == 0xde && bytes[1] == 0xad && bytes[2] == 0xbe && bytes[3] == 0xef,
116             "binary quoted value was not preserved (it's 0x%02X%02X%02X%02X)\n",
117             0xff & bytes[0], 0xff & bytes[1], 0xff & bytes[2], 0xff & bytes[3]);
118
119         size = 4;
120         lret = RegQueryValueExA(key, "binary_unquoted", NULL, NULL, bytes, &size);
121         ok(lret == ERROR_SUCCESS, "RegQueryValueA, failed, error %d\n", lret);
122         ok(bytes[0] == 0xde && bytes[1] == 0xad && bytes[2] == 0xbe && bytes[3] == 0xef,
123             "binary unquoted value was not preserved (it's 0x%02X%02X%02X%02X)\n",
124             0xff & bytes[0], 0xff & bytes[1], 0xff & bytes[2], 0xff & bytes[3]);
125
126         hr = IRegistrar_StringUnregister(registrar, textW);
127         ok(SUCCEEDED(hr), "IRegistrar_StringUnregister failed, hr = 0x%08X\n", hr);
128         RegCloseKey(key);
129
130         HeapFree(GetProcessHeap(), 0, textW);
131     }
132     else
133         skip("allocating memory failed\n");
134
135     IRegistrar_Release(registrar);
136 }
137
138 START_TEST(registrar)
139 {
140     CoInitialize(NULL);
141
142     test_registrar();
143
144     CoUninitialize();
145 }