From d4e7845887de72cd3c36b7b862297cf23ccb0a8d Mon Sep 17 00:00:00 2001 From: Detlef Riekenberg Date: Thu, 23 Oct 2008 23:05:37 +0200 Subject: [PATCH] shdocvw/tests: Add tests for URLSubRegQueryA. --- dlls/shdocvw/tests/Makefile.in | 1 + dlls/shdocvw/tests/shdocvw.c | 168 +++++++++++++++++++++++++++++++++ 2 files changed, 169 insertions(+) create mode 100644 dlls/shdocvw/tests/shdocvw.c diff --git a/dlls/shdocvw/tests/Makefile.in b/dlls/shdocvw/tests/Makefile.in index 635d553260..e20f8edbe1 100644 --- a/dlls/shdocvw/tests/Makefile.in +++ b/dlls/shdocvw/tests/Makefile.in @@ -7,6 +7,7 @@ IMPORTS = shell32 ole32 oleaut32 user32 advapi32 kernel32 CTESTS = \ intshcut.c \ + shdocvw.c \ shortcut.c \ webbrowser.c diff --git a/dlls/shdocvw/tests/shdocvw.c b/dlls/shdocvw/tests/shdocvw.c new file mode 100644 index 0000000000..f80a1112eb --- /dev/null +++ b/dlls/shdocvw/tests/shdocvw.c @@ -0,0 +1,168 @@ +/* + * Unit tests for misc shdocvw functions + * + * Copyright 2008 Detlef Riekenberg + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 of the License, or (at your option) any later version. + * + * This library is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this library; if not, write to the Free Software + * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA + */ + + +#include + +#include "windef.h" +#include "winbase.h" +#include "winreg.h" +#include "wininet.h" + +#include "wine/test.h" + +/* ################ */ + +static HMODULE hshdocvw; +static HRESULT (WINAPI *pURLSubRegQueryA)(LPCSTR, LPCSTR, DWORD, LPVOID, DWORD, DWORD); + +static CHAR appdata[] = "AppData"; +static CHAR common_appdata[] = "Common AppData"; +static CHAR default_page_url[] = "Default_Page_URL"; +static CHAR does_not_exist[] = "does_not_exist"; +static CHAR regpath_iemain[] = "Software\\Microsoft\\Internet Explorer\\Main"; +static CHAR regpath_shellfolders[] = "Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\Shell Folders"; +static CHAR start_page[] = "Start Page"; + +/* ################ */ + +static void init_functions(void) +{ + hshdocvw = LoadLibraryA("shdocvw.dll"); + pURLSubRegQueryA = (void *) GetProcAddress(hshdocvw, (LPSTR) 151); +} + +/* ################ */ + +static void test_URLSubRegQueryA(void) +{ + CHAR buffer[INTERNET_MAX_URL_LENGTH]; + HRESULT hr; + DWORD used; + DWORD len; + + if (!pURLSubRegQueryA) { + skip("URLSubRegQueryA not found\n"); + return; + } + + memset(buffer, '#', sizeof(buffer)-2); + buffer[sizeof(buffer)-1] = '\0'; + /* called by inetcpl.cpl */ + hr = pURLSubRegQueryA(regpath_iemain, default_page_url, REG_SZ, buffer, INTERNET_MAX_URL_LENGTH, -1); + ok(hr == E_FAIL || hr == S_OK, "got 0x%x (expected E_FAIL or S_OK)\n", hr); + + memset(buffer, '#', sizeof(buffer)-2); + buffer[sizeof(buffer)-1] = '\0'; + /* called by inetcpl.cpl */ + hr = pURLSubRegQueryA(regpath_iemain, start_page, REG_SZ, buffer, INTERNET_MAX_URL_LENGTH, -1); + len = lstrlenA(buffer); + /* respect privacy: do not dump the url */ + ok(hr == S_OK, "got 0x%x and %d (expected S_OK)\n", hr, len); + + /* test buffer length: just large enough */ + memset(buffer, '#', sizeof(buffer)-2); + buffer[sizeof(buffer)-1] = '\0'; + hr = pURLSubRegQueryA(regpath_iemain, start_page, REG_SZ, buffer, len+1, -1); + used = lstrlenA(buffer); + /* respect privacy: do not dump the url */ + ok((hr == S_OK) && (used == len), + "got 0x%x and %d (expected S_OK and %d)\n", hr, used, len); + + /* no space for terminating 0: result is truncated */ + memset(buffer, '#', sizeof(buffer)-2); + buffer[sizeof(buffer)-1] = '\0'; + hr = pURLSubRegQueryA(regpath_iemain, start_page, REG_SZ, buffer, len, -1); + used = lstrlenA(buffer); + ok((hr == S_OK) && (used == len - 1), + "got 0x%x and %d (expected S_OK and %d)\n", hr, used, len - 1); + + /* no space for the complete result: truncate another char */ + if (len > 1) { + memset(buffer, '#', sizeof(buffer)-2); + buffer[sizeof(buffer)-1] = '\0'; + hr = pURLSubRegQueryA(regpath_iemain, start_page, REG_SZ, buffer, len-1, -1); + used = lstrlenA(buffer); + ok((hr == S_OK) && (used == (len - 2)), + "got 0x%x and %d (expected S_OK and %d)\n", hr, used, len - 2); + } + + /* only space for the terminating 0: function still succeded */ + memset(buffer, '#', sizeof(buffer)-2); + buffer[sizeof(buffer)-1] = '\0'; + hr = pURLSubRegQueryA(regpath_iemain, start_page, REG_SZ, buffer, 1, -1); + used = lstrlenA(buffer); + ok((hr == S_OK) && !used, + "got 0x%x and %d (expected S_OK and 0)\n", hr, used); + + /* size of buffer is 0, but the function still succeed */ + memset(buffer, '#', sizeof(buffer)-2); + buffer[sizeof(buffer)-1] = '\0'; + hr = pURLSubRegQueryA(regpath_iemain, start_page, REG_SZ, buffer, 0, -1); + used = lstrlenA(buffer); + ok((hr == S_OK) && (used == INTERNET_MAX_URL_LENGTH - 1), + "got 0x%x and %d (expected S_OK and %d)\n", hr, used, INTERNET_MAX_URL_LENGTH-1); + + /* still succeed without a buffer for the result */ + hr = pURLSubRegQueryA(regpath_iemain, start_page, REG_SZ, NULL, 0, -1); + ok(hr == S_OK, "got 0x%x (expected S_OK)\n", hr); + + /* still succeed, when a length is given without a buffer */ + hr = pURLSubRegQueryA(regpath_iemain, start_page, REG_SZ, NULL, INTERNET_MAX_URL_LENGTH, -1); + ok(hr == S_OK, "got 0x%x (expected S_OK)\n", hr); + + /* this value does not exist */ + memset(buffer, '#', sizeof(buffer)-2); + buffer[sizeof(buffer)-1] = '\0'; + hr = pURLSubRegQueryA(regpath_iemain, does_not_exist, REG_SZ, buffer, INTERNET_MAX_URL_LENGTH, -1); + /* random bytes are copied to the buffer */ + ok((hr == E_FAIL), "got 0x%x (expected E_FAIL)\n", hr); + + /* the third parameter is ignored. Is it really a type? (data is REG_SZ) */ + memset(buffer, '#', sizeof(buffer)-2); + buffer[sizeof(buffer)-1] = '\0'; + hr = pURLSubRegQueryA(regpath_iemain, start_page, REG_DWORD, buffer, INTERNET_MAX_URL_LENGTH, -1); + used = lstrlenA(buffer); + ok((hr == S_OK) && (used == len), + "got 0x%x and %d (expected S_OK and %d)\n", hr, used, len); + + /* the function works for HKCU and HKLM */ + memset(buffer, '#', sizeof(buffer)-2); + buffer[sizeof(buffer)-1] = '\0'; + hr = pURLSubRegQueryA(regpath_shellfolders, appdata, REG_SZ, buffer, INTERNET_MAX_URL_LENGTH, -1); + used = lstrlenA(buffer); + ok(hr == S_OK, "got 0x%x and %d (expected S_OK)\n", hr, used); + + memset(buffer, '#', sizeof(buffer)-2); + buffer[sizeof(buffer)-1] = '\0'; + hr = pURLSubRegQueryA(regpath_shellfolders, common_appdata, REG_SZ, buffer, INTERNET_MAX_URL_LENGTH, -1); + used = lstrlenA(buffer); + ok(hr == S_OK, "got 0x%x and %d (expected S_OK)\n", hr, used); + + /* todo: what does the last parameter mean? */ +} + +/* ################ */ + +START_TEST(shdocvw) +{ + init_functions(); + test_URLSubRegQueryA(); +}